From c589e7cafba8edded961812e42fde0836af25f69 Mon Sep 17 00:00:00 2001 From: cthomas Date: Tue, 21 Jan 2025 18:51:42 -0800 Subject: [PATCH] chore: rename /archival to /archival_memory (#719) --- letta/client/client.py | 12 +++++++----- letta/server/rest_api/routers/v1/agents.py | 6 +++--- 2 files changed, 10 insertions(+), 8 deletions(-) diff --git a/letta/client/client.py b/letta/client/client.py index a26d220c..ab7692f7 100644 --- a/letta/client/client.py +++ b/letta/client/client.py @@ -881,7 +881,9 @@ class RESTClient(AbstractClient): params["before"] = str(before) if after: params["after"] = str(after) - response = requests.get(f"{self.base_url}/{self.api_prefix}/agents/{str(agent_id)}/archival", params=params, headers=self.headers) + response = requests.get( + f"{self.base_url}/{self.api_prefix}/agents/{str(agent_id)}/archival_memory", params=params, headers=self.headers + ) assert response.status_code == 200, f"Failed to get archival memory: {response.text}" return [Passage(**passage) for passage in response.json()] @@ -898,7 +900,7 @@ class RESTClient(AbstractClient): """ request = CreateArchivalMemory(text=memory) response = requests.post( - f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/archival", headers=self.headers, json=request.model_dump() + f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/archival_memory", headers=self.headers, json=request.model_dump() ) if response.status_code != 200: raise ValueError(f"Failed to insert archival memory: {response.text}") @@ -912,7 +914,7 @@ class RESTClient(AbstractClient): agent_id (str): ID of the agent memory_id (str): ID of the memory """ - response = requests.delete(f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/archival/{memory_id}", headers=self.headers) + response = requests.delete(f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/archival_memory/{memory_id}", headers=self.headers) assert response.status_code == 200, f"Failed to delete archival memory: {response.text}" # messages (recall memory) @@ -1846,7 +1848,7 @@ class RESTClient(AbstractClient): memory (Memory): The updated memory """ response = requests.post( - f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/core_memory/block", + f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/core_memory/blocks", headers=self.headers, json=create_block.model_dump(), ) @@ -1904,7 +1906,7 @@ class RESTClient(AbstractClient): Returns: blocks (List[Block]): The blocks in the agent's core memory """ - response = requests.get(f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/core_memory/block", headers=self.headers) + response = requests.get(f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/core_memory/blocks", headers=self.headers) if response.status_code != 200: raise ValueError(f"Failed to get agent memory blocks: {response.text}") return [Block(**block) for block in response.json()] diff --git a/letta/server/rest_api/routers/v1/agents.py b/letta/server/rest_api/routers/v1/agents.py index 6e1a9232..caeb3fa7 100644 --- a/letta/server/rest_api/routers/v1/agents.py +++ b/letta/server/rest_api/routers/v1/agents.py @@ -328,7 +328,7 @@ def update_agent_memory_block( return block -@router.get("/{agent_id}/archival", response_model=List[Passage], operation_id="list_agent_archival_memory") +@router.get("/{agent_id}/archival_memory", response_model=List[Passage], operation_id="list_agent_archival_memory") def get_agent_archival_memory( agent_id: str, server: "SyncServer" = Depends(get_letta_server), @@ -354,7 +354,7 @@ def get_agent_archival_memory( ) -@router.post("/{agent_id}/archival", response_model=List[Passage], operation_id="create_agent_archival_memory") +@router.post("/{agent_id}/archival_memory", response_model=List[Passage], operation_id="create_agent_archival_memory") def insert_agent_archival_memory( agent_id: str, request: CreateArchivalMemory = Body(...), @@ -371,7 +371,7 @@ def insert_agent_archival_memory( # TODO(ethan): query or path parameter for memory_id? # @router.delete("/{agent_id}/archival") -@router.delete("/{agent_id}/archival/{memory_id}", response_model=None, operation_id="delete_agent_archival_memory") +@router.delete("/{agent_id}/archival_memory/{memory_id}", response_model=None, operation_id="delete_agent_archival_memory") def delete_agent_archival_memory( agent_id: str, memory_id: str,