encounter/adk-python
0
0
Fork 0
mirror of https://github.com/encounter/adk-python.git synced 2026-07-09 18:19:28 -07:00
Code Issues Packages Projects Releases Wiki Activity

fix: Update API Registry Toolset to prod cloudapiregistry URL now that it is available

Browse Source 
Co-authored-by: Kathy Wu <wukathy@google.com>
PiperOrigin-RevId: 839547174
This commit is contained in:
Kathy Wu
2025-12-02 20:17:38 -08:00
committed by Copybara-Service
parent b807d62fe3
commit b638a48357
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/google/adk/tools/api_registry.py
+1 -2
View File
@@ -29,8 +29,7 @@ from .base_toolset import ToolPredicate
from .mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams from .mcp_tool.mcp_session_manager import StreamableHTTPConnectionParams
from .mcp_tool.mcp_toolset import McpToolset from .mcp_tool.mcp_toolset import McpToolset
# TODO(wukathy): Update to prod URL once it is available. API_REGISTRY_URL = "https://cloudapiregistry.googleapis.com"
API_REGISTRY_URL = "https://staging-cloudapiregistry.sandbox.googleapis.com"
class ApiRegistry: class ApiRegistry:
tests/unittests/tools/test_api_registry.py
+1 -1
View File
@@ -73,7 +73,7 @@ class TestApiRegistry(unittest.IsolatedAsyncioTestCase):
self.assertIn("test-mcp-server-2", api_registry._mcp_servers) self.assertIn("test-mcp-server-2", api_registry._mcp_servers)
self.assertIn("test-mcp-server-no-url", api_registry._mcp_servers) self.assertIn("test-mcp-server-no-url", api_registry._mcp_servers)
mock_client_instance.get.assert_called_once_with( 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={ headers={
"Authorization": "Bearer mock_token", "Authorization": "Bearer mock_token",
"Content-Type": "application/json", "Content-Type": "application/json",