feat: mcp provide datasource

Author: XiaJunjie2020

backend/apps/mcp/mcp.py

@@ -64,7 +64,7 @@ def get_user(session: SessionDep, token: str):
     db_user: UserModel = get_db_user(session=session, user_id=token_data.id)
     session_user = UserInfoDTO.model_validate(db_user.model_dump())
     session_user.isAdmin = session_user.id == 1 and session_user.account == 'admin'
-    session_user.language = 'zh'
+    session_user.language = 'zh-CN'
     if session_user.isAdmin:
         session_user = session_user
     ws_model: UserWsModel = session.exec(
@@ -80,7 +80,7 @@ def get_user(session: SessionDep, token: str):
     return session_user
 
 
-@router.post("/mcp_ds_list", operation_id="mcp_get_datasource_list")
+@router.post("/mcp_ds_list", operation_id="mcp_datasource_list")
 async def datasource_list(session: SessionDep, token: str):
     session_user = get_user(session, token)
     return get_datasource_list(session=session, user=session_user)

backend/main.py

@@ -184,7 +184,7 @@ async def custom_swagger_ui(request: Request):
     description="SQLBot MCP Server",
     describe_all_responses=True,
     describe_full_response_schema=True,
-    include_operations=["get_datasource_list", "get_model_list", "mcp_question", "mcp_start", "mcp_assistant"]
+    include_operations=["mcp_datasource_list", "get_model_list", "mcp_question", "mcp_start", "mcp_assistant"]
 )
 
 mcp.mount(mcp_app)