mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-07-22 01:05:52 -04:00
Daniel Chalef
c7f7788d0b
Hey @hwchase17 - This PR adds a `ZepMemory` class, improves handling of Zep's message metadata, and makes it easier for folks building custom chains to persist metadata alongside their chat history. We've had plenty confused users unfamiliar with ChatMessageHistory classes and how to wrap the `ZepChatMessageHistory` in a `ConversationBufferMemory`. So we've created the `ZepMemory` class as a light wrapper for `ZepChatMessageHistory`. Details: - add ZepMemory, modify notebook to demo use of ZepMemory - Modify summary to be SystemMessage - add metadata argument to add_message; add Zep metadata to Message.additional_kwargs - support passing in metadata
66 lines
2.1 KiB
Python
66 lines
2.1 KiB
Python
from langchain.memory.buffer import (
|
|
ConversationBufferMemory,
|
|
ConversationStringBufferMemory,
|
|
)
|
|
from langchain.memory.buffer_window import ConversationBufferWindowMemory
|
|
from langchain.memory.chat_message_histories import (
|
|
CassandraChatMessageHistory,
|
|
ChatMessageHistory,
|
|
CosmosDBChatMessageHistory,
|
|
DynamoDBChatMessageHistory,
|
|
FileChatMessageHistory,
|
|
MomentoChatMessageHistory,
|
|
MongoDBChatMessageHistory,
|
|
PostgresChatMessageHistory,
|
|
RedisChatMessageHistory,
|
|
SQLChatMessageHistory,
|
|
ZepChatMessageHistory,
|
|
)
|
|
from langchain.memory.combined import CombinedMemory
|
|
from langchain.memory.entity import (
|
|
ConversationEntityMemory,
|
|
InMemoryEntityStore,
|
|
RedisEntityStore,
|
|
SQLiteEntityStore,
|
|
)
|
|
from langchain.memory.kg import ConversationKGMemory
|
|
from langchain.memory.motorhead_memory import MotorheadMemory
|
|
from langchain.memory.readonly import ReadOnlySharedMemory
|
|
from langchain.memory.simple import SimpleMemory
|
|
from langchain.memory.summary import ConversationSummaryMemory
|
|
from langchain.memory.summary_buffer import ConversationSummaryBufferMemory
|
|
from langchain.memory.token_buffer import ConversationTokenBufferMemory
|
|
from langchain.memory.vectorstore import VectorStoreRetrieverMemory
|
|
from langchain.memory.zep_memory import ZepMemory
|
|
|
|
__all__ = [
|
|
"CassandraChatMessageHistory",
|
|
"ChatMessageHistory",
|
|
"CombinedMemory",
|
|
"ConversationBufferMemory",
|
|
"ConversationBufferWindowMemory",
|
|
"ConversationEntityMemory",
|
|
"ConversationKGMemory",
|
|
"ConversationStringBufferMemory",
|
|
"ConversationSummaryBufferMemory",
|
|
"ConversationSummaryMemory",
|
|
"ConversationTokenBufferMemory",
|
|
"CosmosDBChatMessageHistory",
|
|
"DynamoDBChatMessageHistory",
|
|
"FileChatMessageHistory",
|
|
"InMemoryEntityStore",
|
|
"MomentoChatMessageHistory",
|
|
"MongoDBChatMessageHistory",
|
|
"MotorheadMemory",
|
|
"PostgresChatMessageHistory",
|
|
"ReadOnlySharedMemory",
|
|
"RedisChatMessageHistory",
|
|
"RedisEntityStore",
|
|
"SQLChatMessageHistory",
|
|
"SQLiteEntityStore",
|
|
"SimpleMemory",
|
|
"VectorStoreRetrieverMemory",
|
|
"ZepChatMessageHistory",
|
|
"ZepMemory",
|
|
]
|