feat: add model setting

Author: wxg0103

apps/models_provider/api/model.py

@@ -4,7 +4,7 @@
 from rest_framework import serializers
 
 from common.mixins.api_mixin import APIMixin
-from common.result import ResultSerializer
+from common.result import ResultSerializer, DefaultResultSerializer
 from models_provider.serializers.model_serializer import ModelModelSerializer, ModelCreateRequest
 from django.utils.translation import gettext_lazy as _
 
@@ -72,3 +72,9 @@ def get_request():
     @staticmethod
     def get_response():
         return ModelModelSerializer
+
+
+class DefaultModelResponse(APIMixin):
+    @staticmethod
+    def get_response():
+        return DefaultResultSerializer()

apps/models_provider/views/model.py

@@ -14,9 +14,9 @@
 from common.auth import TokenAuth
 from common.auth.authentication import has_permissions
 from common.constants.permission_constants import PermissionConstants
-from common.result import result
+from common.result import result, DefaultResultSerializer
 from common.utils.common import query_params_to_single_dict
-from models_provider.api.model import ModelCreateAPI, GetModelApi, ModelEditApi, ModelListResponse
+from models_provider.api.model import ModelCreateAPI, GetModelApi, ModelEditApi, ModelListResponse, DefaultModelResponse
 from models_provider.api.provide import ProvideApi
 from models_provider.serializers.model_serializer import ModelSerializer
 
@@ -82,6 +82,7 @@ def put(self, request: Request, workspace_id, model_id: str):
                        description=_('Delete model'),
                        operation_id=_('Delete model'),
                        parameters=GetModelApi.get_query_params_api(),
+                       responses=DefaultModelResponse.get_response(),
                        tags=[_('Model')])
         @has_permissions(PermissionConstants.MODEL_DELETE.get_workspace_permission())
         def delete(self, request: Request, workspace_id: str, model_id: str):
@@ -147,6 +148,7 @@ class PauseDownload(APIView):
                        description=_('Pause model download'),
                        operation_id=_('Pause model download'),
                        parameters=GetModelApi.get_query_params_api(),
+                       responses=DefaultModelResponse.get_response(),
                        tags=[_('Model')])
         @has_permissions(PermissionConstants.MODEL_CREATE.get_workspace_permission())
         def put(self, request: Request, workspace_id: str, model_id: str):