Update chatbot/memory.py

chatbot/memory.py

@@ -1,9 +1,16 @@
 from langchain.memory import ConversationSummaryBufferMemory
-from chatbot.llm import gemini_llm  
+from langchain.memory import ChatMessageHistory
+from langchain.schema import AIMessage, HumanMessage
+
+# Khởi tạo chat history
+chat_history = ChatMessageHistory()
 
 memory = ConversationSummaryBufferMemory(
     llm=gemini_llm,
     memory_key="chat_history",  # this key will hold the conversation context
+    input_key="question",    # Key input từ người dùng
     max_token_limit=2000,       # adjust this limit based on your needs
-    return_messages=True
+    prompt=summary_prompt,
+    moving_summary_buffer="summary",  # Lưu summary của hội thoại
+    chat_memory=chat_history  # Lưu trữ toàn bộ hội thoại (bao gồm cả câu hỏi và phản hồi)
 )