rapidata 2.35.1__py3-none-any.whl → 3.0.0a8__py3-none-any.whl

rapidata/api_client/models/file_asset_model_metadata_value.py

@@ -17,63 +17,48 @@ 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.asset_metadata_model import AssetMetadataModel
 from rapidata.api_client.models.classification_metadata_model import ClassificationMetadataModel
 from rapidata.api_client.models.count_metadata_model import CountMetadataModel
-from rapidata.api_client.models.demographic_metadata_model import DemographicMetadataModel
 from rapidata.api_client.models.file_type_metadata_model import FileTypeMetadataModel
 from rapidata.api_client.models.image_dimension_metadata_model import ImageDimensionMetadataModel
 from rapidata.api_client.models.location_metadata_model import LocationMetadataModel
 from rapidata.api_client.models.original_filename_metadata_model import OriginalFilenameMetadataModel
-from rapidata.api_client.models.prompt_metadata_model import PromptMetadataModel
 from rapidata.api_client.models.source_url_metadata_model import SourceUrlMetadataModel
 from rapidata.api_client.models.streams_metadata_model import StreamsMetadataModel
 from rapidata.api_client.models.text_metadata_model import TextMetadataModel
-from rapidata.api_client.models.transcription_metadata_model import TranscriptionMetadataModel
-from rapidata.api_client.models.translated_prompt_metadata_model import TranslatedPromptMetadataModel
 from rapidata.api_client.models.video_duration_metadata_model import VideoDurationMetadataModel
 from pydantic import StrictStr, Field
 from typing import Union, List, Set, Optional, Dict
 from typing_extensions import Literal, Self
 
-FILEASSETMODELMETADATAVALUE_ONE_OF_SCHEMAS = ["AssetMetadataModel", "ClassificationMetadataModel", "CountMetadataModel", "DemographicMetadataModel", "FileTypeMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "PromptMetadataModel", "SourceUrlMetadataModel", "StreamsMetadataModel", "TextMetadataModel", "TranscriptionMetadataModel", "TranslatedPromptMetadataModel", "VideoDurationMetadataModel"]
+FILEASSETMODELMETADATAVALUE_ONE_OF_SCHEMAS = ["ClassificationMetadataModel", "CountMetadataModel", "FileTypeMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "SourceUrlMetadataModel", "StreamsMetadataModel", "TextMetadataModel", "VideoDurationMetadataModel"]
 
 class FileAssetModelMetadataValue(BaseModel):
     """
     FileAssetModelMetadataValue
     """
-    # data type: AssetMetadataModel
-    oneof_schema_1_validator: Optional[AssetMetadataModel] = None
     # data type: ClassificationMetadataModel
-    oneof_schema_2_validator: Optional[ClassificationMetadataModel] = None
+    oneof_schema_1_validator: Optional[ClassificationMetadataModel] = None
     # data type: CountMetadataModel
-    oneof_schema_3_validator: Optional[CountMetadataModel] = None
-    # data type: DemographicMetadataModel
-    oneof_schema_4_validator: Optional[DemographicMetadataModel] = None
+    oneof_schema_2_validator: Optional[CountMetadataModel] = None
     # data type: FileTypeMetadataModel
-    oneof_schema_5_validator: Optional[FileTypeMetadataModel] = None
+    oneof_schema_3_validator: Optional[FileTypeMetadataModel] = None
     # data type: ImageDimensionMetadataModel
-    oneof_schema_6_validator: Optional[ImageDimensionMetadataModel] = None
+    oneof_schema_4_validator: Optional[ImageDimensionMetadataModel] = None
     # data type: LocationMetadataModel
-    oneof_schema_7_validator: Optional[LocationMetadataModel] = None
+    oneof_schema_5_validator: Optional[LocationMetadataModel] = None
     # data type: OriginalFilenameMetadataModel
-    oneof_schema_8_validator: Optional[OriginalFilenameMetadataModel] = None
-    # data type: PromptMetadataModel
-    oneof_schema_9_validator: Optional[PromptMetadataModel] = None
+    oneof_schema_6_validator: Optional[OriginalFilenameMetadataModel] = None
     # data type: SourceUrlMetadataModel
-    oneof_schema_10_validator: Optional[SourceUrlMetadataModel] = None
+    oneof_schema_7_validator: Optional[SourceUrlMetadataModel] = None
     # data type: StreamsMetadataModel
-    oneof_schema_11_validator: Optional[StreamsMetadataModel] = None
+    oneof_schema_8_validator: Optional[StreamsMetadataModel] = None
     # data type: TextMetadataModel
-    oneof_schema_12_validator: Optional[TextMetadataModel] = None
-    # data type: TranscriptionMetadataModel
-    oneof_schema_13_validator: Optional[TranscriptionMetadataModel] = None
-    # data type: TranslatedPromptMetadataModel
-    oneof_schema_14_validator: Optional[TranslatedPromptMetadataModel] = None
+    oneof_schema_9_validator: Optional[TextMetadataModel] = None
     # data type: VideoDurationMetadataModel
-    oneof_schema_15_validator: Optional[VideoDurationMetadataModel] = None
-    actual_instance: Optional[Union[AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel, VideoDurationMetadataModel]] = None
-    one_of_schemas: Set[str] = { "AssetMetadataModel", "ClassificationMetadataModel", "CountMetadataModel", "DemographicMetadataModel", "FileTypeMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "PromptMetadataModel", "SourceUrlMetadataModel", "StreamsMetadataModel", "TextMetadataModel", "TranscriptionMetadataModel", "TranslatedPromptMetadataModel", "VideoDurationMetadataModel" }
+    oneof_schema_10_validator: Optional[VideoDurationMetadataModel] = None
+    actual_instance: Optional[Union[ClassificationMetadataModel, CountMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, VideoDurationMetadataModel]] = None
+    one_of_schemas: Set[str] = { "ClassificationMetadataModel", "CountMetadataModel", "FileTypeMetadataModel", "ImageDimensionMetadataModel", "LocationMetadataModel", "OriginalFilenameMetadataModel", "SourceUrlMetadataModel", "StreamsMetadataModel", "TextMetadataModel", "VideoDurationMetadataModel" }
 
     model_config = ConfigDict(
         validate_assignment=True,
@@ -99,11 +84,6 @@ 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`")
@@ -114,11 +94,6 @@ class FileAssetModelMetadataValue(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `CountMetadataModel`")
         else:
             match += 1
-        # validate data type: DemographicMetadataModel
-        if not isinstance(v, DemographicMetadataModel):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `DemographicMetadataModel`")
-        else:
-            match += 1
         # validate data type: FileTypeMetadataModel
         if not isinstance(v, FileTypeMetadataModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `FileTypeMetadataModel`")
@@ -139,11 +114,6 @@ class FileAssetModelMetadataValue(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `OriginalFilenameMetadataModel`")
         else:
             match += 1
-        # validate data type: PromptMetadataModel
-        if not isinstance(v, PromptMetadataModel):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `PromptMetadataModel`")
-        else:
-            match += 1
         # validate data type: SourceUrlMetadataModel
         if not isinstance(v, SourceUrlMetadataModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `SourceUrlMetadataModel`")
@@ -159,16 +129,6 @@ class FileAssetModelMetadataValue(BaseModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `TextMetadataModel`")
         else:
             match += 1
-        # validate data type: TranscriptionMetadataModel
-        if not isinstance(v, TranscriptionMetadataModel):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `TranscriptionMetadataModel`")
-        else:
-            match += 1
-        # validate data type: TranslatedPromptMetadataModel
-        if not isinstance(v, TranslatedPromptMetadataModel):
-            error_messages.append(f"Error! Input type `{type(v)}` is not `TranslatedPromptMetadataModel`")
-        else:
-            match += 1
         # validate data type: VideoDurationMetadataModel
         if not isinstance(v, VideoDurationMetadataModel):
             error_messages.append(f"Error! Input type `{type(v)}` is not `VideoDurationMetadataModel`")
@@ -176,10 +136,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: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when setting `actual_instance` in FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when setting `actual_instance` in FileAssetModelMetadataValue with oneOf schemas: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when setting `actual_instance` in FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
         else:
             return v
 
@@ -194,12 +154,6 @@ 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)
@@ -212,12 +166,6 @@ class FileAssetModelMetadataValue(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into DemographicMetadataModel
-        try:
-            instance.actual_instance = DemographicMetadataModel.from_json(json_str)
-            match += 1
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
         # deserialize data into FileTypeMetadataModel
         try:
             instance.actual_instance = FileTypeMetadataModel.from_json(json_str)
@@ -242,12 +190,6 @@ class FileAssetModelMetadataValue(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into PromptMetadataModel
-        try:
-            instance.actual_instance = PromptMetadataModel.from_json(json_str)
-            match += 1
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
         # deserialize data into SourceUrlMetadataModel
         try:
             instance.actual_instance = SourceUrlMetadataModel.from_json(json_str)
@@ -266,18 +208,6 @@ class FileAssetModelMetadataValue(BaseModel):
             match += 1
         except (ValidationError, ValueError) as e:
             error_messages.append(str(e))
-        # deserialize data into TranscriptionMetadataModel
-        try:
-            instance.actual_instance = TranscriptionMetadataModel.from_json(json_str)
-            match += 1
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
-        # deserialize data into TranslatedPromptMetadataModel
-        try:
-            instance.actual_instance = TranslatedPromptMetadataModel.from_json(json_str)
-            match += 1
-        except (ValidationError, ValueError) as e:
-            error_messages.append(str(e))
         # deserialize data into VideoDurationMetadataModel
         try:
             instance.actual_instance = VideoDurationMetadataModel.from_json(json_str)
@@ -287,10 +217,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: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("Multiple matches found when deserializing the JSON string into FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
         elif match == 0:
             # no match
-            raise ValueError("No match found when deserializing the JSON string into FileAssetModelMetadataValue with oneOf schemas: AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
+            raise ValueError("No match found when deserializing the JSON string into FileAssetModelMetadataValue with oneOf schemas: ClassificationMetadataModel, CountMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, VideoDurationMetadataModel. Details: " + ", ".join(error_messages))
         else:
             return instance
 
@@ -304,7 +234,7 @@ class FileAssetModelMetadataValue(BaseModel):
         else:
             return json.dumps(self.actual_instance)
 
-    def to_dict(self) -> Optional[Union[Dict[str, Any], AssetMetadataModel, ClassificationMetadataModel, CountMetadataModel, DemographicMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, PromptMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, TranscriptionMetadataModel, TranslatedPromptMetadataModel, VideoDurationMetadataModel]]:
+    def to_dict(self) -> Optional[Union[Dict[str, Any], ClassificationMetadataModel, CountMetadataModel, FileTypeMetadataModel, ImageDimensionMetadataModel, LocationMetadataModel, OriginalFilenameMetadataModel, SourceUrlMetadataModel, StreamsMetadataModel, TextMetadataModel, VideoDurationMetadataModel]]:
         """Returns the dict representation of the actual instance"""
         if self.actual_instance is None:
             return None

rapidata/api_client/models/file_stream_result.py

@@ -0,0 +1,122 @@
+# 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 datetime import datetime
+from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictBytes, StrictStr
+from typing import Any, ClassVar, Dict, List, Optional, Tuple, Union
+from rapidata.api_client.models.entity_tag_header_value import EntityTagHeaderValue
+from typing import Optional, Set
+from typing_extensions import Self
+
+class FileStreamResult(BaseModel):
+    """
+    FileStreamResult
+    """ # noqa: E501
+    file_stream: Optional[Union[StrictBytes, StrictStr, Tuple[StrictStr, StrictBytes]]] = Field(default=None, alias="fileStream")
+    content_type: Optional[StrictStr] = Field(default=None, alias="contentType")
+    file_download_name: Optional[StrictStr] = Field(default=None, alias="fileDownloadName")
+    last_modified: Optional[datetime] = Field(default=None, alias="lastModified")
+    entity_tag: Optional[EntityTagHeaderValue] = Field(default=None, alias="entityTag")
+    enable_range_processing: Optional[StrictBool] = Field(default=None, alias="enableRangeProcessing")
+    __properties: ClassVar[List[str]] = ["fileStream", "contentType", "fileDownloadName", "lastModified", "entityTag", "enableRangeProcessing"]
+
+    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 FileStreamResult from a JSON string"""
+        return cls.from_dict(json.loads(json_str))
+
+    def to_dict(self) -> Dict[str, Any]:
+        """Return the dictionary representation of the model using alias.
+
+        This has the following differences from calling pydantic's
+        `self.model_dump(by_alias=True)`:
+
+        * `None` is only added to the output dict for nullable fields that
+          were set at model initialization. Other fields with value `None`
+          are ignored.
+        """
+        excluded_fields: Set[str] = set([
+        ])
+
+        _dict = self.model_dump(
+            by_alias=True,
+            exclude=excluded_fields,
+            exclude_none=True,
+        )
+        # override the default output from pydantic by calling `to_dict()` of entity_tag
+        if self.entity_tag:
+            _dict['entityTag'] = self.entity_tag.to_dict()
+        # set to None if content_type (nullable) is None
+        # and model_fields_set contains the field
+        if self.content_type is None and "content_type" in self.model_fields_set:
+            _dict['contentType'] = None
+
+        # set to None if file_download_name (nullable) is None
+        # and model_fields_set contains the field
+        if self.file_download_name is None and "file_download_name" in self.model_fields_set:
+            _dict['fileDownloadName'] = None
+
+        # set to None if last_modified (nullable) is None
+        # and model_fields_set contains the field
+        if self.last_modified is None and "last_modified" in self.model_fields_set:
+            _dict['lastModified'] = None
+
+        # set to None if entity_tag (nullable) is None
+        # and model_fields_set contains the field
+        if self.entity_tag is None and "entity_tag" in self.model_fields_set:
+            _dict['entityTag'] = None
+
+        return _dict
+
+    @classmethod
+    def from_dict(cls, obj: Optional[Dict[str, Any]]) -> Optional[Self]:
+        """Create an instance of FileStreamResult from a dict"""
+        if obj is None:
+            return None
+
+        if not isinstance(obj, dict):
+            return cls.model_validate(obj)
+
+        _obj = cls.model_validate({
+            "fileStream": obj.get("fileStream"),
+            "contentType": obj.get("contentType"),
+            "fileDownloadName": obj.get("fileDownloadName"),
+            "lastModified": obj.get("lastModified"),
+            "entityTag": EntityTagHeaderValue.from_dict(obj["entityTag"]) if obj.get("entityTag") is not None else None,
+            "enableRangeProcessing": obj.get("enableRangeProcessing")
+        })
+        return _obj
+
+

rapidata/api_client/models/file_type.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)

rapidata/api_client/models/file_type_metadata.py

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

rapidata/api_client/models/file_type_metadata_model.py

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

rapidata/api_client/models/filter.py

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

rapidata/api_client/models/filter_operator.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)

rapidata/api_client/models/fork_benchmark_result.py

@@ -0,0 +1,87 @@
+# 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, StrictStr
+from typing import Any, ClassVar, Dict, List
+from typing import Optional, Set
+from typing_extensions import Self
+
+class ForkBenchmarkResult(BaseModel):
+    """
+    ForkBenchmarkResult
+    """ # noqa: E501
+    id: StrictStr
+    __properties: ClassVar[List[str]] = ["id"]
+
+    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 ForkBenchmarkResult 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 ForkBenchmarkResult 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")
+        })
+        return _obj
+
+

rapidata/api_client/models/form_file_wrapper.py

@@ -19,6 +19,7 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictBool, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.file_asset_metadata_value import FileAssetMetadataValue
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -31,7 +32,8 @@ class FormFileWrapper(BaseModel):
     content_length: Optional[StrictInt] = Field(default=None, alias="contentLength")
     content_type: Optional[StrictStr] = Field(default=None, alias="contentType")
     is_in_memory: Optional[StrictBool] = Field(default=None, alias="isInMemory")
-    __properties: ClassVar[List[str]] = ["_t", "name", "contentLength", "contentType", "isInMemory"]
+    metadata: Optional[Dict[str, FileAssetMetadataValue]] = None
+    __properties: ClassVar[List[str]] = ["_t", "name", "contentLength", "contentType", "isInMemory", "metadata"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -87,6 +89,13 @@ class FormFileWrapper(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
+        # 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
         # set to None if content_length (nullable) is None
         # and model_fields_set contains the field
         if self.content_length is None and "content_length" in self.model_fields_set:
@@ -113,7 +122,13 @@ class FormFileWrapper(BaseModel):
             "name": obj.get("name"),
             "contentLength": obj.get("contentLength"),
             "contentType": obj.get("contentType"),
-            "isInMemory": obj.get("isInMemory")
+            "isInMemory": obj.get("isInMemory"),
+            "metadata": dict(
+                (_k, FileAssetMetadataValue.from_dict(_v))
+                for _k, _v in obj["metadata"].items()
+            )
+            if obj.get("metadata") is not None
+            else None
         })
         return _obj