Update app.py

app.py

@@ -345,10 +345,27 @@ def qa_prompt():
         )
 
 def on_change_documents_only():
-    st.session_state.toggle_docs = { 
-        'qa_prompt': qa_prompt(),
-        'tooltip': 'The AI answer your questions only considering the documents provided' if st.session_state.documents_only else """The AI answer your questions considering the documents provided, and if it doesn't found the answer in them, try to find in its own internal knowledge"""
-    }
+    if st.session_state.documents_only:
+        st.session_state.toggle_docs = {
+            'qa_prompt': """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, reply with 'no_answer'. Use three sentences maximum and keep the answer concise.
+            
+            Question: {query} 
+            
+            Context: {context} 
+            
+            Answer:""",
+            'tooltip': 'The AI answer your questions only considering the documents provided'
+        }
+    else:
+        st.session_state.toggle_docs = {
+            'qa_prompt': """If the context is not relevant, please answer the question by using your own knowledge about the topic.
+            If you decide to provide information using your own knowledge or general knowledge, write 'knowledge_topic' at the top of your answer.
+        
+            {context}
+
+            Question: {query}""",
+            'tooltip': 'The AI answer your questions considering the documents provided, and if it doesn't found the answer in them, try to find in its own internal knowledge'
+        }
     
     
 if __name__ == '__main__':