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(Application): Application can configure specified model

Browse Source
This commit is contained in:
ulleo
2026-06-02 14:46:55 +08:00
committed by ulleo
parent 46de2322b0
commit 4b0d4ce663
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/system/api/aimodel.py
+6 -6
View File
@@ -192,7 +192,7 @@ async def delete_model(
session.commit()
@router.get("/{id}/ws_mapping", response_model=List[int], summary=f"{PLACEHOLDER_PREFIX}system_model_query",
@router.get("/{id}/ws_mapping", response_model=List[str], summary=f"{PLACEHOLDER_PREFIX}system_model_query",
description=f"{PLACEHOLDER_PREFIX}system_model_query")
@require_permissions(permission=SqlbotPermission(role=['admin']))
async def get_model_ws_mapping_by_id(
@@ -211,21 +211,21 @@ async def get_model_ws_mapping_by_id(
)
ws_ids: List[int] = session.exec(stmt).all()
return ws_ids
return [str(ws_id) for ws_id in ws_ids]
@router.put("/{id}/ws_mapping", response_model=List[int], summary=f"{PLACEHOLDER_PREFIX}system_model_query",
@router.put("/{id}/ws_mapping", response_model=List[str], summary=f"{PLACEHOLDER_PREFIX}system_model_query",
description=f"{PLACEHOLDER_PREFIX}system_model_query")
@require_permissions(permission=SqlbotPermission(role=['admin']))
async def update_model_ws_mapping_by_id(
session: SessionDep,
id: int = Path(description="ID"),
ws_ids: List[int] = Body(description="workspace id list"),
ws_ids: List[str] = Body(description="workspace id list"),
):
if ws_ids is None:
ws_ids = []
# 提前去重
ws_ids = list(set(ws_ids))
ws_ids = list({int(ws_id) for ws_id in ws_ids})
db_model = session.get(AiModelDetail, id)
if not db_model:
@@ -246,7 +246,7 @@ async def update_model_ws_mapping_by_id(
session.commit()
return ws_ids
return [str(ws_id) for ws_id in ws_ids]
@router.get("/list_by_ws", response_model=AiModelBrief, summary=f"{PLACEHOLDER_PREFIX}system_model_query",