From f13a29a61186a9a4aa65ca105fa4dc8be716ae8c Mon Sep 17 00:00:00 2001 From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com> Date: Thu, 26 Jun 2025 18:18:19 +0800 Subject: [PATCH] fix: model share (#3402) --- apps/models_provider/serializers/model_serializer.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/apps/models_provider/serializers/model_serializer.py b/apps/models_provider/serializers/model_serializer.py index 10ec8128..8675870d 100644 --- a/apps/models_provider/serializers/model_serializer.py +++ b/apps/models_provider/serializers/model_serializer.py @@ -344,12 +344,11 @@ class ModelSerializer(serializers.Serializer): if with_valid: self.is_valid(raise_exception=True) queryset = self._build_query_params(workspace_id) - model_workspace_authorization = DatabaseModelManage.get_model("model_workspace_authorization") + get_authorized_model = DatabaseModelManage.get_model("get_authorized_model") shared_queryset = QuerySet(Model).filter(workspace_id='None') - if model_workspace_authorization is not None: - shared_queryset = get_authorized_tool(QuerySet(Model), workspace_id, - model_workspace_authorization=model_workspace_authorization) + if get_authorized_model is not None: + shared_queryset = get_authorized_model(QuerySet(Model), workspace_id) # 构建共享模型和普通模型列表 shared_model = [self._build_model_data(model) for model in shared_queryset.order_by("-create_time")]