Create app/routes/mcp.py

app/routes/mcp.py

@@ -0,0 +1,108 @@
+"""MCP protocol endpoints - SSE and JSON-RPC style."""
+
+from __future__ import annotations
+
+import json
+import logging
+from typing import Any
+
+from fastapi import APIRouter, Request
+from fastapi.responses import JSONResponse
+from sse_starlette.sse import EventSourceResponse
+
+from app.mcp_handler import TOOL_DEFINITIONS, handle_tool_call
+from app.models import MCPListToolsResponse, MCPToolInput, MCPToolResult
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter(prefix="/mcp", tags=["mcp"])
+
+
+@router.get("/tools")
+async def list_tools() -> MCPListToolsResponse:
+    """List all available MCP tools."""
+    return MCPListToolsResponse(tools=TOOL_DEFINITIONS)
+
+
+@router.post("/execute", response_model=MCPToolResult)
+async def execute_tool(tool_input: MCPToolInput) -> MCPToolResult:
+    """Execute an MCP tool call."""
+    logger.info("MCP tool call: %s", tool_input.name)
+    result = await handle_tool_call(tool_input.name, tool_input.arguments)
+    return result
+
+
+@router.post("/execute/stream")
+async def execute_tool_stream(tool_input: MCPToolInput) -> EventSourceResponse:
+    """Execute an MCP tool call with SSE streaming response."""
+
+    async def event_generator():
+        yield {
+            "event": "start",
+            "data": json.dumps({"tool": tool_input.name, "status": "running"}),
+        }
+
+        result = await handle_tool_call(tool_input.name, tool_input.arguments)
+
+        yield {
+            "event": "result",
+            "data": json.dumps(result.model_dump()),
+        }
+
+        yield {
+            "event": "done",
+            "data": json.dumps({"status": "completed"}),
+        }
+
+    return EventSourceResponse(event_generator())
+
+
+# ─── JSON-RPC 2.0 Compatible Endpoint ─────────────────────────────
+
+@router.post("/jsonrpc")
+async def jsonrpc_handler(request: Request) -> JSONResponse:
+    """Handle JSON-RPC 2.0 style MCP requests."""
+    try:
+        body = await request.json()
+    except Exception:
+        return JSONResponse(
+            status_code=400,
+            content={
+                "jsonrpc": "2.0",
+                "error": {"code": -32700, "message": "Parse error"},
+                "id": None,
+            },
+        )
+
+    request_id = body.get("id")
+    method = body.get("method", "")
+    params = body.get("params", {})
+
+    if method == "tools/list":
+        return JSONResponse(
+            content={
+                "jsonrpc": "2.0",
+                "result": {"tools": TOOL_DEFINITIONS},
+                "id": request_id,
+            }
+        )
+    elif method == "tools/call":
+        tool_name = params.get("name", "")
+        tool_args = params.get("arguments", {})
+        result = await handle_tool_call(tool_name, tool_args)
+        return JSONResponse(
+            content={
+                "jsonrpc": "2.0",
+                "result": result.model_dump(),
+                "id": request_id,
+            }
+        )
+    else:
+        return JSONResponse(
+            status_code=400,
+            content={
+                "jsonrpc": "2.0",
+                "error": {"code": -32601, "message": f"Method not found: {method}"},
+                "id": request_id,
+            },
+        )