Update rag_service.py

rag_service.py

@@ -113,10 +113,21 @@ class RAGService:
             for text_element in docs_by_type["texts"]:
                 context_text += str(text_element)
 
-        prompt_template = f"""
-        Answer the question based only on the following context, which can include text, tables, and the below image.
-        Context: {context_text}
-        Question: {user_question}
+       prompt_template = f"""
+       You are a helpful AI assistant. You have access to the following context from uploaded documents:
+       Context: {context_text}
+       Question: {user_question}
+       
+       Instructions:
+       - If the question is related to the provided context, use that information to answer
+       - If the question is not related to the context or is a general query, answer based on your general knowledge
+       - Be clea
+       """
+    
+        # prompt_template = f"""
+        # Answer the question based only on the following context, which can include text, tables, and the below image.
+        # Context: {context_text}
+        # Question: {user_question}
         """
 
         prompt_content = [{"type": "text", "text": prompt_template}]