o qmiD!@sddlZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZddlmZmZdd lmZmZdd lmZmZdd lmZmZdd lmZmZdd lm Z m!Z!ddl"m#Z#m$Z$ddl%m&Z&m'Z'GdddZ(GdddZ)dS)N)AsyncChatClient ChatClient)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncDocumentIntelligenceClientDocumentIntelligenceClient)SarvamAIEnvironment)AsyncSpeechToTextClientSpeechToTextClient)AsyncSpeechToTextJobClientSpeechToTextJobClient) AsyncSpeechToTextStreamingClientSpeechToTextStreamingClient)#AsyncSpeechToTextTranslateJobClientSpeechToTextTranslateJobClient))AsyncSpeechToTextTranslateStreamingClient$SpeechToTextTranslateStreamingClient)AsyncTextClient TextClient)AsyncTextToSpeechClientTextToSpeechClient) AsyncTextToSpeechStreamingClientTextToSpeechStreamingClientc@teZdZdZejedddddddedej e dej ej e e fd ej e d ej e d ej ejf d d ZdS)SarvamAIa9 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. Examples -------- from sarvamai import SarvamAI client = SarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) SARVAM_API_KEYNT environmentapi_subscription_keyheaderstimeoutfollow_redirects httpx_clientrr r!r"r#r$cC|dur|n |dur dn|jj}|durtddt||||dur#|n|dur.tj||dntj|d|d|_t|jd|_t |jd|_ t |jd|_ t |jd|_t|jd|_t|jd|_t|jd|_t|jd|_t|jd|_t|jd|_dSN<zcThe client must be instantiated be either passing in api_subscription_key or setting SARVAM_API_KEY)body)r"r#)r")rr r!r$r")client_wrapper)r"readrrhttpxClient_client_wrapperrtextr speech_to_textrtext_to_speechrchatrspeech_to_text_jobrspeech_to_text_translate_jobr document_intelligencerspeech_to_text_streamingr"speech_to_text_translate_streamingrtext_to_speech_streamingselfrr r!r"r#r$_defaulted_timeoutr;C/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/client.py__init__>:   zSarvamAI.__init__)__name__ __module__ __qualname____doc__r PRODUCTIONosgetenvtypingOptionalstrDictfloatboolr+r,r=r;r;r;r<r*' rc@r) AsyncSarvamAIaH 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. Examples -------- from sarvamai import AsyncSarvamAI client = AsyncSarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) rNTrrr r!r"r#r$cCr%r&)r"r*rrr+ AsyncClientr-rr.r r/rr0rr1r r2rr3rr4rr5rr6rr7r8r;r;r<r=r>zAsyncSarvamAI.__init__)r?r@rArBr rCrDrErFrGrHrIrJrKr+rNr=r;r;r;r<rMhrLrM)*rDrFr+ chat.clientrrcore.api_errorrcore.client_wrapperrrdocument_intelligence.clientrr rr speech_to_text.clientr r speech_to_text_job.clientr rspeech_to_text_streaming.clientrr#speech_to_text_translate_job.clientrr)speech_to_text_translate_streaming.clientrr text.clientrrtext_to_speech.clientrrtext_to_speech_streaming.clientrrrrMr;r;r;r<s$  O