o hD@sddlmZddlZddlmZddlmZddlmZm Z m Z ddl m Z m Z ddlmZmZdd lmZd Zd Ze d Z eGd ddZGdddejZGdddejZeGdddZGdddejZGdddejZdS)) annotationsN) AsyncIterator) dataclass)AnyCallableLiteral)npnpt) AudioInputStreamedAudioInput)get_sentence_based_splitterzYYou will receive partial sentences. Do not complete the sentence, just read out the text.x) alloyashcoralechofableonyxnovasageshimmerc@sveZdZUdZdZded< dZded< ejZ ded < dZ d ed < d Z d ed< e Z ded< dZded<dS)TTSModelSettingszSettings for a TTS model.NzTTSVoice | Nonevoicerint buffer_sizez npt.DTypeLikedtypezYCallable[[npt.NDArray[np.int16 | np.float32]], npt.NDArray[np.int16 | np.float32]] | Nonetransform_datazXYou will receive partial sentences. Do not complete the sentence just read out the text.str instructionsz Callable[[str], tuple[str, str]] text_splitter float | Nonespeed)__name__ __module__ __qualname____doc__r__annotations__rr int16rrrr r r"r)r)S/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/voice/model.pyrs$      rc@s4eZdZdZeejd ddZejdd d Zd S)TTSModelz?A text-to-speech model that can convert text into audio output.returnrcCdS)zThe name of the TTS model.Nr)selfr)r)r* model_nameCzTTSModel.model_nametextsettingsrAsyncIterator[bytes]cCr-)zGiven a text string, produces a stream of audio bytes, in PCM format. Args: text: The text to convert to audio. Returns: An async iterator of audio bytes, in PCM format. Nr))r/r2r3r)r)r*runI z TTSModel.runNr,r)r2rr3rr,r4) r#r$r%r&propertyabcabstractmethodr0r5r)r)r)r*r+@sr+c@s0eZdZdZejd ddZejd ddZd S) StreamedTranscriptionSessionz(A streamed transcription of audio input.r,AsyncIterator[str]cCr-)zYields a stream of text transcriptions. Each transcription is a turn in the conversation. This method is expected to return only after `close()` is called. Nr)r.r)r)r*transcribe_turnsYsz-StreamedTranscriptionSession.transcribe_turnsNonecdS)zCloses the session.Nr)r.r)r)r*closeasz"StreamedTranscriptionSession.closeN)r,r<)r,r>)r#r$r%r&r9r:r=r@r)r)r)r*r;Vs  r;c@sHeZdZUdZdZded< dZded< dZded< dZded <dS) STTModelSettingsz$Settings for a speech-to-text model.N str | Nonepromptlanguager! temperaturezdict[str, Any] | Noneturn_detection) r#r$r%r&rCr'rDrErFr)r)r)r*rAgs     rAc@sDeZdZdZeejdddZejdd dZejdddZ dS)STTModelz>A speech-to-text model that can convert audio input into text.r,rcCr-)zThe name of the STT model.Nr)r.r)r)r*r0{r1zSTTModel.model_nameinputr r3rAtrace_include_sensitive_databool"trace_include_sensitive_audio_datacr?)aGiven an audio input, produces a text transcription. Args: input: The audio input to transcribe. settings: The settings to use for the transcription. trace_include_sensitive_data: Whether to include sensitive data in traces. trace_include_sensitive_audio_data: Whether to include sensitive audio data in traces. Returns: The text transcription of the audio input. Nr)r/rHr3rIrKr)r)r* transcribeszSTTModel.transcriber r;cr?)aCreates a new transcription session, which you can push audio to, and receive a stream of text transcriptions. Args: input: The audio input to transcribe. settings: The settings to use for the transcription. trace_include_sensitive_data: Whether to include sensitive data in traces. trace_include_sensitive_audio_data: Whether to include sensitive audio data in traces. Returns: A new transcription session. Nr)rLr)r)r*create_sessionszSTTModel.create_sessionNr7) rHr r3rArIrJrKrJr,r) rHr r3rArIrJrKrJr,r;) r#r$r%r&r8r9r:r0rMrNr)r)r)r*rGxs rGc@s0eZdZdZejd ddZejd d d Zd S)VoiceModelProviderzThe base interface for a voice model provider. A model provider is responsible for creating speech-to-text and text-to-speech models, given a name. r0rBr,rGcCr-)zGet a speech-to-text model by name. Args: model_name: The name of the model to get. Returns: The speech-to-text model. Nr)r/r0r)r)r* get_stt_modelr6z VoiceModelProvider.get_stt_modelr+cCr-)z#Get a text-to-speech model by name.Nr)rPr)r)r* get_tts_modelsz VoiceModelProvider.get_tts_modelN)r0rBr,rG)r0rBr,r+)r#r$r%r&r9r:rQrRr)r)r)r*rOs   rO) __future__rr9collections.abcr dataclassesrtypingrrrimportsr r rHr r utilsr DEFAULT_TTS_INSTRUCTIONSDEFAULT_TTS_BUFFER_SIZETTSVoicerABCr+r;rArGrOr)r)r)r*s*    *5