temp / DV-AGENT /memory.py
NEXAS's picture
Upload 22 files
182219d verified
raw
history blame contribute delete
No virus
330 Bytes
from langchain.memory import ConversationBufferWindowMemory
from langchain_community.chat_message_histories import StreamlitChatMessageHistory
memory_storage = StreamlitChatMessageHistory(key="chat_messages")
memory = ConversationBufferWindowMemory(memory_key="chat_history", human_prefix="User", chat_memory=memory_storage, k=3)