rapidata 1.2.0__py3-none-any.whl → 1.2.2__py3-none-any.whl

rapidata/api_client/models/__init__.py

@@ -44,6 +44,7 @@ from rapidata.api_client.models.capped_selection import CappedSelection
 from rapidata.api_client.models.capped_selection_selections_inner import CappedSelectionSelectionsInner
 from rapidata.api_client.models.classification_metadata import ClassificationMetadata
 from rapidata.api_client.models.classification_metadata_filter_config import ClassificationMetadataFilterConfig
+from rapidata.api_client.models.classification_metadata_model import ClassificationMetadataModel
 from rapidata.api_client.models.classify_payload import ClassifyPayload
 from rapidata.api_client.models.clone_dataset_model import CloneDatasetModel
 from rapidata.api_client.models.clone_order_model import CloneOrderModel
@@ -54,15 +55,17 @@ from rapidata.api_client.models.compare_result import CompareResult
 from rapidata.api_client.models.compare_truth import CompareTruth
 from rapidata.api_client.models.compare_workflow_config import CompareWorkflowConfig
 from rapidata.api_client.models.compare_workflow_config_model import CompareWorkflowConfigModel
-from rapidata.api_client.models.compare_workflow_config_referee import CompareWorkflowConfigReferee
 from rapidata.api_client.models.compare_workflow_get_result_overview_get200_response import CompareWorkflowGetResultOverviewGet200Response
 from rapidata.api_client.models.compare_workflow_model import CompareWorkflowModel
 from rapidata.api_client.models.compare_workflow_model1 import CompareWorkflowModel1
+from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
 from rapidata.api_client.models.completed_rapid_model import CompletedRapidModel
+from rapidata.api_client.models.completed_rapid_model_asset import CompletedRapidModelAsset
 from rapidata.api_client.models.conditional_validation_selection import ConditionalValidationSelection
 from rapidata.api_client.models.coordinate import Coordinate
 from rapidata.api_client.models.count_classification_metadata_filter_config import CountClassificationMetadataFilterConfig
 from rapidata.api_client.models.count_metadata import CountMetadata
+from rapidata.api_client.models.count_metadata_model import CountMetadataModel
 from rapidata.api_client.models.country_user_filter_model import CountryUserFilterModel
 from rapidata.api_client.models.create_client_model import CreateClientModel
 from rapidata.api_client.models.create_complex_order_model import CreateComplexOrderModel
@@ -72,9 +75,6 @@ from rapidata.api_client.models.create_dataset_artifact_model import CreateDatas
 from rapidata.api_client.models.create_dataset_artifact_model_dataset import CreateDatasetArtifactModelDataset
 from rapidata.api_client.models.create_demographic_rapid_model import CreateDemographicRapidModel
 from rapidata.api_client.models.create_empty_validation_set_result import CreateEmptyValidationSetResult
-from rapidata.api_client.models.create_independent_workflow_model import CreateIndependentWorkflowModel
-from rapidata.api_client.models.create_independent_workflow_model_workflow_config import CreateIndependentWorkflowModelWorkflowConfig
-from rapidata.api_client.models.create_independent_workflow_result import CreateIndependentWorkflowResult
 from rapidata.api_client.models.create_order_model import CreateOrderModel
 from rapidata.api_client.models.create_order_model_referee import CreateOrderModelReferee
 from rapidata.api_client.models.create_order_model_user_filters_inner import CreateOrderModelUserFiltersInner
@@ -95,6 +95,7 @@ from rapidata.api_client.models.datapoint_model import DatapointModel
 from rapidata.api_client.models.datapoint_model_asset import DatapointModelAsset
 from rapidata.api_client.models.dataset_artifact_model import DatasetArtifactModel
 from rapidata.api_client.models.dataset_evaluation_step_model import DatasetEvaluationStepModel
+from rapidata.api_client.models.demographic_metadata_model import DemographicMetadataModel
 from rapidata.api_client.models.demographic_selection import DemographicSelection
 from rapidata.api_client.models.early_stopping_referee_model import EarlyStoppingRefereeModel
 from rapidata.api_client.models.empty_validation_truth import EmptyValidationTruth
@@ -106,6 +107,10 @@ from rapidata.api_client.models.file_artifact_model import FileArtifactModel
 from rapidata.api_client.models.file_asset import FileAsset
 from rapidata.api_client.models.file_asset_metadata_inner import FileAssetMetadataInner
 from rapidata.api_client.models.file_asset_model import FileAssetModel
+from rapidata.api_client.models.file_asset_model1 import FileAssetModel1
+from rapidata.api_client.models.file_asset_model1_metadata_inner import FileAssetModel1MetadataInner
+from rapidata.api_client.models.file_asset_model2 import FileAssetModel2
+from rapidata.api_client.models.file_asset_model_metadata_inner import FileAssetModelMetadataInner
 from rapidata.api_client.models.filter import Filter
 from rapidata.api_client.models.filter_operator import FilterOperator
 from rapidata.api_client.models.free_text_payload import FreeTextPayload
@@ -131,6 +136,7 @@ from rapidata.api_client.models.get_workflow_progress_result import GetWorkflowP
 from rapidata.api_client.models.get_workflow_result_overview_result import GetWorkflowResultOverviewResult
 from rapidata.api_client.models.i_workflow_model_paged_result import IWorkflowModelPagedResult
 from rapidata.api_client.models.image_dimension_metadata import ImageDimensionMetadata
+from rapidata.api_client.models.image_dimension_metadata_model import ImageDimensionMetadataModel
 from rapidata.api_client.models.import_from_file_result import ImportFromFileResult
 from rapidata.api_client.models.import_validation_set_from_file_result import ImportValidationSetFromFileResult
 from rapidata.api_client.models.in_progress_rapid_model import InProgressRapidModel
@@ -153,11 +159,15 @@ from rapidata.api_client.models.locate_rapid_blueprint import LocateRapidBluepri
 from rapidata.api_client.models.locate_result import LocateResult
 from rapidata.api_client.models.location_metadata import LocationMetadata
 from rapidata.api_client.models.location_metadata_exists_filter_config import LocationMetadataExistsFilterConfig
+from rapidata.api_client.models.location_metadata_model import LocationMetadataModel
 from rapidata.api_client.models.logic_operator import LogicOperator
 from rapidata.api_client.models.login_model import LoginModel
 from rapidata.api_client.models.metadata_visibilities import MetadataVisibilities
 from rapidata.api_client.models.multi_asset import MultiAsset
 from rapidata.api_client.models.multi_asset_model import MultiAssetModel
+from rapidata.api_client.models.multi_asset_model1 import MultiAssetModel1
+from rapidata.api_client.models.multi_asset_model1_assets_inner import MultiAssetModel1AssetsInner
+from rapidata.api_client.models.multi_asset_model2 import MultiAssetModel2
 from rapidata.api_client.models.naive_referee_config import NaiveRefereeConfig
 from rapidata.api_client.models.naive_referee_model import NaiveRefereeModel
 from rapidata.api_client.models.named_classification import NamedClassification
@@ -170,10 +180,13 @@ from rapidata.api_client.models.newsletter_model import NewsletterModel
 from rapidata.api_client.models.not_started_rapid_model import NotStartedRapidModel
 from rapidata.api_client.models.null_asset import NullAsset
 from rapidata.api_client.models.null_asset_model import NullAssetModel
+from rapidata.api_client.models.null_asset_model1 import NullAssetModel1
+from rapidata.api_client.models.null_asset_model2 import NullAssetModel2
 from rapidata.api_client.models.order_model import OrderModel
 from rapidata.api_client.models.order_query_get200_response import OrderQueryGet200Response
 from rapidata.api_client.models.order_state import OrderState
 from rapidata.api_client.models.original_filename_metadata import OriginalFilenameMetadata
+from rapidata.api_client.models.original_filename_metadata_model import OriginalFilenameMetadataModel
 from rapidata.api_client.models.page_info import PageInfo
 from rapidata.api_client.models.polygon_payload import PolygonPayload
 from rapidata.api_client.models.polygon_rapid_blueprint import PolygonRapidBlueprint
@@ -183,6 +196,7 @@ from rapidata.api_client.models.private_text_metadata_input import PrivateTextMe
 from rapidata.api_client.models.probabilistic_attach_category_referee_config import ProbabilisticAttachCategoryRefereeConfig
 from rapidata.api_client.models.prompt_metadata import PromptMetadata
 from rapidata.api_client.models.prompt_metadata_input import PromptMetadataInput
+from rapidata.api_client.models.prompt_metadata_model import PromptMetadataModel
 from rapidata.api_client.models.public_text_metadata_input import PublicTextMetadataInput
 from rapidata.api_client.models.query_campaigns_model import QueryCampaignsModel
 from rapidata.api_client.models.query_model import QueryModel
@@ -204,12 +218,12 @@ from rapidata.api_client.models.shape import Shape
 from rapidata.api_client.models.signup_customer_model import SignupCustomerModel
 from rapidata.api_client.models.signup_shadow_customer_model import SignupShadowCustomerModel
 from rapidata.api_client.models.simple_workflow_config import SimpleWorkflowConfig
-from rapidata.api_client.models.simple_workflow_config_blueprint import SimpleWorkflowConfigBlueprint
 from rapidata.api_client.models.simple_workflow_config_model import SimpleWorkflowConfigModel
 from rapidata.api_client.models.simple_workflow_config_model_blueprint import SimpleWorkflowConfigModelBlueprint
 from rapidata.api_client.models.simple_workflow_get_result_overview_get200_response import SimpleWorkflowGetResultOverviewGet200Response
 from rapidata.api_client.models.simple_workflow_model import SimpleWorkflowModel
 from rapidata.api_client.models.simple_workflow_model1 import SimpleWorkflowModel1
+from rapidata.api_client.models.simple_workflow_model1_blueprint import SimpleWorkflowModel1Blueprint
 from rapidata.api_client.models.simple_workflow_model_blueprint import SimpleWorkflowModelBlueprint
 from rapidata.api_client.models.skip_result import SkipResult
 from rapidata.api_client.models.sort_criterion import SortCriterion
@@ -219,14 +233,20 @@ from rapidata.api_client.models.submit_coco_model import SubmitCocoModel
 from rapidata.api_client.models.submit_coco_result import SubmitCocoResult
 from rapidata.api_client.models.text_asset import TextAsset
 from rapidata.api_client.models.text_asset_model import TextAssetModel
+from rapidata.api_client.models.text_asset_model1 import TextAssetModel1
+from rapidata.api_client.models.text_asset_model2 import TextAssetModel2
 from rapidata.api_client.models.text_metadata import TextMetadata
+from rapidata.api_client.models.text_metadata_model import TextMetadataModel
 from rapidata.api_client.models.transcription_metadata import TranscriptionMetadata
 from rapidata.api_client.models.transcription_metadata_input import TranscriptionMetadataInput
+from rapidata.api_client.models.transcription_metadata_model import TranscriptionMetadataModel
 from rapidata.api_client.models.transcription_payload import TranscriptionPayload
 from rapidata.api_client.models.transcription_rapid_blueprint import TranscriptionRapidBlueprint
 from rapidata.api_client.models.transcription_result import TranscriptionResult
 from rapidata.api_client.models.transcription_truth import TranscriptionTruth
 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_access_model import UpdateAccessModel
 from rapidata.api_client.models.update_order_model import UpdateOrderModel

rapidata/api_client/models/add_validation_text_rapid_model.py

@@ -33,9 +33,9 @@ class AddValidationTextRapidModel(BaseModel):
     payload: AddValidationRapidModelPayload
     metadata: List[DatapointMetadataModelMetadataInner] = Field(description="Some metadata to attach to the rapid.")
     truth: AddValidationRapidModelTruth
-    text: StrictStr
+    texts: List[StrictStr]
     random_correct_probability: Optional[Union[StrictFloat, StrictInt]] = Field(description="The probability for an answer to be correct when randomly guessing.", alias="randomCorrectProbability")
-    __properties: ClassVar[List[str]] = ["validationSetId", "payload", "metadata", "truth", "text", "randomCorrectProbability"]
+    __properties: ClassVar[List[str]] = ["validationSetId", "payload", "metadata", "truth", "texts", "randomCorrectProbability"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -110,7 +110,7 @@ class AddValidationTextRapidModel(BaseModel):
             "payload": AddValidationRapidModelPayload.from_dict(obj["payload"]) if obj.get("payload") is not None else None,
             "metadata": [DatapointMetadataModelMetadataInner.from_dict(_item) for _item in obj["metadata"]] if obj.get("metadata") is not None else None,
             "truth": AddValidationRapidModelTruth.from_dict(obj["truth"]) if obj.get("truth") is not None else None,
-            "text": obj.get("text"),
+            "texts": obj.get("texts"),
             "randomCorrectProbability": obj.get("randomCorrectProbability")
         })
         return _obj

rapidata/api_client/models/classification_metadata_model.py

@@ -0,0 +1,98 @@
+# 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 pprint
+import re  # noqa: F401
+import json
+
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ClassificationMetadataModel(BaseModel):
+    """
+    ClassificationMetadataModel
+    """ # noqa: E501
+    t: StrictStr = Field(description="Discriminator value for ClassificationMetadata", alias="_t")
+    classification: StrictStr
+    identifier: StrictStr
+    __properties: ClassVar[List[str]] = ["_t", "classification", "identifier"]
+
+    @field_validator('t')
+    def t_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['ClassificationMetadata']):
+            raise ValueError("must be one of enum values ('ClassificationMetadata')")
+        return value
+
+    model_config = ConfigDict(
+        populate_by_name=True,
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    def to_str(self) -> str:
+        """Returns the string representation of the model using alias"""
+        return pprint.pformat(self.model_dump(by_alias=True))
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the model using alias"""
+        # TODO: pydantic v2: use .model_dump_json(by_alias=True, exclude_unset=True) instead
+        return json.dumps(self.to_dict())
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Optional[Self]:
+        """Create an instance of ClassificationMetadataModel from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of ClassificationMetadataModel from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "_t": obj.get("_t") if obj.get("_t") is not None else 'ClassificationMetadata',
+            "classification": obj.get("classification"),
+            "identifier": obj.get("identifier")
+        })
+        return _obj
+
+

rapidata/api_client/models/compare_workflow_config.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from rapidata.api_client.models.compare_workflow_config_referee import CompareWorkflowConfigReferee
+from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
 from rapidata.api_client.models.feature_flag import FeatureFlag
 from typing import Optional, Set
 from typing_extensions import Self
@@ -35,7 +35,7 @@ class CompareWorkflowConfig(BaseModel):
     match_size: Optional[StrictInt] = Field(default=None, alias="matchSize")
     scaling_factor: Optional[StrictInt] = Field(default=None, alias="scalingFactor")
     matches_until_completed: Optional[StrictInt] = Field(default=None, alias="matchesUntilCompleted")
-    referee: CompareWorkflowConfigReferee
+    referee: CompareWorkflowModel1Referee
     target_country_codes: List[StrictStr] = Field(alias="targetCountryCodes")
     feature_flags: Optional[List[FeatureFlag]] = Field(default=None, alias="featureFlags")
     priority: Optional[StrictStr] = None
@@ -122,7 +122,7 @@ class CompareWorkflowConfig(BaseModel):
             "matchSize": obj.get("matchSize"),
             "scalingFactor": obj.get("scalingFactor"),
             "matchesUntilCompleted": obj.get("matchesUntilCompleted"),
-            "referee": CompareWorkflowConfigReferee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
+            "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
             "targetCountryCodes": obj.get("targetCountryCodes"),
             "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None,
             "priority": obj.get("priority"),

rapidata/api_client/models/compare_workflow_config_model.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from rapidata.api_client.models.compare_workflow_config_referee import CompareWorkflowConfigReferee
+from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
 from rapidata.api_client.models.feature_flag import FeatureFlag
 from typing import Optional, Set
 from typing_extensions import Self
@@ -35,7 +35,7 @@ class CompareWorkflowConfigModel(BaseModel):
     match_size: Optional[StrictInt] = Field(default=None, description="The number of datapoints to match against each other in each match.", alias="matchSize")
     scaling_factor: Optional[StrictInt] = Field(default=None, description="Scaling factor to use in the ELO calculation.  Adjusts the sensitivity of the Elo rating system to differences in player ratings.  It affects how much the rating changes based on the expected outcome versus the actual outcome.", alias="scalingFactor")
     matches_until_completed: Optional[StrictInt] = Field(default=None, description="The number of matches to run before each datapoint is considered \"completed\".", alias="matchesUntilCompleted")
-    referee: CompareWorkflowConfigReferee
+    referee: CompareWorkflowModel1Referee
     target_country_codes: List[StrictStr] = Field(description="A list of country codes that this workflow is targeting.", alias="targetCountryCodes")
     feature_flags: Optional[List[FeatureFlag]] = Field(default=None, alias="featureFlags")
     priority: Optional[StrictStr] = None
@@ -127,7 +127,7 @@ class CompareWorkflowConfigModel(BaseModel):
             "matchSize": obj.get("matchSize"),
             "scalingFactor": obj.get("scalingFactor"),
             "matchesUntilCompleted": obj.get("matchesUntilCompleted"),
-            "referee": CompareWorkflowConfigReferee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
+            "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
             "targetCountryCodes": obj.get("targetCountryCodes"),
             "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None,
             "priority": obj.get("priority"),

rapidata/api_client/models/compare_workflow_model1.py

@@ -19,8 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
-from rapidata.api_client.models.compare_workflow_config_referee import CompareWorkflowConfigReferee
-from rapidata.api_client.models.feature_flag import FeatureFlag
+from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -31,11 +30,8 @@ class CompareWorkflowModel1(BaseModel):
     t: StrictStr = Field(description="Discriminator value for CompareWorkflowModel", alias="_t")
     id: StrictStr
     dataset_id: Optional[StrictStr] = Field(alias="datasetId")
-    target_country_codes: List[StrictStr] = Field(alias="targetCountryCodes")
-    feature_flags: List[FeatureFlag] = Field(alias="featureFlags")
-    referee: CompareWorkflowConfigReferee
+    referee: CompareWorkflowModel1Referee
     state: StrictStr
-    priority: StrictStr
     criteria: StrictStr
     name: StrictStr
     owner_mail: Optional[StrictStr] = Field(alias="ownerMail")
@@ -44,7 +40,7 @@ class CompareWorkflowModel1(BaseModel):
     match_size: StrictInt = Field(alias="matchSize")
     scaling_factor: StrictInt = Field(alias="scalingFactor")
     matches_until_completed: StrictInt = Field(alias="matchesUntilCompleted")
-    __properties: ClassVar[List[str]] = ["_t", "id", "datasetId", "targetCountryCodes", "featureFlags", "referee", "state", "priority", "criteria", "name", "ownerMail", "startingElo", "kFactor", "matchSize", "scalingFactor", "matchesUntilCompleted"]
+    __properties: ClassVar[List[str]] = ["_t", "id", "datasetId", "referee", "state", "criteria", "name", "ownerMail", "startingElo", "kFactor", "matchSize", "scalingFactor", "matchesUntilCompleted"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -92,13 +88,6 @@ class CompareWorkflowModel1(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in feature_flags (list)
-        _items = []
-        if self.feature_flags:
-            for _item_feature_flags in self.feature_flags:
-                if _item_feature_flags:
-                    _items.append(_item_feature_flags.to_dict())
-            _dict['featureFlags'] = _items
         # override the default output from pydantic by calling `to_dict()` of referee
         if self.referee:
             _dict['referee'] = self.referee.to_dict()
@@ -127,11 +116,8 @@ class CompareWorkflowModel1(BaseModel):
             "_t": obj.get("_t") if obj.get("_t") is not None else 'CompareWorkflowModel',
             "id": obj.get("id"),
             "datasetId": obj.get("datasetId"),
-            "targetCountryCodes": obj.get("targetCountryCodes"),
-            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None,
-            "referee": CompareWorkflowConfigReferee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
+            "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
             "state": obj.get("state"),
-            "priority": obj.get("priority"),
             "criteria": obj.get("criteria"),
             "name": obj.get("name"),
             "ownerMail": obj.get("ownerMail"),

rapidata/api_client/models/compare_workflow_model1_referee.py

@@ -0,0 +1,154 @@
+# 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
+import pprint
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
+from typing import Any, List, Optional
+from rapidata.api_client.models.naive_referee_config import NaiveRefereeConfig
+from rapidata.api_client.models.never_ending_referee_config import NeverEndingRefereeConfig
+from rapidata.api_client.models.probabilistic_attach_category_referee_config import ProbabilisticAttachCategoryRefereeConfig
+from pydantic import StrictStr, Field
+from typing import Union, List, Set, Optional, Dict
+from typing_extensions import Literal, Self
+
+COMPAREWORKFLOWMODEL1REFEREE_ONE_OF_SCHEMAS = ["NaiveRefereeConfig", "NeverEndingRefereeConfig", "ProbabilisticAttachCategoryRefereeConfig"]
+
+class CompareWorkflowModel1Referee(BaseModel):
+    """
+    CompareWorkflowModel1Referee
+    """
+    # data type: NaiveRefereeConfig
+    oneof_schema_1_validator: Optional[NaiveRefereeConfig] = None
+    # data type: NeverEndingRefereeConfig
+    oneof_schema_2_validator: Optional[NeverEndingRefereeConfig] = None
+    # data type: ProbabilisticAttachCategoryRefereeConfig
+    oneof_schema_3_validator: Optional[ProbabilisticAttachCategoryRefereeConfig] = None
+    actual_instance: Optional[Union[NaiveRefereeConfig, NeverEndingRefereeConfig, ProbabilisticAttachCategoryRefereeConfig]] = None
+    one_of_schemas: Set[str] = { "NaiveRefereeConfig", "NeverEndingRefereeConfig", "ProbabilisticAttachCategoryRefereeConfig" }
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    discriminator_value_class_map: Dict[str, str] = {
+    }
+
+    def __init__(self, *args, **kwargs) -> None:
+        if args:
+            if len(args) > 1:
+                raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
+            if kwargs:
+                raise ValueError("If a position argument is used, keyword arguments cannot be used.")
+            super().__init__(actual_instance=args[0])
+        else:
+            super().__init__(**kwargs)
+
+    @field_validator('actual_instance')
+    def actual_instance_must_validate_oneof(cls, v):
+        instance = CompareWorkflowModel1Referee.model_construct()
+        error_messages = []
+        match = 0
+        # validate data type: NaiveRefereeConfig
+        if not isinstance(v, NaiveRefereeConfig):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `NaiveRefereeConfig`")
+        else:
+            match += 1
+        # validate data type: NeverEndingRefereeConfig
+        if not isinstance(v, NeverEndingRefereeConfig):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `NeverEndingRefereeConfig`")
+        else:
+            match += 1
+        # validate data type: ProbabilisticAttachCategoryRefereeConfig
+        if not isinstance(v, ProbabilisticAttachCategoryRefereeConfig):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ProbabilisticAttachCategoryRefereeConfig`")
+        else:
+            match += 1
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when setting `actual_instance` in CompareWorkflowModel1Referee with oneOf schemas: NaiveRefereeConfig, NeverEndingRefereeConfig, ProbabilisticAttachCategoryRefereeConfig. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when setting `actual_instance` in CompareWorkflowModel1Referee with oneOf schemas: NaiveRefereeConfig, NeverEndingRefereeConfig, ProbabilisticAttachCategoryRefereeConfig. Details: " + ", ".join(error_messages))
+        else:
+            return v
+
+    @classmethod
+    def from_dict(cls, obj: Union[str, Dict[str, Any]]) -> Self:
+        return cls.from_json(json.dumps(obj))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Returns the object represented by the json string"""
+        instance = cls.model_construct()
+        error_messages = []
+        match = 0
+
+        # deserialize data into NaiveRefereeConfig
+        try:
+            instance.actual_instance = NaiveRefereeConfig.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into NeverEndingRefereeConfig
+        try:
+            instance.actual_instance = NeverEndingRefereeConfig.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into ProbabilisticAttachCategoryRefereeConfig
+        try:
+            instance.actual_instance = ProbabilisticAttachCategoryRefereeConfig.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when deserializing the JSON string into CompareWorkflowModel1Referee with oneOf schemas: NaiveRefereeConfig, NeverEndingRefereeConfig, ProbabilisticAttachCategoryRefereeConfig. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when deserializing the JSON string into CompareWorkflowModel1Referee with oneOf schemas: NaiveRefereeConfig, NeverEndingRefereeConfig, ProbabilisticAttachCategoryRefereeConfig. Details: " + ", ".join(error_messages))
+        else:
+            return instance
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the actual instance"""
+        if self.actual_instance is None:
+            return "null"
+
+        if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
+            return self.actual_instance.to_json()
+        else:
+            return json.dumps(self.actual_instance)
+
+    def to_dict(self) -> Optional[Union[Dict[str, Any], NaiveRefereeConfig, NeverEndingRefereeConfig, ProbabilisticAttachCategoryRefereeConfig]]:
+        """Returns the dict representation of the actual instance"""
+        if self.actual_instance is None:
+            return None
+
+        if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
+            return self.actual_instance.to_dict()
+        else:
+            # primitive type
+            return self.actual_instance
+
+    def to_str(self) -> str:
+        """Returns the string representation of the actual instance"""
+        return pprint.pformat(self.model_dump())
+
+

rapidata/api_client/models/completed_rapid_model.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List
-from rapidata.api_client.models.datapoint_model_asset import DatapointModelAsset
+from rapidata.api_client.models.completed_rapid_model_asset import CompletedRapidModelAsset
 from rapidata.api_client.models.rapid_answer import RapidAnswer
 from typing import Optional, Set
 from typing_extensions import Self
@@ -29,7 +29,7 @@ class CompletedRapidModel(BaseModel):
     CompletedRapidModel
     """ # noqa: E501
     rapid_id: StrictStr = Field(alias="rapidId")
-    asset: DatapointModelAsset
+    asset: CompletedRapidModelAsset
     answers: List[RapidAnswer]
     __properties: ClassVar[List[str]] = ["rapidId", "asset", "answers"]
 
@@ -95,7 +95,7 @@ class CompletedRapidModel(BaseModel):
 
         _obj = cls.model_validate({
             "rapidId": obj.get("rapidId"),
-            "asset": DatapointModelAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
+            "asset": CompletedRapidModelAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
             "answers": [RapidAnswer.from_dict(_item) for _item in obj["answers"]] if obj.get("answers") is not None else None
         })
         return _obj