o oiGB@s2ddlmZddlZddlZddlZddlmZddlmZm Z ddl m Z m Z ddl mZejrddlmZmZdd lmZmZdd lmZmZdd lmZmZdd lmZmZdd lm Z m!Z!ddl"m#Z#m$Z$ddl%m&Z&m'Z'ddl(m)Z)m*Z*ddl+m,Z,m-Z-ddl.m/Z/m0Z0GdddZ1GdddZ2dS)) annotationsN)ApiError)AsyncClientWrapperSyncClientWrapper) LogConfigLogger)SarvamAIEnvironment)AsyncChatClient ChatClient)AsyncDocumentIntelligenceClientDocumentIntelligenceClient)"AsyncPronunciationDictionaryClientPronunciationDictionaryClient)AsyncSpeechToTextClientSpeechToTextClient)AsyncSpeechToTextJobClientSpeechToTextJobClient) AsyncSpeechToTextStreamingClientSpeechToTextStreamingClient)#AsyncSpeechToTextTranslateJobClientSpeechToTextTranslateJobClient))AsyncSpeechToTextTranslateStreamingClient$SpeechToTextTranslateStreamingClient)AsyncTextClient TextClient)AsyncTextToSpeechClientTextToSpeechClient) AsyncTextToSpeechStreamingClientTextToSpeechStreamingClientc@eZdZdZejedddddddd,ddZe ddZ e ddZ e ddZ e ddZ e ddZe d d!Ze d"d#Ze d$d%Ze d&d'Ze d(d)Ze d*d+ZdS)-SarvamAIam Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions. Parameters ---------- environment : SarvamAIEnvironment The environment to use for requests from the client. from .environment import SarvamAIEnvironment Defaults to SarvamAIEnvironment.PRODUCTION api_subscription_key : typing.Optional[str] headers : typing.Optional[typing.Dict[str, str]] Additional headers to send with every request. timeout : typing.Optional[float] The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced. follow_redirects : typing.Optional[bool] Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in. httpx_client : typing.Optional[httpx.Client] The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration. logging : typing.Optional[typing.Union[LogConfig, Logger]] Configure logging for the SDK. Accepts a LogConfig dict with 'level' (debug/info/warn/error), 'logger' (custom logger implementation), and 'silent' (boolean, defaults to True) fields. You can also pass a pre-configured Logger instance. Examples -------- from sarvamai import SarvamAI client = SarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) SARVAM_API_KEYNT environmentapi_subscription_keyheaderstimeoutfollow_redirects httpx_clientloggingr$r r%typing.Optional[str]r&&typing.Optional[typing.Dict[str, str]]r'typing.Optional[float]r(typing.Optional[bool]r)typing.Optional[httpx.Client]r*0typing.Optional[typing.Union[LogConfig, Logger]]c C|dur|n |dur dn|jj}|durtddt||||dur#|n|dur.tj||dntj|d||d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_dSN<zcThe client must be instantiated be either passing in api_subscription_key or setting SARVAM_API_KEY)body)r'r()r')r$r%r&r)r'r*)r'readrrhttpxClient_client_wrapper_text_speech_to_text_text_to_speech_pronunciation_dictionary_chat_speech_to_text_job_speech_to_text_translate_job_document_intelligence_speech_to_text_streaming#_speech_to_text_translate_streaming_text_to_speech_streaming selfr$r%r&r'r(r)r*_defaulted_timeoutrGh/home/aiteam/pcaa-dev/call-proccessing/stt_pipeline/venv/lib/python3.10/site-packages/sarvamai/client.py__init__G:    zSarvamAI.__init__cC*|jdurddlm}||jd|_|jS)Nr)rclient_wrapper)r9 text.clientrr8)rErrGrGrHtextq  z SarvamAI.textcCrK)Nr)rrL)r:speech_to_text.clientrr8)rErrGrGrHspeech_to_textyrPzSarvamAI.speech_to_textcCrK)Nr)rrL)r;text_to_speech.clientrr8)rErrGrGrHtext_to_speechrPzSarvamAI.text_to_speechcCrK)Nr)rrL)r<pronunciation_dictionary.clientrr8)rErrGrGrHpronunciation_dictionaryrPz!SarvamAI.pronunciation_dictionarycCrK)Nr)r rL)r= chat.clientr r8)rEr rGrGrHchatrPz SarvamAI.chatcCrK)Nr)rrL)r>speech_to_text_job.clientrr8)rErrGrGrHspeech_to_text_jobrPzSarvamAI.speech_to_text_jobcCrK)Nr)rrL)r?#speech_to_text_translate_job.clientrr8)rErrGrGrHspeech_to_text_translate_jobrPz%SarvamAI.speech_to_text_translate_jobcCrK)Nr)r rL)r@document_intelligence.clientr r8)rEr rGrGrHdocument_intelligencerPzSarvamAI.document_intelligencecCrK)Nr)rrL)rAspeech_to_text_streaming.clientrr8)rErrGrGrHspeech_to_text_streamingrPz!SarvamAI.speech_to_text_streamingcCrK)Nr)rrL)rB)speech_to_text_translate_streaming.clientrr8)rErrGrGrH"speech_to_text_translate_streaming  z+SarvamAI.speech_to_text_translate_streamingcCrK)Nr)rrL)rCtext_to_speech_streaming.clientrr8)rErrGrGrHtext_to_speech_streamingrPz!SarvamAI.text_to_speech_streaming)r$r r%r+r&r,r'r-r(r.r)r/r*r0__name__ __module__ __qualname____doc__r PRODUCTIONosgetenvrIpropertyrOrRrTrVrXrZr\r^r`rbrerGrGrGrHr!s@**           r!c@r )- AsyncSarvamAIa| Use this class to access the different functions within the SDK. You can instantiate any number of clients with different configuration that will propagate to these functions. Parameters ---------- environment : SarvamAIEnvironment The environment to use for requests from the client. from .environment import SarvamAIEnvironment Defaults to SarvamAIEnvironment.PRODUCTION api_subscription_key : typing.Optional[str] headers : typing.Optional[typing.Dict[str, str]] Additional headers to send with every request. timeout : typing.Optional[float] The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced. follow_redirects : typing.Optional[bool] Whether the default httpx client follows redirects or not, this is irrelevant if a custom httpx client is passed in. httpx_client : typing.Optional[httpx.AsyncClient] The httpx client to use for making requests, a preconfigured client is used by default, however this is useful should you want to pass in any custom httpx configuration. logging : typing.Optional[typing.Union[LogConfig, Logger]] Configure logging for the SDK. Accepts a LogConfig dict with 'level' (debug/info/warn/error), 'logger' (custom logger implementation), and 'silent' (boolean, defaults to True) fields. You can also pass a pre-configured Logger instance. Examples -------- from sarvamai import AsyncSarvamAI client = AsyncSarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) r"NTr#r$r r%r+r&r,r'r-r(r.r)"typing.Optional[httpx.AsyncClient]r*r0c Cr1r2)r'r5rrr6 AsyncClientr8r9r:r;r<r=r>r?r@rArBrCrDrGrGrHrIrJzAsyncSarvamAI.__init__cCrK)Nr)rrL)r9rNrr8)rErrGrGrHrOrPzAsyncSarvamAI.textcCrK)Nr)rrL)r:rQrr8)rErrGrGrHrR&rPzAsyncSarvamAI.speech_to_textcCrK)Nr)rrL)r;rSrr8)rErrGrGrHrT.rPzAsyncSarvamAI.text_to_speechcCrK)Nr)rrL)r<rUrr8)rErrGrGrHrV6rPz&AsyncSarvamAI.pronunciation_dictionarycCrK)Nr)r rL)r=rWr r8)rEr rGrGrHrX>rPzAsyncSarvamAI.chatcCrK)Nr)rrL)r>rYrr8)rErrGrGrHrZFrPz AsyncSarvamAI.speech_to_text_jobcCrK)Nr)rrL)r?r[rr8)rErrGrGrHr\Nrcz*AsyncSarvamAI.speech_to_text_translate_jobcCrK)Nr)r rL)r@r]r r8)rEr rGrGrHr^XrPz#AsyncSarvamAI.document_intelligencecCrK)Nr)rrL)rAr_rr8)rErrGrGrHr``rPz&AsyncSarvamAI.speech_to_text_streamingcCrK)Nr)rrL)rBrarr8)rErrGrGrHrbhs  z0AsyncSarvamAI.speech_to_text_translate_streamingcCrK)Nr)rrL)rCrdrr8)rErrGrGrHretrPz&AsyncSarvamAI.text_to_speech_streaming)r$r r%r+r&r,r'r-r(r.r)rpr*r0rfrGrGrGrHros@**           ro)3 __future__rrltypingr6core.api_errorrcore.client_wrapperrr core.loggingrrr$r TYPE_CHECKINGrWr r r]r r rUrrrQrrrYrrr_rrr[rrrarrrNrrrSrrrdrrr!rorGrGrGrHs.   .