📦 chore(ZepMemory.py): add ZepMemory component to handle interactions with Zep API

The ZepMemory component is added to handle interactions with the Zep API. It provides a build method that takes in configuration parameters such as zep_api_url, api_key, session_id, memory_key, and return_messages. These parameters are used to initialize an instance of the ZepMemory class, which is a subclass of BaseMemory. The ZepMemory class handles the communication with the Zep API and provides methods for storing and retrieving data from the Zep memory.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-22 20:38:41 -03:00
commit da694a1d9b

View file

@ -0,0 +1,39 @@
from langchain.memory.zep_memory import ZepMemory
from langflow import CustomComponent
from langchain.schema.memory import BaseMemory
class ZepMemoryComponent(CustomComponent):
display_name: str = "Zep Memory"
def build_config(self):
return {
"zep_api_url": {
"display_name": "Zep API URL",
"value": "http://localhost:8000",
},
"api_key": {
"password": True,
"display_name": "API Key",
},
"session_id": {
"display_name": "Session ID",
"info": "The session ID to use for the memory.",
},
}
def build(
self,
api_key: str,
session_id: str,
memory_key: str,
return_messages: bool,
zep_api_url: str = "http://localhost:8000",
) -> BaseMemory:
return ZepMemory(
session_id=session_id,
url=zep_api_url,
api_key=api_key,
memory_key=memory_key,
return_messages=return_messages,
)