feat: refactor response serializers for model parameters and lists

Author: liuruibin

apps/models_provider/api/provide.py

@@ -46,6 +46,21 @@ class ModelParamsFormSerializer(serializers.Serializer):
     props_info = serializers.DictField(required=False, label=_("props info"))
 
 
+class ModelParamsFormResponse(ResultSerializer):
+    def get_data(self):
+        return serializers.ListSerializer(child=ModelParamsFormSerializer())
+
+
+class ModelListResponse(ResultSerializer):
+    def get_data(self):
+        return serializers.ListSerializer(child=ModelListSerializer())
+
+
+class ProvideListResponse(ResultSerializer):
+    def get_data(self):
+        return serializers.ListSerializer(child=ProvideListSerializer())
+
+
 class ProvideApi(APIMixin):
     class ModelParamsForm(APIMixin):
         @staticmethod
@@ -73,11 +88,7 @@ def get_query_params_api():
 
         @staticmethod
         def get_response():
-            class ProvideListSerializer(ResultSerializer):
-                def get_data(self):
-                    return serializers.ListSerializer(child=ModelParamsFormSerializer())
-
-            return ProvideListSerializer
+            return ModelParamsFormResponse
 
     class ModelList(APIMixin):
         @staticmethod
@@ -99,11 +110,7 @@ def get_query_params_api():
 
         @staticmethod
         def get_response():
-            class ProvideListSerializer(ResultSerializer):
-                def get_data(self):
-                    return serializers.ListSerializer(child=ModelListSerializer())
-
-            return ProvideListSerializer
+            return ModelListResponse
 
     @staticmethod
     def get_response():
@@ -126,8 +133,4 @@ def get_query_params_api():
 
         @staticmethod
         def get_response():
-            class ProvideListResponse(ResultSerializer):
-                def get_data(self):
-                    return serializers.ListSerializer(child=ProvideListSerializer())
-
             return ProvideListResponse

apps/models_provider/views/model.py

@@ -118,6 +118,7 @@ def get(self, request: Request, workspace_id: str, model_id: str):
                        description=_('Save model parameter form'),
                        operation_id=_('Save model parameter form'),
                        parameters=GetModelApi.get_query_params_api(),
+                       request=GetModelApi.get_request(),
                        responses=ProvideApi.ModelParamsForm.get_response(),
                        tags=[_('Model')])
         @has_permissions(PermissionConstants.MODEL_READ.get_workspace_permission())
@@ -148,6 +149,7 @@ class PauseDownload(APIView):
                        description=_('Pause model download'),
                        operation_id=_('Pause model download'),
                        parameters=GetModelApi.get_query_params_api(),
+                       request=GetModelApi.get_request(),
                        responses=DefaultModelResponse.get_response(),
                        tags=[_('Model')])
         @has_permissions(PermissionConstants.MODEL_CREATE.get_workspace_permission())

apps/models_provider/views/provide.py

@@ -73,7 +73,7 @@ class ModelParamsForm(APIView):
 
         @extend_schema(methods=['GET'],
                        description=_('Get model default parameters'),
-                       operation_id=_('Get the model creation form'),
+                       operation_id=_('Get model default parameters'),
                        parameters=ProvideApi.ModelParamsForm.get_query_params_api(),
                        responses=ProvideApi.ModelParamsForm.get_response(),
                        tags=[_('Model')])