From fc23969b89e3ff9e5bcc39b9094ff9f08b41011d Mon Sep 17 00:00:00 2001 From: Sarah Wooders Date: Sat, 6 Jul 2024 09:21:23 -0700 Subject: [PATCH 1/2] remove duplicate functions --- memgpt/client/client.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/memgpt/client/client.py b/memgpt/client/client.py index d37aad03ae..5af4e8ec1c 100644 --- a/memgpt/client/client.py +++ b/memgpt/client/client.py @@ -698,15 +698,6 @@ def __init__( # create user if does not exist self.server.create_user({"id": self.user_id}, exists_ok=True) - # messages - def send_message(self, agent_id: uuid.UUID, message: str, role: str, stream: Optional[bool] = False) -> UserMessageResponse: - self.interface.clear() - usage = self.server.user_message(user_id=self.user_id, agent_id=agent_id, message=message) - if self.auto_save: - self.save() - else: - return UserMessageResponse(messages=self.interface.to_list(), usage=usage) - # agents def list_agents(self): @@ -804,8 +795,9 @@ def update_agent_core_memory(self, agent_id: str, new_memory_contents: Dict) -> # agent interactions - def send_message(self, agent_id: uuid.UUID, message: str, role: str, stream: Optional[bool] = False) -> UserMessageResponse: + def send_message(self, agent_id: uuid.UUID, message: str, role: str) -> UserMessageResponse: self.interface.clear() + # TODO: implement streaming with stream=True/False if role == "system": usage = self.server.system_message(user_id=self.user_id, agent_id=agent_id, message=message) elif role == "user": From b9c2d70436e7b2e3df73f5ca8cd1de42fb1e64e1 Mon Sep 17 00:00:00 2001 From: Sarah Wooders Date: Sun, 7 Jul 2024 10:11:47 -0700 Subject: [PATCH 2/2] minore fix --- memgpt/client/client.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/memgpt/client/client.py b/memgpt/client/client.py index 5af4e8ec1c..2b2bc316c6 100644 --- a/memgpt/client/client.py +++ b/memgpt/client/client.py @@ -795,9 +795,11 @@ def update_agent_core_memory(self, agent_id: str, new_memory_contents: Dict) -> # agent interactions - def send_message(self, agent_id: uuid.UUID, message: str, role: str) -> UserMessageResponse: + def send_message(self, agent_id: uuid.UUID, message: str, role: str, stream: Optional[bool] = False) -> UserMessageResponse: + if stream: + # TODO: implement streaming with stream=True/False + raise NotImplementedError self.interface.clear() - # TODO: implement streaming with stream=True/False if role == "system": usage = self.server.system_message(user_id=self.user_id, agent_id=agent_id, message=message) elif role == "user":