rapidata 2.29.1__py3-none-any.whl → 2.31.0__py3-none-any.whl

rapidata/api_client/api/validation_set_api.py

@@ -1402,7 +1402,7 @@ class ValidationSetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> AddValidationRapidResult:
-        """Adds a new validation rapid to the specified validation set using files to create the assets.
+        """(Deprecated) Adds a new validation rapid to the specified validation set using files to create the assets.
 
 
         :param validation_set_id: The ID of the validation set to add the rapid to. (required)
@@ -1432,6 +1432,7 @@ class ValidationSetApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /validation-set/{validationSetId}/rapid/files is deprecated.", DeprecationWarning)
 
         _param = self._validation_set_validation_set_id_rapid_files_post_serialize(
             validation_set_id=validation_set_id,
@@ -1476,7 +1477,7 @@ class ValidationSetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> ApiResponse[AddValidationRapidResult]:
-        """Adds a new validation rapid to the specified validation set using files to create the assets.
+        """(Deprecated) Adds a new validation rapid to the specified validation set using files to create the assets.
 
 
         :param validation_set_id: The ID of the validation set to add the rapid to. (required)
@@ -1506,6 +1507,7 @@ class ValidationSetApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /validation-set/{validationSetId}/rapid/files is deprecated.", DeprecationWarning)
 
         _param = self._validation_set_validation_set_id_rapid_files_post_serialize(
             validation_set_id=validation_set_id,
@@ -1550,7 +1552,7 @@ class ValidationSetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Adds a new validation rapid to the specified validation set using files to create the assets.
+        """(Deprecated) Adds a new validation rapid to the specified validation set using files to create the assets.
 
 
         :param validation_set_id: The ID of the validation set to add the rapid to. (required)
@@ -1580,6 +1582,7 @@ class ValidationSetApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /validation-set/{validationSetId}/rapid/files is deprecated.", DeprecationWarning)
 
         _param = self._validation_set_validation_set_id_rapid_files_post_serialize(
             validation_set_id=validation_set_id,
@@ -1688,6 +1691,343 @@ class ValidationSetApi:
 
 
 
+    @validate_call
+    def validation_set_validation_set_id_rapid_post(
+        self,
+        validation_set_id: Annotated[StrictStr, Field(description="The ID of the validation set to add the rapid to.")],
+        model: Optional[AddValidationRapidModel] = None,
+        files: Optional[List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]]] = None,
+        texts: Annotated[Optional[List[StrictStr]], Field(description="The texts to use for the rapid.")] = None,
+        urls: Annotated[Optional[List[StrictStr]], Field(description="The urls to use for the rapid")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> AddValidationRapidResult:
+        """Adds a new validation rapid to the specified validation set using files to create the assets.
+
+
+        :param validation_set_id: The ID of the validation set to add the rapid to. (required)
+        :type validation_set_id: str
+        :param model:
+        :type model: AddValidationRapidModel
+        :param files:
+        :type files: List[bytearray]
+        :param texts: The texts to use for the rapid.
+        :type texts: List[str]
+        :param urls: The urls to use for the rapid
+        :type urls: List[str]
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._validation_set_validation_set_id_rapid_post_serialize(
+            validation_set_id=validation_set_id,
+            model=model,
+            files=files,
+            texts=texts,
+            urls=urls,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AddValidationRapidResult",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        ).data
+
+
+    @validate_call
+    def validation_set_validation_set_id_rapid_post_with_http_info(
+        self,
+        validation_set_id: Annotated[StrictStr, Field(description="The ID of the validation set to add the rapid to.")],
+        model: Optional[AddValidationRapidModel] = None,
+        files: Optional[List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]]] = None,
+        texts: Annotated[Optional[List[StrictStr]], Field(description="The texts to use for the rapid.")] = None,
+        urls: Annotated[Optional[List[StrictStr]], Field(description="The urls to use for the rapid")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> ApiResponse[AddValidationRapidResult]:
+        """Adds a new validation rapid to the specified validation set using files to create the assets.
+
+
+        :param validation_set_id: The ID of the validation set to add the rapid to. (required)
+        :type validation_set_id: str
+        :param model:
+        :type model: AddValidationRapidModel
+        :param files:
+        :type files: List[bytearray]
+        :param texts: The texts to use for the rapid.
+        :type texts: List[str]
+        :param urls: The urls to use for the rapid
+        :type urls: List[str]
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._validation_set_validation_set_id_rapid_post_serialize(
+            validation_set_id=validation_set_id,
+            model=model,
+            files=files,
+            texts=texts,
+            urls=urls,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AddValidationRapidResult",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        response_data.read()
+        return self.api_client.response_deserialize(
+            response_data=response_data,
+            response_types_map=_response_types_map,
+        )
+
+
+    @validate_call
+    def validation_set_validation_set_id_rapid_post_without_preload_content(
+        self,
+        validation_set_id: Annotated[StrictStr, Field(description="The ID of the validation set to add the rapid to.")],
+        model: Optional[AddValidationRapidModel] = None,
+        files: Optional[List[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]]] = None,
+        texts: Annotated[Optional[List[StrictStr]], Field(description="The texts to use for the rapid.")] = None,
+        urls: Annotated[Optional[List[StrictStr]], Field(description="The urls to use for the rapid")] = None,
+        _request_timeout: Union[
+            None,
+            Annotated[StrictFloat, Field(gt=0)],
+            Tuple[
+                Annotated[StrictFloat, Field(gt=0)],
+                Annotated[StrictFloat, Field(gt=0)]
+            ]
+        ] = None,
+        _request_auth: Optional[Dict[StrictStr, Any]] = None,
+        _content_type: Optional[StrictStr] = None,
+        _headers: Optional[Dict[StrictStr, Any]] = None,
+        _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
+    ) -> RESTResponseType:
+        """Adds a new validation rapid to the specified validation set using files to create the assets.
+
+
+        :param validation_set_id: The ID of the validation set to add the rapid to. (required)
+        :type validation_set_id: str
+        :param model:
+        :type model: AddValidationRapidModel
+        :param files:
+        :type files: List[bytearray]
+        :param texts: The texts to use for the rapid.
+        :type texts: List[str]
+        :param urls: The urls to use for the rapid
+        :type urls: List[str]
+        :param _request_timeout: timeout setting for this request. If one
+                                 number provided, it will be total request
+                                 timeout. It can also be a pair (tuple) of
+                                 (connection, read) timeouts.
+        :type _request_timeout: int, tuple(int, int), optional
+        :param _request_auth: set to override the auth_settings for an a single
+                              request; this effectively ignores the
+                              authentication in the spec for a single request.
+        :type _request_auth: dict, optional
+        :param _content_type: force content-type for the request.
+        :type _content_type: str, Optional
+        :param _headers: set to override the headers for a single
+                         request; this effectively ignores the headers
+                         in the spec for a single request.
+        :type _headers: dict, optional
+        :param _host_index: set to override the host_index for a single
+                            request; this effectively ignores the host_index
+                            in the spec for a single request.
+        :type _host_index: int, optional
+        :return: Returns the result object.
+        """ # noqa: E501
+
+        _param = self._validation_set_validation_set_id_rapid_post_serialize(
+            validation_set_id=validation_set_id,
+            model=model,
+            files=files,
+            texts=texts,
+            urls=urls,
+            _request_auth=_request_auth,
+            _content_type=_content_type,
+            _headers=_headers,
+            _host_index=_host_index
+        )
+
+        _response_types_map: Dict[str, Optional[str]] = {
+            '200': "AddValidationRapidResult",
+        }
+        response_data = self.api_client.call_api(
+            *_param,
+            _request_timeout=_request_timeout
+        )
+        return response_data.response
+
+
+    def _validation_set_validation_set_id_rapid_post_serialize(
+        self,
+        validation_set_id,
+        model,
+        files,
+        texts,
+        urls,
+        _request_auth,
+        _content_type,
+        _headers,
+        _host_index,
+    ) -> RequestSerialized:
+
+        _host = None
+
+        _collection_formats: Dict[str, str] = {
+            'files': 'multi',
+            'texts': 'multi',
+            'urls': 'multi',
+        }
+
+        _path_params: Dict[str, str] = {}
+        _query_params: List[Tuple[str, str]] = []
+        _header_params: Dict[str, Optional[str]] = _headers or {}
+        _form_params: List[Tuple[str, str]] = []
+        _files: Dict[
+            str, Union[str, bytes, List[str], List[bytes], List[Tuple[str, bytes]]]
+        ] = {}
+        _body_params: Optional[bytes] = None
+
+        # process the path parameters
+        if validation_set_id is not None:
+            _path_params['validationSetId'] = validation_set_id
+        # process the query parameters
+        # process the header parameters
+        # process the form parameters
+        if model is not None:
+            _form_params.append(('model', model))
+        if files is not None:
+            _files['files'] = files
+        if texts is not None:
+            _form_params.append(('texts', texts))
+        if urls is not None:
+            _form_params.append(('urls', urls))
+        # process the body parameter
+
+
+        # set the HTTP header `Accept`
+        if 'Accept' not in _header_params:
+            _header_params['Accept'] = self.api_client.select_header_accept(
+                [
+                    'text/plain', 
+                    'application/json', 
+                    'text/json'
+                ]
+            )
+
+        # set the HTTP header `Content-Type`
+        if _content_type:
+            _header_params['Content-Type'] = _content_type
+        else:
+            _default_content_type = (
+                self.api_client.select_header_content_type(
+                    [
+                        'multipart/form-data'
+                    ]
+                )
+            )
+            if _default_content_type is not None:
+                _header_params['Content-Type'] = _default_content_type
+
+        # authentication setting
+        _auth_settings: List[str] = [
+            'bearer', 
+            'oauth2'
+        ]
+
+        return self.api_client.param_serialize(
+            method='POST',
+            resource_path='/validation-set/{validationSetId}/rapid',
+            path_params=_path_params,
+            query_params=_query_params,
+            header_params=_header_params,
+            body=_body_params,
+            post_params=_form_params,
+            files=_files,
+            auth_settings=_auth_settings,
+            collection_formats=_collection_formats,
+            _host=_host,
+            _request_auth=_request_auth
+        )
+
+
+
+
     @validate_call
     def validation_set_validation_set_id_rapid_texts_post(
         self,
@@ -1706,7 +2046,7 @@ class ValidationSetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> AddValidationRapidResult:
-        """Adds a new validation rapid to the specified validation set using text sources to create the assets.
+        """(Deprecated) Adds a new validation rapid to the specified validation set using text sources to create the assets.
 
 
         :param validation_set_id: The ID of the validation set to add the rapid to. (required)
@@ -1734,6 +2074,7 @@ class ValidationSetApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /validation-set/{validationSetId}/rapid/texts is deprecated.", DeprecationWarning)
 
         _param = self._validation_set_validation_set_id_rapid_texts_post_serialize(
             validation_set_id=validation_set_id,
@@ -1776,7 +2117,7 @@ class ValidationSetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> ApiResponse[AddValidationRapidResult]:
-        """Adds a new validation rapid to the specified validation set using text sources to create the assets.
+        """(Deprecated) Adds a new validation rapid to the specified validation set using text sources to create the assets.
 
 
         :param validation_set_id: The ID of the validation set to add the rapid to. (required)
@@ -1804,6 +2145,7 @@ class ValidationSetApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /validation-set/{validationSetId}/rapid/texts is deprecated.", DeprecationWarning)
 
         _param = self._validation_set_validation_set_id_rapid_texts_post_serialize(
             validation_set_id=validation_set_id,
@@ -1846,7 +2188,7 @@ class ValidationSetApi:
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
     ) -> RESTResponseType:
-        """Adds a new validation rapid to the specified validation set using text sources to create the assets.
+        """(Deprecated) Adds a new validation rapid to the specified validation set using text sources to create the assets.
 
 
         :param validation_set_id: The ID of the validation set to add the rapid to. (required)
@@ -1874,6 +2216,7 @@ class ValidationSetApi:
         :type _host_index: int, optional
         :return: Returns the result object.
         """ # noqa: E501
+        warnings.warn("POST /validation-set/{validationSetId}/rapid/texts is deprecated.", DeprecationWarning)
 
         _param = self._validation_set_validation_set_id_rapid_texts_post_serialize(
             validation_set_id=validation_set_id,

rapidata/api_client/models/__init__.py

@@ -301,6 +301,9 @@ from rapidata.api_client.models.report_model import ReportModel
 from rapidata.api_client.models.response_count_user_filter_model import ResponseCountUserFilterModel
 from rapidata.api_client.models.retrieval_mode import RetrievalMode
 from rapidata.api_client.models.root_filter import RootFilter
+from rapidata.api_client.models.run_status import RunStatus
+from rapidata.api_client.models.runs_by_leaderboard_result import RunsByLeaderboardResult
+from rapidata.api_client.models.runs_by_leaderboard_result_paged_result import RunsByLeaderboardResultPagedResult
 from rapidata.api_client.models.scrub_payload import ScrubPayload
 from rapidata.api_client.models.scrub_range import ScrubRange
 from rapidata.api_client.models.scrub_rapid_blueprint import ScrubRapidBlueprint
@@ -350,9 +353,11 @@ from rapidata.api_client.models.transcription_word import TranscriptionWord
 from rapidata.api_client.models.translated_prompt_metadata_model import TranslatedPromptMetadataModel
 from rapidata.api_client.models.translated_string import TranslatedString
 from rapidata.api_client.models.unlock_order_result import UnlockOrderResult
+from rapidata.api_client.models.update_benchmark_name_model import UpdateBenchmarkNameModel
 from rapidata.api_client.models.update_campaign_model import UpdateCampaignModel
 from rapidata.api_client.models.update_dataset_name_model import UpdateDatasetNameModel
 from rapidata.api_client.models.update_dimensions_model import UpdateDimensionsModel
+from rapidata.api_client.models.update_leaderboard_name_model import UpdateLeaderboardNameModel
 from rapidata.api_client.models.update_order_name_model import UpdateOrderNameModel
 from rapidata.api_client.models.update_validation_rapid_model import UpdateValidationRapidModel
 from rapidata.api_client.models.update_validation_rapid_model_truth import UpdateValidationRapidModelTruth

rapidata/api_client/models/file_type.py

@@ -26,6 +26,7 @@ class FileType(str, Enum):
     """
     allowed enum values
     """
+    UNKNOWN = 'Unknown'
     IMAGE = 'Image'
     VIDEO = 'Video'
     AUDIO = 'Audio'

rapidata/api_client/models/file_type_metadata.py

@@ -41,8 +41,8 @@ class FileTypeMetadata(BaseModel):
     @field_validator('file_type')
     def file_type_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['Image', 'Video', 'Audio']):
-            raise ValueError("must be one of enum values ('Image', 'Video', 'Audio')")
+        if value not in set(['Unknown', 'Image', 'Video', 'Audio']):
+            raise ValueError("must be one of enum values ('Unknown', 'Image', 'Video', 'Audio')")
         return value
 
     model_config = ConfigDict(

rapidata/api_client/models/file_type_metadata_model.py

@@ -40,8 +40,8 @@ class FileTypeMetadataModel(BaseModel):
     @field_validator('file_type')
     def file_type_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['Image', 'Video', 'Audio']):
-            raise ValueError("must be one of enum values ('Image', 'Video', 'Audio')")
+        if value not in set(['Unknown', 'Image', 'Video', 'Audio']):
+            raise ValueError("must be one of enum values ('Unknown', 'Image', 'Video', 'Audio')")
         return value
 
     model_config = ConfigDict(

rapidata/api_client/models/get_standing_by_id_result.py

@@ -17,7 +17,7 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictFloat, StrictInt, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictFloat, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional, Union
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,10 +31,11 @@ class GetStandingByIdResult(BaseModel):
     benchmark_id: StrictStr = Field(alias="benchmarkId")
     dataset_id: StrictStr = Field(alias="datasetId")
     status: StrictStr
+    is_disabled: StrictBool = Field(alias="isDisabled")
     score: Optional[Union[StrictFloat, StrictInt]] = None
     wins: StrictInt
     total_matches: StrictInt = Field(alias="totalMatches")
-    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId", "status", "score", "wins", "totalMatches"]
+    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId", "status", "isDisabled", "score", "wins", "totalMatches"]
 
     @field_validator('status')
     def status_validate_enum(cls, value):
@@ -104,6 +105,7 @@ class GetStandingByIdResult(BaseModel):
             "benchmarkId": obj.get("benchmarkId"),
             "datasetId": obj.get("datasetId"),
             "status": obj.get("status"),
+            "isDisabled": obj.get("isDisabled"),
             "score": obj.get("score"),
             "wins": obj.get("wins"),
             "totalMatches": obj.get("totalMatches")

rapidata/api_client/models/participant_by_benchmark.py

@@ -36,8 +36,8 @@ class ParticipantByBenchmark(BaseModel):
     @field_validator('status')
     def status_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['Created', 'Submitted']):
-            raise ValueError("must be one of enum values ('Created', 'Submitted')")
+        if value not in set(['Created', 'Submitted', 'Disabled']):
+            raise ValueError("must be one of enum values ('Created', 'Submitted', 'Disabled')")
         return value
 
     model_config = ConfigDict(

rapidata/api_client/models/participant_status.py

@@ -28,6 +28,7 @@ class ParticipantStatus(str, Enum):
     """
     CREATED = 'Created'
     SUBMITTED = 'Submitted'
+    DISABLED = 'Disabled'
 
     @classmethod
     def from_json(cls, json_str: str) -> Self:

rapidata/api_client/models/prompt_by_benchmark_result.py

@@ -19,7 +19,8 @@ import json
 
 from datetime import datetime
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.get_validation_rapids_result_asset import GetValidationRapidsResultAsset
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,10 +28,11 @@ class PromptByBenchmarkResult(BaseModel):
     """
     PromptByBenchmarkResult
     """ # noqa: E501
-    prompt: StrictStr
+    prompt: Optional[StrictStr] = None
+    prompt_asset: Optional[GetValidationRapidsResultAsset] = Field(default=None, alias="promptAsset")
     identifier: StrictStr
     created_at: datetime = Field(alias="createdAt")
-    __properties: ClassVar[List[str]] = ["prompt", "identifier", "createdAt"]
+    __properties: ClassVar[List[str]] = ["prompt", "promptAsset", "identifier", "createdAt"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -71,6 +73,19 @@ class PromptByBenchmarkResult(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of prompt_asset
+        if self.prompt_asset:
+            _dict['promptAsset'] = self.prompt_asset.to_dict()
+        # set to None if prompt (nullable) is None
+        # and model_fields_set contains the field
+        if self.prompt is None and "prompt" in self.model_fields_set:
+            _dict['prompt'] = None
+
+        # set to None if prompt_asset (nullable) is None
+        # and model_fields_set contains the field
+        if self.prompt_asset is None and "prompt_asset" in self.model_fields_set:
+            _dict['promptAsset'] = None
+
         return _dict
 
     @classmethod
@@ -84,6 +99,7 @@ class PromptByBenchmarkResult(BaseModel):
 
         _obj = cls.model_validate({
             "prompt": obj.get("prompt"),
+            "promptAsset": GetValidationRapidsResultAsset.from_dict(obj["promptAsset"]) if obj.get("promptAsset") is not None else None,
             "identifier": obj.get("identifier"),
             "createdAt": obj.get("createdAt")
         })

rapidata/api_client/models/run_status.py

@@ -0,0 +1,39 @@
+# coding: utf-8
+
+"""
+    Rapidata.Dataset
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+from enum import Enum
+from typing_extensions import Self
+
+
+class RunStatus(str, Enum):
+    """
+    RunStatus
+    """
+
+    """
+    allowed enum values
+    """
+    QUEUED = 'Queued'
+    RUNNING = 'Running'
+    COMPLETED = 'Completed'
+    FAILED = 'Failed'
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Create an instance of RunStatus from a JSON string"""
+        return cls(json.loads(json_str))
+
+