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:
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
+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_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:
+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-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",