|
|
@ -2,7 +2,7 @@ from __future__ import annotations
|
|
|
|
|
|
|
|
|
|
|
|
import logging
|
|
|
|
import logging
|
|
|
|
from enum import Enum
|
|
|
|
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.chat_history import BaseChatMessageHistory
|
|
|
|
from langchain_core.messages import (
|
|
|
|
from langchain_core.messages import (
|
|
|
@ -184,6 +184,38 @@ class ZepChatMessageHistory(BaseChatMessageHistory):
|
|
|
|
|
|
|
|
|
|
|
|
self.zep_client.memory.add_memory(self.session_id, zep_memory)
|
|
|
|
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(
|
|
|
|
def search(
|
|
|
|
self,
|
|
|
|
self,
|
|
|
|
query: str,
|
|
|
|
query: str,
|
|
|
@ -218,3 +250,15 @@ class ZepChatMessageHistory(BaseChatMessageHistory):
|
|
|
|
logger.warning(
|
|
|
|
logger.warning(
|
|
|
|
f"Session {self.session_id} not found in Zep. Skipping delete."
|
|
|
|
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."
|
|
|
|
|
|
|
|
)
|
|
|
|