diff --git a/letta/client/client.py b/letta/client/client.py index 5ebab3b03a..8a9d3e700a 100644 --- 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 index 3259d18c7b..15979346c4 100644 --- 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 index 1ddab15e83..2865bb2ec4 100644 --- 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