o qmi"@sddlZddlmZmZddlZddlZddlmm Z ddl m Z ddl mZmZddlmZddlmZmZdd lmZmZdd lmZdd lmZzdd lmZWneyddd lmZYnwGd ddZ GdddZ!dS)N)asynccontextmanagercontextmanager)ApiError)AsyncClientWrapperSyncClientWrapper)RequestOptions)#AsyncRawTextToSpeechStreamingClientRawTextToSpeechStreamingClient)&AsyncTextToSpeechStreamingSocketClient!TextToSpeechStreamingSocketClient)TextToSpeechStreamingModel)(TextToSpeechStreamingSendCompletionEvent)connectc @xeZdZdefddZedefddZedddddd e j e d e j e d e j e d e j ede jef d dZdS)TextToSpeechStreamingClientclient_wrappercCt|d|_dSN)r)r _raw_clientselfrr\/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/text_to_speech_streaming/client.py__init__z$TextToSpeechStreamingClient.__init__returncC|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawTextToSpeechStreamingClient rrrrrwith_raw_response z-TextToSpeechStreamingClient.with_raw_responseNmodelsend_completion_eventapi_subscription_keyrequest_optionsr$r%r&r'c cs |jjjd}t}|dur|d|}|dur"|d|}|d|}|jj}|dur9t||d<|rFd|vrF| |dz"t j ||d}t |d VWdWdS1sawYWdSt jjy} z| j} | d krt| t|d d t| t|d d d} ~ ww)a WebSocket channel for real-time TTS synthesis. **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. **Model-Specific Notes:** - **bulbul:v2:** Supports pitch, loudness, pace (0.3-3.0). Default sample rate: 22050 Hz. - **bulbul:v3-beta:** Does NOT support pitch/loudness. Pace range: 0.5-2.0. Supports temperature parameter. Default sample rate: 24000 Hz. Preprocessing is always enabled. Parameters ---------- model : typing.Optional[TextToSpeechStreamingModel] Text to speech model to use. - **bulbul:v2** (default): Standard TTS model with pitch/loudness support - **bulbul:v3-beta**: Advanced model with temperature control (no pitch/loudness) send_completion_event : typing.Optional[TextToSpeechStreamingSendCompletionEvent] Enable completion event notifications when TTS generation finishes. When set to true, an event message will be sent when the final audio chunk has been generated. api_subscription_key : typing.Optional[str] API subscription key for authentication request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- TextToSpeechStreamingSocketClient /text-to-speech/wsNr$r%?Api-Subscription-Keyadditional_headers)r+ websocket/Websocket initialized with invalid credentials. status_codeheadersbody8Unexpected error when initializing websocket connection.)r_client_wrapperget_environment productionhttpx QueryParamsadd get_headersstrupdatewebsockets_sync_clientrr websockets exceptionsInvalidStatusCoder1rdict rr$r%r&r'ws_url query_paramsr2protocolexcr1rrrr&s@(     &z#TextToSpeechStreamingClient.connect)__name__ __module__ __qualname__rrpropertyr r!rtypingOptionalrrr<rIteratorr rrrrrr( rc @r) AsyncTextToSpeechStreamingClientrcCrr)r rrrrrrmrz)AsyncTextToSpeechStreamingClient.__init__rcCr)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- AsyncRawTextToSpeechStreamingClient rr rrrr!pr"z2AsyncTextToSpeechStreamingClient.with_raw_responseNr#r$r%r&r'c Cs2|jjjd}t}|dur|d|}|dur"|d|}|d|}|jj}|dur9t||d<|rFd|vrF| |dz+t ||d4IdH}t |d VWdIdHWdS1IdHsjwYWdSt j jy} z| j} | d krt| t|d d t| t|d d d} ~ ww)a WebSocket channel for real-time TTS synthesis. **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. **Model-Specific Notes:** - **bulbul:v2:** Supports pitch, loudness, pace (0.3-3.0). Default sample rate: 22050 Hz. - **bulbul:v3-beta:** Does NOT support pitch/loudness. Pace range: 0.5-2.0. Supports temperature parameter. Default sample rate: 24000 Hz. Preprocessing is always enabled. Parameters ---------- model : typing.Optional[TextToSpeechStreamingModel] Text to speech model to use. - **bulbul:v2** (default): Standard TTS model with pitch/loudness support - **bulbul:v3-beta**: Advanced model with temperature control (no pitch/loudness) send_completion_event : typing.Optional[TextToSpeechStreamingSendCompletionEvent] Enable completion event notifications when TTS generation finishes. When set to true, an event message will be sent when the final audio chunk has been generated. api_subscription_key : typing.Optional[str] API subscription key for authentication request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncTextToSpeechStreamingSocketClient r(Nr$r%r)r*r+) extra_headersr,r.r/r0r4)rr5r6r7r8r9r:r;r<r=websockets_client_connectr r?r@rAr1rrBrCrrrr{s@(     2z(AsyncTextToSpeechStreamingClient.connect)rHrIrJrrrKr r!rrLrMrrr<r AsyncIteratorr rrrrrrPlrOrP)"rL contextlibrrr8websockets.exceptionsr?websockets.sync.clientsyncclientr>core.api_errorrcore.client_wrapperrrcore.request_optionsr raw_clientr r socket_clientr r $types.text_to_speech_streaming_modelr4types.text_to_speech_streaming_send_completion_eventrwebsockets.legacy.clientrrR ImportErrorrrPrrrrs&     U