rapidata 2.36.0__py3-none-any.whl → 2.36.2__py3-none-any.whl

rapidata/api_client/models/create_order_model.py

@@ -20,10 +20,13 @@ 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.ab_test_selection_a_inner import AbTestSelectionAInner
+from rapidata.api_client.models.aggregator_type import AggregatorType
 from rapidata.api_client.models.and_user_filter_model_filters_inner import AndUserFilterModelFiltersInner
 from rapidata.api_client.models.create_order_model_referee import CreateOrderModelReferee
 from rapidata.api_client.models.create_order_model_workflow import CreateOrderModelWorkflow
 from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
+from rapidata.api_client.models.retrieval_mode import RetrievalMode
+from rapidata.api_client.models.sticky_state import StickyState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -35,16 +38,16 @@ class CreateOrderModel(BaseModel):
     order_name: StrictStr = Field(description="The name is used as an identifier for an order and can be freely chosen.", alias="orderName")
     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")
+    aggregator: Optional[AggregatorType] = None
     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.")
-    sticky_state: Optional[StrictStr] = Field(default=None, description="Indicates if the underlying campaign should be sticky.", alias="stickyState")
+    sticky_state: Optional[StickyState] = Field(default=None, alias="stickyState")
     user_score_dimensions: Optional[List[StrictStr]] = Field(default=None, description="The user score dimensions are used to determine the score of the responses from the user.", alias="userScoreDimensions")
     demographic_keys: Optional[List[StrictStr]] = Field(default=None, description="The demographic keys are used to determine which demographics to store on the resposnes from the user.", alias="demographicKeys")
     user_filters: Optional[List[AndUserFilterModelFiltersInner]] = Field(default=None, 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[AbTestSelectionAInner]] = Field(default=None, description="The selections are used to determine which tasks are shown to a user.")
-    retrieval_mode: Optional[StrictStr] = Field(default=None, description="The retrieval mode defines how rapids are retrieved from the active labeling pool.", alias="retrievalMode")
+    retrieval_mode: Optional[RetrievalMode] = Field(default=None, alias="retrievalMode")
     max_iterations: Optional[StrictInt] = Field(default=None, description="The maximum number of times a user is allowed to see the same rapid.", alias="maxIterations")
     __properties: ClassVar[List[str]] = ["_t", "orderName", "workflow", "referee", "aggregator", "featureFlags", "priority", "stickyState", "userScoreDimensions", "demographicKeys", "userFilters", "validationSetId", "selections", "retrievalMode", "maxIterations"]
 
@@ -55,36 +58,6 @@ class CreateOrderModel(BaseModel):
             raise ValueError("must be one of enum values ('CreateOrderModel')")
         return value
 
-    @field_validator('aggregator')
-    def aggregator_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['NonCommittal', 'MajorityVote', 'SimpleMatchup', 'LocateCluster', 'Classification', 'Locate', 'BoundingBox', 'Line', 'Transcription', 'SinglePointLocate', 'FreeText', 'Scrub']):
-            raise ValueError("must be one of enum values ('NonCommittal', 'MajorityVote', 'SimpleMatchup', 'LocateCluster', 'Classification', 'Locate', 'BoundingBox', 'Line', 'Transcription', 'SinglePointLocate', 'FreeText', 'Scrub')")
-        return value
-
-    @field_validator('sticky_state')
-    def sticky_state_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['None', 'Temporary', 'Permanent']):
-            raise ValueError("must be one of enum values ('None', 'Temporary', 'Permanent')")
-        return value
-
-    @field_validator('retrieval_mode')
-    def retrieval_mode_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['Random', 'Shuffled', 'Sequential']):
-            raise ValueError("must be one of enum values ('Random', 'Shuffled', 'Sequential')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,
@@ -151,11 +124,6 @@ class CreateOrderModel(BaseModel):
                 if _item_selections:
                     _items.append(_item_selections.to_dict())
             _dict['selections'] = _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 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:
@@ -166,11 +134,6 @@ class CreateOrderModel(BaseModel):
         if self.priority is None and "priority" in self.model_fields_set:
             _dict['priority'] = None
 
-        # set to None if sticky_state (nullable) is None
-        # and model_fields_set contains the field
-        if self.sticky_state is None and "sticky_state" in self.model_fields_set:
-            _dict['stickyState'] = None
-
         # set to None if user_score_dimensions (nullable) is None
         # and model_fields_set contains the field
         if self.user_score_dimensions is None and "user_score_dimensions" in self.model_fields_set:
@@ -196,11 +159,6 @@ class CreateOrderModel(BaseModel):
         if self.selections is None and "selections" in self.model_fields_set:
             _dict['selections'] = None
 
-        # set to None if retrieval_mode (nullable) is None
-        # and model_fields_set contains the field
-        if self.retrieval_mode is None and "retrieval_mode" in self.model_fields_set:
-            _dict['retrievalMode'] = None
-
         # set to None if max_iterations (nullable) is None
         # and model_fields_set contains the field
         if self.max_iterations is None and "max_iterations" in self.model_fields_set:

rapidata/api_client/models/effort_capped_selection.py

@@ -19,6 +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.retrieval_mode import RetrievalMode
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,7 +29,7 @@ class EffortCappedSelection(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for EffortCappedSelection", alias="_t")
     effort_budget: StrictInt = Field(alias="effortBudget")
-    retrieval_mode: Optional[StrictStr] = Field(default=None, alias="retrievalMode")
+    retrieval_mode: Optional[RetrievalMode] = Field(default=None, alias="retrievalMode")
     max_iterations: Optional[StrictInt] = Field(default=None, alias="maxIterations")
     __properties: ClassVar[List[str]] = ["_t", "effortBudget", "retrievalMode", "maxIterations"]
 
@@ -39,16 +40,6 @@ class EffortCappedSelection(BaseModel):
             raise ValueError("must be one of enum values ('EffortCappedSelection')")
         return value
 
-    @field_validator('retrieval_mode')
-    def retrieval_mode_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['Random', 'Shuffled', 'Sequential']):
-            raise ValueError("must be one of enum values ('Random', 'Shuffled', 'Sequential')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/evaluation_workflow_config.py

@@ -18,8 +18,9 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 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
 
@@ -31,7 +32,8 @@ class EvaluationWorkflowConfig(BaseModel):
     validation_set_id: StrictStr = Field(alias="validationSetId")
     referee: CompareWorkflowModel1Referee
     should_accept_incorrect: StrictBool = Field(alias="shouldAcceptIncorrect")
-    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "referee", "shouldAcceptIncorrect"]
+    feature_flags: Optional[List[FeatureFlag]] = Field(default=None, alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "referee", "shouldAcceptIncorrect", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -82,6 +84,13 @@ class EvaluationWorkflowConfig(BaseModel):
         # 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
         return _dict
 
     @classmethod
@@ -97,7 +106,8 @@ class EvaluationWorkflowConfig(BaseModel):
             "_t": obj.get("_t") if obj.get("_t") is not None else 'EvaluationWorkflowConfig',
             "validationSetId": obj.get("validationSetId"),
             "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
-            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect")
+            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect"),
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj
 

rapidata/api_client/models/evaluation_workflow_model.py

@@ -18,7 +18,8 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.feature_flag import FeatureFlag
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,7 +30,8 @@ class EvaluationWorkflowModel(BaseModel):
     t: StrictStr = Field(description="Discriminator value for EvaluationWorkflow", alias="_t")
     validation_set_id: StrictStr = Field(description="The Validation Set id is used to as a source for the tasks that will be sent to the user.", alias="validationSetId")
     should_accept_incorrect: StrictBool = Field(description="Indicates if the user should get feedback on their answers if they answer wrong. If set to true the user will not notice that he was tested.", alias="shouldAcceptIncorrect")
-    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "shouldAcceptIncorrect"]
+    feature_flags: Optional[List[FeatureFlag]] = Field(default=None, description="The list of feature flags that will be applied to the rapids created by this workflow.", alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "validationSetId", "shouldAcceptIncorrect", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -77,6 +79,13 @@ class EvaluationWorkflowModel(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
         return _dict
 
     @classmethod
@@ -91,7 +100,8 @@ class EvaluationWorkflowModel(BaseModel):
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'EvaluationWorkflow',
             "validationSetId": obj.get("validationSetId"),
-            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect")
+            "shouldAcceptIncorrect": obj.get("shouldAcceptIncorrect"),
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj
 

rapidata/api_client/models/file_type_metadata.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.file_type import FileType
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,8 +28,8 @@ class FileTypeMetadata(BaseModel):
     FileTypeMetadata
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for FileTypeMetadata", alias="_t")
-    file_type: StrictStr = Field(alias="fileType")
-    visibilities: Optional[StrictStr] = None
+    file_type: FileType = Field(alias="fileType")
+    visibilities: Optional[List[StrictStr]] = None
     __properties: ClassVar[List[str]] = ["_t", "fileType", "visibilities"]
 
     @field_validator('t')
@@ -38,11 +39,15 @@ class FileTypeMetadata(BaseModel):
             raise ValueError("must be one of enum values ('FileTypeMetadata')")
         return value
 
-    @field_validator('file_type')
-    def file_type_validate_enum(cls, value):
+    @field_validator('visibilities')
+    def visibilities_validate_enum(cls, value):
         """Validates the enum"""
-        if value not in set(['Unknown', 'Image', 'Video', 'Audio']):
-            raise ValueError("must be one of enum values ('Unknown', 'Image', 'Video', 'Audio')")
+        if value is None:
+            return value
+
+        for i in value:
+            if i not in set(['None', 'Users', 'Customers', 'Admins', 'Dashboard', 'All']):
+                raise ValueError("each list item must be one of ('None', 'Users', 'Customers', 'Admins', 'Dashboard', 'All')")
         return value
 
     model_config = ConfigDict(

rapidata/api_client/models/file_type_metadata_model.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List
+from rapidata.api_client.models.file_type import FileType
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,7 +28,7 @@ class FileTypeMetadataModel(BaseModel):
     FileTypeMetadataModel
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for FileTypeMetadata", alias="_t")
-    file_type: StrictStr = Field(alias="fileType")
+    file_type: FileType = Field(alias="fileType")
     __properties: ClassVar[List[str]] = ["_t", "fileType"]
 
     @field_validator('t')
@@ -37,13 +38,6 @@ class FileTypeMetadataModel(BaseModel):
             raise ValueError("must be one of enum values ('FileTypeMetadata')")
         return value
 
-    @field_validator('file_type')
-    def file_type_validate_enum(cls, value):
-        """Validates the enum"""
-        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(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/filter.py

@@ -17,8 +17,10 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.filter_operator import FilterOperator
+from rapidata.api_client.models.logic_operator import LogicOperator
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -28,31 +30,11 @@ class Filter(BaseModel):
     """ # noqa: E501
     var_field: Optional[StrictStr] = Field(default=None, alias="field")
     value: Optional[Any] = None
-    operator: Optional[StrictStr] = None
-    logic: Optional[StrictStr] = None
+    operator: Optional[FilterOperator] = None
+    logic: Optional[LogicOperator] = None
     filters: Optional[List[Filter]] = None
     __properties: ClassVar[List[str]] = ["field", "value", "operator", "logic", "filters"]
 
-    @field_validator('operator')
-    def operator_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['Eq', 'Neq', 'Gt', 'Gte', 'Lt', 'Lte', 'Contains', 'StartsWith', 'EndsWith', 'In']):
-            raise ValueError("must be one of enum values ('Eq', 'Neq', 'Gt', 'Gte', 'Lt', 'Lte', 'Contains', 'StartsWith', 'EndsWith', 'In')")
-        return value
-
-    @field_validator('logic')
-    def logic_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['And', 'Or', 'Not']):
-            raise ValueError("must be one of enum values ('And', 'Or', 'Not')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/get_datapoint_by_id_result.py

@@ -18,9 +18,10 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
 from rapidata.api_client.models.datapoint_asset import DatapointAsset
+from rapidata.api_client.models.datapoint_state import DatapointState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -30,19 +31,12 @@ class GetDatapointByIdResult(BaseModel):
     """ # noqa: E501
     id: StrictStr
     dataset_id: StrictStr = Field(alias="datasetId")
-    state: StrictStr
+    state: DatapointState
     sort_index: Optional[StrictInt] = Field(default=None, alias="sortIndex")
     asset: DatapointAsset
     created_at: datetime = Field(alias="createdAt")
     __properties: ClassVar[List[str]] = ["id", "datasetId", "state", "sortIndex", "asset", "createdAt"]
 
-    @field_validator('state')
-    def state_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Ready', 'Pending', 'Failed']):
-            raise ValueError("must be one of enum values ('Ready', 'Pending', 'Failed')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/get_rapid_responses_result.py

@@ -17,10 +17,11 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List
 from rapidata.api_client.models.datapoint_asset import DatapointAsset
 from rapidata.api_client.models.rapid_response import RapidResponse
+from rapidata.api_client.models.rapid_state import RapidState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -31,16 +32,9 @@ class GetRapidResponsesResult(BaseModel):
     rapid_id: StrictStr = Field(alias="rapidId")
     asset: DatapointAsset
     responses: List[RapidResponse]
-    state: StrictStr
+    state: RapidState
     __properties: ClassVar[List[str]] = ["rapidId", "asset", "responses", "state"]
 
-    @field_validator('state')
-    def state_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None']):
-            raise ValueError("must be one of enum values ('Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/get_recommended_validation_set_result.py

@@ -0,0 +1,95 @@
+# 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
+from typing import Any, ClassVar, Dict, List
+from rapidata.api_client.models.validation_set_model import ValidationSetModel
+from typing import Optional, Set
+from typing_extensions import Self
+
+class GetRecommendedValidationSetResult(BaseModel):
+    """
+    GetRecommendedValidationSetResult
+    """ # noqa: E501
+    validation_sets: List[ValidationSetModel] = Field(alias="validationSets")
+    __properties: ClassVar[List[str]] = ["validationSets"]
+
+    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 GetRecommendedValidationSetResult 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,
+        )
+        # override the default output from pydantic by calling `to_dict()` of each item in validation_sets (list)
+        _items = []
+        if self.validation_sets:
+            for _item_validation_sets in self.validation_sets:
+                if _item_validation_sets:
+                    _items.append(_item_validation_sets.to_dict())
+            _dict['validationSets'] = _items
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of GetRecommendedValidationSetResult from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "validationSets": [ValidationSetModel.from_dict(_item) for _item in obj["validationSets"]] if obj.get("validationSets") is not None else None
+        })
+        return _obj
+
+

rapidata/api_client/models/get_standing_by_id_result.py

@@ -17,8 +17,9 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List
+from rapidata.api_client.models.standing_status import StandingStatus
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -30,17 +31,10 @@ class GetStandingByIdResult(BaseModel):
     name: StrictStr
     benchmark_id: StrictStr = Field(alias="benchmarkId")
     dataset_id: StrictStr = Field(alias="datasetId")
-    status: StrictStr
+    status: StandingStatus
     is_disabled: StrictBool = Field(alias="isDisabled")
     __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "datasetId", "status", "isDisabled"]
 
-    @field_validator('status')
-    def status_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Created', 'Active', 'Idle']):
-            raise ValueError("must be one of enum values ('Created', 'Active', 'Idle')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/get_validation_rapids_result.py

@@ -17,12 +17,13 @@ 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, StrictFloat, StrictInt, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional, Union
 from rapidata.api_client.models.file_asset_model_metadata_value import FileAssetModelMetadataValue
 from rapidata.api_client.models.get_validation_rapids_result_asset import GetValidationRapidsResultAsset
 from rapidata.api_client.models.get_validation_rapids_result_payload import GetValidationRapidsResultPayload
 from rapidata.api_client.models.get_validation_rapids_result_truth import GetValidationRapidsResultTruth
+from rapidata.api_client.models.rapid_state import RapidState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -40,16 +41,9 @@ class GetValidationRapidsResult(BaseModel):
     invalid_validation_count: StrictInt = Field(alias="invalidValidationCount")
     explanation: Optional[StrictStr] = None
     random_correct_probability: Optional[Union[StrictFloat, StrictInt]] = Field(default=None, alias="randomCorrectProbability")
-    state: StrictStr
+    state: RapidState
     __properties: ClassVar[List[str]] = ["id", "type", "asset", "truth", "payload", "metadata", "correctValidationCount", "invalidValidationCount", "explanation", "randomCorrectProbability", "state"]
 
-    @field_validator('state')
-    def state_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None']):
-            raise ValueError("must be one of enum values ('Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/get_workflow_progress_result.py

@@ -17,8 +17,9 @@ 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, StrictFloat, StrictInt
 from typing import Any, ClassVar, Dict, List, Union
+from rapidata.api_client.models.workflow_state import WorkflowState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,16 +30,9 @@ class GetWorkflowProgressResult(BaseModel):
     completion_percentage: Union[StrictFloat, StrictInt] = Field(alias="completionPercentage")
     total: StrictInt
     completed: StrictInt
-    state: StrictStr
+    state: WorkflowState
     __properties: ClassVar[List[str]] = ["completionPercentage", "total", "completed", "state"]
 
-    @field_validator('state')
-    def state_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Created', 'Started', 'Labeling', 'Paused', 'Done', 'Failed']):
-            raise ValueError("must be one of enum values ('Created', 'Started', 'Labeling', 'Paused', 'Done', 'Failed')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/get_workflow_results_result.py

@@ -17,11 +17,12 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List
 from rapidata.api_client.models.datapoint_asset import DatapointAsset
 from rapidata.api_client.models.get_validation_rapids_result_payload import GetValidationRapidsResultPayload
 from rapidata.api_client.models.rapid_response import RapidResponse
+from rapidata.api_client.models.rapid_state import RapidState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -33,16 +34,9 @@ class GetWorkflowResultsResult(BaseModel):
     payload: GetValidationRapidsResultPayload
     asset: DatapointAsset
     responses: List[RapidResponse]
-    state: StrictStr
+    state: RapidState
     __properties: ClassVar[List[str]] = ["rapidId", "payload", "asset", "responses", "state"]
 
-    @field_validator('state')
-    def state_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None']):
-            raise ValueError("must be one of enum values ('Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,