# This file was auto-generated by Fern from our API Definition. import datetime as dt import typing from json.decoder import JSONDecodeError from ..commons.errors.access_denied_error import AccessDeniedError from ..commons.errors.error import Error from ..commons.errors.method_not_allowed_error import MethodNotAllowedError from ..commons.errors.not_found_error import NotFoundError from ..commons.errors.unauthorized_error import UnauthorizedError from ..commons.types.dataset_run_item import DatasetRunItem from ..core.api_error import ApiError from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper from ..core.http_response import AsyncHttpResponse, HttpResponse from ..core.pydantic_utilities import parse_obj_as from ..core.request_options import RequestOptions from .types.paginated_dataset_run_items import PaginatedDatasetRunItems # this is used as the default value for optional parameters OMIT = typing.cast(typing.Any, ...) class RawDatasetRunItemsClient: def __init__(self, *, client_wrapper: SyncClientWrapper): self._client_wrapper = client_wrapper def create( self, *, run_name: str, dataset_item_id: str, run_description: typing.Optional[str] = OMIT, metadata: typing.Optional[typing.Any] = OMIT, observation_id: typing.Optional[str] = OMIT, trace_id: typing.Optional[str] = OMIT, dataset_version: typing.Optional[dt.datetime] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[DatasetRunItem]: """ Create a dataset run item Parameters ---------- run_name : str dataset_item_id : str run_description : typing.Optional[str] Description of the run. If run exists, description will be updated. metadata : typing.Optional[typing.Any] Metadata of the dataset run, updates run if run already exists observation_id : typing.Optional[str] trace_id : typing.Optional[str] traceId should always be provided. For compatibility with older SDK versions it can also be inferred from the provided observationId. dataset_version : typing.Optional[dt.datetime] ISO 8601 timestamp (RFC 3339, Section 5.6) in UTC (e.g., "2026-01-21T14:35:42Z"). Specifies the dataset version to use for this experiment run. If provided, the experiment will use dataset items as they existed at or before this timestamp. If not provided, uses the latest version of dataset items. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[DatasetRunItem] """ _response = self._client_wrapper.httpx_client.request( "api/public/dataset-run-items", method="POST", json={ "runName": run_name, "runDescription": run_description, "metadata": metadata, "datasetItemId": dataset_item_id, "observationId": observation_id, "traceId": trace_id, "datasetVersion": dataset_version, }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( DatasetRunItem, parse_obj_as( type_=DatasetRunItem, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 400: raise Error( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 401: raise UnauthorizedError( 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 AccessDeniedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 405: raise MethodNotAllowedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 404: raise NotFoundError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # 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 list( self, *, dataset_id: str, run_name: str, page: typing.Optional[int] = None, limit: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None, ) -> HttpResponse[PaginatedDatasetRunItems]: """ List dataset run items Parameters ---------- dataset_id : str run_name : str page : typing.Optional[int] page number, starts at 1 limit : typing.Optional[int] limit of items per page request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[PaginatedDatasetRunItems] """ _response = self._client_wrapper.httpx_client.request( "api/public/dataset-run-items", method="GET", params={ "datasetId": dataset_id, "runName": run_name, "page": page, "limit": limit, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( PaginatedDatasetRunItems, parse_obj_as( type_=PaginatedDatasetRunItems, # type: ignore object_=_response.json(), ), ) return HttpResponse(response=_response, data=_data) if _response.status_code == 400: raise Error( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 401: raise UnauthorizedError( 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 AccessDeniedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 405: raise MethodNotAllowedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 404: raise NotFoundError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # 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 AsyncRawDatasetRunItemsClient: def __init__(self, *, client_wrapper: AsyncClientWrapper): self._client_wrapper = client_wrapper async def create( self, *, run_name: str, dataset_item_id: str, run_description: typing.Optional[str] = OMIT, metadata: typing.Optional[typing.Any] = OMIT, observation_id: typing.Optional[str] = OMIT, trace_id: typing.Optional[str] = OMIT, dataset_version: typing.Optional[dt.datetime] = OMIT, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[DatasetRunItem]: """ Create a dataset run item Parameters ---------- run_name : str dataset_item_id : str run_description : typing.Optional[str] Description of the run. If run exists, description will be updated. metadata : typing.Optional[typing.Any] Metadata of the dataset run, updates run if run already exists observation_id : typing.Optional[str] trace_id : typing.Optional[str] traceId should always be provided. For compatibility with older SDK versions it can also be inferred from the provided observationId. dataset_version : typing.Optional[dt.datetime] ISO 8601 timestamp (RFC 3339, Section 5.6) in UTC (e.g., "2026-01-21T14:35:42Z"). Specifies the dataset version to use for this experiment run. If provided, the experiment will use dataset items as they existed at or before this timestamp. If not provided, uses the latest version of dataset items. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[DatasetRunItem] """ _response = await self._client_wrapper.httpx_client.request( "api/public/dataset-run-items", method="POST", json={ "runName": run_name, "runDescription": run_description, "metadata": metadata, "datasetItemId": dataset_item_id, "observationId": observation_id, "traceId": trace_id, "datasetVersion": dataset_version, }, request_options=request_options, omit=OMIT, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( DatasetRunItem, parse_obj_as( type_=DatasetRunItem, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 400: raise Error( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 401: raise UnauthorizedError( 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 AccessDeniedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 405: raise MethodNotAllowedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 404: raise NotFoundError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # 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 list( self, *, dataset_id: str, run_name: str, page: typing.Optional[int] = None, limit: typing.Optional[int] = None, request_options: typing.Optional[RequestOptions] = None, ) -> AsyncHttpResponse[PaginatedDatasetRunItems]: """ List dataset run items Parameters ---------- dataset_id : str run_name : str page : typing.Optional[int] page number, starts at 1 limit : typing.Optional[int] limit of items per page request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[PaginatedDatasetRunItems] """ _response = await self._client_wrapper.httpx_client.request( "api/public/dataset-run-items", method="GET", params={ "datasetId": dataset_id, "runName": run_name, "page": page, "limit": limit, }, request_options=request_options, ) try: if 200 <= _response.status_code < 300: _data = typing.cast( PaginatedDatasetRunItems, parse_obj_as( type_=PaginatedDatasetRunItems, # type: ignore object_=_response.json(), ), ) return AsyncHttpResponse(response=_response, data=_data) if _response.status_code == 400: raise Error( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 401: raise UnauthorizedError( 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 AccessDeniedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 405: raise MethodNotAllowedError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # type: ignore object_=_response.json(), ), ), ) if _response.status_code == 404: raise NotFoundError( headers=dict(_response.headers), body=typing.cast( typing.Any, parse_obj_as( type_=typing.Any, # 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, )