From bc62fe58c1091366054466a4943f2446fc6e0e00 Mon Sep 17 00:00:00 2001 From: junjun Date: Sun, 4 Jan 2026 14:18:29 +0800 Subject: [PATCH] feat: mcp provide datasource --- backend/apps/mcp/mcp.py | 4 ++-- backend/main.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/apps/mcp/mcp.py b/backend/apps/mcp/mcp.py index 7ea52593..b935ffb1 100644 --- a/backend/apps/mcp/mcp.py +++ b/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) diff --git a/backend/main.py b/backend/main.py index 6bd9c1c4..e473af87 100644 --- a/backend/main.py +++ b/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)