anhkhoiphan commited on
Commit
ed3a456
verified
1 Parent(s): d00ea52

Use combined memory

Browse files
Files changed (1) hide show
  1. chatbot/memory.py +7 -2
chatbot/memory.py CHANGED
@@ -1,3 +1,8 @@
1
- from langchain.memory import ConversationBufferMemory
 
 
 
 
 
 
2
 
3
- memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
 
1
+ from langchain.memory import CombinedMemory, ConversationBufferWindowMemory, ConversationSummaryMemory
2
+ from chatbot.llm import gemini_llm
3
+
4
+ buffer_memory = ConversationBufferWindowMemory(k=5)
5
+ summary_memory = ConversationSummaryMemory(llm=gemini_llm)
6
+
7
+ combined_memory = CombinedMemory(memories=[buffer_memory, summary_memory])
8