20 lines
530 B
Python
20 lines
530 B
Python
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()
|