Skip to content

Commit 07caf1f

Browse files
committed
refactor: add share model
1 parent 387967c commit 07caf1f

File tree

3 files changed

+24
-3
lines changed

3 files changed

+24
-3
lines changed

apps/models_provider/serializers/model_serializer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -388,4 +388,4 @@ def save_model_params_form(self, model_params_form, with_valid=True):
388388
model = QuerySet(Model).filter(id=model_id).first()
389389
model.model_params_form = model_params_form
390390
model.save()
391-
return True
391+
return True

apps/models_provider/urls.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,12 @@
1212
path('provider/model_params_form', views.Provide.ModelParamsForm.as_view()),
1313
path('provider/model_form', views.Provide.ModelForm.as_view()),
1414
path('workspace/<str:workspace_id>/model', views.ModelSetting.as_view()),
15-
path('workspace/<str:workspace_id>/model/<str:model_id>/model_params_form', views.ModelSetting.ModelParamsForm.as_view()),
15+
path('workspace/<str:workspace_id>/share_model', views.ModelSetting.Share.as_view()),
16+
path('workspace/<str:workspace_id>/model/<str:model_id>/model_params_form',
17+
views.ModelSetting.ModelParamsForm.as_view()),
1618
path('workspace/<str:workspace_id>/model/<str:model_id>', views.ModelSetting.Operate.as_view()),
17-
path('workspace/<str:workspace_id>/model/<str:model_id>/pause_download', views.ModelSetting.PauseDownload.as_view()),
19+
path('workspace/<str:workspace_id>/model/<str:model_id>/pause_download',
20+
views.ModelSetting.PauseDownload.as_view()),
1821
path('workspace/<str:workspace_id>/model/<str:model_id>/meta', views.ModelSetting.ModelMeta.as_view()),
1922
]
2023

apps/models_provider/views/model.py

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -210,3 +210,21 @@ class PauseDownload(APIView):
210210
def put(self, request: Request, workspace_id: str, model_id: str):
211211
return result.success(
212212
ModelSerializer.Operate(data={'id': model_id}).pause_download())
213+
214+
class Share(APIView):
215+
authentication_classes = [TokenAuth]
216+
217+
@extend_schema(methods=['Get'],
218+
summary=_('Get Share model'),
219+
description=_('Get Share model'),
220+
operation_id=_('Get Share model'), # type: ignore
221+
parameters=GetModelApi.get_query_params_api(),
222+
request=GetModelApi.get_request(),
223+
responses=DefaultModelResponse.get_response(),
224+
tags=[_('Model')]) # type: ignore
225+
@has_permissions(PermissionConstants.MODEL_READ.get_workspace_permission())
226+
def get(self, request: Request, workspace_id: str):
227+
return result.success(
228+
ModelSerializer.Query(
229+
data={**query_params_to_single_dict(request.query_params)}).list(workspace_id='None',
230+
with_valid=True))

0 commit comments

Comments
 (0)