feat: mcp

Author: XiaJunjie2020

backend/apps/datasource/api/datasource.py

@@ -18,7 +18,7 @@
 path = "/opt/sqlbot/data/excel"
 
 
-@router.get("/list")
+@router.get("/list", operation_id="get_datasource_list")
 async def datasource_list(session: SessionDep):
     return get_datasource_list(session=session)
 

backend/common/utils/whitelist.py

@@ -20,7 +20,9 @@
     "*.woff2",
     "*.ttf",
     "*.eot",
-    "*.otf"
+    "*.otf",
+    "/mcp",
+    "/mcp/message"
 ]
 
 class WhitelistChecker:

backend/main.py

@@ -14,15 +14,20 @@
 from common.core.response_middleware import ResponseMiddleware, exception_handler
 from alembic.config import Config
 from alembic import command
+from fastapi_mcp import FastApiMCP
+
 
 def run_migrations():
     alembic_cfg = Config("alembic.ini")
     command.upgrade(alembic_cfg, "head")
+
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     run_migrations()
     yield
 
+
 def custom_generate_unique_id(route: APIRoute) -> str:
     tag = route.tags[0] if route.tags and len(route.tags) > 0 else ""
     return f"{tag}-{route.name}"
@@ -38,6 +43,17 @@ def custom_generate_unique_id(route: APIRoute) -> str:
     lifespan=lifespan
 )
 
+mcp_app = FastAPI()
+
+mcp = FastApiMCP(
+    app,
+    name="SQLBot MCP Server",
+    description="SQLBot MCP Server",
+    include_operations=["get_datasource_list"]
+)
+
+mcp.mount(mcp_app)
+
 # Set all CORS enabled origins
 if settings.all_cors_origins:
     app.add_middleware(
@@ -66,9 +82,13 @@ def custom_generate_unique_id(route: APIRoute) -> str:
     @app.get("/", include_in_schema=False)
     async def read_index():
         return FileResponse(path=os.path.join(frontend_dist, "index.html"))
-    
+
+
     app.mount("/", StaticFiles(directory=frontend_dist), name="static")
 
+mcp.setup_server()
+
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)
+
+    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)

backend/pyproject.toml

@@ -34,6 +34,7 @@ dependencies = [
     "psycopg2-binary (>=2.9.10,<3.0.0)",
     "oracledb (>=3.1.1,<4.0.0)",
     "pyyaml (>=6.0.2,<7.0.0)",
+    "fastapi-mcp (>=0.3.4,<0.4.0)",
 ]
 [[tool.uv.index]]
 url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"