From f7fdf7cc8f0a54e499cfa1676e8334f0eaaa3557 Mon Sep 17 00:00:00 2001 From: Charles Packer Date: Thu, 5 Sep 2024 11:06:31 -0700 Subject: [PATCH] fix: Add remove tool from Agent API (#1714) --- memgpt/server/rest_api/tools/index.py | 2 +- memgpt/server/server.py | 21 --------------------- 2 files changed, 1 insertion(+), 22 deletions(-) diff --git a/memgpt/server/rest_api/tools/index.py b/memgpt/server/rest_api/tools/index.py index b5c20d7d..6072b133 100644 --- a/memgpt/server/rest_api/tools/index.py +++ b/memgpt/server/rest_api/tools/index.py @@ -24,7 +24,7 @@ def setup_user_tools_index_router(server: SyncServer, interface: QueuingInterfac """ # Clear the interface interface.clear() - server.delete_tool(id) + server.delete_tool(tool_id) @router.get("/tools/{tool_id}", tags=["tools"], response_model=Tool) async def get_tool( diff --git a/memgpt/server/server.py b/memgpt/server/server.py index 4610128f..4fa7869a 100644 --- a/memgpt/server/server.py +++ b/memgpt/server/server.py @@ -780,27 +780,6 @@ class SyncServer(Server): # TODO: probably reload the agent somehow? return memgpt_agent.agent_state - def delete_agent( - self, - user_id: str, - agent_id: str, - ): - # TODO: delete agent data - - if self.ms.get_user(user_id=user_id) is None: - raise ValueError(f"User user_id={user_id} does not exist") - if self.ms.get_agent(agent_id=agent_id, user_id=user_id) is None: - raise ValueError(f"Agent agent_id={agent_id} does not exist") - - # TODO: delete related tables (recall/archival memory) - - # TODO: Make sure the user owns the agent - # TODO: consider clean up any blocks associated with this agent - # and no other agents (e.g. not templates, but not used as shared blocks either) - agent = self.ms.get_agent(agent_id=agent_id, user_id=user_id) - if agent is not None: - self.ms.delete_agent(agent_id=agent_id) - def _agent_state_to_config(self, agent_state: AgentState) -> dict: """Convert AgentState to a dict for a JSON response""" assert agent_state is not None