Update rag_service.py

rag_service.py

@@ -105,39 +105,39 @@ class RAGService:
 
 
     def build_prompt(self, kwargs):
-    """Build prompt with context and images"""
-    docs_by_type = kwargs["context"]
-    user_question = kwargs["question"]
-
-    context_text = ""
-    if len(docs_by_type["texts"]) > 0:
-        for text_element in docs_by_type["texts"]:
-            context_text += str(text_element)
-
-    # Always use this flexible prompt
-    prompt_template = f"""
-    You are a helpful AI assistant. 
+        """Build prompt with context and images"""
+        docs_by_type = kwargs["context"]
+        user_question = kwargs["question"]
     
-    Context from documents (use if relevant): {context_text}
+        context_text = ""
+        if len(docs_by_type["texts"]) > 0:
+            for text_element in docs_by_type["texts"]:
+                context_text += str(text_element)
     
-    Question: {user_question}
+        # Always use this flexible prompt
+        prompt_template = f"""
+        You are a helpful AI assistant. 
+        
+        Context from documents (use if relevant): {context_text}
+        
+        Question: {user_question}
+        
+        Instructions: Answer the question. If the provided context is relevant to the question, use it. If not, answer using your general knowledge.
+        """
     
-    Instructions: Answer the question. If the provided context is relevant to the question, use it. If not, answer using your general knowledge.
-    """
-
-    prompt_content = [{"type": "text", "text": prompt_template}]
+        prompt_content = [{"type": "text", "text": prompt_template}]
+        
+        # Add images only if context exists
+        if len(docs_by_type["images"]) > 0:
+            for image in docs_by_type["images"]:
+                prompt_content.append(
+                    {
+                        "type": "image_url",
+                        "image_url": {"url": f"data:image/jpeg;base64,{image}"},
+                    }
+                )
     
-    # Add images only if context exists
-    if len(docs_by_type["images"]) > 0:
-        for image in docs_by_type["images"]:
-            prompt_content.append(
-                {
-                    "type": "image_url",
-                    "image_url": {"url": f"data:image/jpeg;base64,{image}"},
-                }
-            )
-
-    return ChatPromptTemplate.from_messages([HumanMessage(content=prompt_content)])
+        return ChatPromptTemplate.from_messages([HumanMessage(content=prompt_content)])
 
     def ask_question(self, question: str):
         """Process a question and return response"""