Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
cthomas
2025-01-21 18:51:42 -08:00
committed by GitHub
parent cabb04c005
commit c589e7cafb
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
letta/client/client.py
View File

@@ -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()]

6
letta/server/rest_api/routers/v1/agents.py
View File

@@ -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,