Update app.py

app.py

@@ -386,7 +386,14 @@ if __name__ == '__main__':
     st.title("Multipurpose AI Agent")
 
     if "qa_prompt" not in st.session_state:
-        st.session_state.qa_prompt = None
+        st.session_state.qa_prompt = lambda query, context:
+        f"""You are an assistant for question-answering tasks. Use the following pieces of retrieved context to answer the question. If you don't know the answer, just say that you don't know. Use three sentences maximum and keep the answer concise.
+        
+        Question: {query} 
+              
+        Context: {context} 
+              
+        Answer:"""
 
     if "tooltip" not in st.session_state:
         st.session_state.tooltip = 'The AI answer your questions only considering the documents provided'