@@ -416,17 +416,6 @@ def save_model_params_form(self, model_params_form, with_valid=True):
416416 return True
417417
418418
419- def get_authorized_tool (tool_query_set , workspace_id ):
420- model_id_list = select_list (get_file_content (
421- os .path .join (PROJECT_DIR , "apps" , "models_provider" , 'sql' ,
422- 'list_share_authorized_model.sql'
423- )), [workspace_id , workspace_id ])
424- tool_query_set = tool_query_set .filter (
425- id__in = [k .get ('model_id' ) for k in model_id_list ]
426- )
427- return tool_query_set
428-
429-
430419class WorkspaceSharedModelSerializer (serializers .Serializer ):
431420 workspace_id = serializers .CharField (required = True , label = _ ('workspace id' ))
432421 name = serializers .CharField (required = False , max_length = 64 , label = _ ('model name' ))
@@ -459,9 +448,9 @@ def get_share_model_list(self):
459448 def _build_queryset (self , workspace_id ):
460449 queryset = QuerySet (Model )
461450 if workspace_id :
462- model_workspace_authorization = DatabaseModelManage .get_model ("model_workspace_authorization " )
463- if model_workspace_authorization is not None :
464- queryset = get_authorized_tool (queryset , workspace_id )
451+ get_authorized_model = DatabaseModelManage .get_model ("get_authorized_model " )
452+ if get_authorized_model is not None :
453+ queryset = get_authorized_model (queryset , workspace_id )
465454
466455 for field in ['name' , 'model_type' , 'model_name' , 'provider' , 'create_user' ]:
467456 value = self .data .get (field )
0 commit comments