Skip to content

Commit ae07051

Browse files
committed
feat(Application): Application can configure specified model
1 parent 74fe71d commit ae07051

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

backend/apps/system/api/aimodel.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from apps.ai_model.model_factory import LLMConfig, LLMFactory
99
from apps.swagger.i18n import PLACEHOLDER_PREFIX
1010
from apps.system.crud.aimodel_manage import get_ai_model_list_by_workspace
11-
from apps.system.models.system_model import AiModelDetail, AiModelWorkspaceMapping
11+
from apps.system.models.system_model import AiModelDetail, AiModelWorkspaceMapping, AiModelBrief
1212
from apps.system.schemas.ai_model_schema import AiModelConfigItem, AiModelCreator, AiModelEditor, AiModelGridItem
1313
from apps.system.schemas.permission import SqlbotPermission, require_permissions
1414
from common.core.deps import SessionDep, Trans, CurrentUser
@@ -192,7 +192,7 @@ async def delete_model(
192192
session.commit()
193193

194194

195-
@router.get("/{id}/ws_mapping", response_model=AiModelEditor, summary=f"{PLACEHOLDER_PREFIX}system_model_query",
195+
@router.get("/{id}/ws_mapping", response_model=List[int], summary=f"{PLACEHOLDER_PREFIX}system_model_query",
196196
description=f"{PLACEHOLDER_PREFIX}system_model_query")
197197
@require_permissions(permission=SqlbotPermission(role=['admin']))
198198
async def get_model_ws_mapping_by_id(
@@ -214,7 +214,7 @@ async def get_model_ws_mapping_by_id(
214214
return ws_ids
215215

216216

217-
@router.put("/{id}/ws_mapping", response_model=AiModelEditor, summary=f"{PLACEHOLDER_PREFIX}system_model_query",
217+
@router.put("/{id}/ws_mapping", response_model=List[int], summary=f"{PLACEHOLDER_PREFIX}system_model_query",
218218
description=f"{PLACEHOLDER_PREFIX}system_model_query")
219219
@require_permissions(permission=SqlbotPermission(role=['admin']))
220220
async def update_model_ws_mapping_by_id(
@@ -249,9 +249,9 @@ async def update_model_ws_mapping_by_id(
249249
return ws_ids
250250

251251

252-
@router.get("/list_by_ws", response_model=AiModelEditor, summary=f"{PLACEHOLDER_PREFIX}system_model_query",
252+
@router.get("/list_by_ws", response_model=AiModelBrief, summary=f"{PLACEHOLDER_PREFIX}system_model_query",
253253
description=f"{PLACEHOLDER_PREFIX}system_model_query")
254-
@require_permissions(permission=SqlbotPermission(role=['admin']))
254+
@require_permissions(permission=SqlbotPermission(role=['ws_admin']))
255255
async def get_model_by_ws(
256256
session: SessionDep,
257257
current_user: CurrentUser

0 commit comments

Comments
 (0)