rapidata 0.4.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

rapidata/api_client/models/create_independent_workflow_result.py

@@ -0,0 +1,89 @@
+# 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
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class CreateIndependentWorkflowResult(BaseModel):
+    """
+    CreateIndependentWorkflowResult
+    """ # noqa: E501
+    workflow_id: StrictStr = Field(alias="workflowId")
+    dataset_id: StrictStr = Field(alias="datasetId")
+    __properties: ClassVar[List[str]] = ["workflowId", "datasetId"]
+
+    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 CreateIndependentWorkflowResult 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 CreateIndependentWorkflowResult from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "workflowId": obj.get("workflowId"),
+            "datasetId": obj.get("datasetId")
+        })
+        return _obj
+
+

rapidata/api_client/models/create_order_model.py

@@ -31,16 +31,24 @@ class CreateOrderModel(BaseModel):
     """
     This model is used to create a simple order
     """ # noqa: E501
+    t: StrictStr = Field(description="Discriminator value for CreateOrderModel", alias="_t")
     order_name: StrictStr = Field(description="The name is used as an identifier for an order and can be freely chosen.", alias="orderName")
-    user_filters: List[CreateOrderModelUserFiltersInner] = Field(description="The user filters are used to restrict the order to only collect votes from a specific demographic.", alias="userFilters")
     workflow: CreateOrderModelWorkflow
     referee: CreateOrderModelReferee
     aggregator: Optional[StrictStr] = Field(default=None, description="The aggregator is used to determine how the data will be aggregated. The default behavior is enough for most cases")
-    validation_set_id: Optional[StrictStr] = Field(default=None, description="The validation set id can be changed to point to a specific validation set. if not provided a sane default will be used.", alias="validationSetId")
-    selections: Optional[List[CreateOrderModelSelectionsInner]] = Field(default=None, description="The selections are used to determine which tasks are shown to a user.")
     feature_flags: Optional[List[FeatureFlagModel]] = Field(default=None, description="The feature flags are used to enable or disable certain features.", alias="featureFlags")
     priority: Optional[StrictInt] = Field(default=None, description="The priority is used to prioritize over other orders.")
-    __properties: ClassVar[List[str]] = ["orderName", "userFilters", "workflow", "referee", "aggregator", "validationSetId", "selections", "featureFlags", "priority"]
+    user_filters: List[CreateOrderModelUserFiltersInner] = Field(description="The user filters are used to restrict the order to only collect votes from a specific demographic.", alias="userFilters")
+    validation_set_id: Optional[StrictStr] = Field(default=None, description="The validation set id can be changed to point to a specific validation set. if not provided a sane default will be  used.", alias="validationSetId")
+    selections: Optional[List[CreateOrderModelSelectionsInner]] = Field(default=None, description="The selections are used to determine which tasks are shown to a user.")
+    __properties: ClassVar[List[str]] = ["_t", "orderName", "workflow", "referee", "aggregator", "featureFlags", "priority", "userFilters", "validationSetId", "selections"]
+
+    @field_validator('t')
+    def t_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['CreateOrderModel']):
+            raise ValueError("must be one of enum values ('CreateOrderModel')")
+        return value
 
     @field_validator('aggregator')
     def aggregator_validate_enum(cls, value):
@@ -91,6 +99,19 @@ class CreateOrderModel(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # override the default output from pydantic by calling `to_dict()` of workflow
+        if self.workflow:
+            _dict['workflow'] = self.workflow.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of referee
+        if self.referee:
+            _dict['referee'] = self.referee.to_dict()
+        # 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 each item in user_filters (list)
         _items = []
         if self.user_filters:
@@ -98,12 +119,6 @@ class CreateOrderModel(BaseModel):
                 if _item_user_filters:
                     _items.append(_item_user_filters.to_dict())
             _dict['userFilters'] = _items
-        # override the default output from pydantic by calling `to_dict()` of workflow
-        if self.workflow:
-            _dict['workflow'] = self.workflow.to_dict()
-        # override the default output from pydantic by calling `to_dict()` of referee
-        if self.referee:
-            _dict['referee'] = self.referee.to_dict()
         # override the default output from pydantic by calling `to_dict()` of each item in selections (list)
         _items = []
         if self.selections:
@@ -111,28 +126,11 @@ class CreateOrderModel(BaseModel):
                 if _item_selections:
                     _items.append(_item_selections.to_dict())
             _dict['selections'] = _items
-        # 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
         # set to None if aggregator (nullable) is None
         # and model_fields_set contains the field
         if self.aggregator is None and "aggregator" in self.model_fields_set:
             _dict['aggregator'] = None
 
-        # set to None if validation_set_id (nullable) is None
-        # and model_fields_set contains the field
-        if self.validation_set_id is None and "validation_set_id" in self.model_fields_set:
-            _dict['validationSetId'] = None
-
-        # set to None if selections (nullable) is None
-        # and model_fields_set contains the field
-        if self.selections is None and "selections" in self.model_fields_set:
-            _dict['selections'] = None
-
         # set to None if feature_flags (nullable) is None
         # and model_fields_set contains the field
         if self.feature_flags is None and "feature_flags" in self.model_fields_set:
@@ -143,6 +141,16 @@ class CreateOrderModel(BaseModel):
         if self.priority is None and "priority" in self.model_fields_set:
             _dict['priority'] = None
 
+        # set to None if validation_set_id (nullable) is None
+        # and model_fields_set contains the field
+        if self.validation_set_id is None and "validation_set_id" in self.model_fields_set:
+            _dict['validationSetId'] = None
+
+        # set to None if selections (nullable) is None
+        # and model_fields_set contains the field
+        if self.selections is None and "selections" in self.model_fields_set:
+            _dict['selections'] = None
+
         return _dict
 
     @classmethod
@@ -155,15 +163,16 @@ class CreateOrderModel(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
+            "_t": obj.get("_t") if obj.get("_t") is not None else 'CreateOrderModel',
             "orderName": obj.get("orderName"),
-            "userFilters": [CreateOrderModelUserFiltersInner.from_dict(_item) for _item in obj["userFilters"]] if obj.get("userFilters") is not None else None,
             "workflow": CreateOrderModelWorkflow.from_dict(obj["workflow"]) if obj.get("workflow") is not None else None,
             "referee": CreateOrderModelReferee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
             "aggregator": obj.get("aggregator"),
-            "validationSetId": obj.get("validationSetId"),
-            "selections": [CreateOrderModelSelectionsInner.from_dict(_item) for _item in obj["selections"]] if obj.get("selections") is not None else None,
             "featureFlags": [FeatureFlagModel.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None,
-            "priority": obj.get("priority")
+            "priority": obj.get("priority"),
+            "userFilters": [CreateOrderModelUserFiltersInner.from_dict(_item) for _item in obj["userFilters"]] if obj.get("userFilters") is not None else None,
+            "validationSetId": obj.get("validationSetId"),
+            "selections": [CreateOrderModelSelectionsInner.from_dict(_item) for _item in obj["selections"]] if obj.get("selections") is not None else None
         })
         return _obj
 

rapidata/api_client/models/create_order_model_selections_inner.py

@@ -17,27 +17,33 @@ 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.conditional_validation_selection import ConditionalValidationSelection
 from rapidata.api_client.models.demographic_selection import DemographicSelection
 from rapidata.api_client.models.labeling_selection import LabelingSelection
+from rapidata.api_client.models.static_selection import StaticSelection
 from rapidata.api_client.models.validation_selection import ValidationSelection
 from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-CREATEORDERMODELSELECTIONSINNER_ONE_OF_SCHEMAS = ["DemographicSelection", "LabelingSelection", "ValidationSelection"]
+CREATEORDERMODELSELECTIONSINNER_ONE_OF_SCHEMAS = ["ConditionalValidationSelection", "DemographicSelection", "LabelingSelection", "StaticSelection", "ValidationSelection"]
 
 class CreateOrderModelSelectionsInner(BaseModel):
     """
-    A selection is used to determine which tasks are shown to a user.
+    CreateOrderModelSelectionsInner
     """
+    # data type: ConditionalValidationSelection
+    oneof_schema_1_validator: Optional[ConditionalValidationSelection] = None
     # data type: DemographicSelection
-    oneof_schema_1_validator: Optional[DemographicSelection] = None
+    oneof_schema_2_validator: Optional[DemographicSelection] = None
     # data type: LabelingSelection
-    oneof_schema_2_validator: Optional[LabelingSelection] = None
+    oneof_schema_3_validator: Optional[LabelingSelection] = None
+    # data type: StaticSelection
+    oneof_schema_4_validator: Optional[StaticSelection] = None
     # data type: ValidationSelection
-    oneof_schema_3_validator: Optional[ValidationSelection] = None
-    actual_instance: Optional[Union[DemographicSelection, LabelingSelection, ValidationSelection]] = None
-    one_of_schemas: Set[str] = { "DemographicSelection", "LabelingSelection", "ValidationSelection" }
+    oneof_schema_5_validator: Optional[ValidationSelection] = None
+    actual_instance: Optional[Union[ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection]] = None
+    one_of_schemas: Set[str] = { "ConditionalValidationSelection", "DemographicSelection", "LabelingSelection", "StaticSelection", "ValidationSelection" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -63,6 +69,11 @@ class CreateOrderModelSelectionsInner(BaseModel):
         instance = CreateOrderModelSelectionsInner.model_construct()
         error_messages = []
         match = 0
+        # validate data type: ConditionalValidationSelection
+        if not isinstance(v, ConditionalValidationSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ConditionalValidationSelection`")
+        else:
+            match += 1
         # validate data type: DemographicSelection
         if not isinstance(v, DemographicSelection):
             error_messages.append(f"Error! Input type `{type(v)}` is not `DemographicSelection`")
@@ -73,6 +84,11 @@ class CreateOrderModelSelectionsInner(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `LabelingSelection`")
         else:
             match += 1
+        # validate data type: StaticSelection
+        if not isinstance(v, StaticSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `StaticSelection`")
+        else:
+            match += 1
         # validate data type: ValidationSelection
         if not isinstance(v, ValidationSelection):
             error_messages.append(f"Error! Input type `{type(v)}` is not `ValidationSelection`")
@@ -80,10 +96,10 @@ class CreateOrderModelSelectionsInner(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in CreateOrderModelSelectionsInner with oneOf schemas: DemographicSelection, LabelingSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in CreateOrderModelSelectionsInner with oneOf schemas: ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in CreateOrderModelSelectionsInner with oneOf schemas: DemographicSelection, LabelingSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in CreateOrderModelSelectionsInner with oneOf schemas: ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -98,6 +114,12 @@ class CreateOrderModelSelectionsInner(BaseModel):
         error_messages = []
         match = 0
 
+        # deserialize data into ConditionalValidationSelection
+        try:
+            instance.actual_instance = ConditionalValidationSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into DemographicSelection
         try:
             instance.actual_instance = DemographicSelection.from_json(json_str)
@@ -110,6 +132,12 @@ class CreateOrderModelSelectionsInner(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into StaticSelection
+        try:
+            instance.actual_instance = StaticSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into ValidationSelection
         try:
             instance.actual_instance = ValidationSelection.from_json(json_str)
@@ -119,10 +147,10 @@ class CreateOrderModelSelectionsInner(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into CreateOrderModelSelectionsInner with oneOf schemas: DemographicSelection, LabelingSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into CreateOrderModelSelectionsInner with oneOf schemas: ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into CreateOrderModelSelectionsInner with oneOf schemas: DemographicSelection, LabelingSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into CreateOrderModelSelectionsInner with oneOf schemas: ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -136,7 +164,7 @@ class CreateOrderModelSelectionsInner(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], DemographicSelection, LabelingSelection, ValidationSelection]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

rapidata/api_client/models/create_order_model_user_filters_inner.py

@@ -31,20 +31,20 @@ CREATEORDERMODELUSERFILTERSINNER_ONE_OF_SCHEMAS = ["AgeUserFilterModel", "Campai
 
 class CreateOrderModelUserFiltersInner(BaseModel):
     """
-    The filter model can be used to restrict users to specific demographics.
+    CreateOrderModelUserFiltersInner
     """
-    # data type: UserScoreUserFilterModel
-    oneof_schema_1_validator: Optional[UserScoreUserFilterModel] = None
-    # data type: LanguageUserFilterModel
-    oneof_schema_2_validator: Optional[LanguageUserFilterModel] = None
+    # data type: AgeUserFilterModel
+    oneof_schema_1_validator: Optional[AgeUserFilterModel] = None
     # data type: CampaignUserFilterModel
-    oneof_schema_3_validator: Optional[CampaignUserFilterModel] = None
+    oneof_schema_2_validator: Optional[CampaignUserFilterModel] = None
     # data type: CountryUserFilterModel
-    oneof_schema_4_validator: Optional[CountryUserFilterModel] = None
-    # data type: AgeUserFilterModel
-    oneof_schema_5_validator: Optional[AgeUserFilterModel] = None
+    oneof_schema_3_validator: Optional[CountryUserFilterModel] = None
     # data type: GenderUserFilterModel
-    oneof_schema_6_validator: Optional[GenderUserFilterModel] = None
+    oneof_schema_4_validator: Optional[GenderUserFilterModel] = None
+    # data type: LanguageUserFilterModel
+    oneof_schema_5_validator: Optional[LanguageUserFilterModel] = None
+    # data type: UserScoreUserFilterModel
+    oneof_schema_6_validator: Optional[UserScoreUserFilterModel] = None
     actual_instance: Optional[Union[AgeUserFilterModel, CampaignUserFilterModel, CountryUserFilterModel, GenderUserFilterModel, LanguageUserFilterModel, UserScoreUserFilterModel]] = None
     one_of_schemas: Set[str] = { "AgeUserFilterModel", "CampaignUserFilterModel", "CountryUserFilterModel", "GenderUserFilterModel", "LanguageUserFilterModel", "UserScoreUserFilterModel" }
 
@@ -72,14 +72,9 @@ class CreateOrderModelUserFiltersInner(BaseModel):
         instance = CreateOrderModelUserFiltersInner.model_construct()
         error_messages = []
         match = 0
-        # validate data type: UserScoreUserFilterModel
-        if not isinstance(v, UserScoreUserFilterModel):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `UserScoreUserFilterModel`")
-        else:
-            match += 1
-        # validate data type: LanguageUserFilterModel
-        if not isinstance(v, LanguageUserFilterModel):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `LanguageUserFilterModel`")
+        # validate data type: AgeUserFilterModel
+        if not isinstance(v, AgeUserFilterModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AgeUserFilterModel`")
         else:
             match += 1
         # validate data type: CampaignUserFilterModel
@@ -92,16 +87,21 @@ class CreateOrderModelUserFiltersInner(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `CountryUserFilterModel`")
         else:
             match += 1
-        # validate data type: AgeUserFilterModel
-        if not isinstance(v, AgeUserFilterModel):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `AgeUserFilterModel`")
-        else:
-            match += 1
         # validate data type: GenderUserFilterModel
         if not isinstance(v, GenderUserFilterModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `GenderUserFilterModel`")
         else:
             match += 1
+        # validate data type: LanguageUserFilterModel
+        if not isinstance(v, LanguageUserFilterModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `LanguageUserFilterModel`")
+        else:
+            match += 1
+        # validate data type: UserScoreUserFilterModel
+        if not isinstance(v, UserScoreUserFilterModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `UserScoreUserFilterModel`")
+        else:
+            match += 1
         if match > 1:
             # more than 1 match
             raise ValueError("Multiple matches found when setting `actual_instance` in CreateOrderModelUserFiltersInner with oneOf schemas: AgeUserFilterModel, CampaignUserFilterModel, CountryUserFilterModel, GenderUserFilterModel, LanguageUserFilterModel, UserScoreUserFilterModel. Details: " + ", ".join(error_messages))
@@ -122,15 +122,9 @@ class CreateOrderModelUserFiltersInner(BaseModel):
         error_messages = []
         match = 0
 
-        # deserialize data into UserScoreUserFilterModel
-        try:
-            instance.actual_instance = UserScoreUserFilterModel.from_json(json_str)
-            match += 1
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # deserialize data into LanguageUserFilterModel
+        # deserialize data into AgeUserFilterModel
         try:
-            instance.actual_instance = LanguageUserFilterModel.from_json(json_str)
+            instance.actual_instance = AgeUserFilterModel.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
@@ -146,15 +140,21 @@ class CreateOrderModelUserFiltersInner(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into AgeUserFilterModel
+        # deserialize data into GenderUserFilterModel
         try:
-            instance.actual_instance = AgeUserFilterModel.from_json(json_str)
+            instance.actual_instance = GenderUserFilterModel.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into GenderUserFilterModel
+        # deserialize data into LanguageUserFilterModel
         try:
-            instance.actual_instance = GenderUserFilterModel.from_json(json_str)
+            instance.actual_instance = LanguageUserFilterModel.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into UserScoreUserFilterModel
+        try:
+            instance.actual_instance = UserScoreUserFilterModel.from_json(json_str)
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))

rapidata/api_client/models/customer_order_model.py

@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
@@ -34,6 +34,13 @@ class CustomerOrderModel(BaseModel):
     is_public: StrictBool = Field(alias="isPublic")
     __properties: ClassVar[List[str]] = ["id", "orderDate", "state", "orderName", "isPublic"]
 
+    @field_validator('state')
+    def state_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['Created', 'Submitted', 'ManualReview', 'Processing', 'Paused', 'Completed', 'Cancelled', 'Failed']):
+            raise ValueError("must be one of enum values ('Created', 'Submitted', 'ManualReview', 'Processing', 'Paused', 'Completed', 'Cancelled', 'Failed')")
+        return value
+
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/dataset_artifact_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 DatasetArtifactModel(BaseModel):
+    """
+    DatasetArtifactModel
+    """ # noqa: E501
+    t: StrictStr = Field(description="Discriminator value for DatasetArtifactModel", alias="_t")
+    dataset_id: StrictStr = Field(alias="datasetId")
+    identifier: StrictStr
+    __properties: ClassVar[List[str]] = ["_t", "datasetId", "identifier"]
+
+    @field_validator('t')
+    def t_validate_enum(cls, value):
+        """Validates the enum"""
+        if value not in set(['DatasetArtifactModel']):
+            raise ValueError("must be one of enum values ('DatasetArtifactModel')")
+        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 DatasetArtifactModel 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 DatasetArtifactModel 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 'DatasetArtifactModel',
+            "datasetId": obj.get("datasetId"),
+            "identifier": obj.get("identifier")
+        })
+        return _obj
+
+

rapidata/api_client/models/demographic_rapid_selection_config.py

@@ -17,8 +17,8 @@ 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 pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,8 +27,9 @@ class DemographicRapidSelectionConfig(BaseModel):
     DemographicRapidSelectionConfig
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for DemographicRapidSelectionConfig", alias="_t")
-    rapid_id: StrictStr = Field(alias="rapidId")
-    __properties: ClassVar[List[str]] = ["_t", "rapidId"]
+    keys: List[StrictStr]
+    max_rapids: Optional[StrictInt] = Field(default=None, alias="maxRapids")
+    __properties: ClassVar[List[str]] = ["_t", "keys", "maxRapids"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -89,7 +90,8 @@ class DemographicRapidSelectionConfig(BaseModel):
 
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'DemographicRapidSelectionConfig',
-            "rapidId": obj.get("rapidId")
+            "keys": obj.get("keys"),
+            "maxRapids": obj.get("maxRapids")
         })
         return _obj
 

rapidata/api_client/models/demographic_selection.py

@@ -17,18 +17,19 @@ 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 pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
+from typing import Any, ClassVar, Dict, List, Optional
 from typing import Optional, Set
 from typing_extensions import Self
 
 class DemographicSelection(BaseModel):
     """
-    The demographic selection can be used to show a specific demographic rapid to the users.
+    DemographicSelection
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for DemographicSelection", alias="_t")
-    rapid_id: StrictStr = Field(description="The rapid id is used to determine which rapid is shown to the user.", alias="rapidId")
-    __properties: ClassVar[List[str]] = ["_t", "rapidId"]
+    keys: List[StrictStr]
+    max_rapids: Optional[StrictInt] = Field(default=None, alias="maxRapids")
+    __properties: ClassVar[List[str]] = ["_t", "keys", "maxRapids"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -89,7 +90,8 @@ class DemographicSelection(BaseModel):
 
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'DemographicSelection',
-            "rapidId": obj.get("rapidId")
+            "keys": obj.get("keys"),
+            "maxRapids": obj.get("maxRapids")
         })
         return _obj