Skip to content

Commit 6d4f5dc

Browse files
committed
feat: add model setting
1 parent 3f7fb88 commit 6d4f5dc

File tree

2 files changed

+22
-6
lines changed

2 files changed

+22
-6
lines changed

apps/models_provider/api/model.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,11 @@ def get_data(self):
1717
class ModelListResponse(APIMixin):
1818
@staticmethod
1919
def get_response():
20-
return serializers.ListSerializer(child=ModelModelSerializer())
20+
class ModelListResult(ResultSerializer):
21+
def get_data(self):
22+
return serializers.ListSerializer(child=ModelModelSerializer())
23+
24+
return ModelListResult
2125

2226

2327
class ModelCreateAPI(APIMixin):
@@ -61,7 +65,7 @@ def get_query_params_api():
6165

6266
@staticmethod
6367
def get_response():
64-
return ModelModelSerializer
68+
return ModelCreateResponse
6569

6670

6771
class ModelEditApi(APIMixin):
@@ -71,7 +75,7 @@ def get_request():
7175

7276
@staticmethod
7377
def get_response():
74-
return ModelModelSerializer
78+
return ModelCreateResponse
7579

7680

7781
class DefaultModelResponse(APIMixin):

apps/models_provider/api/provide.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,11 @@ def get_query_params_api():
7373

7474
@staticmethod
7575
def get_response():
76-
return serializers.ListSerializer(child=ModelParamsFormSerializer())
76+
class ProvideListSerializer(ResultSerializer):
77+
def get_data(self):
78+
return serializers.ListSerializer(child=ModelParamsFormSerializer())
79+
80+
return ProvideListSerializer
7781

7882
class ModelList(APIMixin):
7983
@staticmethod
@@ -95,7 +99,11 @@ def get_query_params_api():
9599

96100
@staticmethod
97101
def get_response():
98-
return serializers.ListSerializer(child=ModelListSerializer())
102+
class ProvideListSerializer(ResultSerializer):
103+
def get_data(self):
104+
return serializers.ListSerializer(child=ModelListSerializer())
105+
106+
return ProvideListSerializer
99107

100108
@staticmethod
101109
def get_response():
@@ -118,4 +126,8 @@ def get_query_params_api():
118126

119127
@staticmethod
120128
def get_response():
121-
return serializers.ListSerializer(child=ProvideListSerializer())
129+
class ProvideListResponse(ResultSerializer):
130+
def get_data(self):
131+
return serializers.ListSerializer(child=ProvideListSerializer())
132+
133+
return ProvideListResponse

0 commit comments

Comments
 (0)