from rest_framework import serializers
from gerar_documento.serializer import ResumoCursorCompeltoSerializer


class RagasSerializer(ResumoCursorCompeltoSerializer):
    files = serializers.ListField(child=serializers.FileField(), required=True)
    id_modelo_do_usuario = serializers.IntegerField(required=False)
    hf_embedding = serializers.CharField(required=False, default="all-MiniLM-L6-v2")


class RagasFromTextSerializer(ResumoCursorCompeltoSerializer):
    files = None
    id_modelo_do_usuario = serializers.IntegerField(required=False, default=9)
    user_message = serializers.CharField(required=True)
    context_provided = serializers.CharField(required=False)