benchling-v3-sdk-python 5.0.0__py3-none-any.whl

benchling_v3_sdk/prediction_batch/raw_client.py

@@ -0,0 +1,1751 @@
+# 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 encode_path_param
+from ..core.pagination import AsyncPager, SyncPager
+from ..core.parse_error import ParsingError
+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.bad_request_error import BadRequestError
+from ..errors.forbidden_error import ForbiddenError
+from ..errors.internal_server_error import InternalServerError
+from ..errors.not_found_error import NotFoundError
+from ..errors.too_many_requests_error import TooManyRequestsError
+from ..types.async_task_link import AsyncTaskLink
+from ..types.general_error import GeneralError
+from ..types.internal_server_error_body import InternalServerErrorBody
+from ..types.prediction_batch import PredictionBatch
+from ..types.prediction_batch_input import PredictionBatchInput
+from ..types.prediction_batch_job_config_input import PredictionBatchJobConfigInput
+from ..types.prediction_batch_paginated_list import PredictionBatchPaginatedList
+from ..types.prediction_job import PredictionJob
+from ..types.prediction_job_paginated_list import PredictionJobPaginatedList
+from .types.batch_create_prediction_batch_response import BatchCreatePredictionBatchResponse
+from .types.list_prediction_batch_request_sort import ListPredictionBatchRequestSort
+from .types.list_prediction_jobs_prediction_batch_request_sort import ListPredictionJobsPredictionBatchRequestSort
+from pydantic import ValidationError
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawPredictionBatchClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def create(
+        self,
+        *,
+        config_schema_version: int,
+        folder_id: str,
+        job_configs: typing.Sequence[PredictionBatchJobConfigInput],
+        model: str,
+        name: str,
+        model_implementation_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[PredictionBatch]:
+        """
+        Create PredictionBatch.
+
+        Parameters
+        ----------
+        config_schema_version : int
+
+        folder_id : str
+
+        job_configs : typing.Sequence[PredictionBatchJobConfigInput]
+
+        model : str
+
+        name : str
+
+        model_implementation_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[PredictionBatch]
+            Created
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch",
+            method="POST",
+            json={
+                "configSchemaVersion": config_schema_version,
+                "folderId": folder_id,
+                "jobConfigs": convert_and_respect_annotation_metadata(
+                    object_=job_configs, annotation=typing.Sequence[PredictionBatchJobConfigInput], direction="write"
+                ),
+                "model": model,
+                "modelImplementationVersion": model_implementation_version,
+                "name": name,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PredictionBatch,
+                    parse_obj_as(
+                        type_=PredictionBatch,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def list(
+        self,
+        *,
+        archive_reason_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        archived_any_of: typing.Optional[typing.Union[bool, typing.Sequence[bool]]] = None,
+        created_at_gt: typing.Optional[str] = None,
+        created_at_gte: typing.Optional[str] = None,
+        created_at_lt: typing.Optional[str] = None,
+        created_at_lte: typing.Optional[str] = None,
+        creator_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        id_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        model_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        modified_at_gt: typing.Optional[str] = None,
+        modified_at_gte: typing.Optional[str] = None,
+        modified_at_lt: typing.Optional[str] = None,
+        modified_at_lte: typing.Optional[str] = None,
+        next_token: typing.Optional[str] = None,
+        omit: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        page_size: typing.Optional[int] = None,
+        returning: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        sort: typing.Optional[ListPredictionBatchRequestSort] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SyncPager[PredictionBatch, PredictionBatchPaginatedList]:
+        """
+        List PredictionBatch items.
+
+        Parameters
+        ----------
+        archive_reason_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts items to those with any of the specified archive reasons. Use "NOT_ARCHIVED" to filter for unarchived items. Use "ANY_ARCHIVED" to filter for archived items regardless of reason. Use "ANY_ARCHIVED_OR_NOT_ARCHIVED" to return items for both archived and unarchived. Comma-separated list.
+
+        archived_any_of : typing.Optional[typing.Union[bool, typing.Sequence[bool]]]
+            If true, returns archived items. If false, returns unarchived items. If both true and false, returns archived and unarchived items. Comma-separated list.
+
+        created_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created after the specified time. e.g. > 2017-04-30.
+
+        created_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or after the specified time. e.g. >= 2017-04-30.
+
+        created_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created before the specified time. e.g. < 2017-04-30.
+
+        created_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or before the specified time. e.g. <= 2017-04-30.
+
+        creator_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to those created by any of the specified user IDs. Comma-separated list.
+
+        id_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to those matching any of the specified IDs. Comma-separated list.
+
+        model_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to prediction batches using any of the specified model names. Comma-separated list.
+
+        modified_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified after the specified time. e.g. > 2017-04-30.
+
+        modified_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or after the specified time. e.g. >= 2017-04-30.
+
+        modified_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified before the specified time. e.g. < 2017-04-30.
+
+        modified_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or before the specified time. e.g. <= 2017-04-30.
+
+        next_token : typing.Optional[str]
+            Token for pagination
+
+        omit : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to omit from each returned item. Cannot overlap with returning.
+
+        page_size : typing.Optional[int]
+            Number of results to return. Defaults to 50, maximum of 100.
+
+        returning : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to include in each returned item. Cannot overlap with omit.
+
+        sort : typing.Optional[ListPredictionBatchRequestSort]
+            Method by which to order results. Valid sorts are: createdAt (created time, oldest first) and modifiedAt (modified time, oldest first). Use :asc or :desc to specify ascending or descending order. Default is modifiedAt:desc.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SyncPager[PredictionBatch, PredictionBatchPaginatedList]
+            OK
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch/items",
+            method="GET",
+            params={
+                "archiveReason.anyOf": ",".join(map(str, archive_reason_any_of))
+                if isinstance(archive_reason_any_of, (list, tuple, set))
+                else archive_reason_any_of,
+                "archived.anyOf": ",".join(map(str, archived_any_of))
+                if isinstance(archived_any_of, (list, tuple, set))
+                else archived_any_of,
+                "createdAt.gt": created_at_gt,
+                "createdAt.gte": created_at_gte,
+                "createdAt.lt": created_at_lt,
+                "createdAt.lte": created_at_lte,
+                "creator.anyOf": ",".join(map(str, creator_any_of))
+                if isinstance(creator_any_of, (list, tuple, set))
+                else creator_any_of,
+                "id.anyOf": ",".join(map(str, id_any_of)) if isinstance(id_any_of, (list, tuple, set)) else id_any_of,
+                "model.anyOf": ",".join(map(str, model_any_of))
+                if isinstance(model_any_of, (list, tuple, set))
+                else model_any_of,
+                "modifiedAt.gt": modified_at_gt,
+                "modifiedAt.gte": modified_at_gte,
+                "modifiedAt.lt": modified_at_lt,
+                "modifiedAt.lte": modified_at_lte,
+                "nextToken": next_token,
+                "omit": ",".join(map(str, omit)) if isinstance(omit, (list, tuple, set)) else omit,
+                "pageSize": page_size,
+                "returning": ",".join(map(str, returning)) if isinstance(returning, (list, tuple, set)) else returning,
+                "sort": sort,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _parsed_response = typing.cast(
+                    PredictionBatchPaginatedList,
+                    parse_obj_as(
+                        type_=PredictionBatchPaginatedList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                _items = _parsed_response.items
+                _parsed_next = _parsed_response.next_token
+                _has_next = _parsed_next is not None and _parsed_next != ""
+                _get_next = lambda: self.list(
+                    archive_reason_any_of=archive_reason_any_of,
+                    archived_any_of=archived_any_of,
+                    created_at_gt=created_at_gt,
+                    created_at_gte=created_at_gte,
+                    created_at_lt=created_at_lt,
+                    created_at_lte=created_at_lte,
+                    creator_any_of=creator_any_of,
+                    id_any_of=id_any_of,
+                    model_any_of=model_any_of,
+                    modified_at_gt=modified_at_gt,
+                    modified_at_gte=modified_at_gte,
+                    modified_at_lt=modified_at_lt,
+                    modified_at_lte=modified_at_lte,
+                    next_token=_parsed_next,
+                    omit=omit,
+                    page_size=page_size,
+                    returning=returning,
+                    sort=sort,
+                    request_options=request_options,
+                )
+                return SyncPager(has_next=_has_next, items=_items, get_next=_get_next, response=_parsed_response)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def get(
+        self,
+        prediction_batch_id: str,
+        *,
+        returning: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        omit: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[PredictionBatch]:
+        """
+        Get a single PredictionBatch by ID.
+
+        Parameters
+        ----------
+        prediction_batch_id : str
+            ID of the PredictionBatch.
+
+        returning : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to include in each returned item. Cannot overlap with omit.
+
+        omit : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to omit from each returned item. Cannot overlap with returning.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[PredictionBatch]
+            OK
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/v3/prediction-batch/{encode_path_param(prediction_batch_id)}",
+            method="GET",
+            params={
+                "returning": ",".join(map(str, returning)) if isinstance(returning, (list, tuple, set)) else returning,
+                "omit": ",".join(map(str, omit)) if isinstance(omit, (list, tuple, set)) else omit,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PredictionBatch,
+                    parse_obj_as(
+                        type_=PredictionBatch,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def list_prediction_jobs(
+        self,
+        prediction_batch_id: str,
+        *,
+        created_at_gt: typing.Optional[str] = None,
+        created_at_gte: typing.Optional[str] = None,
+        created_at_lt: typing.Optional[str] = None,
+        created_at_lte: typing.Optional[str] = None,
+        id_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        modified_at_gt: typing.Optional[str] = None,
+        modified_at_gte: typing.Optional[str] = None,
+        modified_at_lt: typing.Optional[str] = None,
+        modified_at_lte: typing.Optional[str] = None,
+        next_token: typing.Optional[str] = None,
+        page_size: typing.Optional[int] = None,
+        sort: typing.Optional[ListPredictionJobsPredictionBatchRequestSort] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> SyncPager[PredictionJob, PredictionJobPaginatedList]:
+        """
+        List PredictionJob items.
+
+        Parameters
+        ----------
+        prediction_batch_id : str
+            ID of the PredictionBatch.
+
+        created_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created after the specified time. e.g. > 2017-04-30.
+
+        created_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or after the specified time. e.g. >= 2017-04-30.
+
+        created_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created before the specified time. e.g. < 2017-04-30.
+
+        created_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or before the specified time. e.g. <= 2017-04-30.
+
+        id_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to those matching any of the specified IDs. Comma-separated list.
+
+        modified_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified after the specified time. e.g. > 2017-04-30.
+
+        modified_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or after the specified time. e.g. >= 2017-04-30.
+
+        modified_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified before the specified time. e.g. < 2017-04-30.
+
+        modified_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or before the specified time. e.g. <= 2017-04-30.
+
+        next_token : typing.Optional[str]
+            Token for pagination
+
+        page_size : typing.Optional[int]
+            Number of results to return. Defaults to 50, maximum of 100.
+
+        sort : typing.Optional[ListPredictionJobsPredictionBatchRequestSort]
+            Method by which to order results. Valid sorts are: createdAt (created time, oldest first) and modifiedAt (modified time, oldest first). Use :asc or :desc to specify ascending or descending order. Default is modifiedAt:desc.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        SyncPager[PredictionJob, PredictionJobPaginatedList]
+            OK
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/v3/prediction-batch/{encode_path_param(prediction_batch_id)}/prediction-jobs/items",
+            method="GET",
+            params={
+                "createdAt.gt": created_at_gt,
+                "createdAt.gte": created_at_gte,
+                "createdAt.lt": created_at_lt,
+                "createdAt.lte": created_at_lte,
+                "id.anyOf": ",".join(map(str, id_any_of)) if isinstance(id_any_of, (list, tuple, set)) else id_any_of,
+                "modifiedAt.gt": modified_at_gt,
+                "modifiedAt.gte": modified_at_gte,
+                "modifiedAt.lt": modified_at_lt,
+                "modifiedAt.lte": modified_at_lte,
+                "nextToken": next_token,
+                "pageSize": page_size,
+                "sort": sort,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _parsed_response = typing.cast(
+                    PredictionJobPaginatedList,
+                    parse_obj_as(
+                        type_=PredictionJobPaginatedList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                _items = _parsed_response.items
+                _parsed_next = _parsed_response.next_token
+                _has_next = _parsed_next is not None and _parsed_next != ""
+                _get_next = lambda: self.list_prediction_jobs(
+                    prediction_batch_id,
+                    created_at_gt=created_at_gt,
+                    created_at_gte=created_at_gte,
+                    created_at_lt=created_at_lt,
+                    created_at_lte=created_at_lte,
+                    id_any_of=id_any_of,
+                    modified_at_gt=modified_at_gt,
+                    modified_at_gte=modified_at_gte,
+                    modified_at_lt=modified_at_lt,
+                    modified_at_lte=modified_at_lte,
+                    next_token=_parsed_next,
+                    page_size=page_size,
+                    sort=sort,
+                    request_options=request_options,
+                )
+                return SyncPager(has_next=_has_next, items=_items, get_next=_get_next, response=_parsed_response)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def batch_create(
+        self, *, items: typing.Sequence[PredictionBatchInput], request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[BatchCreatePredictionBatchResponse]:
+        """
+        Batch create PredictionBatch synchronously in one transaction. Maximum 25 items per request.
+
+        Parameters
+        ----------
+        items : typing.Sequence[PredictionBatchInput]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[BatchCreatePredictionBatchResponse]
+            Created
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch:batch-create",
+            method="POST",
+            json={
+                "items": convert_and_respect_annotation_metadata(
+                    object_=items, annotation=typing.Sequence[PredictionBatchInput], direction="write"
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    BatchCreatePredictionBatchResponse,
+                    parse_obj_as(
+                        type_=BatchCreatePredictionBatchResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    def bulk_create(
+        self, *, file_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[AsyncTaskLink]:
+        """
+        Bulk create PredictionBatch.
+
+        Parameters
+        ----------
+        file_id : str
+            The API ID of the scratch file (`scrfile_XXXXXXXX`) containing the items to import. The referenced file must be a scratch file whose upload has completed successfully.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[AsyncTaskLink]
+            Task started
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch:bulk-create",
+            method="POST",
+            json={
+                "fileId": file_id,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AsyncTaskLink,
+                    parse_obj_as(
+                        type_=AsyncTaskLink,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawPredictionBatchClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def create(
+        self,
+        *,
+        config_schema_version: int,
+        folder_id: str,
+        job_configs: typing.Sequence[PredictionBatchJobConfigInput],
+        model: str,
+        name: str,
+        model_implementation_version: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[PredictionBatch]:
+        """
+        Create PredictionBatch.
+
+        Parameters
+        ----------
+        config_schema_version : int
+
+        folder_id : str
+
+        job_configs : typing.Sequence[PredictionBatchJobConfigInput]
+
+        model : str
+
+        name : str
+
+        model_implementation_version : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[PredictionBatch]
+            Created
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch",
+            method="POST",
+            json={
+                "configSchemaVersion": config_schema_version,
+                "folderId": folder_id,
+                "jobConfigs": convert_and_respect_annotation_metadata(
+                    object_=job_configs, annotation=typing.Sequence[PredictionBatchJobConfigInput], direction="write"
+                ),
+                "model": model,
+                "modelImplementationVersion": model_implementation_version,
+                "name": name,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PredictionBatch,
+                    parse_obj_as(
+                        type_=PredictionBatch,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def list(
+        self,
+        *,
+        archive_reason_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        archived_any_of: typing.Optional[typing.Union[bool, typing.Sequence[bool]]] = None,
+        created_at_gt: typing.Optional[str] = None,
+        created_at_gte: typing.Optional[str] = None,
+        created_at_lt: typing.Optional[str] = None,
+        created_at_lte: typing.Optional[str] = None,
+        creator_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        id_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        model_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        modified_at_gt: typing.Optional[str] = None,
+        modified_at_gte: typing.Optional[str] = None,
+        modified_at_lt: typing.Optional[str] = None,
+        modified_at_lte: typing.Optional[str] = None,
+        next_token: typing.Optional[str] = None,
+        omit: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        page_size: typing.Optional[int] = None,
+        returning: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        sort: typing.Optional[ListPredictionBatchRequestSort] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncPager[PredictionBatch, PredictionBatchPaginatedList]:
+        """
+        List PredictionBatch items.
+
+        Parameters
+        ----------
+        archive_reason_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts items to those with any of the specified archive reasons. Use "NOT_ARCHIVED" to filter for unarchived items. Use "ANY_ARCHIVED" to filter for archived items regardless of reason. Use "ANY_ARCHIVED_OR_NOT_ARCHIVED" to return items for both archived and unarchived. Comma-separated list.
+
+        archived_any_of : typing.Optional[typing.Union[bool, typing.Sequence[bool]]]
+            If true, returns archived items. If false, returns unarchived items. If both true and false, returns archived and unarchived items. Comma-separated list.
+
+        created_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created after the specified time. e.g. > 2017-04-30.
+
+        created_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or after the specified time. e.g. >= 2017-04-30.
+
+        created_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created before the specified time. e.g. < 2017-04-30.
+
+        created_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or before the specified time. e.g. <= 2017-04-30.
+
+        creator_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to those created by any of the specified user IDs. Comma-separated list.
+
+        id_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to those matching any of the specified IDs. Comma-separated list.
+
+        model_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to prediction batches using any of the specified model names. Comma-separated list.
+
+        modified_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified after the specified time. e.g. > 2017-04-30.
+
+        modified_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or after the specified time. e.g. >= 2017-04-30.
+
+        modified_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified before the specified time. e.g. < 2017-04-30.
+
+        modified_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or before the specified time. e.g. <= 2017-04-30.
+
+        next_token : typing.Optional[str]
+            Token for pagination
+
+        omit : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to omit from each returned item. Cannot overlap with returning.
+
+        page_size : typing.Optional[int]
+            Number of results to return. Defaults to 50, maximum of 100.
+
+        returning : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to include in each returned item. Cannot overlap with omit.
+
+        sort : typing.Optional[ListPredictionBatchRequestSort]
+            Method by which to order results. Valid sorts are: createdAt (created time, oldest first) and modifiedAt (modified time, oldest first). Use :asc or :desc to specify ascending or descending order. Default is modifiedAt:desc.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncPager[PredictionBatch, PredictionBatchPaginatedList]
+            OK
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch/items",
+            method="GET",
+            params={
+                "archiveReason.anyOf": ",".join(map(str, archive_reason_any_of))
+                if isinstance(archive_reason_any_of, (list, tuple, set))
+                else archive_reason_any_of,
+                "archived.anyOf": ",".join(map(str, archived_any_of))
+                if isinstance(archived_any_of, (list, tuple, set))
+                else archived_any_of,
+                "createdAt.gt": created_at_gt,
+                "createdAt.gte": created_at_gte,
+                "createdAt.lt": created_at_lt,
+                "createdAt.lte": created_at_lte,
+                "creator.anyOf": ",".join(map(str, creator_any_of))
+                if isinstance(creator_any_of, (list, tuple, set))
+                else creator_any_of,
+                "id.anyOf": ",".join(map(str, id_any_of)) if isinstance(id_any_of, (list, tuple, set)) else id_any_of,
+                "model.anyOf": ",".join(map(str, model_any_of))
+                if isinstance(model_any_of, (list, tuple, set))
+                else model_any_of,
+                "modifiedAt.gt": modified_at_gt,
+                "modifiedAt.gte": modified_at_gte,
+                "modifiedAt.lt": modified_at_lt,
+                "modifiedAt.lte": modified_at_lte,
+                "nextToken": next_token,
+                "omit": ",".join(map(str, omit)) if isinstance(omit, (list, tuple, set)) else omit,
+                "pageSize": page_size,
+                "returning": ",".join(map(str, returning)) if isinstance(returning, (list, tuple, set)) else returning,
+                "sort": sort,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _parsed_response = typing.cast(
+                    PredictionBatchPaginatedList,
+                    parse_obj_as(
+                        type_=PredictionBatchPaginatedList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                _items = _parsed_response.items
+                _parsed_next = _parsed_response.next_token
+                _has_next = _parsed_next is not None and _parsed_next != ""
+
+                async def _get_next():
+                    return await self.list(
+                        archive_reason_any_of=archive_reason_any_of,
+                        archived_any_of=archived_any_of,
+                        created_at_gt=created_at_gt,
+                        created_at_gte=created_at_gte,
+                        created_at_lt=created_at_lt,
+                        created_at_lte=created_at_lte,
+                        creator_any_of=creator_any_of,
+                        id_any_of=id_any_of,
+                        model_any_of=model_any_of,
+                        modified_at_gt=modified_at_gt,
+                        modified_at_gte=modified_at_gte,
+                        modified_at_lt=modified_at_lt,
+                        modified_at_lte=modified_at_lte,
+                        next_token=_parsed_next,
+                        omit=omit,
+                        page_size=page_size,
+                        returning=returning,
+                        sort=sort,
+                        request_options=request_options,
+                    )
+
+                return AsyncPager(has_next=_has_next, items=_items, get_next=_get_next, response=_parsed_response)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def get(
+        self,
+        prediction_batch_id: str,
+        *,
+        returning: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        omit: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[PredictionBatch]:
+        """
+        Get a single PredictionBatch by ID.
+
+        Parameters
+        ----------
+        prediction_batch_id : str
+            ID of the PredictionBatch.
+
+        returning : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to include in each returned item. Cannot overlap with omit.
+
+        omit : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Comma-separated list of top-level fields to omit from each returned item. Cannot overlap with returning.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[PredictionBatch]
+            OK
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/v3/prediction-batch/{encode_path_param(prediction_batch_id)}",
+            method="GET",
+            params={
+                "returning": ",".join(map(str, returning)) if isinstance(returning, (list, tuple, set)) else returning,
+                "omit": ",".join(map(str, omit)) if isinstance(omit, (list, tuple, set)) else omit,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    PredictionBatch,
+                    parse_obj_as(
+                        type_=PredictionBatch,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def list_prediction_jobs(
+        self,
+        prediction_batch_id: str,
+        *,
+        created_at_gt: typing.Optional[str] = None,
+        created_at_gte: typing.Optional[str] = None,
+        created_at_lt: typing.Optional[str] = None,
+        created_at_lte: typing.Optional[str] = None,
+        id_any_of: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
+        modified_at_gt: typing.Optional[str] = None,
+        modified_at_gte: typing.Optional[str] = None,
+        modified_at_lt: typing.Optional[str] = None,
+        modified_at_lte: typing.Optional[str] = None,
+        next_token: typing.Optional[str] = None,
+        page_size: typing.Optional[int] = None,
+        sort: typing.Optional[ListPredictionJobsPredictionBatchRequestSort] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncPager[PredictionJob, PredictionJobPaginatedList]:
+        """
+        List PredictionJob items.
+
+        Parameters
+        ----------
+        prediction_batch_id : str
+            ID of the PredictionBatch.
+
+        created_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created after the specified time. e.g. > 2017-04-30.
+
+        created_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or after the specified time. e.g. >= 2017-04-30.
+
+        created_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created before the specified time. e.g. < 2017-04-30.
+
+        created_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those created at or before the specified time. e.g. <= 2017-04-30.
+
+        id_any_of : typing.Optional[typing.Union[str, typing.Sequence[str]]]
+            Restricts results to those matching any of the specified IDs. Comma-separated list.
+
+        modified_at_gt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified after the specified time. e.g. > 2017-04-30.
+
+        modified_at_gte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or after the specified time. e.g. >= 2017-04-30.
+
+        modified_at_lt : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified before the specified time. e.g. < 2017-04-30.
+
+        modified_at_lte : typing.Optional[str]
+            Datetime, in RFC 3339 format. Time zone defaults to UTC. Restricts results to those modified at or before the specified time. e.g. <= 2017-04-30.
+
+        next_token : typing.Optional[str]
+            Token for pagination
+
+        page_size : typing.Optional[int]
+            Number of results to return. Defaults to 50, maximum of 100.
+
+        sort : typing.Optional[ListPredictionJobsPredictionBatchRequestSort]
+            Method by which to order results. Valid sorts are: createdAt (created time, oldest first) and modifiedAt (modified time, oldest first). Use :asc or :desc to specify ascending or descending order. Default is modifiedAt:desc.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncPager[PredictionJob, PredictionJobPaginatedList]
+            OK
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/v3/prediction-batch/{encode_path_param(prediction_batch_id)}/prediction-jobs/items",
+            method="GET",
+            params={
+                "createdAt.gt": created_at_gt,
+                "createdAt.gte": created_at_gte,
+                "createdAt.lt": created_at_lt,
+                "createdAt.lte": created_at_lte,
+                "id.anyOf": ",".join(map(str, id_any_of)) if isinstance(id_any_of, (list, tuple, set)) else id_any_of,
+                "modifiedAt.gt": modified_at_gt,
+                "modifiedAt.gte": modified_at_gte,
+                "modifiedAt.lt": modified_at_lt,
+                "modifiedAt.lte": modified_at_lte,
+                "nextToken": next_token,
+                "pageSize": page_size,
+                "sort": sort,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _parsed_response = typing.cast(
+                    PredictionJobPaginatedList,
+                    parse_obj_as(
+                        type_=PredictionJobPaginatedList,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                _items = _parsed_response.items
+                _parsed_next = _parsed_response.next_token
+                _has_next = _parsed_next is not None and _parsed_next != ""
+
+                async def _get_next():
+                    return await self.list_prediction_jobs(
+                        prediction_batch_id,
+                        created_at_gt=created_at_gt,
+                        created_at_gte=created_at_gte,
+                        created_at_lt=created_at_lt,
+                        created_at_lte=created_at_lte,
+                        id_any_of=id_any_of,
+                        modified_at_gt=modified_at_gt,
+                        modified_at_gte=modified_at_gte,
+                        modified_at_lt=modified_at_lt,
+                        modified_at_lte=modified_at_lte,
+                        next_token=_parsed_next,
+                        page_size=page_size,
+                        sort=sort,
+                        request_options=request_options,
+                    )
+
+                return AsyncPager(has_next=_has_next, items=_items, get_next=_get_next, response=_parsed_response)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def batch_create(
+        self, *, items: typing.Sequence[PredictionBatchInput], request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[BatchCreatePredictionBatchResponse]:
+        """
+        Batch create PredictionBatch synchronously in one transaction. Maximum 25 items per request.
+
+        Parameters
+        ----------
+        items : typing.Sequence[PredictionBatchInput]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[BatchCreatePredictionBatchResponse]
+            Created
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch:batch-create",
+            method="POST",
+            json={
+                "items": convert_and_respect_annotation_metadata(
+                    object_=items, annotation=typing.Sequence[PredictionBatchInput], direction="write"
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    BatchCreatePredictionBatchResponse,
+                    parse_obj_as(
+                        type_=BatchCreatePredictionBatchResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+    async def bulk_create(
+        self, *, file_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[AsyncTaskLink]:
+        """
+        Bulk create PredictionBatch.
+
+        Parameters
+        ----------
+        file_id : str
+            The API ID of the scratch file (`scrfile_XXXXXXXX`) containing the items to import. The referenced file must be a scratch file whose upload has completed successfully.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[AsyncTaskLink]
+            Task started
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v3/prediction-batch:bulk-create",
+            method="POST",
+            json={
+                "fileId": file_id,
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    AsyncTaskLink,
+                    parse_obj_as(
+                        type_=AsyncTaskLink,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Any,
+                        parse_obj_as(
+                            type_=typing.Any,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        GeneralError,
+                        parse_obj_as(
+                            type_=GeneralError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        InternalServerErrorBody,
+                        parse_obj_as(
+                            type_=InternalServerErrorBody,  # 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)
+        except ValidationError as e:
+            raise ParsingError(
+                status_code=_response.status_code, headers=dict(_response.headers), body=_response.json(), cause=e
+            )
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)