o il@s~dZddlmZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZeGdd d e jZGd d d e Zd S) zNSambaNova's Speech-to-Text service implementation for real-time transcription.) dataclass)AnyOptional)logger)SAMBANOVA_TTFS_P99)BaseWhisperSTTService Transcription)Languagec@seZdZdZdS)SambaNovaSTTSettingsz'Settings for the SambaNova STT service.N)__name__ __module__ __qualname____doc__rrR/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/services/sambanova/stt.pyr sr cseZdZdZeZdddddddeddeedeededee d eed ee d eed ee d e ddffddZ de defddZZS)SambaNovaSTTServicezSambaNova Whisper speech-to-text service. Uses SambaNova's Whisper API to convert audio to text. Requires a SambaNova API key set via the api_key parameter or SAMBANOVA_API_KEY environment variable. Nzhttps://api.sambanova.ai/v1)modelapi_keybase_urllanguageprompt temperaturesettingsttfs_p99_latencyrrrrrrrrkwargsreturnc  s|jdtjddd} |dur|dd|| _|dur$|dd|| _|dur1|dd|| _|dur>|dd|| _|durG| |t j d ||| |d| dS) aInitialize SambaNova STT service. Args: model: Whisper model to use. .. deprecated:: 0.0.105 Use ``settings=SambaNovaSTTService.Settings(model=...)`` instead. api_key: SambaNova API key. Defaults to None. base_url: API base URL. Defaults to "https://api.sambanova.ai/v1". language: Language of the audio input. .. deprecated:: 0.0.105 Use ``settings=SambaNovaSTTService.Settings(language=...)`` instead. prompt: Optional text to guide the model's style or continue a previous segment. .. deprecated:: 0.0.105 Use ``settings=SambaNovaSTTService.Settings(prompt=...)`` instead. temperature: Optional sampling temperature between 0 and 1. .. deprecated:: 0.0.105 Use ``settings=SambaNovaSTTService.Settings(temperature=...)`` instead. settings: Runtime-updatable settings. When provided alongside deprecated parameters, ``settings`` values take precedence. ttfs_p99_latency: P99 latency from speech end to final transcript in seconds. Override for your deployment. See https://github.com/pipecat-ai/stt-benchmark **kwargs: Additional arguments passed to `pipecat.services.whisper.base_stt.BaseWhisperSTTService`. zWhisper-Large-v3N)rrrrrrrr)rrrrr) Settingsr EN"_warn_init_param_moved_to_settingsrrrr apply_updatesuper__init__) selfrrrrrrrrrdefault_settings __class__rrr!&s8-      zSambaNovaSTTService.__init__audiocs|jjdus J|jrtdd|df|jjd|jjd}|jjdur+|jj|d<|jjdur7|jj|d<|jj j j di|IdHS) NzSambaNova STT does not support probability metrics (include_prob_metrics parameter has no effect). Check their docs: https://docs.sambanova.ai/docs/en/features/audio#request-parameters for more details.z audio.wavz audio/wavjson)filerresponse_formatrrrr) _settingsr_include_prob_metricsrwarningrrr_clientr&transcriptionscreate)r"r&rrrr _transcribevs     zSambaNovaSTTService._transcribe)r r r rr rrrstrr floatrr!bytesrr0 __classcell__rrr$rrsB    PrN)r dataclassesrtypingrrlogururpipecat.services.stt_latencyr!pipecat.services.whisper.base_sttrrpipecat.transcriptions.languager rr rrrrrs