rapidata 2.23.1__py3-none-any.whl → 2.24.0__py3-none-any.whl

rapidata/api_client/models/asset_metadata_model_asset.py

@@ -0,0 +1,170 @@
+# 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 pydantic import StrictStr, Field
+from typing import Union, List, Set, Optional, Dict
+from typing_extensions import Literal, Self
+
+ASSETMETADATAMODELASSET_ONE_OF_SCHEMAS = ["FileAssetModel", "MultiAssetModel", "NullAssetModel", "TextAssetModel"]
+
+class AssetMetadataModelAsset(BaseModel):
+    """
+    AssetMetadataModelAsset
+    """
+    # data type: FileAssetModel
+    oneof_schema_1_validator: Optional[FileAssetModel] = None
+    # data type: MultiAssetModel
+    oneof_schema_2_validator: Optional[MultiAssetModel] = None
+    # data type: NullAssetModel
+    oneof_schema_3_validator: Optional[NullAssetModel] = None
+    # data type: TextAssetModel
+    oneof_schema_4_validator: Optional[TextAssetModel] = None
+    actual_instance: Optional[Union[FileAssetModel, MultiAssetModel, NullAssetModel, TextAssetModel]] = None
+    one_of_schemas: Set[str] = { "FileAssetModel", "MultiAssetModel", "NullAssetModel", "TextAssetModel" }
+
+    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 = AssetMetadataModelAsset.model_construct()
+        error_messages = []
+        match = 0
+        # validate data type: FileAssetModel
+        if not isinstance(v, FileAssetModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `FileAssetModel`")
+        else:
+            match += 1
+        # validate data type: MultiAssetModel
+        if not isinstance(v, MultiAssetModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `MultiAssetModel`")
+        else:
+            match += 1
+        # validate data type: NullAssetModel
+        if not isinstance(v, NullAssetModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `NullAssetModel`")
+        else:
+            match += 1
+        # validate data type: TextAssetModel
+        if not isinstance(v, TextAssetModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `TextAssetModel`")
+        else:
+            match += 1
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when setting `actual_instance` in AssetMetadataModelAsset with oneOf schemas: FileAssetModel, MultiAssetModel, NullAssetModel, TextAssetModel. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when setting `actual_instance` in AssetMetadataModelAsset with oneOf schemas: FileAssetModel, MultiAssetModel, NullAssetModel, TextAssetModel. 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 FileAssetModel
+        try:
+            instance.actual_instance = FileAssetModel.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into MultiAssetModel
+        try:
+            instance.actual_instance = MultiAssetModel.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into NullAssetModel
+        try:
+            instance.actual_instance = NullAssetModel.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into TextAssetModel
+        try:
+            instance.actual_instance = TextAssetModel.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 AssetMetadataModelAsset with oneOf schemas: FileAssetModel, MultiAssetModel, NullAssetModel, TextAssetModel. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when deserializing the JSON string into AssetMetadataModelAsset with oneOf schemas: FileAssetModel, MultiAssetModel, NullAssetModel, TextAssetModel. 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], FileAssetModel, MultiAssetModel, NullAssetModel, TextAssetModel]]:
+        """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())
+
+from rapidata.api_client.models.file_asset_model import FileAssetModel
+from rapidata.api_client.models.multi_asset_model import MultiAssetModel
+from rapidata.api_client.models.null_asset_model import NullAssetModel
+from rapidata.api_client.models.text_asset_model import TextAssetModel
+# TODO: Rewrite to not use raise_errors
+AssetMetadataModelAsset.model_rebuild(raise_errors=False)
+

rapidata/api_client/models/compare_workflow_config_metadata_value.py

@@ -30,32 +30,34 @@ from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-COMPAREWORKFLOWCONFIGMETADATAVALUE_ONE_OF_SCHEMAS = ["ClassificationMetadata", "CountMetadata", "ImageDimensionMetadata", "LocationMetadata", "OriginalFilenameMetadata", "PromptMetadata", "SourceUrlMetadata", "TextMetadata", "TranscriptionMetadata"]
+COMPAREWORKFLOWCONFIGMETADATAVALUE_ONE_OF_SCHEMAS = ["AssetMetadata", "ClassificationMetadata", "CountMetadata", "ImageDimensionMetadata", "LocationMetadata", "OriginalFilenameMetadata", "PromptMetadata", "SourceUrlMetadata", "TextMetadata", "TranscriptionMetadata"]
 
 class CompareWorkflowConfigMetadataValue(BaseModel):
     """
     CompareWorkflowConfigMetadataValue
     """
+    # data type: AssetMetadata
+    oneof_schema_1_validator: Optional[AssetMetadata] = None
     # data type: ClassificationMetadata
-    oneof_schema_1_validator: Optional[ClassificationMetadata] = None
+    oneof_schema_2_validator: Optional[ClassificationMetadata] = None
     # data type: CountMetadata
-    oneof_schema_2_validator: Optional[CountMetadata] = None
+    oneof_schema_3_validator: Optional[CountMetadata] = None
     # data type: ImageDimensionMetadata
-    oneof_schema_3_validator: Optional[ImageDimensionMetadata] = None
+    oneof_schema_4_validator: Optional[ImageDimensionMetadata] = None
     # data type: LocationMetadata
-    oneof_schema_4_validator: Optional[LocationMetadata] = None
+    oneof_schema_5_validator: Optional[LocationMetadata] = None
     # data type: OriginalFilenameMetadata
-    oneof_schema_5_validator: Optional[OriginalFilenameMetadata] = None
+    oneof_schema_6_validator: Optional[OriginalFilenameMetadata] = None
     # data type: PromptMetadata
-    oneof_schema_6_validator: Optional[PromptMetadata] = None
+    oneof_schema_7_validator: Optional[PromptMetadata] = None
     # data type: SourceUrlMetadata
-    oneof_schema_7_validator: Optional[SourceUrlMetadata] = None
+    oneof_schema_8_validator: Optional[SourceUrlMetadata] = None
     # data type: TextMetadata
-    oneof_schema_8_validator: Optional[TextMetadata] = None
+    oneof_schema_9_validator: Optional[TextMetadata] = None
     # data type: TranscriptionMetadata
-    oneof_schema_9_validator: Optional[TranscriptionMetadata] = None
-    actual_instance: Optional[Union[ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata]] = None
-    one_of_schemas: Set[str] = { "ClassificationMetadata", "CountMetadata", "ImageDimensionMetadata", "LocationMetadata", "OriginalFilenameMetadata", "PromptMetadata", "SourceUrlMetadata", "TextMetadata", "TranscriptionMetadata" }
+    oneof_schema_10_validator: Optional[TranscriptionMetadata] = None
+    actual_instance: Optional[Union[AssetMetadata, ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata]] = None
+    one_of_schemas: Set[str] = { "AssetMetadata", "ClassificationMetadata", "CountMetadata", "ImageDimensionMetadata", "LocationMetadata", "OriginalFilenameMetadata", "PromptMetadata", "SourceUrlMetadata", "TextMetadata", "TranscriptionMetadata" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -81,6 +83,11 @@ class CompareWorkflowConfigMetadataValue(BaseModel):
         instance = CompareWorkflowConfigMetadataValue.model_construct()
         error_messages = []
         match = 0
+        # validate data type: AssetMetadata
+        if not isinstance(v, AssetMetadata):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AssetMetadata`")
+        else:
+            match += 1
         # validate data type: ClassificationMetadata
         if not isinstance(v, ClassificationMetadata):
             error_messages.append(f"Error! Input type `{type(v)}` is not `ClassificationMetadata`")
@@ -128,10 +135,10 @@ class CompareWorkflowConfigMetadataValue(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in CompareWorkflowConfigMetadataValue with oneOf schemas: ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in CompareWorkflowConfigMetadataValue with oneOf schemas: AssetMetadata, ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in CompareWorkflowConfigMetadataValue with oneOf schemas: ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in CompareWorkflowConfigMetadataValue with oneOf schemas: AssetMetadata, ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -146,6 +153,12 @@ class CompareWorkflowConfigMetadataValue(BaseModel):
         error_messages = []
         match = 0
 
+        # deserialize data into AssetMetadata
+        try:
+            instance.actual_instance = AssetMetadata.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into ClassificationMetadata
         try:
             instance.actual_instance = ClassificationMetadata.from_json(json_str)
@@ -203,10 +216,10 @@ class CompareWorkflowConfigMetadataValue(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into CompareWorkflowConfigMetadataValue with oneOf schemas: ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into CompareWorkflowConfigMetadataValue with oneOf schemas: AssetMetadata, ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into CompareWorkflowConfigMetadataValue with oneOf schemas: ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into CompareWorkflowConfigMetadataValue with oneOf schemas: AssetMetadata, ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -220,7 +233,7 @@ class CompareWorkflowConfigMetadataValue(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], AssetMetadata, ClassificationMetadata, CountMetadata, ImageDimensionMetadata, LocationMetadata, OriginalFilenameMetadata, PromptMetadata, SourceUrlMetadata, TextMetadata, TranscriptionMetadata]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None
@@ -235,4 +248,7 @@ class CompareWorkflowConfigMetadataValue(BaseModel):
         """Returns the string representation of the actual instance"""
         return pprint.pformat(self.model_dump())
 
+from rapidata.api_client.models.asset_metadata import AssetMetadata
+# TODO: Rewrite to not use raise_errors
+CompareWorkflowConfigMetadataValue.model_rebuild(raise_errors=False)
 

rapidata/api_client/models/create_datapoint_from_files_model_metadata_inner.py

@@ -18,6 +18,7 @@ import pprint
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
 from typing import Any, List, Optional
 from rapidata.api_client.models.private_text_metadata_input import PrivateTextMetadataInput
+from rapidata.api_client.models.prompt_asset_metadata_input import PromptAssetMetadataInput
 from rapidata.api_client.models.prompt_metadata_input import PromptMetadataInput
 from rapidata.api_client.models.public_text_metadata_input import PublicTextMetadataInput
 from rapidata.api_client.models.transcription_metadata_input import TranscriptionMetadataInput
@@ -25,7 +26,7 @@ from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-CREATEDATAPOINTFROMFILESMODELMETADATAINNER_ONE_OF_SCHEMAS = ["PrivateTextMetadataInput", "PromptMetadataInput", "PublicTextMetadataInput", "TranscriptionMetadataInput"]
+CREATEDATAPOINTFROMFILESMODELMETADATAINNER_ONE_OF_SCHEMAS = ["PrivateTextMetadataInput", "PromptAssetMetadataInput", "PromptMetadataInput", "PublicTextMetadataInput", "TranscriptionMetadataInput"]
 
 class CreateDatapointFromFilesModelMetadataInner(BaseModel):
     """
@@ -33,14 +34,16 @@ class CreateDatapointFromFilesModelMetadataInner(BaseModel):
     """
     # data type: PrivateTextMetadataInput
     oneof_schema_1_validator: Optional[PrivateTextMetadataInput] = None
+    # data type: PromptAssetMetadataInput
+    oneof_schema_2_validator: Optional[PromptAssetMetadataInput] = None
     # data type: PromptMetadataInput
-    oneof_schema_2_validator: Optional[PromptMetadataInput] = None
+    oneof_schema_3_validator: Optional[PromptMetadataInput] = None
     # data type: PublicTextMetadataInput
-    oneof_schema_3_validator: Optional[PublicTextMetadataInput] = None
+    oneof_schema_4_validator: Optional[PublicTextMetadataInput] = None
     # data type: TranscriptionMetadataInput
-    oneof_schema_4_validator: Optional[TranscriptionMetadataInput] = None
-    actual_instance: Optional[Union[PrivateTextMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput]] = None
-    one_of_schemas: Set[str] = { "PrivateTextMetadataInput", "PromptMetadataInput", "PublicTextMetadataInput", "TranscriptionMetadataInput" }
+    oneof_schema_5_validator: Optional[TranscriptionMetadataInput] = None
+    actual_instance: Optional[Union[PrivateTextMetadataInput, PromptAssetMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput]] = None
+    one_of_schemas: Set[str] = { "PrivateTextMetadataInput", "PromptAssetMetadataInput", "PromptMetadataInput", "PublicTextMetadataInput", "TranscriptionMetadataInput" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -71,6 +74,11 @@ class CreateDatapointFromFilesModelMetadataInner(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `PrivateTextMetadataInput`")
         else:
             match += 1
+        # validate data type: PromptAssetMetadataInput
+        if not isinstance(v, PromptAssetMetadataInput):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `PromptAssetMetadataInput`")
+        else:
+            match += 1
         # validate data type: PromptMetadataInput
         if not isinstance(v, PromptMetadataInput):
             error_messages.append(f"Error! Input type `{type(v)}` is not `PromptMetadataInput`")
@@ -88,10 +96,10 @@ class CreateDatapointFromFilesModelMetadataInner(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptAssetMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptAssetMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -112,6 +120,12 @@ class CreateDatapointFromFilesModelMetadataInner(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
+        # deserialize data into PromptAssetMetadataInput
+        try:
+            instance.actual_instance = PromptAssetMetadataInput.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into PromptMetadataInput
         try:
             instance.actual_instance = PromptMetadataInput.from_json(json_str)
@@ -133,10 +147,10 @@ class CreateDatapointFromFilesModelMetadataInner(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptAssetMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into CreateDatapointFromFilesModelMetadataInner with oneOf schemas: PrivateTextMetadataInput, PromptAssetMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -150,7 +164,7 @@ class CreateDatapointFromFilesModelMetadataInner(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], PrivateTextMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], PrivateTextMetadataInput, PromptAssetMetadataInput, PromptMetadataInput, PublicTextMetadataInput, TranscriptionMetadataInput]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

rapidata/api_client/models/datapoint.py

@@ -20,7 +20,7 @@ import json
 from datetime import datetime
 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.asset_metadata_model_asset import AssetMetadataModelAsset
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -31,7 +31,7 @@ class Datapoint(BaseModel):
     id: StrictStr
     dataset_id: StrictStr = Field(alias="datasetId")
     sort_index: Optional[StrictInt] = Field(default=None, alias="sortIndex")
-    asset: DatapointAsset
+    asset: AssetMetadataModelAsset
     created_at: datetime = Field(alias="createdAt")
     __properties: ClassVar[List[str]] = ["id", "datasetId", "sortIndex", "asset", "createdAt"]
 
@@ -97,7 +97,7 @@ class Datapoint(BaseModel):
             "id": obj.get("id"),
             "datasetId": obj.get("datasetId"),
             "sortIndex": obj.get("sortIndex"),
-            "asset": DatapointAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
+            "asset": AssetMetadataModelAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None,
             "createdAt": obj.get("createdAt")
         })
         return _obj

rapidata/api_client/models/datapoint_model.py

@@ -19,7 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List
-from rapidata.api_client.models.datapoint_asset import DatapointAsset
+from rapidata.api_client.models.asset_metadata_model_asset import AssetMetadataModelAsset
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,7 +29,7 @@ class DatapointModel(BaseModel):
     """ # noqa: E501
     id: StrictStr
     dataset_id: StrictStr = Field(alias="datasetId")
-    asset: DatapointAsset
+    asset: AssetMetadataModelAsset
     __properties: ClassVar[List[str]] = ["id", "datasetId", "asset"]
 
     model_config = ConfigDict(
@@ -88,7 +88,7 @@ class DatapointModel(BaseModel):
         _obj = cls.model_validate({
             "id": obj.get("id"),
             "datasetId": obj.get("datasetId"),
-            "asset": DatapointAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None
+            "asset": AssetMetadataModelAsset.from_dict(obj["asset"]) if obj.get("asset") is not None else None
         })
         return _obj
 

rapidata/api_client/models/file_asset.py

@@ -19,7 +19,6 @@ 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_asset_metadata_inner import FileAssetMetadataInner
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -29,9 +28,8 @@ class FileAsset(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for FileAsset", alias="_t")
     file_name: StrictStr = Field(alias="fileName")
-    original_file_name: StrictStr = Field(alias="originalFileName")
-    metadata: Optional[List[FileAssetMetadataInner]] = None
-    __properties: ClassVar[List[str]] = ["_t", "fileName", "originalFileName", "metadata"]
+    metadata: Optional[Dict[str, CompareWorkflowConfigMetadataValue]] = None
+    __properties: ClassVar[List[str]] = ["_t", "fileName", "metadata"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -79,13 +77,13 @@ class FileAsset(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # override the default output from pydantic by calling `to_dict()` of each item in metadata (list)
-        _items = []
+        # override the default output from pydantic by calling `to_dict()` of each value in metadata (dict)
+        _field_dict = {}
         if self.metadata:
-            for _item_metadata in self.metadata:
-                if _item_metadata:
-                    _items.append(_item_metadata.to_dict())
-            _dict['metadata'] = _items
+            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
         return _dict
 
     @classmethod
@@ -100,9 +98,16 @@ class FileAsset(BaseModel):
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'FileAsset',
             "fileName": obj.get("fileName"),
-            "originalFileName": obj.get("originalFileName"),
-            "metadata": [FileAssetMetadataInner.from_dict(_item) for _item in obj["metadata"]] if obj.get("metadata") is not None else None
+            "metadata": dict(
+                (_k, CompareWorkflowConfigMetadataValue.from_dict(_v))
+                for _k, _v in obj["metadata"].items()
+            )
+            if obj.get("metadata") is not None
+            else None
         })
         return _obj
 
+from rapidata.api_client.models.compare_workflow_config_metadata_value import CompareWorkflowConfigMetadataValue
+# TODO: Rewrite to not use raise_errors
+FileAsset.model_rebuild(raise_errors=False)
 

rapidata/api_client/models/file_asset_model.py

@@ -19,7 +19,6 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List
-from rapidata.api_client.models.file_asset_model_metadata_value import FileAssetModelMetadataValue
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -110,4 +109,7 @@ class FileAssetModel(BaseModel):
         })
         return _obj
 
+from rapidata.api_client.models.file_asset_model_metadata_value import FileAssetModelMetadataValue
+# TODO: Rewrite to not use raise_errors
+FileAssetModel.model_rebuild(raise_errors=False)
 

rapidata/api_client/models/file_asset_model_metadata_value.py

@@ -32,36 +32,38 @@ from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-FILEASSETMODELMETADATAVALUE_ONE_OF_SCHEMAS = ["ClassificationMetadataModel", "CountMetadataModel", "DemographicMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "PromptMetadataModel", "SourceUrlMetadataModel", "TextMetadataModel", "TranscriptionMetadataModel", "TranslatedPromptMetadataModel"]
+FILEASSETMODELMETADATAVALUE_ONE_OF_SCHEMAS = ["AssetMetadataModel", "ClassificationMetadataModel", "CountMetadataModel", "DemographicMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "PromptMetadataModel", "SourceUrlMetadataModel", "TextMetadataModel", "TranscriptionMetadataModel", "TranslatedPromptMetadataModel"]
 
 class FileAssetModelMetadataValue(BaseModel):
     """
     FileAssetModelMetadataValue
     """
+    # data type: AssetMetadataModel
+    oneof_schema_1_validator: Optional[AssetMetadataModel] = None
     # data type: ClassificationMetadataModel
-    oneof_schema_1_validator: Optional[ClassificationMetadataModel] = None
+    oneof_schema_2_validator: Optional[ClassificationMetadataModel] = None
     # data type: CountMetadataModel
-    oneof_schema_2_validator: Optional[CountMetadataModel] = None
+    oneof_schema_3_validator: Optional[CountMetadataModel] = None
     # data type: DemographicMetadataModel
-    oneof_schema_3_validator: Optional[DemographicMetadataModel] = None
+    oneof_schema_4_validator: Optional[DemographicMetadataModel] = None
     # data type: ImageDimensionMetadataModel
-    oneof_schema_4_validator: Optional[ImageDimensionMetadataModel] = None
+    oneof_schema_5_validator: Optional[ImageDimensionMetadataModel] = None
     # data type: LocationMetadataModel
-    oneof_schema_5_validator: Optional[LocationMetadataModel] = None
+    oneof_schema_6_validator: Optional[LocationMetadataModel] = None
     # data type: OriginalFilenameMetadataModel
-    oneof_schema_6_validator: Optional[OriginalFilenameMetadataModel] = None
+    oneof_schema_7_validator: Optional[OriginalFilenameMetadataModel] = None
     # data type: PromptMetadataModel
-    oneof_schema_7_validator: Optional[PromptMetadataModel] = None
+    oneof_schema_8_validator: Optional[PromptMetadataModel] = None
     # data type: SourceUrlMetadataModel
-    oneof_schema_8_validator: Optional[SourceUrlMetadataModel] = None
+    oneof_schema_9_validator: Optional[SourceUrlMetadataModel] = None
     # data type: TextMetadataModel
-    oneof_schema_9_validator: Optional[TextMetadataModel] = None
+    oneof_schema_10_validator: Optional[TextMetadataModel] = None
     # data type: TranscriptionMetadataModel
-    oneof_schema_10_validator: Optional[TranscriptionMetadataModel] = None
+    oneof_schema_11_validator: Optional[TranscriptionMetadataModel] = None
     # data type: TranslatedPromptMetadataModel
-    oneof_schema_11_validator: Optional[TranslatedPromptMetadataModel] = None
-    actual_instance: Optional[Union[ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel]] = None
-    one_of_schemas: Set[str] = { "ClassificationMetadataModel", "CountMetadataModel", "DemographicMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "PromptMetadataModel", "SourceUrlMetadataModel", "TextMetadataModel", "TranscriptionMetadataModel", "TranslatedPromptMetadataModel" }
+    oneof_schema_12_validator: Optional[TranslatedPromptMetadataModel] = None
+    actual_instance: Optional[Union[AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel]] = None
+    one_of_schemas: Set[str] = { "AssetMetadataModel", "ClassificationMetadataModel", "CountMetadataModel", "DemographicMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "PromptMetadataModel", "SourceUrlMetadataModel", "TextMetadataModel", "TranscriptionMetadataModel", "TranslatedPromptMetadataModel" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -87,6 +89,11 @@ class FileAssetModelMetadataValue(BaseModel):
         instance = FileAssetModelMetadataValue.model_construct()
         error_messages = []
         match = 0
+        # validate data type: AssetMetadataModel
+        if not isinstance(v, AssetMetadataModel):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `AssetMetadataModel`")
+        else:
+            match += 1
         # validate data type: ClassificationMetadataModel
         if not isinstance(v, ClassificationMetadataModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `ClassificationMetadataModel`")
@@ -144,10 +151,10 @@ class FileAssetModelMetadataValue(BaseModel):
             match += 1
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when setting `actual_instance` in FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in FileAssetModelMetadataValue with oneOf schemas: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in FileAssetModelMetadataValue with oneOf schemas: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -162,6 +169,12 @@ class FileAssetModelMetadataValue(BaseModel):
         error_messages = []
         match = 0
 
+        # deserialize data into AssetMetadataModel
+        try:
+            instance.actual_instance = AssetMetadataModel.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
         # deserialize data into ClassificationMetadataModel
         try:
             instance.actual_instance = ClassificationMetadataModel.from_json(json_str)
@@ -231,10 +244,10 @@ class FileAssetModelMetadataValue(BaseModel):
 
         if match > 1:
             # more than 1 match
-            raise ValueError("Multiple matches found when deserializing the JSON string into FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into FileAssetModelMetadataValue with oneOf schemas: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into FileAssetModelMetadataValue with oneOf schemas: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -248,7 +261,7 @@ class FileAssetModelMetadataValue(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None
@@ -263,4 +276,7 @@ class FileAssetModelMetadataValue(BaseModel):
         """Returns the string representation of the actual instance"""
         return pprint.pformat(self.model_dump())
 
+from rapidata.api_client.models.asset_metadata_model import AssetMetadataModel
+# TODO: Rewrite to not use raise_errors
+FileAssetModelMetadataValue.model_rebuild(raise_errors=False)