o qmi%.@sddlZddlmZmZddlZddlZddlmm Z ddl m Z ddl mZmZddlmZddlmZmZdd lmZmZdd lmZdd lmZdd lmZdd lmZzddl m!Z"Wne#ypddlm!Z"YnwGdddZ$GdddZ%dS)N)asynccontextmanagercontextmanager)ApiError)AsyncClientWrapperSyncClientWrapper)RequestOptions),AsyncRawSpeechToTextTranslateStreamingClient'RawSpeechToTextTranslateStreamingClient)/AsyncSpeechToTextTranslateStreamingSocketClient*SpeechToTextTranslateStreamingSocketClient))SpeechToTextTranslateStreamingFlushSignal)0SpeechToTextTranslateStreamingHighVadSensitivity)-SpeechToTextTranslateStreamingInputAudioCodec)(SpeechToTextTranslateStreamingVadSignals)connectc@eZdZdefddZedefddZedddddddddd e j e j d d e j e d e j e d e j ede j ede j ede j e de j ede jefddZdS)$SpeechToTextTranslateStreamingClientclient_wrappercCt|d|_dSN)r)r _raw_clientselfrrf/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/speech_to_text_translate_streaming/client.py__init__z-SpeechToTextTranslateStreamingClient.__init__returncC|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawSpeechToTextTranslateStreamingClient rrrrrwith_raw_response z6SpeechToTextTranslateStreamingClient.with_raw_responseNmodel sample_ratehigh_vad_sensitivity vad_signals flush_signalinput_audio_codecapi_subscription_keyrequest_optionsr& saaras:v2.5r'r(r)r*r+r,r-c csp|jjjd} t} |dur| d|} |dur"| d|} |dur,| d|} |dur6| d|} |dur@| d|} |durJ| d|} | d | } |jj} |durat|| d <|rnd |vrn| |d z"t j | | d } t | d VWdWdS1swYWdSt jjy} z| j}|dkrt|t| ddt|t| ddd} ~ ww)u WebSocket channel for real-time speech to text streaming with English translation. **Note:** This API Reference page is provided for informational purposes only. The Try It playground may not provide the best experience for streaming audio. For optimal streaming performance, please use the SDK or implement your own WebSocket client. Parameters ---------- model : typing.Optional[typing.Literal["saaras:v2.5"]] Model to be used for speech to text translation. - **saaras:v2.5** (default): Translation model that translates audio from any spoken Indic language to English. - Example: Hindi audio → English text output sample_rate : typing.Optional[str] Audio sample rate for the WebSocket connection. When specified as a connection parameter, only 16kHz and 8kHz are supported. 8kHz is only available via this connection parameter. If not specified, defaults to 16kHz. high_vad_sensitivity : typing.Optional[SpeechToTextTranslateStreamingHighVadSensitivity] Enable high VAD (Voice Activity Detection) sensitivity vad_signals : typing.Optional[SpeechToTextTranslateStreamingVadSignals] Enable VAD signals in response flush_signal : typing.Optional[SpeechToTextTranslateStreamingFlushSignal] Signal to flush the audio buffer and finalize transcription and translation input_audio_codec : typing.Optional[SpeechToTextTranslateStreamingInputAudioCodec] Audio codec/format of the input stream. Use this when sending raw PCM audio. Supported values: wav, pcm_s16le, pcm_l16, pcm_raw. api_subscription_key : typing.Optional[str] API subscription key for authentication request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- SpeechToTextTranslateStreamingSocketClient /speech-to-text-translate/wsNr&r'r(r)r*r+?Api-Subscription-Keyadditional_headers)r2 websocket/Websocket initialized with invalid credentials. status_codeheadersbody8Unexpected error when initializing websocket connection.)r_client_wrapperget_environment productionhttpx QueryParamsadd get_headersstrupdatewebsockets_sync_clientrr websockets exceptionsInvalidStatusCoder8rdictrr&r'r(r)r*r+r,r-ws_url query_paramsr9protocolexcr8rrrr*sP6         &z,SpeechToTextTranslateStreamingClient.connect)__name__ __module__ __qualname__rrpropertyr r#rtypingOptionalLiteralrCrrrrrIteratorr rrrrrr@    rc@r))AsyncSpeechToTextTranslateStreamingClientrcCrr)r rrrrrrrz2AsyncSpeechToTextTranslateStreamingClient.__init__rcCr )z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawSpeechToTextTranslateStreamingClient r!r"rrrr#r$z;AsyncSpeechToTextTranslateStreamingClient.with_raw_responseNr%r&r.r'r(r)r*r+r,r-c Cs|jjjd} t} |dur| d|} |dur"| d|} |dur,| d|} |dur6| d|} |dur@| d|} |durJ| d|} | d | } |jj} |durat|| d <|rnd |vrn| |d z+t | | d 4IdH} t | d VWdIdHWdS1IdHswYWdSt j jy} z| j}|dkrt|t| ddt|t| ddd} ~ ww)u WebSocket channel for real-time speech to text streaming with English translation. **Note:** This API Reference page is provided for informational purposes only. The Try It playground may not provide the best experience for streaming audio. For optimal streaming performance, please use the SDK or implement your own WebSocket client. Parameters ---------- model : typing.Optional[typing.Literal["saaras:v2.5"]] Model to be used for speech to text translation. - **saaras:v2.5** (default): Translation model that translates audio from any spoken Indic language to English. - Example: Hindi audio → English text output sample_rate : typing.Optional[str] Audio sample rate for the WebSocket connection. When specified as a connection parameter, only 16kHz and 8kHz are supported. 8kHz is only available via this connection parameter. If not specified, defaults to 16kHz. high_vad_sensitivity : typing.Optional[SpeechToTextTranslateStreamingHighVadSensitivity] Enable high VAD (Voice Activity Detection) sensitivity vad_signals : typing.Optional[SpeechToTextTranslateStreamingVadSignals] Enable VAD signals in response flush_signal : typing.Optional[SpeechToTextTranslateStreamingFlushSignal] Signal to flush the audio buffer and finalize transcription and translation input_audio_codec : typing.Optional[SpeechToTextTranslateStreamingInputAudioCodec] Audio codec/format of the input stream. Use this when sending raw PCM audio. Supported values: wav, pcm_s16le, pcm_l16, pcm_raw. api_subscription_key : typing.Optional[str] API subscription key for authentication request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncSpeechToTextTranslateStreamingSocketClient r/Nr&r'r(r)r*r+r0r1r2) extra_headersr3r5r6r7r;)rr<r=r>r?r@rArBrCrDwebsockets_client_connectr rFrGrHr8rrIrJrrrrsP6         2z1AsyncSpeechToTextTranslateStreamingClient.connect)rOrPrQrrrRr r#rrSrTrUrCrrrrr AsyncIteratorr rrrrrrXrWrX)&rS contextlibrrr?websockets.exceptionsrFwebsockets.sync.clientsyncclientrEcore.api_errorrcore.client_wrapperrrcore.request_optionsr raw_clientr r socket_clientr r 5types.speech_to_text_translate_streaming_flush_signalr=types.speech_to_text_translate_streaming_high_vad_sensitivityr:types.speech_to_text_translate_streaming_input_audio_codecr4types.speech_to_text_translate_streaming_vad_signalsrwebsockets.legacy.clientrrZ ImportErrorrrXrrrrs*       k