o qmœi½>ã@sddlZddlmZmZddlZddlZddlmm Z ddl m Z ddl mZmZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"zddl#m$Z%Wne&yzddlm$Z%YnwGdd„dƒZ'Gdd„dƒZ(dS)éN)ÚasynccontextmanagerÚcontextmanageré)ÚApiError)ÚAsyncClientWrapperÚSyncClientWrapper)ÚRequestOptionsé)Ú&AsyncSpeechToTextStreamingSocketClientÚ!SpeechToTextStreamingSocketClient)Ú SpeechToTextStreamingFlushSignal)Ú'SpeechToTextStreamingHighVadSensitivity)Ú$SpeechToTextStreamingInputAudioCodec)Ú!SpeechToTextStreamingLanguageCode)ÚSpeechToTextStreamingMode)ÚSpeechToTextStreamingModel)ÚSpeechToTextStreamingVadSignals)Úconnectc@ó¦eZdZdefdd„Zeddddddddddœ dedeje deje d eje d eje d eje d ejed ejedeje dejedejefdd„ƒZdS)ÚRawSpeechToTextStreamingClientÚclient_wrappercCó ||_dS©N©Ú_client_wrapper©Úselfr©rú`/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/speech_to_text_streaming/raw_client.pyÚ__init__ó z'RawSpeechToTextStreamingClient.__init__N© ÚmodelÚmodeÚ sample_rateÚhigh_vad_sensitivityÚ vad_signalsÚ flush_signalÚinput_audio_codecÚapi_subscription_keyÚrequest_optionsÚ language_coder"r#r$r%r&r'r(r)r*Úreturnc  cs”|j ¡jd} t ¡} |dur|  d|¡} |dur!|  d|¡} |dur+|  d|¡} |dur5|  d|¡} |dur?|  d|¡} |durI|  d|¡} |durS|  d |¡} |dur]|  d |¡} | d | ›} |j ¡} | durst| ƒ| d <| r€d | vr€|  | d ¡z"t j | | d}t |dVWdƒWdS1s›wYWdSt j jyÉ}z|j}|dkr¼t|t| ƒdd‚t|t| ƒdd‚d}~ww)u– WebSocket channel for real-time speech to text streaming. **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 ---------- language_code : SpeechToTextStreamingLanguageCode Specifies the language of the input audio in BCP-47 format. **Available Options (saarika:v2.5):** - `unknown` (default): Use when the language is not known; the API will auto-detect. - `hi-IN`: Hindi - `bn-IN`: Bengali - `gu-IN`: Gujarati - `kn-IN`: Kannada - `ml-IN`: Malayalam - `mr-IN`: Marathi - `od-IN`: Odia - `pa-IN`: Punjabi - `ta-IN`: Tamil - `te-IN`: Telugu - `en-IN`: English **Additional Options (saaras:v3 only):** - `as-IN`: Assamese - `ur-IN`: Urdu - `ne-IN`: Nepali - `kok-IN`: Konkani - `ks-IN`: Kashmiri - `sd-IN`: Sindhi - `sa-IN`: Sanskrit - `sat-IN`: Santali - `mni-IN`: Manipuri - `brx-IN`: Bodo - `mai-IN`: Maithili - `doi-IN`: Dogri model : typing.Optional[SpeechToTextStreamingModel] Specifies the model to use for speech-to-text conversion. - **saarika:v2.5** (default): Transcribes audio in the spoken language. - **saaras:v3**: State-of-the-art model with flexible output formats. Supports multiple modes via the `mode` parameter: transcribe, translate, verbatim, translit, codemix. mode : typing.Optional[SpeechToTextStreamingMode] Mode of operation. **Only applicable when using saaras:v3 model.** Example audio: 'मेरा फोन नंबर है 9840950950' - **transcribe** (default): Standard transcription in the original language with proper formatting and number normalization. - Output: `मेरा फोन नंबर है 9840950950` - **translate**: Translates speech from any supported Indic language to English. - Output: `My phone number is 9840950950` - **verbatim**: Exact word-for-word transcription without normalization, preserving filler words and spoken numbers as-is. - Output: `मेरा फोन नंबर है नौ आठ चार zero नौ पांच zero नौ पांच zero` - **translit**: Romanization - Transliterates speech to Latin/Roman script only. - Output: `mera phone number hai 9840950950` - **codemix**: Code-mixed text with English words in English and Indic words in native script. - Output: `मेरा phone number है 9840950950` 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[SpeechToTextStreamingHighVadSensitivity] Enable high VAD (Voice Activity Detection) sensitivity vad_signals : typing.Optional[SpeechToTextStreamingVadSignals] Enable VAD signals in response flush_signal : typing.Optional[SpeechToTextStreamingFlushSignal] Signal to flush the audio buffer and finalize transcription input_audio_codec : typing.Optional[SpeechToTextStreamingInputAudioCodec] 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 ------- SpeechToTextStreamingSocketClient ú/speech-to-text/wsNú language-coder"r#r$r%r&r'r(ú?úApi-Subscription-KeyÚadditional_headers)r1©Ú websocketé‘ú/Websocket initialized with invalid credentials.©Ú status_codeÚheadersÚbodyú8Unexpected error when initializing websocket connection.)rÚget_environmentÚ productionÚhttpxÚ QueryParamsÚaddÚ get_headersÚstrÚupdateÚwebsockets_sync_clientrr Ú websocketsÚ exceptionsÚInvalidStatusCoder7rÚdict©rr+r"r#r$r%r&r'r(r)r*Úws_urlÚ query_paramsr8ÚprotocolÚexcr7rrrrsX€l           &ÿýý€øz&RawSpeechToTextStreamingClient.connect)Ú__name__Ú __module__Ú __qualname__rrrrÚtypingÚOptionalrrrAr rr rrÚIteratorr rrrrrróFôýüûúùø ÷ ö õ ô órc@r)Ú#AsyncRawSpeechToTextStreamingClientrcCrrrrrrrr¶r z,AsyncRawSpeechToTextStreamingClient.__init__Nr!r+r"r#r$r%r&r'r(r)r*r,c  Cs¦|j ¡jd} t ¡} |dur|  d|¡} |dur!|  d|¡} |dur+|  d|¡} |dur5|  d|¡} |dur?|  d|¡} |durI|  d|¡} |durS|  d |¡} |dur]|  d |¡} | d | ›} |j ¡} | durst| ƒ| d <| r€d | vr€|  | d ¡z+t | | d4IdHš}t |dVWdƒIdHWdS1IdHs¤wYWdSt j j yÒ}z|j}|dkrÅt|t| ƒdd‚t|t| ƒdd‚d}~ww)u› WebSocket channel for real-time speech to text streaming. **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 ---------- language_code : SpeechToTextStreamingLanguageCode Specifies the language of the input audio in BCP-47 format. **Available Options (saarika:v2.5):** - `unknown` (default): Use when the language is not known; the API will auto-detect. - `hi-IN`: Hindi - `bn-IN`: Bengali - `gu-IN`: Gujarati - `kn-IN`: Kannada - `ml-IN`: Malayalam - `mr-IN`: Marathi - `od-IN`: Odia - `pa-IN`: Punjabi - `ta-IN`: Tamil - `te-IN`: Telugu - `en-IN`: English **Additional Options (saaras:v3 only):** - `as-IN`: Assamese - `ur-IN`: Urdu - `ne-IN`: Nepali - `kok-IN`: Konkani - `ks-IN`: Kashmiri - `sd-IN`: Sindhi - `sa-IN`: Sanskrit - `sat-IN`: Santali - `mni-IN`: Manipuri - `brx-IN`: Bodo - `mai-IN`: Maithili - `doi-IN`: Dogri model : typing.Optional[SpeechToTextStreamingModel] Specifies the model to use for speech-to-text conversion. - **saarika:v2.5** (default): Transcribes audio in the spoken language. - **saaras:v3**: State-of-the-art model with flexible output formats. Supports multiple modes via the `mode` parameter: transcribe, translate, verbatim, translit, codemix. mode : typing.Optional[SpeechToTextStreamingMode] Mode of operation. **Only applicable when using saaras:v3 model.** Example audio: 'मेरा फोन नंबर है 9840950950' - **transcribe** (default): Standard transcription in the original language with proper formatting and number normalization. - Output: `मेरा फोन नंबर है 9840950950` - **translate**: Translates speech from any supported Indic language to English. - Output: `My phone number is 9840950950` - **verbatim**: Exact word-for-word transcription without normalization, preserving filler words and spoken numbers as-is. - Output: `मेरा फोन नंबर है नौ आठ चार zero नौ पांच zero नौ पांच zero` - **translit**: Romanization - Transliterates speech to Latin/Roman script only. - Output: `mera phone number hai 9840950950` - **codemix**: Code-mixed text with English words in English and Indic words in native script. - Output: `मेरा phone number है 9840950950` 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[SpeechToTextStreamingHighVadSensitivity] Enable high VAD (Voice Activity Detection) sensitivity vad_signals : typing.Optional[SpeechToTextStreamingVadSignals] Enable VAD signals in response flush_signal : typing.Optional[SpeechToTextStreamingFlushSignal] Signal to flush the audio buffer and finalize transcription input_audio_codec : typing.Optional[SpeechToTextStreamingInputAudioCodec] 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 ------- AsyncSpeechToTextStreamingSocketClient r-Nr.r"r#r$r%r&r'r(r/r0r1)Ú extra_headersr2r4r5r6r:)rr;r<r=r>r?r@rArBÚwebsockets_client_connectr rDrErFr7rrGrHrrrr¹sX€l           2ÿýý€øz+AsyncRawSpeechToTextStreamingClient.connect)rMrNrOrrrrrPrQrrrAr rr rrÚ AsyncIteratorr rrrrrrTµrSrT))rPÚ contextlibrrr=Úwebsockets.exceptionsrDÚwebsockets.sync.clientÚsyncÚclientrCÚcore.api_errorrÚcore.client_wrapperrrÚcore.request_optionsrÚ socket_clientr r Ú+types.speech_to_text_streaming_flush_signalr Ú3types.speech_to_text_streaming_high_vad_sensitivityr Ú0types.speech_to_text_streaming_input_audio_codecrÚ,types.speech_to_text_streaming_language_coderÚ#types.speech_to_text_streaming_moderÚ$types.speech_to_text_streaming_modelrÚ*types.speech_to_text_streaming_vad_signalsrÚwebsockets.legacy.clientrrVÚ ImportErrorrrTrrrrÚs0          ÿ