chore: rename /archival to /archival_memory (#719)

letta-ai · Jan 22, 2025 · 434b94f · 434b94f
1 parent b0e7291
commit 434b94f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/letta/client/client.py b/letta/client/client.py
@@ -881,7 +881,9 @@ def get_archival_memory(
             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 @@ def insert_archival_memory(self, agent_id: str, memory: str) -> List[Passage]:
         """
         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 @@ def delete_archival_memory(self, agent_id: str, memory_id: str):
             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 @@ def add_agent_memory_block(self, agent_id: str, create_block: CreateBlock) -> Me
             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 @@ def list_agent_memory_blocks(self, agent_id: str) -> List[Block]:
         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
@@ -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,