o c۷i@s^ddlZddlmZddlmZddlmZmZddlm Z er%ddl Z Gddde Z dS)N)Any)Audio)assert_soundfile_installedis_soundfile_installed)BaseCompletionRequestc@seZdZUdZdZedBed<dZedBed<eed<dZedBed<dZ ee BdBed<de d e ee ffd d Z edd e ee fded dfddZdS) SpeechRequestaaRequest for text-to-speech synthesis. Supports both preset voices and voice cloning via reference audio. Attributes: id: Optional unique identifier for the speech request. model: Optional model identifier for the speech synthesis. input: Text input to be converted to speech. voice: Optional preset voice identifier (e.g., 'Neutral Male', 'Neutral Female') to use for speech synthesis. ref_audio: Optional reference audio for voice cloning, provided as a base64-encoded string or raw bytes. Takes precedence over voice when both are provided. Nidmodelinputvoice ref_audiokwargsreturncKs|jdhd}t|jdur;t|jtrt|j}nt|j}t}t j ||j |j |j d|d||d<|||S)zConvert this SpeechRequest to an OpenAI-compatible request dictionary. Args: **kwargs: Additional key-value pairs to include in the request dictionary. Returns: An OpenAI-compatible request dictionary. r )excludeN)formatr) model_dumprr isinstancebytesioBytesIOr from_base64sfwrite audio_array sampling_raterseekupdate)selfr openai_requestbufferaudior!\/home/ubuntu/vllm_env/lib/python3.10/site-packages/mistral_common/protocol/speech/request.py to_openai s      zSpeechRequest.to_openaiFrstrictcsfdd|D}|d}durRt|tjr|}n t|dr*|j}n|}t|t rNt j ||d}|j dusEJd|j | |j |d<n||d<|d}t|trb|d |d<d i|S) a:Create a SpeechRequest instance from an OpenAI-compatible request dictionary. Args: openai_request: The OpenAI request dictionary. strict: A flag indicating whether to perform strict validation of the audio data. Returns: An instance of SpeechRequest. cs i|] \}}|jvr||qSr!) model_fields).0kvclsr!r" Hs z-SpeechRequest.from_openai..r Nfile)r$z+Audio format must be set, got audio.format=r rr!)itemsgetrrrgetvaluehasattrr,readrr from_bytesr to_base64dict)r*rr$converted_dictr audio_bytesr r r!r)r" from_openai=s          zSpeechRequest.from_openai)F)__name__ __module__ __qualname____doc__rstr__annotations__r r r rrr4r# classmethodboolr7r!r!r!r"r s  &r) rtypingrmistral_common.audiormistral_common.importsrrmistral_common.protocol.baser soundfilerrr!r!r!r"s