sync: fix: keep MCP server cards aligned with registered tools (#735)

Synced from monorepo directory: wan

Author: acedatacloud-dev

core/server_card.py

@@ -0,0 +1,9 @@
+"""Build public server cards from the registered MCP tools."""
+
+from typing import Any
+
+
+def registered_tool_cards(mcp: Any) -> list[dict[str, str]]:
+    """Return a stable public view of the registered tools."""
+    tools = sorted(mcp._tool_manager.list_tools(), key=lambda tool: tool.name)
+    return [{"name": tool.name, "description": tool.description or ""} for tool in tools]

main.py

@@ -18,6 +18,7 @@
 
 from core.config import settings
 from core.server import mcp
+from core.server_card import registered_tool_cards
 
 # Configure logging
 logging.basicConfig(
@@ -163,33 +164,7 @@ async def server_card(_request: Request) -> JSONResponse:
                     {
                         "serverInfo": {"name": "MCP Wan"},
                         "authentication": {"required": True, "schemes": ["bearer"]},
-                        "tools": [
-                            {
-                                "name": "wan_generate_video",
-                                "description": "Generate video from text",
-                            },
-                            {
-                                "name": "wan_generate_video_from_image",
-                                "description": "Generate video from image",
-                            },
-                            {"name": "wan_get_task", "description": "Query task status"},
-                            {
-                                "name": "wan_get_tasks_batch",
-                                "description": "Query multiple tasks",
-                            },
-                            {
-                                "name": "wan_list_models",
-                                "description": "List available models",
-                            },
-                            {
-                                "name": "wan_list_resolutions",
-                                "description": "List available resolutions",
-                            },
-                            {
-                                "name": "wan_list_actions",
-                                "description": "List available actions",
-                            },
-                        ],
+                        "tools": registered_tool_cards(mcp),
                         "prompts": [
                             {
                                 "name": "wan_video_generation_guide",

tests/test_server_card.py

@@ -0,0 +1,21 @@
+"""Server-card tools must match the MCP registry exactly."""
+
+from pathlib import Path
+
+import tools  # noqa: F401, I001
+from core.server import mcp
+from core.server_card import registered_tool_cards
+
+
+def test_server_card_tools_match_registered_tools():
+    registered = {tool.name for tool in mcp._tool_manager.list_tools()}
+    advertised = {tool["name"] for tool in registered_tool_cards(mcp)}
+
+    assert advertised == registered
+    assert len(advertised) == len(registered_tool_cards(mcp))
+
+
+def test_http_server_card_uses_registered_tools():
+    source = (Path(__file__).parents[1] / "main.py").read_text()
+
+    assert '"tools": registered_tool_cards(mcp)' in source