mirror of
https://github.com/encounter/adk-python.git
synced 2026-07-09 18:19:28 -07:00
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
This commit is contained in:
committed by
Copybara-Service
parent
b807d62fe3
commit
b638a48357
@@ -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:
|
||||
|
||||
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user