Spaces:
Running
Running
File size: 659 Bytes
c66a7e7 7eb86f7 c66a7e7 7eb86f7 c66a7e7 7eb86f7 c66a7e7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from rest_framework import serializers
from gerar_documento.serializer import GerarDocumentoSerializer
class RagasSerializer(GerarDocumentoSerializer):
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(GerarDocumentoSerializer):
files = None
id_modelo_do_usuario = serializers.IntegerField(required=False, default=9)
user_message = serializers.CharField(required=True)
context_provided = serializers.CharField(required=False)
|