Skip to content

Commit a3fefcf

Browse files
committed
fix: support new and old hud domain for MCP patching
1 parent 57ad2ac commit a3fefcf

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

hud/utils/mcp.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,23 @@ class MCPConfigPatch(BaseModel):
1717
meta: dict[str, Any] | None = Field(default_factory=dict, alias="meta")
1818

1919

20+
def _is_hud_server(url: str) -> bool:
21+
"""Check if a URL is a HUD MCP server.
22+
23+
Matches any mcp.hud.* domain (including .ai, .so, and future domains).
24+
"""
25+
if not url:
26+
return False
27+
return "mcp.hud." in url.lower()
28+
29+
2030
def patch_mcp_config(mcp_config: dict[str, dict[str, Any]], patch: MCPConfigPatch) -> None:
2131
"""Patch MCP config with additional values."""
22-
hud_mcp_url = settings.hud_mcp_url
23-
2432
for server_cfg in mcp_config.values():
2533
url = server_cfg.get("url", "")
2634

2735
# 1) HTTP header lane (only for hud MCP servers)
28-
if hud_mcp_url in url and patch.headers:
36+
if _is_hud_server(url) and patch.headers:
2937
for key, value in patch.headers.items():
3038
headers = server_cfg.setdefault("headers", {})
3139
headers.setdefault(key, value)
@@ -50,9 +58,8 @@ def setup_hud_telemetry(
5058
raise ValueError("Please run initialize() before setting up client-side telemetry")
5159

5260
# Check if there are any HUD servers to setup telemetry for
53-
hud_mcp_url = settings.hud_mcp_url
5461
has_hud_servers = any(
55-
hud_mcp_url in server_cfg.get("url", "") for server_cfg in mcp_config.values()
62+
_is_hud_server(server_cfg.get("url", "")) for server_cfg in mcp_config.values()
5663
)
5764

5865
# If no HUD servers, no need for telemetry setup

0 commit comments

Comments
 (0)