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