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

fix: Fix MCP oid missing validation

Browse Source
This commit is contained in:
junjun
2026-06-11 17:51:37 +08:00
committed by Junjun
parent cc45f47c12
commit 447592b767
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/mcp/mcp.py
+2 -2
View File
@@ -111,7 +111,7 @@ async def datasource_list(session: SessionDep, trans: Trans, mcp_ds: McpDs):
w_list = await user_ws_options(session, session_user.id, trans)
oid_list = [item.id for item in w_list]
if int(mcp_ds.oid) not in oid_list:
raise HTTPException(status_code=400, detail="This user not in current workspace")
raise HTTPException(status_code=400, detail="The current user is not in the selected workspace")
session_user.oid = int(mcp_ds.oid)
ds_list = get_datasource_list(session=session, user=session_user)
@@ -143,7 +143,7 @@ async def mcp_question(session: SessionDep, trans: Trans, chat: McpQuestion):
w_list = await user_ws_options(session, session_user.id, trans)
oid_list = [item.id for item in w_list]
if int(chat.oid) not in oid_list:
raise HTTPException(status_code=400, detail="This user not in current workspace")
raise HTTPException(status_code=400, detail="The current user is not in the selected workspace")
session_user.oid = int(chat.oid)
ds_id: Optional[int] = None