Skip to content

Commit 8ce3a44

Browse files
authored
fix: list shard model (#3399)
1 parent cf674ec commit 8ce3a44

File tree

1 file changed

+3
-14
lines changed

1 file changed

+3
-14
lines changed

apps/models_provider/serializers/model_serializer.py

Lines changed: 3 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -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-
430419
class 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

Comments
 (0)