Kumiko_v1 / chatbot /memory.py
anhkhoiphan's picture
Use combined memory
ed3a456 verified
raw
history blame
329 Bytes
from langchain.memory import CombinedMemory, ConversationBufferWindowMemory, ConversationSummaryMemory
from chatbot.llm import gemini_llm
buffer_memory = ConversationBufferWindowMemory(k=5)
summary_memory = ConversationSummaryMemory(llm=gemini_llm)
combined_memory = CombinedMemory(memories=[buffer_memory, summary_memory])