o qmiI@sddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZddlmZmZddlmZmZeejdZ GdddZ!GdddZ"dS)N)AsyncClientWrapperSyncClientWrapper)RequestOptions)BulkJobCallbackParams)SpeechToTextJobParametersParams)BulkJobInitResponseV1)FilesDownloadResponse)FilesUploadResponse)JobStatusV1Response)SpeechToTextModel)SpeechToTextLanguage)Mode)AsyncRawSpeechToTextJobClientRawSpeechToTextJobClient)AsyncSpeechToTextJobSpeechToTextJob.c@sreZdZdefddZedefddZeddd e d e j e d e j e defd d Zddded e j e defddZddddede j ed e j e defddZdddede jed e j e defddZdddede jed e j e defddZddddddedfdede j eded ed!e j ed"e j ed e j e d e j e defd#d$Zdedefd%d&Z dS)'SpeechToTextJobClientclient_wrappercCt|d|_dSN)r)r _raw_clientselfrrV/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/speech_to_text_job/client.py__init__zSpeechToTextJobClient.__init__returncC|jS)z Retrieves a raw implementation of this client that returns raw responses. Returns ------- RawSpeechToTextJobClient rrrrrwith_raw_response z'SpeechToTextJobClient.with_raw_responseNcallbackrequest_optionsjob_parametersr&r'cC|jj|||d}|jS)a Get a job uuid, and storage folder details for speech to text bulk job v1 Parameters ---------- job_parameters : SpeechToTextJobParametersParams Job Parameters for the bulk job callback : typing.Optional[BulkJobCallbackParams] Parameters for callback URL request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- BulkJobInitResponseV1 Successful Response Examples -------- from sarvamai import SarvamAI client = SarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) client.speech_to_text_job.initialise( job_parameters={}, ) r(r&r'r initialisedatarr(r&r' _responserrrr,&s %z SpeechToTextJobClient.initialiser'job_idcCs|jj||d}|jS)ao Get the status of a speech to text bulk job V1 Parameters ---------- job_id : str The unique identifier of the job request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- JobStatusV1Response Successful Response Examples -------- from sarvamai import SarvamAI client = SarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) client.speech_to_text_job.get_status( job_id="job_id", ) r0r get_statusr-rr1r'r/rrrr3Rsz SpeechToTextJobClient.get_statusptu_idr'r6cCr))a Start a speech to text bulk job V1 Parameters ---------- job_id : str The unique identifier of the job ptu_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- JobStatusV1Response Successful Response Examples -------- from sarvamai import SarvamAI client = SarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) client.speech_to_text_job.start( job_id="job_id", ) r5rstartr-rr1r6r'r/rrrr8s$zSpeechToTextJobClient.startfilescCr))a Start a speech to text bulk job V1 Parameters ---------- job_id : str files : typing.Sequence[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- FilesUploadResponse Successful Response Examples -------- from sarvamai import SarvamAI client = SarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) client.speech_to_text_job.get_upload_links( job_id="job_id", files=["files"], ) r1r;r'rget_upload_linksr-rr1r;r'r/rrrr>r:z&SpeechToTextJobClient.get_upload_linkscCr))a Start a speech to text bulk job V1 Parameters ---------- job_id : str files : typing.Sequence[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- FilesDownloadResponse Successful Response Examples -------- from sarvamai import SarvamAI client = SarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) client.speech_to_text_job.get_download_links( job_id="job_id", files=["files"], ) r<rget_download_linksr-r?rrrrAr:z(SpeechToTextJobClient.get_download_links saarika:v2.5Fmodelmodewith_diarizationwith_timestamps language_code num_speakersc Cs.|jt||||||d||d} t| j|dS)aQ Create a new Speech-to-Text bulk job. Parameters ---------- model : SpeechToTextModel, default="saarika:v2.5" The model to use for transcription. mode : typing.Optional[Mode], default=None Mode of operation. Only applicable for saaras:v3 model. Options: transcribe, translate, indic-en, verbatim, translit, codemix with_diarization : typing.Optional[bool], default=False Whether to enable speaker diarization (distinguishing who said what). with_timestamps : typing.Optional[bool], default=False Whether to include word-level timestamps in the transcription output. language_code : typing.Optional[SpeechToTextLanguage], default=None The language code of the input audio (e.g., "hi-IN", "bn-IN"). num_speakers : typing.Optional[int], default=None The number of distinct speakers in the audio, if known. callback : typing.Optional[BulkJobCallbackParams], default=OMIT Optional callback configuration to receive job completion events. request_options : typing.Optional[RequestOptions], default=None Request-specific configuration. Returns ------- SpeechToTextJob A handle to the newly created Speech-to-Text job. )rGrCrDrHrErFr*r1client)r,rrr1 rrCrDrErFrGrHr&r'responserrr create_jobs. z SpeechToTextJobClient.create_jobcCs t||dS)aP Get an existing Speech-to-Text job handle by job ID. Parameters ---------- job_id : str The job ID of the previously created Speech-to-Text job. Returns ------- SpeechToTextJob A job handle which can be used to check status or retrieve results. rI)rrr1rrrget_job*s zSpeechToTextJobClient.get_job)!__name__ __module__ __qualname__rrpropertyrr#OMITrtypingOptionalrrrr,strr r3intr8Sequencer r>r rAr rboolr rrMrOrrrrrs - % . . +   None: await client.speech_to_text_job.initialise( job_parameters={}, ) asyncio.run(main()) r*Nr+r.rrrr,Js- z%AsyncSpeechToTextJobClient.initialiser0r1cs|jj||dIdH}|jS)a Get the status of a speech to text bulk job V1 Parameters ---------- job_id : str The unique identifier of the job request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- JobStatusV1Response Successful Response Examples -------- import asyncio from sarvamai import AsyncSarvamAI client = AsyncSarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) async def main() -> None: await client.speech_to_text_job.get_status( job_id="job_id", ) asyncio.run(main()) r0Nr2r4rrrr3~s & z%AsyncSpeechToTextJobClient.get_statusr5r6cr\)a Start a speech to text bulk job V1 Parameters ---------- job_id : str The unique identifier of the job ptu_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- JobStatusV1Response Successful Response Examples -------- import asyncio from sarvamai import AsyncSarvamAI client = AsyncSarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) async def main() -> None: await client.speech_to_text_job.start( job_id="job_id", ) asyncio.run(main()) r5Nr7r9rrrr8 , z AsyncSpeechToTextJobClient.startr;cr\)a Start a speech to text bulk job V1 Parameters ---------- job_id : str files : typing.Sequence[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- FilesUploadResponse Successful Response Examples -------- import asyncio from sarvamai import AsyncSarvamAI client = AsyncSarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) async def main() -> None: await client.speech_to_text_job.get_upload_links( job_id="job_id", files=["files"], ) asyncio.run(main()) r<Nr=r?rrrr>r]z+AsyncSpeechToTextJobClient.get_upload_linkscr\)a Start a speech to text bulk job V1 Parameters ---------- job_id : str files : typing.Sequence[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- FilesDownloadResponse Successful Response Examples -------- import asyncio from sarvamai import AsyncSarvamAI client = AsyncSarvamAI( api_subscription_key="YOUR_API_SUBSCRIPTION_KEY", ) async def main() -> None: await client.speech_to_text_job.get_download_links( job_id="job_id", files=["files"], ) asyncio.run(main()) r<Nr@r?rrrrA r]z-AsyncSpeechToTextJobClient.get_download_linksrBFrCrDrErFrGrHrc s6|jt||||||d||dIdH} t| j|dS)a@ Create a new Speech-to-Text bulk job. Parameters ---------- model : SpeechToTextModel, default="saarika:v2.5" The model to use for transcription. mode : typing.Optional[Mode], default=None Mode of operation. Only applicable for saaras:v3 model. Options: transcribe, translate, indic-en, verbatim, translit, codemix with_diarization : typing.Optional[bool], default=False Whether to enable speaker diarization (distinguishing who said what). with_timestamps : typing.Optional[bool], default=False Whether to include word-level timestamps in the transcription output. language_code : typing.Optional[SpeechToTextLanguage], default=None The language code of the input audio (e.g., "hi-IN", "bn-IN"). num_speakers : typing.Optional[int] = None The number of distinct speakers in the audio, if known. callback : typing.Optional[BulkJobCallbackParams], default=OMIT Optional callback configuration to receive job completion events. request_options : typing.Optional[RequestOptions], default=None Request-specific configuration. Returns ------- AsyncSpeechToTextJob A handle to the newly created job. )rGrCrDrErFrHr*NrI)r,rrr1rKrrrrM<s.  z%AsyncSpeechToTextJobClient.create_jobcst||dS)aU Get an existing Speech-to-Text job handle by job ID. Parameters ---------- job_id : str The job ID of the previously created speech-to-text job. Returns ------- AsyncSpeechToTextJob A job handle which can be used to check status or retrieve results. rI)rrNrrrrOxs z"AsyncSpeechToTextJobClient.get_job) rPrQrRrrrSrr#rTrrUrVrrrr,rWr r3rXr8rYr r>r rAr rrZr rMrOrrrrr[;s 5 / 6 6 3  s&          &