45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
from rest_framework import serializers
|
||
|
|
|
||
|
|
from .models import Conversation, Message
|
||
|
|
|
||
|
|
|
||
|
|
class ConversationListSerializer(serializers.ModelSerializer):
|
||
|
|
conversation_id = serializers.UUIDField(source="uuid", read_only=True)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Conversation
|
||
|
|
fields = [
|
||
|
|
"conversation_id",
|
||
|
|
"title",
|
||
|
|
"updated_at",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class MessageSerializer(serializers.ModelSerializer):
|
||
|
|
message_id = serializers.UUIDField(source="uuid", read_only=True)
|
||
|
|
conversation_id = serializers.UUIDField(source="conversation.uuid", read_only=True)
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
model = Message
|
||
|
|
fields = [
|
||
|
|
"message_id",
|
||
|
|
"conversation_id",
|
||
|
|
"role",
|
||
|
|
"content",
|
||
|
|
"images",
|
||
|
|
"raw_response",
|
||
|
|
"created_at",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class ChatPostSerializer(serializers.Serializer):
|
||
|
|
content = serializers.CharField(required=False, allow_blank=True, default="")
|
||
|
|
images = serializers.ListField(
|
||
|
|
child=serializers.CharField(),
|
||
|
|
required=False,
|
||
|
|
default=list,
|
||
|
|
)
|
||
|
|
conversation_id = serializers.UUIDField(required=False)
|
||
|
|
title = serializers.CharField(required=False, allow_blank=True, max_length=255)
|
||
|
|
farm_context = serializers.JSONField(required=False)
|