29 lines
887 B
Python
29 lines
887 B
Python
from rest_framework import serializers
|
|
|
|
from ai.models import AIModel
|
|
from utils.serializers import CustomModelSerializer
|
|
from utils.custom_model_viewSet import CustomModelViewSet
|
|
|
|
class AIModelSerializer(CustomModelSerializer):
|
|
api_key_name = serializers.CharField(source='key.name', read_only=True)
|
|
"""
|
|
AI 模型 序列化器
|
|
"""
|
|
class Meta:
|
|
model = AIModel
|
|
fields = '__all__'
|
|
read_only_fields = ['id', 'create_time', 'update_time']
|
|
|
|
|
|
class AIModelViewSet(CustomModelViewSet):
|
|
"""
|
|
AI 模型 视图集
|
|
"""
|
|
queryset = AIModel.objects.filter(is_deleted=False).select_related('key')
|
|
serializer_class = AIModelSerializer
|
|
filterset_fields = ['name', 'status', 'platform', 'model', 'model_type', ]
|
|
search_fields = ['name'] # 根据实际字段调整
|
|
ordering_fields = ['create_time', 'id']
|
|
ordering = ['sort']
|
|
|