Update app.py

app.py

@@ -229,6 +229,14 @@ Question: %s
 
 Document:
         """ % message
+        retriever_system="""
+        You are an AI assistant that answers questions based on documents.
+
+%s
+
+Question: %s
+Answer:
+        """ % (text,message)
 #[Wait for user's document]
         
         # Create the agent
@@ -260,7 +268,9 @@ Document:
 
         # Get the response stream
         stream = agent.get_chat_response(
-            retriever_system+text,
+            retriever_system,
+            #retriever_system+text,
+            #retriever_system+text,
             llm_sampling_settings=settings,
             chat_history=messages,
             returns_streaming_generator=True,