From 6d4f5dc1c3c4202dfa021163cec886cdabadf6d3 Mon Sep 17 00:00:00 2001 From: wxg0103 <727495428@qq.com> Date: Tue, 22 Apr 2025 10:45:11 +0800 Subject: [PATCH] feat: add model setting --- apps/models_provider/api/model.py | 10 +++++++--- apps/models_provider/api/provide.py | 18 +++++++++++++++--- 2 files changed, 22 insertions(+), 6 deletions(-) diff --git a/apps/models_provider/api/model.py b/apps/models_provider/api/model.py index e76499595..6a7bc44c0 100644 --- a/apps/models_provider/api/model.py +++ b/apps/models_provider/api/model.py @@ -17,7 +17,11 @@ class ModelCreateResponse(ResultSerializer): class ModelListResponse(APIMixin): @staticmethod def get_response(): - return serializers.ListSerializer(child=ModelModelSerializer()) + class ModelListResult(ResultSerializer): + def get_data(self): + return serializers.ListSerializer(child=ModelModelSerializer()) + + return ModelListResult class ModelCreateAPI(APIMixin): @@ -61,7 +65,7 @@ class GetModelApi(APIMixin): @staticmethod def get_response(): - return ModelModelSerializer + return ModelCreateResponse class ModelEditApi(APIMixin): @@ -71,7 +75,7 @@ class ModelEditApi(APIMixin): @staticmethod def get_response(): - return ModelModelSerializer + return ModelCreateResponse class DefaultModelResponse(APIMixin): diff --git a/apps/models_provider/api/provide.py b/apps/models_provider/api/provide.py index 8c2ddcc2c..90551096e 100644 --- a/apps/models_provider/api/provide.py +++ b/apps/models_provider/api/provide.py @@ -73,7 +73,11 @@ class ProvideApi(APIMixin): @staticmethod def get_response(): - return serializers.ListSerializer(child=ModelParamsFormSerializer()) + class ProvideListSerializer(ResultSerializer): + def get_data(self): + return serializers.ListSerializer(child=ModelParamsFormSerializer()) + + return ProvideListSerializer class ModelList(APIMixin): @staticmethod @@ -95,7 +99,11 @@ class ProvideApi(APIMixin): @staticmethod def get_response(): - return serializers.ListSerializer(child=ModelListSerializer()) + class ProvideListSerializer(ResultSerializer): + def get_data(self): + return serializers.ListSerializer(child=ModelListSerializer()) + + return ProvideListSerializer @staticmethod def get_response(): @@ -118,4 +126,8 @@ class ProvideApi(APIMixin): @staticmethod def get_response(): - return serializers.ListSerializer(child=ProvideListSerializer()) + class ProvideListResponse(ResultSerializer): + def get_data(self): + return serializers.ListSerializer(child=ProvideListSerializer()) + + return ProvideListResponse