# This file was auto-generated by Fern from our API Definition. import typing from json.decoder import JSONDecodeError from ..core.api_error import ApiError from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper from ..core.http_response import AsyncHttpResponse, HttpResponse from ..core.jsonable_encoder import jsonable_encoder from ..core.pydantic_utilities import parse_obj_as from ..core.request_options import RequestOptions from ..core.serialization import convert_and_respect_annotation_metadata from ..errors.unprocessable_entity_error import UnprocessableEntityError from ..types.http_validation_error import HttpValidationError from ..types.languages import Languages from ..types.orchestrator_pipeline_call_result import OrchestratorPipelineCallResult from ..types.orchestrator_pipeline_result import OrchestratorPipelineResult from ..types.run_i_ds_request_payload import RunIDsRequestPayload from ..types.subtitle_formatting_options import SubtitleFormattingOptions from ..types.transcript_data_type import TranscriptDataType from ..types.transcript_file_format import TranscriptFileFormat # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class RawSubtitlesClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def create_subtitle( self, *, media_url: str, source_language: Languages, target_languages: typing.Sequence[Languages], project_name: typing.Optional[str] = OMIT, project_description: typing.Optional[str] = OMIT, folder_id: typing.Optional[int] = OMIT, transcription_mode: typing.Optional[typing.Literal["fast", "slow"]] = "fast", formatting_options: typing.Optional[typing.Union[SubtitleFormattingOptions, typing.Dict[str, typing.Any]]] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[OrchestratorPipelineCallResult]: """ Parameters ---------- media_url : str source_language : Languages target_languages : typing.Sequence[Languages] project_name : typing.Optional[str] project_description : typing.Optional[str] folder_id : typing.Optional[int] transcription_mode : typing.Optional[typing.Literal["fast", "slow"]] Transcription mode: `fast` (default) or `slow`. formatting_options : typing.Optional[typing.Union[SubtitleFormattingOptions, typing.Dict[str, typing.Any]]] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[OrchestratorPipelineCallResult] Successful Response """ _response = self._client_wrapper.httpx_client.request( "sub", method="POST", json={ "project_name": project_name, "project_description": project_description, "folder_id": folder_id, "media_url": media_url, "source_language": source_language, "target_languages": target_languages, "transcription_mode": transcription_mode, "formatting_options": _serialize_formatting_options(formatting_options), }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( OrchestratorPipelineCallResult, parse_obj_as( type_=OrchestratorPipelineCallResult, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def get_subtitle_task_status( self, task_id: str, *, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[OrchestratorPipelineResult]: """ Parameters ---------- task_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[OrchestratorPipelineResult] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"sub/{jsonable_encoder(task_id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( OrchestratorPipelineResult, parse_obj_as( type_=OrchestratorPipelineResult, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def get_subtitle_result( self, run_id: typing.Optional[int], *, format_type: typing.Optional[TranscriptFileFormat] = None, data_type: typing.Optional[TranscriptDataType] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.Any]: """ Parameters ---------- run_id : typing.Optional[int] format_type : typing.Optional[TranscriptFileFormat] data_type : typing.Optional[TranscriptDataType] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"sub-result/{jsonable_encoder(run_id)}", method="GET", params={ "format_type": format_type, "data_type": data_type, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def get_subtitle_result_for_language( self, run_id: typing.Optional[int], language: Languages, *, format_type: typing.Optional[TranscriptFileFormat] = None, data_type: typing.Optional[TranscriptDataType] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.Any]: """ Parameters ---------- run_id : typing.Optional[int] language : Languages format_type : typing.Optional[TranscriptFileFormat] data_type : typing.Optional[TranscriptDataType] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"sub-result/{jsonable_encoder(run_id)}/{jsonable_encoder(language)}", method="GET", params={ "format_type": format_type, "data_type": data_type, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def get_subtitle_results( self, *, run_ids: typing.Sequence[int], format_type: typing.Optional[TranscriptFileFormat] = None, data_type: typing.Optional[TranscriptDataType] = None, traceparent: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.Dict[str, typing.Any]]: """ Parameters ---------- run_ids : typing.Sequence[int] Between 2 and 5 unique run IDs. For a single run, use ``get_subtitle_result`` instead. format_type : typing.Optional[TranscriptFileFormat] data_type : typing.Optional[TranscriptDataType] traceparent : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Dict[str, typing.Any]] Successful Response """ _response = self._client_wrapper.httpx_client.request( "sub-results", method="POST", params={ "format_type": format_type, "data_type": data_type, }, json=jsonable_encoder( convert_and_respect_annotation_metadata( object_=RunIDsRequestPayload(run_ids=list(run_ids)), annotation=RunIDsRequestPayload, direction="write", ) ), headers={ "content-type": "application/json", "traceparent": str(traceparent) if traceparent is not None else None, }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.Dict[str, typing.Any], parse_obj_as( type_=typing.Dict[str, typing.Any], # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) class AsyncRawSubtitlesClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def create_subtitle( self, *, media_url: str, source_language: Languages, target_languages: typing.Sequence[Languages], project_name: typing.Optional[str] = OMIT, project_description: typing.Optional[str] = OMIT, folder_id: typing.Optional[int] = OMIT, transcription_mode: typing.Optional[typing.Literal["fast", "slow"]] = "fast", formatting_options: typing.Optional[typing.Union[SubtitleFormattingOptions, typing.Dict[str, typing.Any]]] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[OrchestratorPipelineCallResult]: """ Parameters ---------- media_url : str source_language : Languages target_languages : typing.Sequence[Languages] project_name : typing.Optional[str] project_description : typing.Optional[str] folder_id : typing.Optional[int] transcription_mode : typing.Optional[typing.Literal["fast", "slow"]] Transcription mode: `fast` (default) or `slow`. formatting_options : typing.Optional[typing.Union[SubtitleFormattingOptions, typing.Dict[str, typing.Any]]] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[OrchestratorPipelineCallResult] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "sub", method="POST", json={ "project_name": project_name, "project_description": project_description, "folder_id": folder_id, "media_url": media_url, "source_language": source_language, "target_languages": target_languages, "transcription_mode": transcription_mode, "formatting_options": _serialize_formatting_options(formatting_options), }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( OrchestratorPipelineCallResult, parse_obj_as( type_=OrchestratorPipelineCallResult, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def get_subtitle_task_status( self, task_id: str, *, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[OrchestratorPipelineResult]: """ Parameters ---------- task_id : str request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[OrchestratorPipelineResult] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"sub/{jsonable_encoder(task_id)}", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( OrchestratorPipelineResult, parse_obj_as( type_=OrchestratorPipelineResult, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def get_subtitle_result( self, run_id: typing.Optional[int], *, format_type: typing.Optional[TranscriptFileFormat] = None, data_type: typing.Optional[TranscriptDataType] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.Any]: """ Parameters ---------- run_id : typing.Optional[int] format_type : typing.Optional[TranscriptFileFormat] data_type : typing.Optional[TranscriptDataType] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Any] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"sub-result/{jsonable_encoder(run_id)}", method="GET", params={ "format_type": format_type, "data_type": data_type, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def get_subtitle_result_for_language( self, run_id: typing.Optional[int], language: Languages, *, format_type: typing.Optional[TranscriptFileFormat] = None, data_type: typing.Optional[TranscriptDataType] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.Any]: """ Parameters ---------- run_id : typing.Optional[int] language : Languages format_type : typing.Optional[TranscriptFileFormat] data_type : typing.Optional[TranscriptDataType] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Any] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"sub-result/{jsonable_encoder(run_id)}/{jsonable_encoder(language)}", method="GET", params={ "format_type": format_type, "data_type": data_type, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) async def get_subtitle_results( self, *, run_ids: typing.Sequence[int], format_type: typing.Optional[TranscriptFileFormat] = None, data_type: typing.Optional[TranscriptDataType] = None, traceparent: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.Dict[str, typing.Any]]: """ Parameters ---------- run_ids : typing.Sequence[int] Between 2 and 5 unique run IDs. For a single run, use ``get_subtitle_result`` instead. format_type : typing.Optional[TranscriptFileFormat] data_type : typing.Optional[TranscriptDataType] traceparent : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Dict[str, typing.Any]] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "sub-results", method="POST", params={ "format_type": format_type, "data_type": data_type, }, json=jsonable_encoder( convert_and_respect_annotation_metadata( object_=RunIDsRequestPayload(run_ids=list(run_ids)), annotation=RunIDsRequestPayload, direction="write", ) ), headers={ "content-type": "application/json", "traceparent": str(traceparent) if traceparent is not None else None, }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( typing.Dict[str, typing.Any], parse_obj_as( type_=typing.Dict[str, typing.Any], # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 422: raise UnprocessableEntityError( headers=dict(_response.headers), body=typing.cast( HttpValidationError, parse_obj_as( type_=HttpValidationError, # type: ignore object_=_response.json(), ), ), ) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) def _serialize_formatting_options( formatting_options: typing.Optional[typing.Union[SubtitleFormattingOptions, typing.Dict[str, typing.Any]]], ) -> typing.Any: if formatting_options is OMIT or formatting_options is None: return formatting_options if isinstance(formatting_options, dict): return formatting_options return convert_and_respect_annotation_metadata( object_=formatting_options, annotation=SubtitleFormattingOptions, direction="write" )