Update chatbot/memory.py

chatbot/memory.py

@@ -1,8 +1,9 @@
-from langchain.memory import CombinedMemory, ConversationBufferWindowMemory, ConversationSummaryMemory
+from langchain.memory import ConversationSummaryBufferMemory
 from chatbot.llm import gemini_llm  
 
-buffer_memory = ConversationBufferWindowMemory(k=5, memory_key="buffer_history")
-summary_memory = ConversationSummaryMemory(llm=gemini_llm, memory_key="summary_history")
-
-memory = CombinedMemory(memories=[buffer_memory, summary_memory])
-
+memory = ConversationSummaryBufferMemory(
+    llm=gemini_llm,
+    memory_key="chat_history",  # this key will hold the conversation context
+    max_token_limit=2000,       # adjust this limit based on your needs
+    return_messages=True
+)