rapidata 2.27.0__py3-none-any.whl → 2.44.4__py3-none-any.whl

rapidata/api_client/models/get_simple_workflow_results_model.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -17,23 +17,33 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
 from rapidata.api_client.models.page_info import PageInfo
-from rapidata.api_client.models.rapid_state import RapidState
 from rapidata.api_client.models.sort_criterion import SortCriterion
 from typing import Optional, Set
 from typing_extensions import Self
 
 class GetSimpleWorkflowResultsModel(BaseModel):
     """
-    Model for getting the overview of a simple workflow result.
+    GetSimpleWorkflowResultsModel
     """ # noqa: E501
     page: Optional[PageInfo] = None
-    states: Optional[List[RapidState]] = Field(default=None, description="An optional list of states to filter the rapids by.")
-    sort_criteria: Optional[List[SortCriterion]] = Field(default=None, description="A list of criteria to sort the results by.", alias="sortCriteria")
+    states: Optional[List[StrictStr]] = None
+    sort_criteria: Optional[List[SortCriterion]] = Field(default=None, alias="sortCriteria")
     __properties: ClassVar[List[str]] = ["page", "states", "sortCriteria"]
 
+    @field_validator('states')
+    def states_validate_enum(cls, value):
+        """Validates the enum"""
+        if value is None:
+            return value
+
+        for i in value:
+            if i not in set(['Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None']):
+                raise ValueError("each list item must be one of ('Labeling', 'Paused', 'Incomplete', 'Flagged', 'Done', 'None')")
+        return value
+
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,
@@ -83,15 +93,10 @@ class GetSimpleWorkflowResultsModel(BaseModel):
                 if _item_sort_criteria:
                     _items.append(_item_sort_criteria.to_dict())
             _dict['sortCriteria'] = _items
-        # set to None if states (nullable) is None
-        # and model_fields_set contains the field
-        if self.states is None and "states" in self.model_fields_set:
-            _dict['states'] = None
-
-        # set to None if sort_criteria (nullable) is None
+        # set to None if page (nullable) is None
         # and model_fields_set contains the field
-        if self.sort_criteria is None and "sort_criteria" in self.model_fields_set:
-            _dict['sortCriteria'] = None
+        if self.page is None and "page" in self.model_fields_set:
+            _dict['page'] = None
 
         return _dict
 

rapidata/api_client/models/get_standing_by_id_result.py

@@ -0,0 +1,96 @@
+# coding: utf-8
+
+"""
+    Rapidata Asset API
+
+    The API for the Rapidata Asset service
+
+    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, 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
+
+class GetStandingByIdResult(BaseModel):
+    """
+    GetStandingByIdResult
+    """ # noqa: E501
+    id: StrictStr
+    name: StrictStr
+    benchmark_id: StrictStr = Field(alias="benchmarkId")
+    status: StandingStatus
+    is_disabled: StrictBool = Field(alias="isDisabled")
+    __properties: ClassVar[List[str]] = ["id", "name", "benchmarkId", "status", "isDisabled"]
+
+    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 GetStandingByIdResult 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 GetStandingByIdResult from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "id": obj.get("id"),
+            "name": obj.get("name"),
+            "benchmarkId": obj.get("benchmarkId"),
+            "status": obj.get("status"),
+            "isDisabled": obj.get("isDisabled")
+        })
+        return _obj
+
+

rapidata/api_client/models/get_validation_rapids_result.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -17,12 +17,12 @@ 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.i_asset_model import IAssetModel
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
+from rapidata.api_client.models.i_validation_truth import IValidationTruth
+from rapidata.api_client.models.rapid_state import RapidState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -32,23 +32,17 @@ class GetValidationRapidsResult(BaseModel):
     """ # noqa: E501
     id: StrictStr
     type: StrictStr
-    asset: Optional[GetValidationRapidsResultAsset] = None
-    truth: Optional[GetValidationRapidsResultTruth] = None
-    payload: GetValidationRapidsResultPayload
-    metadata: Dict[str, FileAssetModelMetadataValue]
+    asset: Optional[IAssetModel]
+    truth: Optional[IValidationTruth] = None
+    payload: IRapidPayload
+    context: Optional[StrictStr] = None
+    context_asset: Optional[IAssetModel] = Field(default=None, alias="contextAsset")
     correct_validation_count: StrictInt = Field(alias="correctValidationCount")
     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
-    __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
+    state: RapidState
+    __properties: ClassVar[List[str]] = ["id", "type", "asset", "truth", "payload", "context", "contextAsset", "correctValidationCount", "invalidValidationCount", "explanation", "randomCorrectProbability", "state"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -98,13 +92,9 @@ class GetValidationRapidsResult(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of payload
         if self.payload:
             _dict['payload'] = self.payload.to_dict()
-        # override the default output from pydantic by calling `to_dict()` of each value in metadata (dict)
-        _field_dict = {}
-        if self.metadata:
-            for _key_metadata in self.metadata:
-                if self.metadata[_key_metadata]:
-                    _field_dict[_key_metadata] = self.metadata[_key_metadata].to_dict()
-            _dict['metadata'] = _field_dict
+        # override the default output from pydantic by calling `to_dict()` of context_asset
+        if self.context_asset:
+            _dict['contextAsset'] = self.context_asset.to_dict()
         # set to None if asset (nullable) is None
         # and model_fields_set contains the field
         if self.asset is None and "asset" in self.model_fields_set:
@@ -115,16 +105,21 @@ class GetValidationRapidsResult(BaseModel):
         if self.truth is None and "truth" in self.model_fields_set:
             _dict['truth'] = None
 
+        # set to None if context (nullable) is None
+        # and model_fields_set contains the field
+        if self.context is None and "context" in self.model_fields_set:
+            _dict['context'] = None
+
+        # set to None if context_asset (nullable) is None
+        # and model_fields_set contains the field
+        if self.context_asset is None and "context_asset" in self.model_fields_set:
+            _dict['contextAsset'] = None
+
         # set to None if explanation (nullable) is None
         # and model_fields_set contains the field
         if self.explanation is None and "explanation" in self.model_fields_set:
             _dict['explanation'] = None
 
-        # set to None if random_correct_probability (nullable) is None
-        # and model_fields_set contains the field
-        if self.random_correct_probability is None and "random_correct_probability" in self.model_fields_set:
-            _dict['randomCorrectProbability'] = None
-
         return _dict
 
     @classmethod
@@ -139,15 +134,11 @@ class GetValidationRapidsResult(BaseModel):
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "type": obj.get("type"),
-            "asset": GetValidationRapidsResultAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
-            "truth": GetValidationRapidsResultTruth.from_dict(obj["truth"]) if obj.get("truth") is not None else None,
-            "payload": GetValidationRapidsResultPayload.from_dict(obj["payload"]) if obj.get("payload") is not None else None,
-            "metadata": dict(
-                (_k, FileAssetModelMetadataValue.from_dict(_v))
-                for _k, _v in obj["metadata"].items()
-            )
-            if obj.get("metadata") is not None
-            else None,
+            "asset": IAssetModel.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
+            "truth": IValidationTruth.from_dict(obj["truth"]) if obj.get("truth") is not None else None,
+            "payload": IRapidPayload.from_dict(obj["payload"]) if obj.get("payload") is not None else None,
+            "context": obj.get("context"),
+            "contextAsset": IAssetModel.from_dict(obj["contextAsset"]) if obj.get("contextAsset") is not None else None,
             "correctValidationCount": obj.get("correctValidationCount"),
             "invalidValidationCount": obj.get("invalidValidationCount"),
             "explanation": obj.get("explanation"),

rapidata/api_client/models/get_validation_set_by_id_result.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -18,7 +18,7 @@ import re  # noqa: F401
 import json
 
 from datetime import datetime
-from pydantic import BaseModel, ConfigDict, Field, StrictStr
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictStr
 from typing import Any, ClassVar, Dict, List
 from typing import Optional, Set
 from typing_extensions import Self
@@ -29,9 +29,12 @@ class GetValidationSetByIdResult(BaseModel):
     """ # noqa: E501
     id: StrictStr
     name: StrictStr
+    is_public: StrictBool = Field(alias="isPublic")
     owner_mail: StrictStr = Field(alias="ownerMail")
+    owner_id: StrictStr = Field(alias="ownerId")
+    dimensions: List[StrictStr]
     created_at: datetime = Field(alias="createdAt")
-    __properties: ClassVar[List[str]] = ["id", "name", "ownerMail", "createdAt"]
+    __properties: ClassVar[List[str]] = ["id", "name", "isPublic", "ownerMail", "ownerId", "dimensions", "createdAt"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -86,7 +89,10 @@ class GetValidationSetByIdResult(BaseModel):
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "name": obj.get("name"),
+            "isPublic": obj.get("isPublic"),
             "ownerMail": obj.get("ownerMail"),
+            "ownerId": obj.get("ownerId"),
+            "dimensions": obj.get("dimensions"),
             "createdAt": obj.get("createdAt")
         })
         return _obj

rapidata/api_client/models/get_workflow_by_id_result.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict
 from typing import Any, ClassVar, Dict, List
-from rapidata.api_client.models.get_workflow_by_id_result_workflow import GetWorkflowByIdResultWorkflow
+from rapidata.api_client.models.i_workflow_model import IWorkflowModel
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,7 +27,7 @@ class GetWorkflowByIdResult(BaseModel):
     """
     GetWorkflowByIdResult
     """ # noqa: E501
-    workflow: GetWorkflowByIdResultWorkflow
+    workflow: IWorkflowModel
     __properties: ClassVar[List[str]] = ["workflow"]
 
     model_config = ConfigDict(
@@ -84,7 +84,7 @@ class GetWorkflowByIdResult(BaseModel):
             return cls.model_validate(obj)
 
         _obj = cls.model_validate({
-            "workflow": GetWorkflowByIdResultWorkflow.from_dict(obj["workflow"]) if obj.get("workflow") is not None else None
+            "workflow": IWorkflowModel.from_dict(obj["workflow"]) if obj.get("workflow") is not None else None
         })
         return _obj
 

rapidata/api_client/models/get_workflow_by_id_result_workflow.py

@@ -19,12 +19,13 @@ from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, f
 from typing import Any, List, Optional
 from rapidata.api_client.models.compare_workflow_model1 import CompareWorkflowModel1
 from rapidata.api_client.models.evaluation_workflow_model1 import EvaluationWorkflowModel1
+from rapidata.api_client.models.grouped_ranking_workflow_model1 import GroupedRankingWorkflowModel1
 from rapidata.api_client.models.simple_workflow_model1 import SimpleWorkflowModel1
 from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-GETWORKFLOWBYIDRESULTWORKFLOW_ONE_OF_SCHEMAS = ["CompareWorkflowModel1", "EvaluationWorkflowModel1", "SimpleWorkflowModel1"]
+GETWORKFLOWBYIDRESULTWORKFLOW_ONE_OF_SCHEMAS = ["CompareWorkflowModel1", "EvaluationWorkflowModel1", "GroupedRankingWorkflowModel1", "SimpleWorkflowModel1"]
 
 class GetWorkflowByIdResultWorkflow(BaseModel):
     """
@@ -34,10 +35,12 @@ class GetWorkflowByIdResultWorkflow(BaseModel):
     oneof_schema_1_validator: Optional[CompareWorkflowModel1] = None
     # data type: EvaluationWorkflowModel1
     oneof_schema_2_validator: Optional[EvaluationWorkflowModel1] = None
+    # data type: GroupedRankingWorkflowModel1
+    oneof_schema_3_validator: Optional[GroupedRankingWorkflowModel1] = None
     # data type: SimpleWorkflowModel1
-    oneof_schema_3_validator: Optional[SimpleWorkflowModel1] = None
-    actual_instance: Optional[Union[CompareWorkflowModel1, EvaluationWorkflowModel1, SimpleWorkflowModel1]] = None
-    one_of_schemas: Set[str] = { "CompareWorkflowModel1", "EvaluationWorkflowModel1", "SimpleWorkflowModel1" }
+    oneof_schema_4_validator: Optional[SimpleWorkflowModel1] = None
+    actual_instance: Optional[Union[CompareWorkflowModel1, EvaluationWorkflowModel1, GroupedRankingWorkflowModel1, SimpleWorkflowModel1]] = None
+    one_of_schemas: Set[str] = { "CompareWorkflowModel1", "EvaluationWorkflowModel1", "GroupedRankingWorkflowModel1", "SimpleWorkflowModel1" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -73,6 +76,11 @@ class GetWorkflowByIdResultWorkflow(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `EvaluationWorkflowModel1`")
         else:
             match += 1
+        # validate data type: GroupedRankingWorkflowModel1
+        if not isinstance(v, GroupedRankingWorkflowModel1):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `GroupedRankingWorkflowModel1`")
+        else:
+            match += 1
         # validate data type: SimpleWorkflowModel1
         if not isinstance(v, SimpleWorkflowModel1):
             error_messages.append(f"Error! Input type `{type(v)}` is not `SimpleWorkflowModel1`")
@@ -80,10 +88,10 @@ class GetWorkflowByIdResultWorkflow(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, GroupedRankingWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, GroupedRankingWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -110,6 +118,12 @@ class GetWorkflowByIdResultWorkflow(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into GroupedRankingWorkflowModel1
+        try:
+            instance.actual_instance = GroupedRankingWorkflowModel1.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into SimpleWorkflowModel1
         try:
             instance.actual_instance = SimpleWorkflowModel1.from_json(json_str)
@@ -119,10 +133,10 @@ class GetWorkflowByIdResultWorkflow(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, GroupedRankingWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into GetWorkflowByIdResultWorkflow with oneOf schemas: CompareWorkflowModel1, EvaluationWorkflowModel1, GroupedRankingWorkflowModel1, SimpleWorkflowModel1. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -136,7 +150,7 @@ class GetWorkflowByIdResultWorkflow(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], CompareWorkflowModel1, EvaluationWorkflowModel1, SimpleWorkflowModel1]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], CompareWorkflowModel1, EvaluationWorkflowModel1, GroupedRankingWorkflowModel1, SimpleWorkflowModel1]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

rapidata/api_client/models/get_workflow_progress_result.py

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -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

@@ -1,9 +1,9 @@
 # coding: utf-8
 
 """
-    Rapidata.Dataset
+    Rapidata Asset API
 
-    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+    The API for the Rapidata Asset service
 
     The version of the OpenAPI document: v1
     Generated by OpenAPI Generator (https://openapi-generator.tech)
@@ -17,11 +17,12 @@ 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 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 pydantic import BaseModel, ConfigDict, Field, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.get_workflow_results_result_response import GetWorkflowResultsResultResponse
+from rapidata.api_client.models.i_asset_model import IAssetModel
+from rapidata.api_client.models.i_rapid_payload import IRapidPayload
+from rapidata.api_client.models.rapid_state import RapidState
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -30,18 +31,13 @@ class GetWorkflowResultsResult(BaseModel):
     GetWorkflowResultsResult
     """ # noqa: E501
     rapid_id: StrictStr = Field(alias="rapidId")
-    payload: GetValidationRapidsResultPayload
-    asset: DatapointAsset
-    responses: List[RapidResponse]
-    state: StrictStr
-    __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
+    payload: IRapidPayload
+    asset: IAssetModel
+    responses: List[GetWorkflowResultsResultResponse]
+    state: RapidState
+    context: Optional[StrictStr] = None
+    context_asset: Optional[IAssetModel] = Field(default=None, alias="contextAsset")
+    __properties: ClassVar[List[str]] = ["rapidId", "payload", "asset", "responses", "state", "context", "contextAsset"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -95,6 +91,19 @@ class GetWorkflowResultsResult(BaseModel):
                 if _item_responses:
                     _items.append(_item_responses.to_dict())
             _dict['responses'] = _items
+        # override the default output from pydantic by calling `to_dict()` of context_asset
+        if self.context_asset:
+            _dict['contextAsset'] = self.context_asset.to_dict()
+        # set to None if context (nullable) is None
+        # and model_fields_set contains the field
+        if self.context is None and "context" in self.model_fields_set:
+            _dict['context'] = None
+
+        # set to None if context_asset (nullable) is None
+        # and model_fields_set contains the field
+        if self.context_asset is None and "context_asset" in self.model_fields_set:
+            _dict['contextAsset'] = None
+
         return _dict
 
     @classmethod
@@ -108,10 +117,12 @@ class GetWorkflowResultsResult(BaseModel):
 
         _obj = cls.model_validate({
             "rapidId": obj.get("rapidId"),
-            "payload": GetValidationRapidsResultPayload.from_dict(obj["payload"]) if obj.get("payload") is not None else None,
-            "asset": DatapointAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
-            "responses": [RapidResponse.from_dict(_item) for _item in obj["responses"]] if obj.get("responses") is not None else None,
-            "state": obj.get("state")
+            "payload": IRapidPayload.from_dict(obj["payload"]) if obj.get("payload") is not None else None,
+            "asset": IAssetModel.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
+            "responses": [GetWorkflowResultsResultResponse.from_dict(_item) for _item in obj["responses"]] if obj.get("responses") is not None else None,
+            "state": obj.get("state"),
+            "context": obj.get("context"),
+            "contextAsset": IAssetModel.from_dict(obj["contextAsset"]) if obj.get("contextAsset") is not None else None
         })
         return _obj