rapidata 2.1.3__tar.gz → 2.2.0__tar.gz

{rapidata-2.1.3 → rapidata-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rapidata
-Version: 2.1.3
+Version: 2.2.0
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 Author: Rapidata AG
@@ -19,6 +19,7 @@ Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)
 Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
+Requires-Dist: tinytag (>=2.0.0,<3.0.0)
 Requires-Dist: tqdm (>=4.66.5,<5.0.0)
 Description-Content-Type: text/markdown
 

{rapidata-2.1.3 → rapidata-2.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "2.1.3"
+version = "2.2.0"
 description = "Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way."
 authors = ["Rapidata AG <info@rapidata.ai>"]
 license = "Apache-2.0"
@@ -17,6 +17,7 @@ python-dateutil = "^2.9.0.post0"
 tqdm = "^4.66.5"
 deprecated = "^1.2.14"
 colorama = "0.4.6"
+tinytag = "^2.0.0"
 
 [tool.poetry.group.dev.dependencies]
 python-dotenv = "^1.0.1"

{rapidata-2.1.3 → rapidata-2.2.0}/src/rapidata/api_client/__init__.py

@@ -247,6 +247,8 @@ from rapidata.api_client.models.query_model import QueryModel
 from rapidata.api_client.models.query_validation_rapids_result import QueryValidationRapidsResult
 from rapidata.api_client.models.query_validation_rapids_result_asset import QueryValidationRapidsResultAsset
 from rapidata.api_client.models.query_validation_rapids_result_paged_result import QueryValidationRapidsResultPagedResult
+from rapidata.api_client.models.query_validation_rapids_result_payload import QueryValidationRapidsResultPayload
+from rapidata.api_client.models.query_validation_rapids_result_truth import QueryValidationRapidsResultTruth
 from rapidata.api_client.models.query_validation_set_model import QueryValidationSetModel
 from rapidata.api_client.models.query_workflows_model import QueryWorkflowsModel
 from rapidata.api_client.models.ranked_datapoint_model import RankedDatapointModel
@@ -259,6 +261,11 @@ from rapidata.api_client.models.read_bridge_token_keys_result import ReadBridgeT
 from rapidata.api_client.models.register_temporary_customer_model import RegisterTemporaryCustomerModel
 from rapidata.api_client.models.register_temporary_customer_result import RegisterTemporaryCustomerResult
 from rapidata.api_client.models.root_filter import RootFilter
+from rapidata.api_client.models.scrub_payload import ScrubPayload
+from rapidata.api_client.models.scrub_range import ScrubRange
+from rapidata.api_client.models.scrub_rapid_blueprint import ScrubRapidBlueprint
+from rapidata.api_client.models.scrub_result import ScrubResult
+from rapidata.api_client.models.scrub_truth import ScrubTruth
 from rapidata.api_client.models.send_completion_mail_step_model import SendCompletionMailStepModel
 from rapidata.api_client.models.shape import Shape
 from rapidata.api_client.models.simple_workflow_config import SimpleWorkflowConfig

{rapidata-2.1.3 → rapidata-2.2.0}/src/rapidata/api_client/models/__init__.py

@@ -215,6 +215,8 @@ from rapidata.api_client.models.query_model import QueryModel
 from rapidata.api_client.models.query_validation_rapids_result import QueryValidationRapidsResult
 from rapidata.api_client.models.query_validation_rapids_result_asset import QueryValidationRapidsResultAsset
 from rapidata.api_client.models.query_validation_rapids_result_paged_result import QueryValidationRapidsResultPagedResult
+from rapidata.api_client.models.query_validation_rapids_result_payload import QueryValidationRapidsResultPayload
+from rapidata.api_client.models.query_validation_rapids_result_truth import QueryValidationRapidsResultTruth
 from rapidata.api_client.models.query_validation_set_model import QueryValidationSetModel
 from rapidata.api_client.models.query_workflows_model import QueryWorkflowsModel
 from rapidata.api_client.models.ranked_datapoint_model import RankedDatapointModel
@@ -227,6 +229,11 @@ from rapidata.api_client.models.read_bridge_token_keys_result import ReadBridgeT
 from rapidata.api_client.models.register_temporary_customer_model import RegisterTemporaryCustomerModel
 from rapidata.api_client.models.register_temporary_customer_result import RegisterTemporaryCustomerResult
 from rapidata.api_client.models.root_filter import RootFilter
+from rapidata.api_client.models.scrub_payload import ScrubPayload
+from rapidata.api_client.models.scrub_range import ScrubRange
+from rapidata.api_client.models.scrub_rapid_blueprint import ScrubRapidBlueprint
+from rapidata.api_client.models.scrub_result import ScrubResult
+from rapidata.api_client.models.scrub_truth import ScrubTruth
 from rapidata.api_client.models.send_completion_mail_step_model import SendCompletionMailStepModel
 from rapidata.api_client.models.shape import Shape
 from rapidata.api_client.models.simple_workflow_config import SimpleWorkflowConfig

{rapidata-2.1.3 → rapidata-2.2.0}/src/rapidata/api_client/models/add_validation_rapid_model_payload.py

@@ -25,12 +25,13 @@ from rapidata.api_client.models.line_payload import LinePayload
 from rapidata.api_client.models.locate_payload import LocatePayload
 from rapidata.api_client.models.named_entity_payload import NamedEntityPayload
 from rapidata.api_client.models.polygon_payload import PolygonPayload
+from rapidata.api_client.models.scrub_payload import ScrubPayload
 from rapidata.api_client.models.transcription_payload import TranscriptionPayload
 from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-ADDVALIDATIONRAPIDMODELPAYLOAD_ONE_OF_SCHEMAS = ["BoundingBoxPayload", "ClassifyPayload", "ComparePayload", "FreeTextPayload", "LinePayload", "LocatePayload", "NamedEntityPayload", "PolygonPayload", "TranscriptionPayload"]
+ADDVALIDATIONRAPIDMODELPAYLOAD_ONE_OF_SCHEMAS = ["BoundingBoxPayload", "ClassifyPayload", "ComparePayload", "FreeTextPayload", "LinePayload", "LocatePayload", "NamedEntityPayload", "PolygonPayload", "ScrubPayload", "TranscriptionPayload"]
 
 class AddValidationRapidModelPayload(BaseModel):
     """
@@ -38,24 +39,26 @@ class AddValidationRapidModelPayload(BaseModel):
     """
     # data type: TranscriptionPayload
     oneof_schema_1_validator: Optional[TranscriptionPayload] = None
+    # data type: ScrubPayload
+    oneof_schema_2_validator: Optional[ScrubPayload] = None
     # data type: PolygonPayload
-    oneof_schema_2_validator: Optional[PolygonPayload] = None
+    oneof_schema_3_validator: Optional[PolygonPayload] = None
     # data type: NamedEntityPayload
-    oneof_schema_3_validator: Optional[NamedEntityPayload] = None
+    oneof_schema_4_validator: Optional[NamedEntityPayload] = None
     # data type: LocatePayload
-    oneof_schema_4_validator: Optional[LocatePayload] = None
+    oneof_schema_5_validator: Optional[LocatePayload] = None
     # data type: LinePayload
-    oneof_schema_5_validator: Optional[LinePayload] = None
+    oneof_schema_6_validator: Optional[LinePayload] = None
     # data type: FreeTextPayload
-    oneof_schema_6_validator: Optional[FreeTextPayload] = None
+    oneof_schema_7_validator: Optional[FreeTextPayload] = None
     # data type: ComparePayload
-    oneof_schema_7_validator: Optional[ComparePayload] = None
+    oneof_schema_8_validator: Optional[ComparePayload] = None
     # data type: ClassifyPayload
-    oneof_schema_8_validator: Optional[ClassifyPayload] = None
+    oneof_schema_9_validator: Optional[ClassifyPayload] = None
     # data type: BoundingBoxPayload
-    oneof_schema_9_validator: Optional[BoundingBoxPayload] = None
-    actual_instance: Optional[Union[BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, TranscriptionPayload]] = None
-    one_of_schemas: Set[str] = { "BoundingBoxPayload", "ClassifyPayload", "ComparePayload", "FreeTextPayload", "LinePayload", "LocatePayload", "NamedEntityPayload", "PolygonPayload", "TranscriptionPayload" }
+    oneof_schema_10_validator: Optional[BoundingBoxPayload] = None
+    actual_instance: Optional[Union[BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload]] = None
+    one_of_schemas: Set[str] = { "BoundingBoxPayload", "ClassifyPayload", "ComparePayload", "FreeTextPayload", "LinePayload", "LocatePayload", "NamedEntityPayload", "PolygonPayload", "ScrubPayload", "TranscriptionPayload" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -86,6 +89,11 @@ class AddValidationRapidModelPayload(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `TranscriptionPayload`")
         else:
             match += 1
+        # validate data type: ScrubPayload
+        if not isinstance(v, ScrubPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ScrubPayload`")
+        else:
+            match += 1
         # validate data type: PolygonPayload
         if not isinstance(v, PolygonPayload):
             error_messages.append(f"Error! Input type `{type(v)}` is not `PolygonPayload`")
@@ -128,10 +136,10 @@ class AddValidationRapidModelPayload(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -152,6 +160,12 @@ class AddValidationRapidModelPayload(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into ScrubPayload
+        try:
+            instance.actual_instance = ScrubPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into PolygonPayload
         try:
             instance.actual_instance = PolygonPayload.from_json(json_str)
@@ -203,10 +217,10 @@ class AddValidationRapidModelPayload(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into AddValidationRapidModelPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -220,7 +234,7 @@ class AddValidationRapidModelPayload(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, TranscriptionPayload]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

{rapidata-2.1.3 → rapidata-2.2.0}/src/rapidata/api_client/models/add_validation_rapid_model_truth.py

@@ -25,12 +25,13 @@ from rapidata.api_client.models.line_truth import LineTruth
 from rapidata.api_client.models.locate_box_truth import LocateBoxTruth
 from rapidata.api_client.models.named_entity_truth import NamedEntityTruth
 from rapidata.api_client.models.polygon_truth import PolygonTruth
+from rapidata.api_client.models.scrub_truth import ScrubTruth
 from rapidata.api_client.models.transcription_truth import TranscriptionTruth
 from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-ADDVALIDATIONRAPIDMODELTRUTH_ONE_OF_SCHEMAS = ["AttachCategoryTruth", "BoundingBoxTruth", "CompareTruth", "EmptyValidationTruth", "LineTruth", "LocateBoxTruth", "NamedEntityTruth", "PolygonTruth", "TranscriptionTruth"]
+ADDVALIDATIONRAPIDMODELTRUTH_ONE_OF_SCHEMAS = ["AttachCategoryTruth", "BoundingBoxTruth", "CompareTruth", "EmptyValidationTruth", "LineTruth", "LocateBoxTruth", "NamedEntityTruth", "PolygonTruth", "ScrubTruth", "TranscriptionTruth"]
 
 class AddValidationRapidModelTruth(BaseModel):
     """
@@ -38,24 +39,26 @@ class AddValidationRapidModelTruth(BaseModel):
     """
     # data type: TranscriptionTruth
     oneof_schema_1_validator: Optional[TranscriptionTruth] = None
+    # data type: ScrubTruth
+    oneof_schema_2_validator: Optional[ScrubTruth] = None
     # data type: PolygonTruth
-    oneof_schema_2_validator: Optional[PolygonTruth] = None
+    oneof_schema_3_validator: Optional[PolygonTruth] = None
     # data type: NamedEntityTruth
-    oneof_schema_3_validator: Optional[NamedEntityTruth] = None
+    oneof_schema_4_validator: Optional[NamedEntityTruth] = None
     # data type: LocateBoxTruth
-    oneof_schema_4_validator: Optional[LocateBoxTruth] = None
+    oneof_schema_5_validator: Optional[LocateBoxTruth] = None
     # data type: LineTruth
-    oneof_schema_5_validator: Optional[LineTruth] = None
+    oneof_schema_6_validator: Optional[LineTruth] = None
     # data type: EmptyValidationTruth
-    oneof_schema_6_validator: Optional[EmptyValidationTruth] = None
+    oneof_schema_7_validator: Optional[EmptyValidationTruth] = None
     # data type: CompareTruth
-    oneof_schema_7_validator: Optional[CompareTruth] = None
+    oneof_schema_8_validator: Optional[CompareTruth] = None
     # data type: AttachCategoryTruth
-    oneof_schema_8_validator: Optional[AttachCategoryTruth] = None
+    oneof_schema_9_validator: Optional[AttachCategoryTruth] = None
     # data type: BoundingBoxTruth
-    oneof_schema_9_validator: Optional[BoundingBoxTruth] = None
-    actual_instance: Optional[Union[AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, TranscriptionTruth]] = None
-    one_of_schemas: Set[str] = { "AttachCategoryTruth", "BoundingBoxTruth", "CompareTruth", "EmptyValidationTruth", "LineTruth", "LocateBoxTruth", "NamedEntityTruth", "PolygonTruth", "TranscriptionTruth" }
+    oneof_schema_10_validator: Optional[BoundingBoxTruth] = None
+    actual_instance: Optional[Union[AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, ScrubTruth, TranscriptionTruth]] = None
+    one_of_schemas: Set[str] = { "AttachCategoryTruth", "BoundingBoxTruth", "CompareTruth", "EmptyValidationTruth", "LineTruth", "LocateBoxTruth", "NamedEntityTruth", "PolygonTruth", "ScrubTruth", "TranscriptionTruth" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -86,6 +89,11 @@ class AddValidationRapidModelTruth(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `TranscriptionTruth`")
         else:
             match += 1
+        # validate data type: ScrubTruth
+        if not isinstance(v, ScrubTruth):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ScrubTruth`")
+        else:
+            match += 1
         # validate data type: PolygonTruth
         if not isinstance(v, PolygonTruth):
             error_messages.append(f"Error! Input type `{type(v)}` is not `PolygonTruth`")
@@ -128,10 +136,10 @@ class AddValidationRapidModelTruth(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, ScrubTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, ScrubTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -152,6 +160,12 @@ class AddValidationRapidModelTruth(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into ScrubTruth
+        try:
+            instance.actual_instance = ScrubTruth.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into PolygonTruth
         try:
             instance.actual_instance = PolygonTruth.from_json(json_str)
@@ -203,10 +217,10 @@ class AddValidationRapidModelTruth(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, ScrubTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into AddValidationRapidModelTruth with oneOf schemas: AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, ScrubTruth, TranscriptionTruth. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -220,7 +234,7 @@ class AddValidationRapidModelTruth(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, TranscriptionTruth]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, ScrubTruth, TranscriptionTruth]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

{rapidata-2.1.3 → rapidata-2.2.0}/src/rapidata/api_client/models/query_validation_rapids_result.py

@@ -19,7 +19,10 @@ import json
 
 from pydantic import BaseModel, ConfigDict, StrictStr
 from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.file_asset_model1_metadata_inner import FileAssetModel1MetadataInner
 from rapidata.api_client.models.query_validation_rapids_result_asset import QueryValidationRapidsResultAsset
+from rapidata.api_client.models.query_validation_rapids_result_payload import QueryValidationRapidsResultPayload
+from rapidata.api_client.models.query_validation_rapids_result_truth import QueryValidationRapidsResultTruth
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,7 +32,10 @@ class QueryValidationRapidsResult(BaseModel):
     """ # noqa: E501
     type: StrictStr
     asset: Optional[QueryValidationRapidsResultAsset]
-    __properties: ClassVar[List[str]] = ["type", "asset"]
+    truth: Optional[QueryValidationRapidsResultTruth] = None
+    payload: QueryValidationRapidsResultPayload
+    metadata: List[FileAssetModel1MetadataInner]
+    __properties: ClassVar[List[str]] = ["type", "asset", "truth", "payload", "metadata"]
 
     model_config = ConfigDict(
         populate_by_name=True,
@@ -73,11 +79,29 @@ class QueryValidationRapidsResult(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of asset
         if self.asset:
             _dict['asset'] = self.asset.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of truth
+        if self.truth:
+            _dict['truth'] = self.truth.to_dict()
+        # 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 item in metadata (list)
+        _items = []
+        if self.metadata:
+            for _item_metadata in self.metadata:
+                if _item_metadata:
+                    _items.append(_item_metadata.to_dict())
+            _dict['metadata'] = _items
         # 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:
             _dict['asset'] = None
 
+        # set to None if truth (nullable) is None
+        # and model_fields_set contains the field
+        if self.truth is None and "truth" in self.model_fields_set:
+            _dict['truth'] = None
+
         return _dict
 
     @classmethod
@@ -91,7 +115,10 @@ class QueryValidationRapidsResult(BaseModel):
 
         _obj = cls.model_validate({
             "type": obj.get("type"),
-            "asset": QueryValidationRapidsResultAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None
+            "asset": QueryValidationRapidsResultAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
+            "truth": QueryValidationRapidsResultTruth.from_dict(obj["truth"]) if obj.get("truth") is not None else None,
+            "payload": QueryValidationRapidsResultPayload.from_dict(obj["payload"]) if obj.get("payload") is not None else None,
+            "metadata": [FileAssetModel1MetadataInner.from_dict(_item) for _item in obj["metadata"]] if obj.get("metadata") is not None else None
         })
         return _obj
 

rapidata-2.2.0/src/rapidata/api_client/models/query_validation_rapids_result_payload.py

@@ -0,0 +1,252 @@
+# coding: utf-8
+
+"""
+    Rapidata.Dataset
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+import pprint
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
+from typing import Any, List, Optional
+from rapidata.api_client.models.bounding_box_payload import BoundingBoxPayload
+from rapidata.api_client.models.classify_payload import ClassifyPayload
+from rapidata.api_client.models.compare_payload import ComparePayload
+from rapidata.api_client.models.free_text_payload import FreeTextPayload
+from rapidata.api_client.models.line_payload import LinePayload
+from rapidata.api_client.models.locate_payload import LocatePayload
+from rapidata.api_client.models.named_entity_payload import NamedEntityPayload
+from rapidata.api_client.models.polygon_payload import PolygonPayload
+from rapidata.api_client.models.scrub_payload import ScrubPayload
+from rapidata.api_client.models.transcription_payload import TranscriptionPayload
+from pydantic import StrictStr, Field
+from typing import Union, List, Set, Optional, Dict
+from typing_extensions import Literal, Self
+
+QUERYVALIDATIONRAPIDSRESULTPAYLOAD_ONE_OF_SCHEMAS = ["BoundingBoxPayload", "ClassifyPayload", "ComparePayload", "FreeTextPayload", "LinePayload", "LocatePayload", "NamedEntityPayload", "PolygonPayload", "ScrubPayload", "TranscriptionPayload"]
+
+class QueryValidationRapidsResultPayload(BaseModel):
+    """
+    QueryValidationRapidsResultPayload
+    """
+    # data type: TranscriptionPayload
+    oneof_schema_1_validator: Optional[TranscriptionPayload] = None
+    # data type: ScrubPayload
+    oneof_schema_2_validator: Optional[ScrubPayload] = None
+    # data type: PolygonPayload
+    oneof_schema_3_validator: Optional[PolygonPayload] = None
+    # data type: NamedEntityPayload
+    oneof_schema_4_validator: Optional[NamedEntityPayload] = None
+    # data type: LocatePayload
+    oneof_schema_5_validator: Optional[LocatePayload] = None
+    # data type: LinePayload
+    oneof_schema_6_validator: Optional[LinePayload] = None
+    # data type: FreeTextPayload
+    oneof_schema_7_validator: Optional[FreeTextPayload] = None
+    # data type: ComparePayload
+    oneof_schema_8_validator: Optional[ComparePayload] = None
+    # data type: ClassifyPayload
+    oneof_schema_9_validator: Optional[ClassifyPayload] = None
+    # data type: BoundingBoxPayload
+    oneof_schema_10_validator: Optional[BoundingBoxPayload] = None
+    actual_instance: Optional[Union[BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload]] = None
+    one_of_schemas: Set[str] = { "BoundingBoxPayload", "ClassifyPayload", "ComparePayload", "FreeTextPayload", "LinePayload", "LocatePayload", "NamedEntityPayload", "PolygonPayload", "ScrubPayload", "TranscriptionPayload" }
+
+    model_config = ConfigDict(
+        validate_assignment=True,
+        protected_namespaces=(),
+    )
+
+
+    discriminator_value_class_map: Dict[str, str] = {
+    }
+
+    def __init__(self, *args, **kwargs) -> None:
+        if args:
+            if len(args) > 1:
+                raise ValueError("If a position argument is used, only 1 is allowed to set `actual_instance`")
+            if kwargs:
+                raise ValueError("If a position argument is used, keyword arguments cannot be used.")
+            super().__init__(actual_instance=args[0])
+        else:
+            super().__init__(**kwargs)
+
+    @field_validator('actual_instance')
+    def actual_instance_must_validate_oneof(cls, v):
+        instance = QueryValidationRapidsResultPayload.model_construct()
+        error_messages = []
+        match = 0
+        # validate data type: TranscriptionPayload
+        if not isinstance(v, TranscriptionPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `TranscriptionPayload`")
+        else:
+            match += 1
+        # validate data type: ScrubPayload
+        if not isinstance(v, ScrubPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ScrubPayload`")
+        else:
+            match += 1
+        # validate data type: PolygonPayload
+        if not isinstance(v, PolygonPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `PolygonPayload`")
+        else:
+            match += 1
+        # validate data type: NamedEntityPayload
+        if not isinstance(v, NamedEntityPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `NamedEntityPayload`")
+        else:
+            match += 1
+        # validate data type: LocatePayload
+        if not isinstance(v, LocatePayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `LocatePayload`")
+        else:
+            match += 1
+        # validate data type: LinePayload
+        if not isinstance(v, LinePayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `LinePayload`")
+        else:
+            match += 1
+        # validate data type: FreeTextPayload
+        if not isinstance(v, FreeTextPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `FreeTextPayload`")
+        else:
+            match += 1
+        # validate data type: ComparePayload
+        if not isinstance(v, ComparePayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ComparePayload`")
+        else:
+            match += 1
+        # validate data type: ClassifyPayload
+        if not isinstance(v, ClassifyPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ClassifyPayload`")
+        else:
+            match += 1
+        # validate data type: BoundingBoxPayload
+        if not isinstance(v, BoundingBoxPayload):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `BoundingBoxPayload`")
+        else:
+            match += 1
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when setting `actual_instance` in QueryValidationRapidsResultPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when setting `actual_instance` in QueryValidationRapidsResultPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+        else:
+            return v
+
+    @classmethod
+    def from_dict(cls, obj: Union[str, Dict[str, Any]]) -> Self:
+        return cls.from_json(json.dumps(obj))
+
+    @classmethod
+    def from_json(cls, json_str: str) -> Self:
+        """Returns the object represented by the json string"""
+        instance = cls.model_construct()
+        error_messages = []
+        match = 0
+
+        # deserialize data into TranscriptionPayload
+        try:
+            instance.actual_instance = TranscriptionPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into ScrubPayload
+        try:
+            instance.actual_instance = ScrubPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into PolygonPayload
+        try:
+            instance.actual_instance = PolygonPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into NamedEntityPayload
+        try:
+            instance.actual_instance = NamedEntityPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into LocatePayload
+        try:
+            instance.actual_instance = LocatePayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into LinePayload
+        try:
+            instance.actual_instance = LinePayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into FreeTextPayload
+        try:
+            instance.actual_instance = FreeTextPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into ComparePayload
+        try:
+            instance.actual_instance = ComparePayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into ClassifyPayload
+        try:
+            instance.actual_instance = ClassifyPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into BoundingBoxPayload
+        try:
+            instance.actual_instance = BoundingBoxPayload.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when deserializing the JSON string into QueryValidationRapidsResultPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when deserializing the JSON string into QueryValidationRapidsResultPayload with oneOf schemas: BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload. Details: " + ", ".join(error_messages))
+        else:
+            return instance
+
+    def to_json(self) -> str:
+        """Returns the JSON representation of the actual instance"""
+        if self.actual_instance is None:
+            return "null"
+
+        if hasattr(self.actual_instance, "to_json") and callable(self.actual_instance.to_json):
+            return self.actual_instance.to_json()
+        else:
+            return json.dumps(self.actual_instance)
+
+    def to_dict(self) -> Optional[Union[Dict[str, Any], BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, ScrubPayload, TranscriptionPayload]]:
+        """Returns the dict representation of the actual instance"""
+        if self.actual_instance is None:
+            return None
+
+        if hasattr(self.actual_instance, "to_dict") and callable(self.actual_instance.to_dict):
+            return self.actual_instance.to_dict()
+        else:
+            # primitive type
+            return self.actual_instance
+
+    def to_str(self) -> str:
+        """Returns the string representation of the actual instance"""
+        return pprint.pformat(self.model_dump())
+
+