from rest_framework import serializers
from .models import AiConfig


class AiConfigSerializer(serializers.ModelSerializer):
    class Meta:
        model = AiConfig
        fields = ['name', 'model_id_or_path']


class DiscordStatusSerializer(serializers.Serializer):
    online = serializers.BooleanField()
    react_to_chat = serializers.BooleanField()


class AiInstanceSerializer(serializers.Serializer):
    configuration = AiConfigSerializer()
    messages = serializers.ListField()
    discord = DiscordStatusSerializer()