Skip to content

Commit 13c5833

Browse files
authored
fix: model permission (#3299)
1 parent 7207e20 commit 13c5833

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

apps/models_provider/views/model.py

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ class ModelSetting(APIView):
6565
request=ModelCreateAPI.get_request(),
6666
responses=ModelCreateAPI.get_response())
6767
@has_permissions(PermissionConstants.MODEL_CREATE.get_workspace_permission(),
68-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
68+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
6969
@log(menu='model', operate='Create model',
7070
get_operation_object=lambda r, k: {'name': r.date.get('name')},
7171
get_details=get_edit_model_details,
@@ -95,7 +95,7 @@ def post(self, request: Request, workspace_id: str):
9595
responses=ModelListResponse.get_response(),
9696
tags=[_('Model')]) # type: ignore
9797
@has_permissions(PermissionConstants.MODEL_READ.get_workspace_permission(),
98-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
98+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
9999
def get(self, request: Request, workspace_id: str):
100100
return result.success(
101101
ModelSerializer.Query(
@@ -114,7 +114,7 @@ class Operate(APIView):
114114
responses=ModelEditApi.get_response(),
115115
tags=[_('Model')]) # type: ignore
116116
@has_permissions(PermissionConstants.MODEL_EDIT.get_workspace_permission(),
117-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
117+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
118118
@log(menu='model', operate='Update model',
119119
get_operation_object=lambda r, k: get_model_operation_object(k.get('model_id')),
120120
get_details=get_edit_model_details,
@@ -132,7 +132,7 @@ def put(self, request: Request, workspace_id, model_id: str):
132132
responses=DefaultModelResponse.get_response(),
133133
tags=[_('Model')]) # type: ignore
134134
@has_permissions(PermissionConstants.MODEL_DELETE.get_workspace_permission(),
135-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
135+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
136136
@log(menu='model', operate='Delete model',
137137
get_operation_object=lambda r, k: get_model_operation_object(k.get('model_id')),
138138
)
@@ -148,7 +148,7 @@ def delete(self, request: Request, workspace_id: str, model_id: str):
148148
responses=GetModelApi.get_response(),
149149
tags=[_('Model')]) # type: ignore
150150
@has_permissions(PermissionConstants.MODEL_READ.get_workspace_permission(),
151-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
151+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
152152
def get(self, request: Request, workspace_id: str, model_id: str):
153153
return result.success(
154154
ModelSerializer.Operate(data={'id': model_id, 'user_id': request.user.id}).one(with_valid=True))
@@ -164,7 +164,7 @@ class ModelParamsForm(APIView):
164164
responses=ProvideApi.ModelParamsForm.get_response(),
165165
tags=[_('Model')]) # type: ignore
166166
@has_permissions(PermissionConstants.MODEL_READ.get_workspace_permission(),
167-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
167+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
168168
def get(self, request: Request, workspace_id: str, model_id: str):
169169
return result.success(
170170
ModelSerializer.ModelParams(data={'id': model_id}).get_model_params())
@@ -178,7 +178,7 @@ def get(self, request: Request, workspace_id: str, model_id: str):
178178
responses=ProvideApi.ModelParamsForm.get_response(),
179179
tags=[_('Model')]) # type: ignore
180180
@has_permissions(PermissionConstants.MODEL_READ.get_workspace_permission(),
181-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
181+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
182182
@log(menu='model', operate='Save model parameter form',
183183
get_operation_object=lambda r, k: get_model_operation_object(k.get('model_id')),
184184
)
@@ -200,7 +200,7 @@ class ModelMeta(APIView):
200200
responses=GetModelApi.get_response(),
201201
tags=[_('Model')]) # type: ignore
202202
@has_permissions(PermissionConstants.MODEL_READ.get_workspace_permission(),
203-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
203+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
204204
def get(self, request: Request, workspace_id: str, model_id: str):
205205
return result.success(
206206
ModelSerializer.Operate(data={'id': model_id}).one_meta(with_valid=True))
@@ -217,7 +217,7 @@ class PauseDownload(APIView):
217217
responses=DefaultModelResponse.get_response(),
218218
tags=[_('Model')]) # type: ignore
219219
@has_permissions(PermissionConstants.MODEL_CREATE.get_workspace_permission(),
220-
RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
220+
RoleConstants.WORKSPACE_MANAGE.get_workspace_role())
221221
def put(self, request: Request, workspace_id: str, model_id: str):
222222
return result.success(
223223
ModelSerializer.Operate(data={'id': model_id}).pause_download())
@@ -235,7 +235,7 @@ class SharedModel(APIView):
235235
responses=ModelListResponse.get_response(),
236236
tags=[_('Shared Model')]
237237
) # type: ignore
238-
@has_permissions(PermissionConstants.MODEL_READ, RoleConstants.WORKSPACE_MANAGE.get_workspace_role)
238+
@has_permissions(PermissionConstants.MODEL_READ, RoleConstants.WORKSPACE_MANAGE.get_workspace_role()())
239239
def get(self, request: Request, workspace_id: str):
240240
return result.success(
241-
SharedModelSerializer(data={'workspace_id': workspace_id}).get_share_model_list())
241+
SharedModelSerializer(data={'workspace_id': workspace_id}).get_share_model_list())

0 commit comments

Comments
 (0)