15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 10:12:51 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: refactor response serializers for model parameters and lists
Some checks are pending
sync2gitee / repo-sync (push) Waiting to run
Details

This commit is contained in:
CaptainB 2025-04-24 13:42:58 +08:00
parent 4c38e8a82b
commit f05a4a67c5
3 changed files with 20 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
apps/models_provider/api/provide.py
View File

@ -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 @@ class ProvideApi(APIMixin):
@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 @@ class ProvideApi(APIMixin):
@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 @@ class ProvideApi(APIMixin):
@staticmethod
def get_response():
class ProvideListResponse(ResultSerializer):
def get_data(self):
return serializers.ListSerializer(child=ProvideListSerializer())
return ProvideListResponse

2
apps/models_provider/views/model.py
View File

@ -118,6 +118,7 @@ class Model(APIView):
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 Model(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())

2
apps/models_provider/views/provide.py
View File

@ -73,7 +73,7 @@ class Provide(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')])