o qmi@s.ddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*e+ej,dZ-GdddZ.GdddZ/dS)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder) parse_obj_as)RequestOptions)'convert_and_respect_annotation_metadata)BadRequestError)ForbiddenError)InternalServerError)ServiceUnavailableError)TooManyRequestsError)UnprocessableEntityError)BulkJobCallbackParams)SpeechToTextJobParametersParams)BulkJobInitResponseV1)FilesDownloadResponse)FilesUploadResponse)JobStatusV1Response.c @eZdZdefddZedddedeje deje d e e fd d Z dd d edeje d e efddZdddd edejedeje d e efddZdd d edejedeje d e efddZdd d edejedeje d e efddZdS)RawSpeechToTextJobClientclient_wrappercC ||_dSN_client_wrapperselfrr"Z/home/ubuntu/.local/lib/python3.10/site-packages/sarvamai/speech_to_text_job/raw_client.py__init__ z!RawSpeechToTextJobClient.__init__Ncallbackrequest_optionsjob_parametersr'r(returnc CsP|jjjd|jjdt|tddt|tdddddi|td}zd |j kr-d krBnnt t t t |d }t||d WS|j d krbtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|}Wntyt|j t|j|jdwt|j t|j|d)aO Create a new speech to text bulk job and receive a job UUID and storage folder details for processing multiple audio files 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 ------- HttpResponse[BulkJobInitResponseV1] Successful Response speech-to-text/job/v1POSTwriteobject_ annotation directionr)r' content-typeapplication/jsonbase_urlmethodjsonheadersr(omit,type_r/responsedatar9body status_coder9rD)r httpx_clientrequestget_environmentbaser rrOMITrKtypingcastrr r8rr dictr9OptionalAnyrrrrrrrtextr!r)r'r( _response_data_response_jsonr"r"r# initialise"s                     z#RawSpeechToTextJobClient.initialiser(job_idc Cs6|jjjdt|d|jjd|d}zd|jkr!dkr6nntt t t | d}t ||dWS|jd krVt t|jttjtjt tjtj| dd |jd krvtt|jttjtjt tjtj| dd |jd krtt|jttjtjt tjtj| dd |jd krtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd | }Wntyt|jt|j|jdwt|jt|j|d)a Retrieve the current status and details of a speech to text bulk job, including progress and file-level information. **Rate Limiting Best Practice:** To prevent rate limit errors and ensure optimal server performance, we recommend implementing a minimum 5-millisecond delay between consecutive status polling requests. This helps maintain system stability while still providing timely status updates. Parameters ---------- job_id : str The unique identifier of the job request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[JobStatusV1Response] Successful Response speech-to-text/job/v1//statusGETr6r7r(r;r<r=r?rBrCrErFrGrHrIrJrrLrMr rNrOrKrQrRrr r8rr rSr9rTrUrrrrrrrrVr!r]r(rXrYrZr"r"r# get_statuss                    z#RawSpeechToTextJobClient.get_statusptu_idr(rfc Cs<|jjjdt|d|jjdd|i|d}zd|jkr$dkr9nntt t t | d}t ||d WS|jd krYt t|jttjtjt tjtj| dd |jd krytt|jttjtjt tjtj| dd |jd krtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd | }Wntyt|jt|j|jdwt|jt|j|d)a Start processing a speech to text bulk job after all audio files have been uploaded Parameters ---------- job_id : str The unique identifier of the job ptu_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[JobStatusV1Response] Successful Response r^/startr,rfr6r7paramsr(r;r<r=r?rBrCrErFrGrHrIrJrbr!r]rfr(rXrYrZr"r"r#start s                     zRawSpeechToTextJobClient.startfilesc C:|jjjd|jjd||dddi|td}zd|jkr#dkr8nntt t t | d }t ||d WS|jd krXt t|jttjtjt tjtj| d d |jd krxtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d | }Wntyt|jt|j|jdwt|jt|j|d)a Generate presigned upload URLs for audio files that will be processed in a speech to text bulk job Parameters ---------- job_id : str files : typing.Sequence[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[FilesUploadResponse] Successful Response "speech-to-text/job/v1/upload-filesr,r]rlr3r4r5r;r<r=r?rBrCrErFrGrHrIrJ)rrLrMrNrOrPrKrQrRrr r8rr rSr9rTrUrrrrrrrrVr!r]rlr(rXrYrZr"r"r#get_upload_links~                    z)RawSpeechToTextJobClient.get_upload_linksc Crm)a Generate presigned download URLs for the transcription output files of a completed speech to text bulk job Parameters ---------- job_id : str files : typing.Sequence[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[FilesDownloadResponse] Successful Response $speech-to-text/job/v1/download-filesr,ror3r4r5r;r<r=r?rBrCrErFrGrHrIrJ)rrLrMrNrOrPrKrQrRrr r8rr rSr9rTrUrrrrrrrrVrpr"r"r#get_download_linksrrz+RawSpeechToTextJobClient.get_download_links)__name__ __module__ __qualname__rr$rPrrQrTrr rrr[strrrdintrkSequencerrqrrtr"r"r"r#rh ~ p t trc @r)AsyncRawSpeechToTextJobClientrcCrrrr r"r"r#r$fr%z&AsyncRawSpeechToTextJobClient.__init__Nr&r)r'r(r*c sX|jjjd|jjdt|tddt|tdddddi|tdId H}zd |j kr1d krFnnt t t t |d }t||d WS|j dkrftt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|j dkrtt|jt t jt jt t jt j|d d|}Wnty t|j t|j|jdwt|j t|j|d)aT Create a new speech to text bulk job and receive a job UUID and storage folder details for processing multiple audio files 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 ------- AsyncHttpResponse[BulkJobInitResponseV1] Successful Response r+r,r-r.r2r3r4r5Nr;r<r=r?rBrCrErFrGrHrIrJ)rrLrMrNrOr rrrPrKrQrRrr r8rr rSr9rTrUrrrrrrrrVrWr"r"r#r[is                      z(AsyncRawSpeechToTextJobClient.initialiser\r]c s>|jjjdt|d|jjd|dIdH}zd|jkr%dkr:nntt t t | d}t ||d WS|jd krZt t|jttjtjt tjtj| dd |jd krztt|jttjtjt tjtj| dd |jd krtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd |jdkrtt|jttjtjt tjtj| dd | }Wntyt|jt|j|jdwt|jt|j|d)a Retrieve the current status and details of a speech to text bulk job, including progress and file-level information. **Rate Limiting Best Practice:** To prevent rate limit errors and ensure optimal server performance, we recommend implementing a minimum 5-millisecond delay between consecutive status polling requests. This helps maintain system stability while still providing timely status updates. Parameters ---------- job_id : str The unique identifier of the job request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[JobStatusV1Response] Successful Response r^r_r`raNr;r<r=r?rBrCrErFrGrHrIrJrrLrMr rNrOrKrQrRrr r8rr rSr9rTrUrrrrrrrrVrcr"r"r#rds                     z(AsyncRawSpeechToTextJobClient.get_statusrerfc sD|jjjdt|d|jjdd|i|dIdH}zd|jkr(dkr=nntt t t | d }t ||d WS|jd kr]t t|jttjtjt tjtj| d d |jd kr}tt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d |jdkrtt|jttjtjt tjtj| d d | }Wntyt|jt|j|jdwt|jt|j|d)a Start processing a speech to text bulk job after all audio files have been uploaded Parameters ---------- job_id : str The unique identifier of the job ptu_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[JobStatusV1Response] Successful Response r^rgr,rfrhNr;r<r=r?rBrCrErFrGrHrIrJr}rjr"r"r#rkRs                      z#AsyncRawSpeechToTextJobClient.startrlc B|jjjd|jjd||dddi|tdIdH}zd|jkr'd krs8                  K