Skip to content

Commit b638a48

Browse files
wukathcopybara-github
authored andcommitted
fix: Update API Registry Toolset to prod cloudapiregistry URL now that it is available
Co-authored-by: Kathy Wu <wukathy@google.com> PiperOrigin-RevId: 839547174
1 parent b807d62 commit b638a48

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/google/adk/tools/api_registry.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,8 +29,7 @@
2929
from .mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams
3030
from .mcp_tool.mcp_toolset import McpToolset
3131

32-
# TODO(wukathy): Update to prod URL once it is available.
33-
API_REGISTRY_URL = "https://staging-cloudapiregistry.sandbox.googleapis.com"
32+
API_REGISTRY_URL = "https://cloudapiregistry.googleapis.com"
3433

3534

3635
class ApiRegistry:

tests/unittests/tools/test_api_registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ def test_init_success(self, MockHttpClient):
7373
self.assertIn("test-mcp-server-2", api_registry._mcp_servers)
7474
self.assertIn("test-mcp-server-no-url", api_registry._mcp_servers)
7575
mock_client_instance.get.assert_called_once_with(
76-
f"https://staging-cloudapiregistry.sandbox.googleapis.com/v1beta/projects/{self.project_id}/locations/{self.location}/mcpServers",
76+
f"https://cloudapiregistry.googleapis.com/v1beta/projects/{self.project_id}/locations/{self.location}/mcpServers",
7777
headers={
7878
"Authorization": "Bearer mock_token",
7979
"Content-Type": "application/json",

0 commit comments

Comments
 (0)