Fix tests

letta/client/client.py

@@ -233,7 +233,7 @@ def delete_tool(self, id: str):
     def get_tool_id(self, name: str) -> Optional[str]:
         raise NotImplementedError
 
-    def add_base_tools(self) -> List[Tool]:
+    def upsert_base_tools(self) -> List[Tool]:
         raise NotImplementedError
 
     def load_data(self, connector: DataConnector, source_name: str):
@@ -1466,7 +1466,7 @@ def get_tool_id(self, tool_name: str):
             raise ValueError(f"Failed to get tool: {response.text}")
         return response.json()
 
-    def add_base_tools(self) -> List[Tool]:
+    def upsert_base_tools(self) -> List[Tool]:
         response = requests.post(f"{self.base_url}/{self.api_prefix}/tools/upsert-base-tools/", headers=self.headers)
         if response.status_code != 200:
             raise ValueError(f"Failed to add base tools: {response.text}")

tests/test_managers.py

@@ -1461,7 +1461,7 @@ def test_delete_tool_by_id(server: SyncServer, print_tool, default_user):
     assert len(tools) == 0
 
 
-def test_add_base_tools(server: SyncServer, default_user):
+def test_upsert_base_tools(server: SyncServer, default_user):
     tools = server.tool_manager.upsert_base_tools(actor=default_user)
     expected_tool_names = sorted(BASE_TOOLS + BASE_MEMORY_TOOLS)
     assert sorted([t.name for t in tools]) == expected_tool_names

tests/test_v1_routes.py

@@ -272,7 +272,7 @@ def test_update_tool(client, mock_sync_server, update_integers_tool, add_integer
     )
 
 
-def test_add_base_tools(client, mock_sync_server, add_integers_tool):
+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"})