# 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 ..errors.unprocessable_entity_error import UnprocessableEntityError from ..types.formalities import Formalities from ..types.gender import Gender from ..types.http_validation_error import HttpValidationError from ..types.languages import Languages from ..types.orchestrator_pipeline_result import OrchestratorPipelineResult from ..types.translation_result import TranslationResult # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class RawTranslationClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def translation_stream( self, *, source_language: Languages, target_language: Languages, text: str, traceparent: typing.Optional[str] = None, formality: typing.Optional[Formalities] = OMIT, gender: typing.Optional[Gender] = OMIT, team_id: typing.Optional[int] = OMIT, project_name: typing.Optional[str] = OMIT, project_description: typing.Optional[str] = OMIT, folder_id: typing.Optional[int] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.Any]: """ Parameters ---------- source_language : Languages target_language : Languages text : str traceparent : typing.Optional[str] formality : typing.Optional[Formalities] gender : typing.Optional[Gender] team_id : typing.Optional[int] project_name : typing.Optional[str] project_description : typing.Optional[str] folder_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Text translation stream """ _response = self._client_wrapper.httpx_client.request( "translation/stream", method="POST", json={ "source_language": source_language, "target_language": target_language, "text": text, "formality": formality, "gender": gender, "team_id": team_id, "project_name": project_name, "project_description": project_description, "folder_id": folder_id, }, headers={ "content-type": "application/json", "traceparent": str(traceparent) if traceparent is not None else None, }, request_options=request_options, omit=OMIT, ) try: if _response is None or not _response.text.strip(): return HttpResponse(response=_response, data=None) 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 create_translation( self, *, texts: typing.Sequence[str], source_language: Languages, target_language: Languages, run_id: typing.Optional[int] = None, project_name: typing.Optional[str] = OMIT, project_description: typing.Optional[str] = OMIT, folder_id: typing.Optional[int] = OMIT, age: typing.Optional[int] = OMIT, formality: typing.Optional[Formalities] = OMIT, gender: typing.Optional[Gender] = OMIT, chosen_dictionaries: typing.Optional[typing.Sequence[int]] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.Any]: """ Parameters ---------- texts : typing.Sequence[str] source_language : Languages target_language : Languages run_id : typing.Optional[int] project_name : typing.Optional[str] project_description : typing.Optional[str] folder_id : typing.Optional[int] age : typing.Optional[int] formality : typing.Optional[Formalities] gender : typing.Optional[Gender] chosen_dictionaries : typing.Optional[typing.Sequence[int]] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response """ _response = self._client_wrapper.httpx_client.request( "translate", method="POST", params={ "run_id": run_id, }, json={ "project_name": project_name, "project_description": project_description, "folder_id": folder_id, "texts": texts, "age": age, "formality": formality, "gender": gender, "source_language": source_language, "target_language": target_language, "chosen_dictionaries": chosen_dictionaries, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if _response is None or not _response.text.strip(): return HttpResponse(response=_response, data=None) 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_translation_task_status( self, task_id: str, *, run_id: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[OrchestratorPipelineResult]: """ Parameters ---------- task_id : str run_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[OrchestratorPipelineResult] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"translate/{jsonable_encoder(task_id)}", method="GET", params={ "run_id": run_id, }, 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_translation_result( self, run_id: typing.Optional[int], *, traceparent: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[TranslationResult]: """ NOTE: This endpoint should be called only by the users to get values for their runs via API. Further we need to validate if the user has access to the run_id, otherwise we should not return the output urls. Parameters ---------- run_id : typing.Optional[int] traceparent : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[TranslationResult] Successful Response """ _response = self._client_wrapper.httpx_client.request( f"translation-result/{jsonable_encoder(run_id)}", method="GET", headers={ "traceparent": str(traceparent) if traceparent is not None else None, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( TranslationResult, parse_obj_as( type_=TranslationResult, # 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_translation_results( self, *, run_ids: typing.Sequence[int], run_id: typing.Optional[int] = None, traceparent: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[typing.Dict[str, TranslationResult]]: """ Parameters ---------- run_ids : typing.Sequence[int] run_id : typing.Optional[int] traceparent : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Dict[str, TranslationResult]] Successful Response """ _response = self._client_wrapper.httpx_client.request( "translation-results", method="POST", params={ "run_id": run_id, }, json={ "run_ids": run_ids, }, 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, TranslationResult], parse_obj_as( type_=typing.Dict[str, TranslationResult], # 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 AsyncRawTranslationClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def translation_stream( self, *, source_language: Languages, target_language: Languages, text: str, traceparent: typing.Optional[str] = None, formality: typing.Optional[Formalities] = OMIT, gender: typing.Optional[Gender] = OMIT, team_id: typing.Optional[int] = OMIT, project_name: typing.Optional[str] = OMIT, project_description: typing.Optional[str] = OMIT, folder_id: typing.Optional[int] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.Any]: """ Parameters ---------- source_language : Languages target_language : Languages text : str traceparent : typing.Optional[str] formality : typing.Optional[Formalities] gender : typing.Optional[Gender] team_id : typing.Optional[int] project_name : typing.Optional[str] project_description : typing.Optional[str] folder_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Any] Text translation stream """ _response = await self._client_wrapper.httpx_client.request( "translation/stream", method="POST", json={ "source_language": source_language, "target_language": target_language, "text": text, "formality": formality, "gender": gender, "team_id": team_id, "project_name": project_name, "project_description": project_description, "folder_id": folder_id, }, headers={ "content-type": "application/json", "traceparent": str(traceparent) if traceparent is not None else None, }, request_options=request_options, omit=OMIT, ) try: if _response is None or not _response.text.strip(): return AsyncHttpResponse(response=_response, data=None) 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 create_translation( self, *, texts: typing.Sequence[str], source_language: Languages, target_language: Languages, run_id: typing.Optional[int] = None, project_name: typing.Optional[str] = OMIT, project_description: typing.Optional[str] = OMIT, folder_id: typing.Optional[int] = OMIT, age: typing.Optional[int] = OMIT, formality: typing.Optional[Formalities] = OMIT, gender: typing.Optional[Gender] = OMIT, chosen_dictionaries: typing.Optional[typing.Sequence[int]] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.Any]: """ Parameters ---------- texts : typing.Sequence[str] source_language : Languages target_language : Languages run_id : typing.Optional[int] project_name : typing.Optional[str] project_description : typing.Optional[str] folder_id : typing.Optional[int] age : typing.Optional[int] formality : typing.Optional[Formalities] gender : typing.Optional[Gender] chosen_dictionaries : typing.Optional[typing.Sequence[int]] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Any] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "translate", method="POST", params={ "run_id": run_id, }, json={ "project_name": project_name, "project_description": project_description, "folder_id": folder_id, "texts": texts, "age": age, "formality": formality, "gender": gender, "source_language": source_language, "target_language": target_language, "chosen_dictionaries": chosen_dictionaries, }, headers={ "content-type": "application/json", }, request_options=request_options, omit=OMIT, ) try: if _response is None or not _response.text.strip(): return AsyncHttpResponse(response=_response, data=None) 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_translation_task_status( self, task_id: str, *, run_id: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[OrchestratorPipelineResult]: """ Parameters ---------- task_id : str run_id : typing.Optional[int] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[OrchestratorPipelineResult] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"translate/{jsonable_encoder(task_id)}", method="GET", params={ "run_id": run_id, }, 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_translation_result( self, run_id: typing.Optional[int], *, traceparent: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[TranslationResult]: """ NOTE: This endpoint should be called only by the users to get values for their runs via API. Further we need to validate if the user has access to the run_id, otherwise we should not return the output urls. Parameters ---------- run_id : typing.Optional[int] traceparent : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[TranslationResult] Successful Response """ _response = await self._client_wrapper.httpx_client.request( f"translation-result/{jsonable_encoder(run_id)}", method="GET", headers={ "traceparent": str(traceparent) if traceparent is not None else None, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( TranslationResult, parse_obj_as( type_=TranslationResult, # 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_translation_results( self, *, run_ids: typing.Sequence[int], run_id: typing.Optional[int] = None, traceparent: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[typing.Dict[str, TranslationResult]]: """ Parameters ---------- run_ids : typing.Sequence[int] run_id : typing.Optional[int] traceparent : typing.Optional[str] request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Dict[str, TranslationResult]] Successful Response """ _response = await self._client_wrapper.httpx_client.request( "translation-results", method="POST", params={ "run_id": run_id, }, json={ "run_ids": run_ids, }, 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, TranslationResult], parse_obj_as( type_=typing.Dict[str, TranslationResult], # 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)