From 39841cb3440cd16cd72aa05c367d3a20eb869a52 Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Mon, 21 Apr 2025 10:24:13 +0800
Subject: [PATCH] feat: add model setting

---
 apps/models_provider/api/model.py   | 8 +++++++-
 apps/models_provider/views/model.py | 6 ++++--
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/apps/models_provider/api/model.py b/apps/models_provider/api/model.py
index 363a82b32..e76499595 100644
--- a/apps/models_provider/api/model.py
+++ b/apps/models_provider/api/model.py
@@ -4,7 +4,7 @@ from drf_spectacular.utils import OpenApiParameter
 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 @@ class ModelEditApi(APIMixin):
     @staticmethod
     def get_response():
         return ModelModelSerializer
+
+
+class DefaultModelResponse(APIMixin):
+    @staticmethod
+    def get_response():
+        return DefaultResultSerializer()
diff --git a/apps/models_provider/views/model.py b/apps/models_provider/views/model.py
index f3df85cac..ffea535c7 100644
--- a/apps/models_provider/views/model.py
+++ b/apps/models_provider/views/model.py
@@ -14,9 +14,9 @@ from rest_framework.request import Request
 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 @@ class Model(APIView):
                        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 Model(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):