From 767cdfc51776ae9a689e1be280f4b05070151bff Mon Sep 17 00:00:00 2001 From: CaptainB Date: Fri, 18 Apr 2025 18:29:50 +0800 Subject: [PATCH] feat: add serializer_class to Operate classes in model, module, and tool APIs --- apps/models_provider/views/model.py | 3 +++ apps/modules/views/module.py | 1 + apps/tools/views/tool.py | 1 + 3 files changed, 5 insertions(+) diff --git a/apps/models_provider/views/model.py b/apps/models_provider/views/model.py index b3ea3fd9de0..fe270aab140 100644 --- a/apps/models_provider/views/model.py +++ b/apps/models_provider/views/model.py @@ -64,6 +64,7 @@ def get(self, request: Request): class Operate(APIView): authentication_classes = [TokenAuth] + serializer_class = ModelSerializer @extend_schema(methods=['PUT'], description=_('Update model'), @@ -101,6 +102,7 @@ def get(self, request: Request, workspace_id: str, model_id: str): class ModelParamsForm(APIView): authentication_classes = [TokenAuth] + serializer_class = ModelSerializer @extend_schema(methods=['GET'], description=_('Get model parameter form'), @@ -142,6 +144,7 @@ def get(self, request: Request, workspace_id: str, model_id: str): class PauseDownload(APIView): authentication_classes = [TokenAuth] + serializer_class = ModelSerializer @extend_schema(methods=['PUT'], description=_('Pause model download'), diff --git a/apps/modules/views/module.py b/apps/modules/views/module.py index b0eed2fb512..072bd394fdb 100644 --- a/apps/modules/views/module.py +++ b/apps/modules/views/module.py @@ -33,6 +33,7 @@ def post(self, request: Request, workspace_id: str, source: str): class Operate(APIView): authentication_classes = [TokenAuth] + serializer_class = ModuleSerializer @extend_schema(methods=['PUT'], description=_('Update module'), diff --git a/apps/tools/views/tool.py b/apps/tools/views/tool.py index 863a0befcf7..456b1d877ba 100644 --- a/apps/tools/views/tool.py +++ b/apps/tools/views/tool.py @@ -30,6 +30,7 @@ def post(self, request: Request, workspace_id: str): class Operate(APIView): authentication_classes = [TokenAuth] + serializer_class = ToolSerializer @extend_schema(methods=['PUT'], description=_('Update tool'),