From 2d7dc6da3d26c37c2930322b6e3ede7aefd6e815 Mon Sep 17 00:00:00 2001
From: junjun <junjie.xia@fit2cloud.com>
Date: Mon, 23 Jun 2025 14:22:09 +0800
Subject: [PATCH] feat: mcp server

---
 backend/main.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/backend/main.py b/backend/main.py
index d9cbb6ff..e468c859 100644
--- a/backend/main.py
+++ b/backend/main.py
@@ -43,16 +43,20 @@ app = FastAPI(
     lifespan=lifespan
 )
 
-# mcp_app = FastAPI()
+mcp_app = FastAPI()
+mcp_app.add_middleware(TokenMiddleware)
+mcp_app.include_router(api_router, prefix=settings.API_V1_STR)
 
 mcp = FastApiMCP(
     app,
     name="SQLBot MCP Server",
     description="SQLBot MCP Server",
+    describe_all_responses=True,
+    describe_full_response_schema=True,
     include_operations=["get_datasource_list", "get_model_list", "question"]
 )
 
-mcp.mount(app)
+mcp.mount(mcp_app)
 
 # Set all CORS enabled origins
 if settings.all_cors_origins: