Update document_chat.py

document_chat.py

@@ -48,6 +48,8 @@ def process_query_with_memory(query, chat_memory):
         retriever=retriever,
         memory=chat_memory
     )
-    chat_history = chat_memory.chat_memory.messages if chat_memory else []
 
-    return conversation_chain.run({"question": query, "chat_history": chat_memory.memory if chat_memory else []})
+    # Fix: Properly load chat history
+    chat_history = chat_memory.load_memory_variables({}).get("chat_history", [])
+
+    return conversation_chain.run({"question": query, "chat_history": chat_history})