From b638a48357c9fd5089b4d59dcd3f2102d1d2b3b8 Mon Sep 17 00:00:00 2001 From: Kathy Wu Date: Tue, 2 Dec 2025 20:16:58 -0800 Subject: [PATCH] fix: Update API Registry Toolset to prod cloudapiregistry URL now that it is available Co-authored-by: Kathy Wu PiperOrigin-RevId: 839547174 --- src/google/adk/tools/api_registry.py | 3 +-- tests/unittests/tools/test_api_registry.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/google/adk/tools/api_registry.py b/src/google/adk/tools/api_registry.py index 941c6f0d..e3f00764 100644 --- a/src/google/adk/tools/api_registry.py +++ b/src/google/adk/tools/api_registry.py @@ -29,8 +29,7 @@ from .base_toolset import ToolPredicate from .mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams from .mcp_tool.mcp_toolset import McpToolset -# TODO(wukathy): Update to prod URL once it is available. -API_REGISTRY_URL = "https://staging-cloudapiregistry.sandbox.googleapis.com" +API_REGISTRY_URL = "https://cloudapiregistry.googleapis.com" class ApiRegistry: diff --git a/tests/unittests/tools/test_api_registry.py b/tests/unittests/tools/test_api_registry.py index d1131eed..df547860 100644 --- a/tests/unittests/tools/test_api_registry.py +++ b/tests/unittests/tools/test_api_registry.py @@ -73,7 +73,7 @@ class TestApiRegistry(unittest.IsolatedAsyncioTestCase): self.assertIn("test-mcp-server-2", api_registry._mcp_servers) self.assertIn("test-mcp-server-no-url", api_registry._mcp_servers) mock_client_instance.get.assert_called_once_with( - f"https://staging-cloudapiregistry.sandbox.googleapis.com/v1beta/projects/{self.project_id}/locations/{self.location}/mcpServers", + f"https://cloudapiregistry.googleapis.com/v1beta/projects/{self.project_id}/locations/{self.location}/mcpServers", headers={ "Authorization": "Bearer mock_token", "Content-Type": "application/json",