Try to pass letta compat tests

letta-ai · Dec 18, 2024 · ba13fd4 · ba13fd4
1 parent d09efac
commit ba13fd4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/letta/client/client.py b/letta/client/client.py
@@ -1467,7 +1467,7 @@ def get_tool_id(self, tool_name: str):
         return response.json()
 
     def upsert_base_tools(self) -> List[Tool]:
-        response = requests.post(f"{self.base_url}/{self.api_prefix}/tools/upsert-base-tools/", headers=self.headers)
+        response = requests.post(f"{self.base_url}/{self.api_prefix}/tools/add-base-tools/", headers=self.headers)
         if response.status_code != 200:
             raise ValueError(f"Failed to add base tools: {response.text}")
 

diff --git a/letta/server/rest_api/routers/v1/tools.py b/letta/server/rest_api/routers/v1/tools.py
@@ -151,7 +151,7 @@ def update_tool(
     return server.tool_manager.update_tool_by_id(tool_id=tool_id, tool_update=request, actor=actor)
 
 
-@router.post("/upsert-base-tools", response_model=List[Tool], operation_id="upsert_base_tools")
+@router.post("/add-base-tools", response_model=List[Tool], operation_id="add_base_tools")
 def upsert_base_tools(
     server: SyncServer = Depends(get_letta_server),
     user_id: Optional[str] = Header(None, alias="user_id"),  # Extract user_id from header, default to None if not present

diff --git a/tests/test_v1_routes.py b/tests/test_v1_routes.py
@@ -275,7 +275,7 @@ def test_update_tool(client, mock_sync_server, update_integers_tool, add_integer
 def test_upsert_base_tools(client, mock_sync_server, add_integers_tool):
     mock_sync_server.tool_manager.upsert_base_tools.return_value = [add_integers_tool]
 
-    response = client.post("/v1/tools/upsert-base-tools", headers={"user_id": "test_user"})
+    response = client.post("/v1/tools/add-base-tools", headers={"user_id": "test_user"})
 
     assert response.status_code == 200
     assert len(response.json()) == 1