Skip to content

Commit deb6402

Browse files
committed
feat: add ModelWorkspaceAuthorization deletion in ModelSerializer
--bug=1059596 --user=刘瑞斌 【共享模型】删除共享模型失败 https://www.tapd.cn/62980211/s/1742883
1 parent ca070d7 commit deb6402

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

apps/models_provider/serializers/model_serializer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
from system_manage.models import WorkspaceUserResourcePermission, AuthTargetType
2929
from system_manage.serializers.user_resource_permission import UserResourcePermissionSerializer
3030
from users.serializers.user import is_workspace_manage
31-
31+
from shared.models.shared_model import ModelWorkspaceAuthorization
3232

3333
def get_default_model_params_setting(provider, model_type, model_name):
3434
credential = get_model_credential(provider, model_type, model_name)
@@ -172,6 +172,7 @@ def delete(self, with_valid=True):
172172
if model is None:
173173
return True
174174
QuerySet(WorkspaceUserResourcePermission).filter(target=model_id).delete()
175+
QuerySet(ModelWorkspaceAuthorization).filter(model_id=model_id).delete()
175176
# TODO : 这里可以添加模型删除的逻辑,需要注意删除模型时的权限和关联关系
176177
# if model.model_type == 'LLM':
177178
# application_count = Application.objects.filter(model_id=model_id).count()

0 commit comments

Comments
 (0)