o i,@sddlZddlZddlmZddlmZddlmZmZddl m Z m Z ddl m Z ddlmZdd lmZd d lmZd d lmZd d lmZd dlmZeejdZGdddZGdddZdS)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)RequestOptions)construct_type)BadRequestError)"AudioGenerateRequestCallbackMethod)AudioGenerateRequestContainer)AudioGenerateRequestEncoding)AudioGenerateRequestModel.c@eZdZdefddZejddddddddddd dedej edej e d ej e d ej ej eej efd ej ed ej ed ej edej edej edej edejeejefddZdS)RawAudioClientclient_wrappercC ||_dSN_client_wrapperselfrrV/home/ubuntu/.local/lib/python3.10/site-packages/deepgram/speak/v1/audio/raw_client.py__init__ zRawAudioClient.__init__N callbackcallback_method mip_opt_outtagbit_rate containerencodingmodel sample_raterequest_optionstextrr r!r"r#r$r%r&r'r(returnc #s|jjjd|jjd|||||||| | d d|idditddttjt ffd d } | VWd d S1sAwYd S) a Convert text into natural-sounding speech using Deepgram's TTS REST API Parameters ---------- text : str The text content to be converted to speech callback : typing.Optional[str] URL to which we'll make the callback request callback_method : typing.Optional[AudioGenerateRequestCallbackMethod] HTTP method by which the callback request will be made mip_opt_out : typing.Optional[bool] Opts out requests from the Deepgram Model Improvement Program. Refer to our Docs for pricing impacts before setting this to true. https://dpgr.am/deepgram-mip tag : typing.Optional[typing.Union[str, typing.Sequence[str]]] Label your requests for the purpose of identification during usage reporting bit_rate : typing.Optional[float] The bitrate of the audio in bits per second. Choose from predefined ranges or specific values based on the encoding type. container : typing.Optional[AudioGenerateRequestContainer] Container specifies the file format wrapper for the output audio. The available options depend on the encoding type. encoding : typing.Optional[AudioGenerateRequestEncoding] Encoding allows you to specify the expected encoding of your audio output model : typing.Optional[AudioGenerateRequestModel] AI model used to process submitted text sample_rate : typing.Optional[float] Sample Rate specifies the sample rate for the output audio. Based on the encoding, different sample rates are supported. For some encodings, the sample rate is not configurable request_options : typing.Optional[RequestOptions] Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response. Returns ------- typing.Iterator[HttpResponse[typing.Iterator[bytes]]] Successful text-to-speech transformation v1/speakPOST rr r!r"r#r$r%r&r'r) content-typeapplication/jsonbase_urlmethodparamsjsonheadersr(omitr*c szMdjkr dkr*nndurddnd}tddj|dDdWSjdkrHttjt tj t tj d d  }Wnt y`tjtjjd wtjtj|d ) N, chunk_sizecss|]}|VqdSrr.0_chunkrrr ssz;RawAudioClient.generate.._stream..r9responsedatatype_object_r5body status_coder5rG)rIgetr iter_bytesreadr dictr5typingcastAnyr r4rrr) _chunk_size_response_json _responser(rr_streamns2  z(RawAudioClient.generate.._streamN) r httpx_clientstreamget_environmentbaseOMITrrNIteratorbytes rr)rr r!r"r#r$r%r&r'r(rVrrTrgenerates4;   "zRawAudioClient.generate)__name__ __module__ __qualname__rr contextlibcontextmanagerstrrNOptionalr boolUnionSequencefloatrrrr r\rr]r_rrrrrL     rc@r)AsyncRawAudioClientrcCrrrrrrrrrzAsyncRawAudioClient.__init__Nrr)rr r!r"r#r$r%r&r'r(r*c s|jjjd|jjd|||||||| | d d|idditd4IdH"d ttjt ffd d } | IdHVWdIdHdS1IdHsNwYdS) a Convert text into natural-sounding speech using Deepgram's TTS REST API Parameters ---------- text : str The text content to be converted to speech callback : typing.Optional[str] URL to which we'll make the callback request callback_method : typing.Optional[AudioGenerateRequestCallbackMethod] HTTP method by which the callback request will be made mip_opt_out : typing.Optional[bool] Opts out requests from the Deepgram Model Improvement Program. Refer to our Docs for pricing impacts before setting this to true. https://dpgr.am/deepgram-mip tag : typing.Optional[typing.Union[str, typing.Sequence[str]]] Label your requests for the purpose of identification during usage reporting bit_rate : typing.Optional[float] The bitrate of the audio in bits per second. Choose from predefined ranges or specific values based on the encoding type. container : typing.Optional[AudioGenerateRequestContainer] Container specifies the file format wrapper for the output audio. The available options depend on the encoding type. encoding : typing.Optional[AudioGenerateRequestEncoding] Encoding allows you to specify the expected encoding of your audio output model : typing.Optional[AudioGenerateRequestModel] AI model used to process submitted text sample_rate : typing.Optional[float] Sample Rate specifies the sample rate for the output audio. Based on the encoding, different sample rates are supported. For some encodings, the sample rate is not configurable request_options : typing.Optional[RequestOptions] Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response. Returns ------- typing.AsyncIterator[AsyncHttpResponse[typing.AsyncIterator[bytes]]] Successful text-to-speech transformation r+r,r-r)r.r/r0Nr*c szPdjkr dkr+nndurddnd}tddj|d2dWSIdHjdkrLttjt tj t tj d d  }Wnt ydtjtjjd wtjtj|d ) Nr7r8r9cSs|z 3dHW}|Vq6dSrrr:rrrr=sz@AsyncRawAudioClient.generate.._stream..r>r?rBrCrFrH)rIrJr aiter_bytesareadr rMr5rNrOrPr r4rrr)rQrTrrrVs6  z-AsyncRawAudioClient.generate.._stream) rrWrXrYrZr[rrN AsyncIteratorr]r^rrTrr_s4;  .zAsyncRawAudioClient.generate)r`rarbrrrcasynccontextmanagerrerNrfr rgrhrirjrrrr rorr]r_rrrrrlrkrl)rcrN json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.request_optionsr core.unchecked_base_modelr errors.bad_request_errorr ,types.audio_generate_request_callback_methodr &types.audio_generate_request_containerr%types.audio_generate_request_encodingr"types.audio_generate_request_modelrrOrPr[rrlrrrrs          u