File size: 692 Bytes
c66a7e7
1286e81
c66a7e7
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
from rest_framework import serializers
from gerar_relatorio_modelo_usuario.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)