Update app.py

app.py

@@ -155,23 +155,20 @@ def create_conversational_rag_chain(vectorstore):
         ]
     )
 
-
     question_answer_chain = create_stuff_documents_chain(llm, qa_prompt)
-
     rag_chain = create_retrieval_chain(ha_retriever, question_answer_chain)
     msgs = StreamlitChatMessageHistory(key="special_app_key")
 
     conversational_rag_chain = RunnableWithMessageHistory(
-    rag_chain,
-    lambda session_id: msgs,
-    input_messages_key="input",
-    history_messages_key="chat_history",
-    output_messages_key="answer",
-    return_source_documents = True
-)
-
+        rag_chain,
+        lambda session_id: msgs,
+        input_messages_key="input",
+        history_messages_key="chat_history",
+        output_messages_key="answer",
+        return_source_documents=True
+    )
 
-    return rag_chain
+    return conversational_rag_chain