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

Revert "feat: add route for editing a memory blocks limit (via Memory, instead of via Blocks)" (#2085)

Browse Source
This commit is contained in:
Charles Packer
2024-11-21 20:08:17 -08:00
committed by GitHub
parent 17e39a57fe
commit 06744c9193
7 changed files with 3 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
letta/client/client.py
View File

@@ -1565,18 +1565,6 @@ class RESTClient(AbstractClient):
# Parse and return the deleted organization
return Organization(**response.json())
def update_agent_memory_limit(self, agent_id: str, block_label: str, limit: int) -> Memory:
# @router.patch("/{agent_id}/memory/limit", response_model=Memory, operation_id="update_agent_memory_limit")
response = requests.patch(
f"{self.base_url}/{self.api_prefix}/agents/{agent_id}/memory/limit",
headers=self.headers,
json={"label": block_label, "limit": limit},
)
if response.status_code != 200:
raise ValueError(f"Failed to update agent memory limit: {response.text}")
return Memory(**response.json())
class LocalClient(AbstractClient):
"""
@@ -2785,6 +2773,3 @@ class LocalClient(AbstractClient):
def delete_org(self, org_id: str) -> Organization:
return self.server.organization_manager.delete_organization_by_id(org_id=org_id)
def update_agent_memory_limit(self, agent_id: str, block_label: str, limit: int) -> Memory:
return self.server.update_agent_memory_limit(user_id=self.user_id, agent_id=agent_id, block_label=block_label, limit=limit)