Upload 60 files

gateway_mcp/registry.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import TYPE_CHECKING
 
+from fastmcp.tools import Tool
 from pydantic import BaseModel
 
 from gateway_mcp.schemas import (
@@ -154,11 +155,13 @@ class ToolRegistry:
             handler = getattr(handlers, definition.handler_name, None)
             if not callable(handler):
                 raise ValueError(f"Missing MCP handler: {definition.handler_name}")
-            registered = server.tool(
-                handler,
-                name=definition.name,
-                description=definition.description,
-                output_schema=definition.output_schema,
+            registered = server.add_tool(
+                Tool.from_function(
+                    handler,
+                    name=definition.name,
+                    description=definition.description,
+                    output_schema=definition.output_schema,
+                )
             )
             actual = set(registered.parameters.get("properties", {}))
             expected = set(definition.input_schema.get("properties", {}))