@@ -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+
2030def 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