U Xd@sddlZddlmZmZddlmZmZddlmZGdddeZ e GdddZ e Gd d d Z e Gd d d Z e Gd ddZdS)N)OptionalList) BaseModelField) worker_configc@s4eZdZUdZedejdZee d<dZ ee d<dS)Requestz' A dataclass to store requests .) max_lengthtextZetlanguageN) __name__ __module__ __qualname____doc__rrZmax_input_lengthr str__annotations__r rr-/home/juku/oma/23/05/asendused/dataclasses.pyr s rc@s&eZdZUeed<eed<eed<dS)SpanstartendvalueN)r r r intrrrrrrrs rc@seZdZUdZeeed<dS) ReplacementNr)r r r rrrrrrrrrs rc@s"eZdZUeed<eeed<dS) CorrectionspanZ replacementsN)r r r rrrrrrrrrs rc@speZdZUdZeedZeege d<dZ e e e d<dZ e e e d<dZee d<d Ze e d <ed d d ZdS)Responsezm A dataclass that can be used to store responses and transfer them over the message queue if needed. )default_factory correctionsNcorrected_text original_text status_codeZOKstatus)returncCstt|S)N)jsondumpsZasdictencode)selfrrrr&/szResponse.encode)r r r rfieldlistrrrrrrrrr!rr"bytesr&rrrrr$s   r)r$typingrrZpydanticrrZ gec_workerrr dataclassrrrrrrrrs