klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-08 01:06:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: mcp provide datasource

Browse Source
This commit is contained in:
junjun
2026-01-04 14:19:42 +08:00
committed by Junjun
parent b7669321d1
commit cdb9f892cd
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/mcp/mcp.py
+2 -2
View File
@@ -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
+1 -1
View File
@@ -184,7 +184,7 @@ mcp = FastApiMCP(
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)