diff --git a/ai/serializers.py b/ai/serializers.py index e69de29..4d0effe 100644 --- a/ai/serializers.py +++ b/ai/serializers.py @@ -0,0 +1,21 @@ +from rest_framework import serializers + +from .models import AIAnalysis + + +class AIAnalysisListSerializer(serializers.ModelSerializer): + username = serializers.CharField(source="user.username") + class_name = serializers.CharField(source="user.class_name") + + class Meta: + model = AIAnalysis + fields = ["id", "create_time", "username", "class_name"] + + +class AIAnalysisDetailSerializer(serializers.ModelSerializer): + username = serializers.CharField(source="user.username") + class_name = serializers.CharField(source="user.class_name") + + class Meta: + model = AIAnalysis + fields = ["id", "create_time", "username", "class_name", "analysis"]