*`iddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZe rddlZGd d eeZGd d eZdS)N)Enum)Any)Field)LanguageAlpha2)Audio)assert_soundfile_installedis_soundfile_installed)BaseCompletionRequest)RawAudioceZdZdZdZdZdS) StreamingModedisabledonlineofflineN)__name__ __module__ __qualname__DISABLEDONLINEOFFLINE/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mistral_common/protocol/transcription/request.pyr r sH FGGGrr c $eZdZUdZdZedzed<dZedzed<eed<e ddZ e dzed <d Z e ed <e ejd Zeed<ddededeeeeeefffdZeddeeefde ddfdZdS)TranscriptionRequestavA class representing a request for audio transcription. This class handles the conversion of audio data into a format suitable for transcription using the OpenAI API. It includes methods to convert the request to and from the OpenAI format. Attributes: id: An optional identifier for the transcription request. model: The model to be used for transcription. audio: The audio data to be transcribed. language: The language of the input audio in ISO-639-1 format (optional). strict_audio_validation: A flag indicating whether to perform strict validation of the audio data. Nidmodelaudio.zrThe language of the input audio. Supplying the input language in ISO-639-1 format will improve language adherence.) descriptionlanguageTstrict_audio_validationzWhether to enable streaming for the transcription request. Online streaming means the audio is streamed to the server and the transcription is streamed back. Offline streaming means the audio is passed in one go to the server.)defaultr streamingrexcludekwargsreturnc |dh}tt|jjt rt j|jj}nt|jjtsJtj |jj}t j}tj ||j |j|j|d||d<|d|d<||d}||z }|D]}||d |S) a9Convert the transcription request into the OpenAI format. This method prepares the transcription request data for compatibility with the OpenAI API. It handles the conversion of audio data and additional parameters into the required format. Args: exclude: Fields to exclude from the conversion. kwargs: Additional parameters to be added to the request. Returns: The request in the OpenAI format. Raises: ImportError: If the required soundfile library is not installed. r)r$)formatrfile random_seedseed)r max_tokensr!r#N) model_dumpr isinstancerdatabytesioBytesIOstrr from_base64sfwrite audio_array sampling_rater(seekpopupdate)selfr$r%openai_requestbufferrdefault_excludeks r to_openaizTranscriptionRequest.to_openai9s/ *.')K)K"$$$ djou - - Z 00FFdjos33 3 33%djo66EZ\\F HVU.0CEL Y Y Y Y KKNNN!'v!/!3!3M!B!Bvf%%%W7"  ( (A   q$ ' ' ' 'rFr=strictc|d}|d}fd|D}| J|t|tjr|}n|j}tj ||}| |j }t||j } | |d<||d<d i|S) aCreate a TranscriptionRequest instance from an OpenAI request dictionary. This method converts an OpenAI request dictionary into a TranscriptionRequest instance, handling the conversion of audio data and other parameters. Args: openai_request: The OpenAI request dictionary. strict: A flag indicating whether to perform strict validation of the audio data. Returns: An instance of TranscriptionRequest. r)r+c:i|]\}}|jv||dv||S)N) temperaturetop_p) model_fields).0r@vclss r z4TranscriptionRequest.from_openai..tsF   1S%%%qyQBZ=Z=Z q=Z=Z=ZrN)rB)r/r(rr*r) getitemsr.r1r2getvaluer)readr from_bytes to_base64r(r ) rJr=rBr)r+converted_dict audio_bytesr audio_str raw_audios ` r from_openaiz TranscriptionRequest.from_openaids!!&))!!&))    &,,..     dBJ ' ' +--//KK)..**K V<<<OOEL11 )ELAAA "+w(,}%s$$^$$$r)r)F)rrr__doc__rr3__annotations__rr rr rr!boolr rr#tuplerdictlistrA classmethodrVrrrrrsZ  Bd E3: OOO&+e  C'''Hnt#%)T((($u& b   I})))s)tCdSVX[S[nI]D]?^))))V"%"%c3h"%"%Rh"%"%"%["%"%"%rr)r1enumrtypingrpydanticr"pydantic_extra_types.language_codermistral_common.audiormistral_common.importsrr mistral_common.protocol.baser &mistral_common.protocol.instruct.chunkr soundfiler5r3r rrrrrgs, ======&&&&&&UUUUUUUU>>>>>>;;;;;;C p%p%p%p%p%0p%p%p%p%p%r