mirror of
https://github.com/hwchase17/langchain.git
synced 2025-06-24 15:43:54 +00:00
zep[patch]: implement add_messages and aadd_messages (#20099)
This PR implement `add_messages` and `aadd_messages` to avoid unnecessary round-trips.
This commit is contained in:
parent
55dd349472
commit
80679ab906
@ -2,7 +2,7 @@ from __future__ import annotations
|
||||
|
||||
import logging
|
||||
from enum import Enum
|
||||
from typing import TYPE_CHECKING, Any, Dict, List, Optional
|
||||
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Sequence
|
||||
|
||||
from langchain_core.chat_history import BaseChatMessageHistory
|
||||
from langchain_core.messages import (
|
||||
@ -184,6 +184,38 @@ class ZepChatMessageHistory(BaseChatMessageHistory):
|
||||
|
||||
self.zep_client.memory.add_memory(self.session_id, zep_memory)
|
||||
|
||||
def add_messages(self, messages: Sequence[BaseMessage]) -> None:
|
||||
"""Append the messages to the Zep memory history"""
|
||||
from zep_python import Memory, Message
|
||||
|
||||
zep_messages = [
|
||||
Message(
|
||||
content=message.content,
|
||||
role=message.type,
|
||||
metadata=message.additional_kwargs.get("metadata", None),
|
||||
)
|
||||
for message in messages
|
||||
]
|
||||
zep_memory = Memory(messages=zep_messages)
|
||||
|
||||
self.zep_client.memory.add_memory(self.session_id, zep_memory)
|
||||
|
||||
async def aadd_messages(self, messages: Sequence[BaseMessage]) -> None:
|
||||
"""Append the messages to the Zep memory history asynchronously"""
|
||||
from zep_python import Memory, Message
|
||||
|
||||
zep_messages = [
|
||||
Message(
|
||||
content=message.content,
|
||||
role=message.type,
|
||||
metadata=message.additional_kwargs.get("metadata", None),
|
||||
)
|
||||
for message in messages
|
||||
]
|
||||
zep_memory = Memory(messages=zep_messages)
|
||||
|
||||
await self.zep_client.memory.aadd_memory(self.session_id, zep_memory)
|
||||
|
||||
def search(
|
||||
self,
|
||||
query: str,
|
||||
@ -218,3 +250,15 @@ class ZepChatMessageHistory(BaseChatMessageHistory):
|
||||
logger.warning(
|
||||
f"Session {self.session_id} not found in Zep. Skipping delete."
|
||||
)
|
||||
|
||||
async def aclear(self) -> None:
|
||||
"""Clear session memory from Zep asynchronously.
|
||||
Note that Zep is long-term storage for memory and this is not advised
|
||||
unless you have specific data retention requirements.
|
||||
"""
|
||||
try:
|
||||
await self.zep_client.memory.adelete_memory(self.session_id)
|
||||
except NotFoundError:
|
||||
logger.warning(
|
||||
f"Session {self.session_id} not found in Zep. Skipping delete."
|
||||
)
|
||||
|
Loading…
Reference in New Issue
Block a user