diff --git a/apps/common/constants/permission_constants.py b/apps/common/constants/permission_constants.py index 4c068a643..2617472ce 100644 --- a/apps/common/constants/permission_constants.py +++ b/apps/common/constants/permission_constants.py @@ -842,6 +842,23 @@ class PermissionConstants(Enum): parent_group=[SystemGroup.SHARED_KNOWLEDGE] ) + SHARED_MODEL_CREATE = Permission( + group=Group.MODEL, operate=Operate.CREATE, role_list=[RoleConstants.ADMIN, RoleConstants.USER], + parent_group=[SystemGroup.SHARED_MODEL] + ) + SHARED_MODEL_READ = Permission( + group=Group.MODEL, operate=Operate.READ, role_list=[RoleConstants.ADMIN, RoleConstants.USER], + parent_group=[SystemGroup.SHARED_MODEL] + ) + SHARED_MODEL_EDIT = Permission( + group=Group.MODEL, operate=Operate.EDIT, role_list=[RoleConstants.ADMIN, RoleConstants.USER], + parent_group=[SystemGroup.SHARED_MODEL] + ) + SHARED_MODEL_DELETE = Permission( + group=Group.MODEL, operate=Operate.DELETE, role_list=[RoleConstants.ADMIN, RoleConstants.USER], + parent_group=[SystemGroup.SHARED_MODEL] + ) + def get_workspace_application_permission(self): return lambda r, kwargs: Permission(group=self.value.group, operate=self.value.operate, resource_path= diff --git a/apps/tools/serializers/tool.py b/apps/tools/serializers/tool.py index 1b6420a42..64b951801 100644 --- a/apps/tools/serializers/tool.py +++ b/apps/tools/serializers/tool.py @@ -208,6 +208,7 @@ class ToolSerializer(serializers.Serializer): desc=instance.get('desc'), code=instance.get('code'), user_id=self.data.get('user_id'), + workspace_id=self.data.get('workspace_id'), input_field_list=instance.get('input_field_list', []), init_field_list=instance.get('init_field_list', []), scope=instance.get('scope', ToolScope.WORKSPACE), @@ -387,6 +388,7 @@ class ToolSerializer(serializers.Serializer): desc=tool.get('desc'), code=tool.get('code'), user_id=user_id, + workspace_id=self.data.get('workspace_id'), input_field_list=tool.get('input_field_list'), init_field_list=tool.get('init_field_list', []), scope=scope,