zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: model share (#3402)

Browse Source
This commit is contained in:
shaohuzhang1 2025-06-26 18:18:19 +08:00 committed by GitHub
parent 15bfe239a4
commit f13a29a611
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/models_provider/serializers/model_serializer.py
View File

@ -344,12 +344,11 @@ class ModelSerializer(serializers.Serializer):
if with_valid: if with_valid:
self.is_valid(raise_exception=True) self.is_valid(raise_exception=True)
queryset = self._build_query_params(workspace_id) 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') shared_queryset = QuerySet(Model).filter(workspace_id='None')
if model_workspace_authorization is not None: if get_authorized_model is not None:
shared_queryset = get_authorized_tool(QuerySet(Model), workspace_id, shared_queryset = get_authorized_model(QuerySet(Model), workspace_id)
model_workspace_authorization=model_workspace_authorization)
# 构建共享模型和普通模型列表 # 构建共享模型和普通模型列表
shared_model = [self._build_model_data(model) for model in shared_queryset.order_by("-create_time")] shared_model = [self._build_model_data(model) for model in shared_queryset.order_by("-create_time")]