Browse Source

feat: API docs for mcp (#24424)

Signed-off-by: -LAN- <laipz8200@outlook.com>
tags/1.8.0
-LAN- 2 months ago
parent
commit
3075d415e1
No account linked to committer's email address
2 changed files with 67 additions and 20 deletions
  1. 13
    1
      api/controllers/mcp/__init__.py
  2. 54
    19
      api/controllers/mcp/mcp.py

+ 13
- 1
api/controllers/mcp/__init__.py View File

@@ -1,8 +1,20 @@
from flask import Blueprint
from flask_restx import Namespace

from libs.external_api import ExternalApi

bp = Blueprint("mcp", __name__, url_prefix="/mcp")
api = ExternalApi(bp)

api = ExternalApi(
bp,
version="1.0",
title="MCP API",
description="API for Model Context Protocol operations",
doc="/docs", # Enable Swagger UI at /mcp/docs
)

mcp_ns = Namespace("mcp", description="MCP operations")

from . import mcp

api.add_namespace(mcp_ns)

+ 54
- 19
api/controllers/mcp/mcp.py View File

@@ -1,8 +1,10 @@
from typing import Optional, Union

from flask_restx import Resource, reqparse
from pydantic import ValidationError

from controllers.console.app.mcp_server import AppMCPServerStatus
from controllers.mcp import api
from controllers.mcp import mcp_ns
from core.app.app_config.entities import VariableEntity
from core.mcp import types
from core.mcp.server.streamable_http import MCPServerStreamableHTTPRequestHandler
@@ -13,22 +15,58 @@ from libs import helper
from models.model import App, AppMCPServer, AppMode


def int_or_str(value):
"""Validate that a value is either an integer or string."""
if isinstance(value, (int, str)):
return value
else:
return None


# Define parser for both documentation and validation
mcp_request_parser = reqparse.RequestParser()
mcp_request_parser.add_argument(
"jsonrpc", type=str, required=True, location="json", help="JSON-RPC version (should be '2.0')"
)
mcp_request_parser.add_argument("method", type=str, required=True, location="json", help="The method to invoke")
mcp_request_parser.add_argument("params", type=dict, required=False, location="json", help="Parameters for the method")
mcp_request_parser.add_argument(
"id", type=int_or_str, required=False, location="json", help="Request ID for tracking responses"
)


@mcp_ns.route("/server/<string:server_code>/mcp")
class MCPAppApi(Resource):
def post(self, server_code):
def int_or_str(value):
if isinstance(value, (int, str)):
return value
else:
return None

parser = reqparse.RequestParser()
parser.add_argument("jsonrpc", type=str, required=True, location="json")
parser.add_argument("method", type=str, required=True, location="json")
parser.add_argument("params", type=dict, required=False, location="json")
parser.add_argument("id", type=int_or_str, required=False, location="json")
args = parser.parse_args()

request_id = args.get("id")
@mcp_ns.expect(mcp_request_parser)
@mcp_ns.doc("handle_mcp_request")
@mcp_ns.doc(description="Handle Model Context Protocol (MCP) requests for a specific server")
@mcp_ns.doc(params={"server_code": "Unique identifier for the MCP server"})
@mcp_ns.doc(
responses={
200: "MCP response successfully processed",
400: "Invalid MCP request or parameters",
404: "Server or app not found",
}
)
def post(self, server_code: str):
"""Handle MCP requests for a specific server.

Processes JSON-RPC formatted requests according to the Model Context Protocol specification.
Validates the server status and associated app before processing the request.

Args:
server_code: Unique identifier for the MCP server

Returns:
dict: JSON-RPC response from the MCP handler

Raises:
ValidationError: Invalid request format or parameters
"""
# Parse and validate all arguments
args = mcp_request_parser.parse_args()

request_id: Optional[Union[int, str]] = args.get("id")

server = db.session.query(AppMCPServer).where(AppMCPServer.server_code == server_code).first()
if not server:
@@ -99,6 +137,3 @@ class MCPAppApi(Resource):
mcp_server_handler = MCPServerStreamableHTTPRequestHandler(app, request, converted_user_input_form)
response = mcp_server_handler.handle()
return helper.compact_generate_response(response)


api.add_resource(MCPAppApi, "/server/<string:server_code>/mcp")

Loading…
Cancel
Save