aditya
ee82d6101e
Implement MCPToolAdapter to connect to external MCP servers, enumerate tools, and register them in ToolRegistry with source="mcp". Includes connect/reconnect/disconnect lifecycle with timeout enforcement, input validation on invoke, capability inference, Behave/Robot/ASV tests, and docs/reference/mcp_adapter.md. ISSUES CLOSED: #159
147 lines
4.4 KiB
Python
147 lines
4.4 KiB
Python
"""Robot Framework helper for MCP Tool Adapter smoke tests.
|
|
|
|
Usage:
|
|
python robot/helper_mcp_adapter.py create-adapter
|
|
python robot/helper_mcp_adapter.py discover-tools
|
|
python robot/helper_mcp_adapter.py invoke-tool
|
|
python robot/helper_mcp_adapter.py register-tools
|
|
python robot/helper_mcp_adapter.py reject-no-command
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
_SRC = str(Path(__file__).resolve().parents[1] / "src")
|
|
if _SRC not in sys.path:
|
|
sys.path.insert(0, _SRC)
|
|
|
|
from cleveragents.mcp.adapter import ( # noqa: E402
|
|
MCPServerConfig,
|
|
MCPToolAdapter,
|
|
MCPTransport,
|
|
)
|
|
from cleveragents.tool.registry import ToolRegistry # noqa: E402
|
|
|
|
|
|
class _MockTransport(MCPTransport):
|
|
"""Minimal mock transport for smoke tests."""
|
|
|
|
def __init__(self, tools: list[dict[str, Any]] | None = None) -> None:
|
|
self._tools = tools or []
|
|
self._results: dict[str, dict[str, Any]] = {}
|
|
|
|
def connect(self, config: MCPServerConfig) -> dict[str, Any]:
|
|
return {"capabilities": {"tools": True}}
|
|
|
|
def call(self, method: str, params: dict[str, Any]) -> dict[str, Any]:
|
|
if method == "tools/list":
|
|
return {"tools": list(self._tools)}
|
|
if method == "tools/call":
|
|
name = params.get("name", "")
|
|
if name in self._results:
|
|
return {"content": self._results[name]}
|
|
return {"content": {"result": "ok"}}
|
|
return {}
|
|
|
|
def close(self) -> None:
|
|
pass
|
|
|
|
|
|
def _mock_tool(name: str) -> dict[str, Any]:
|
|
return {"name": name, "description": f"Mock {name}", "inputSchema": {}}
|
|
|
|
|
|
def main() -> int:
|
|
if len(sys.argv) < 2:
|
|
print("Usage: helper_mcp_adapter.py <command>")
|
|
return 1
|
|
|
|
dispatch = {
|
|
"create-adapter": _create_adapter,
|
|
"discover-tools": _discover_tools,
|
|
"invoke-tool": _invoke_tool,
|
|
"register-tools": _register_tools,
|
|
"reject-no-command": _reject_no_command,
|
|
}
|
|
handler = dispatch.get(sys.argv[1])
|
|
if handler is None:
|
|
print(f"Unknown command: {sys.argv[1]}")
|
|
return 1
|
|
return handler()
|
|
|
|
|
|
def _create_adapter() -> int:
|
|
config = MCPServerConfig(name="github", transport="stdio", command="echo")
|
|
adapter = MCPToolAdapter(config=config)
|
|
print(f"server-name: {adapter.server_name}")
|
|
print(f"transport: {adapter.transport_type}")
|
|
print(f"connected: {adapter.is_connected}")
|
|
return 0
|
|
|
|
|
|
def _discover_tools() -> int:
|
|
tools = [
|
|
_mock_tool("create_issue"),
|
|
_mock_tool("list_repos"),
|
|
_mock_tool("search"),
|
|
]
|
|
config = MCPServerConfig(name="test", transport="stdio", command="echo")
|
|
transport = _MockTransport(tools=tools)
|
|
adapter = MCPToolAdapter(config=config, transport=transport)
|
|
adapter.connect()
|
|
discovered = adapter.discover_tools()
|
|
print(f"discovered: {len(discovered)}")
|
|
for i, t in enumerate(discovered):
|
|
print(f"tool-{i}: {t.name}")
|
|
return 0
|
|
|
|
|
|
def _invoke_tool() -> int:
|
|
tools = [_mock_tool("create_issue")]
|
|
config = MCPServerConfig(name="test", transport="stdio", command="echo")
|
|
transport = _MockTransport(tools=tools)
|
|
transport._results["create_issue"] = {"id": 42}
|
|
adapter = MCPToolAdapter(config=config, transport=transport)
|
|
adapter.connect()
|
|
adapter.discover_tools()
|
|
result = adapter.invoke("create_issue", {"title": "Bug"})
|
|
print(f"success: {result.success}")
|
|
print(f"has-data: {bool(result.data)}")
|
|
return 0
|
|
|
|
|
|
def _register_tools() -> int:
|
|
tools = [_mock_tool("tool_a"), _mock_tool("tool_b")]
|
|
config = MCPServerConfig(name="test", transport="stdio", command="echo")
|
|
transport = _MockTransport(tools=tools)
|
|
adapter = MCPToolAdapter(config=config, transport=transport)
|
|
adapter.connect()
|
|
registry = ToolRegistry()
|
|
names = adapter.register_tools(registry, namespace="mcp-test")
|
|
print(f"registered: {len(names)}")
|
|
if names and "/" in names[0]:
|
|
print(f"prefix: {names[0].split('/')[0]}/")
|
|
return 0
|
|
|
|
|
|
def _reject_no_command() -> int:
|
|
config = MCPServerConfig(name="bad", transport="stdio")
|
|
try:
|
|
MCPToolAdapter.from_server_config(config)
|
|
print("unexpected-success")
|
|
return 1
|
|
except ValueError as exc:
|
|
err = str(exc).lower()
|
|
if "command" in err:
|
|
print("validation-error: command")
|
|
else:
|
|
print(f"validation-error-other: {exc}")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|