88from apps .ai_model .model_factory import LLMConfig , LLMFactory
99from apps .swagger .i18n import PLACEHOLDER_PREFIX
1010from 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
1212from apps .system .schemas .ai_model_schema import AiModelConfigItem , AiModelCreator , AiModelEditor , AiModelGridItem
1313from apps .system .schemas .permission import SqlbotPermission , require_permissions
1414from 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' ]))
198198async 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' ]))
220220async 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 ' ]))
255255async def get_model_by_ws (
256256 session : SessionDep ,
257257 current_user : CurrentUser
0 commit comments