Update rag_service.py

rag_service.py

@@ -114,18 +114,31 @@ class RAGService:
             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. 
+            # 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.
+            """
         
-        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.
-        """
-    
-        prompt_content = [{"type": "text", "text": prompt_template}]
+            prompt_content = [{"type": "text", "text": prompt_template}]
+
+        else:
+            # No context found - use direct LLM prompt
+            prompt_template = f"""
+            You are a helpful AI assistant. Answer the following question using your general knowledge:
+            
+            Question: {user_question}
+            """
+            
+            prompt_content = [{"type": "text", "text": prompt_template}]
+
+            
+            
         
         # Add images only if context exists
         if len(docs_by_type["images"]) > 0: