Spaces:
Sleeping
Sleeping
Use combined memory
Browse files- chatbot/memory.py +7 -2
chatbot/memory.py
CHANGED
@@ -1,3 +1,8 @@
|
|
1 |
-
from langchain.memory import
|
|
|
|
|
|
|
|
|
|
|
|
|
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 |
|
|