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

rapidata/api_client/models/simple_workflow_config.py

@@ -18,8 +18,9 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
 from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
+from rapidata.api_client.models.feature_flag import FeatureFlag
 from rapidata.api_client.models.validation_set_zip_post_request_blueprint import ValidationSetZipPostRequestBlueprint
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,7 +32,8 @@ class SimpleWorkflowConfig(BaseModel):
     t: StrictStr = Field(description="Discriminator value for SimpleWorkflowConfig", alias="_t")
     referee: CompareWorkflowModel1Referee
     blueprint: ValidationSetZipPostRequestBlueprint
-    __properties: ClassVar[List[str]] = ["_t", "referee", "blueprint"]
+    feature_flags: Optional[List[FeatureFlag]] = Field(default=None, alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "referee", "blueprint", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -85,6 +87,13 @@ class SimpleWorkflowConfig(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of blueprint
         if self.blueprint:
             _dict['blueprint'] = self.blueprint.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in feature_flags (list)
+        _items = []
+        if self.feature_flags:
+            for _item_feature_flags in self.feature_flags:
+                if _item_feature_flags:
+                    _items.append(_item_feature_flags.to_dict())
+            _dict['featureFlags'] = _items
         return _dict
 
     @classmethod
@@ -99,7 +108,8 @@ class SimpleWorkflowConfig(BaseModel):
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'SimpleWorkflowConfig',
             "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
-            "blueprint": ValidationSetZipPostRequestBlueprint.from_dict(obj["blueprint"]) if obj.get("blueprint") is not None else None
+            "blueprint": ValidationSetZipPostRequestBlueprint.from_dict(obj["blueprint"]) if obj.get("blueprint") is not None else None,
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj
 

rapidata/api_client/models/simple_workflow_config_model.py

@@ -20,6 +20,7 @@ import json
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List
 from rapidata.api_client.models.compare_workflow_model1_referee import CompareWorkflowModel1Referee
+from rapidata.api_client.models.feature_flag import FeatureFlag
 from rapidata.api_client.models.simple_workflow_config_model_blueprint import SimpleWorkflowConfigModelBlueprint
 from typing import Optional, Set
 from typing_extensions import Self
@@ -31,8 +32,8 @@ class SimpleWorkflowConfigModel(BaseModel):
     t: StrictStr = Field(description="Discriminator value for SimpleWorkflowConfig", alias="_t")
     referee: CompareWorkflowModel1Referee
     blueprint: SimpleWorkflowConfigModelBlueprint
-    target_country_codes: List[StrictStr] = Field(description="A list of country codes that this workflow is targeting.", alias="targetCountryCodes")
-    __properties: ClassVar[List[str]] = ["_t", "referee", "blueprint", "targetCountryCodes"]
+    feature_flags: List[FeatureFlag] = Field(description="The list of feature flags that will be applied to the rapids created by this workflow.", alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "referee", "blueprint", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -86,6 +87,13 @@ class SimpleWorkflowConfigModel(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of blueprint
         if self.blueprint:
             _dict['blueprint'] = self.blueprint.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in feature_flags (list)
+        _items = []
+        if self.feature_flags:
+            for _item_feature_flags in self.feature_flags:
+                if _item_feature_flags:
+                    _items.append(_item_feature_flags.to_dict())
+            _dict['featureFlags'] = _items
         return _dict
 
     @classmethod
@@ -101,7 +109,7 @@ class SimpleWorkflowConfigModel(BaseModel):
             "_t": obj.get("_t") if obj.get("_t") is not None else 'SimpleWorkflowConfig',
             "referee": CompareWorkflowModel1Referee.from_dict(obj["referee"]) if obj.get("referee") is not None else None,
             "blueprint": SimpleWorkflowConfigModelBlueprint.from_dict(obj["blueprint"]) if obj.get("blueprint") is not None else None,
-            "targetCountryCodes": obj.get("targetCountryCodes")
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj
 

rapidata/api_client/models/simple_workflow_model.py

@@ -18,7 +18,8 @@ import re  # noqa: F401
 import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List
+from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.feature_flag import FeatureFlag
 from rapidata.api_client.models.simple_workflow_model_blueprint import SimpleWorkflowModelBlueprint
 from typing import Optional, Set
 from typing_extensions import Self
@@ -29,7 +30,8 @@ class SimpleWorkflowModel(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for SimpleWorkflow", alias="_t")
     blueprint: SimpleWorkflowModelBlueprint
-    __properties: ClassVar[List[str]] = ["_t", "blueprint"]
+    feature_flags: Optional[List[FeatureFlag]] = Field(default=None, description="The list of feature flags that will be applied to the rapids created by this workflow.", alias="featureFlags")
+    __properties: ClassVar[List[str]] = ["_t", "blueprint", "featureFlags"]
 
     @field_validator('t')
     def t_validate_enum(cls, value):
@@ -80,6 +82,13 @@ class SimpleWorkflowModel(BaseModel):
         # override the default output from pydantic by calling `to_dict()` of blueprint
         if self.blueprint:
             _dict['blueprint'] = self.blueprint.to_dict()
+        # override the default output from pydantic by calling `to_dict()` of each item in feature_flags (list)
+        _items = []
+        if self.feature_flags:
+            for _item_feature_flags in self.feature_flags:
+                if _item_feature_flags:
+                    _items.append(_item_feature_flags.to_dict())
+            _dict['featureFlags'] = _items
         return _dict
 
     @classmethod
@@ -93,7 +102,8 @@ class SimpleWorkflowModel(BaseModel):
 
         _obj = cls.model_validate({
             "_t": obj.get("_t") if obj.get("_t") is not None else 'SimpleWorkflow',
-            "blueprint": SimpleWorkflowModelBlueprint.from_dict(obj["blueprint"]) if obj.get("blueprint") is not None else None
+            "blueprint": SimpleWorkflowModelBlueprint.from_dict(obj["blueprint"]) if obj.get("blueprint") is not None else None,
+            "featureFlags": [FeatureFlag.from_dict(_item) for _item in obj["featureFlags"]] if obj.get("featureFlags") is not None else None
         })
         return _obj
 

rapidata/api_client/models/sort_criterion.py

@@ -17,8 +17,9 @@ import pprint
 import re  # noqa: F401
 import json
 
-from pydantic import BaseModel, ConfigDict, Field, StrictStr, field_validator
+from pydantic import BaseModel, ConfigDict, Field, StrictStr
 from typing import Any, ClassVar, Dict, List
+from rapidata.api_client.models.sort_direction import SortDirection
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -27,16 +28,9 @@ class SortCriterion(BaseModel):
     SortCriterion
     """ # noqa: E501
     property_name: StrictStr = Field(alias="propertyName")
-    direction: StrictStr
+    direction: SortDirection
     __properties: ClassVar[List[str]] = ["propertyName", "direction"]
 
-    @field_validator('direction')
-    def direction_validate_enum(cls, value):
-        """Validates the enum"""
-        if value not in set(['Asc', 'Desc']):
-            raise ValueError("must be one of enum values ('Asc', 'Desc')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/source_url_metadata.py

@@ -28,7 +28,7 @@ class SourceUrlMetadata(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for SourceUrlMetadata", alias="_t")
     url: StrictStr
-    visibilities: Optional[StrictStr] = None
+    visibilities: Optional[List[StrictStr]] = None
     __properties: ClassVar[List[str]] = ["_t", "url", "visibilities"]
 
     @field_validator('t')
@@ -38,6 +38,17 @@ class SourceUrlMetadata(BaseModel):
             raise ValueError("must be one of enum values ('SourceUrlMetadata')")
         return value
 
+    @field_validator('visibilities')
+    def visibilities_validate_enum(cls, value):
+        """Validates the enum"""
+        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(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/standing_by_leaderboard.py

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

rapidata/api_client/models/streams_metadata.py

@@ -30,7 +30,7 @@ class StreamsMetadata(BaseModel):
     has_audio: Optional[StrictBool] = Field(default=None, alias="hasAudio")
     has_video: Optional[StrictBool] = Field(default=None, alias="hasVideo")
     has_subtitles: Optional[StrictBool] = Field(default=None, alias="hasSubtitles")
-    visibilities: Optional[StrictStr] = None
+    visibilities: Optional[List[StrictStr]] = None
     __properties: ClassVar[List[str]] = ["_t", "hasAudio", "hasVideo", "hasSubtitles", "visibilities"]
 
     @field_validator('t')
@@ -40,6 +40,17 @@ class StreamsMetadata(BaseModel):
             raise ValueError("must be one of enum values ('StreamsMetadata')")
         return value
 
+    @field_validator('visibilities')
+    def visibilities_validate_enum(cls, value):
+        """Validates the enum"""
+        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(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/text_metadata.py

@@ -28,7 +28,7 @@ class TextMetadata(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for TextMetadata", alias="_t")
     text: StrictStr
-    visibilities: Optional[StrictStr] = None
+    visibilities: Optional[List[StrictStr]] = None
     __properties: ClassVar[List[str]] = ["_t", "text", "visibilities"]
 
     @field_validator('t')
@@ -38,6 +38,17 @@ class TextMetadata(BaseModel):
             raise ValueError("must be one of enum values ('TextMetadata')")
         return value
 
+    @field_validator('visibilities')
+    def visibilities_validate_enum(cls, value):
+        """Validates the enum"""
+        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(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/transcription_metadata.py

@@ -28,7 +28,7 @@ class TranscriptionMetadata(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for TranscriptionMetadata", alias="_t")
     transcription: StrictStr
-    visibilities: StrictStr
+    visibilities: List[StrictStr]
     __properties: ClassVar[List[str]] = ["_t", "transcription", "visibilities"]
 
     @field_validator('t')
@@ -38,6 +38,14 @@ class TranscriptionMetadata(BaseModel):
             raise ValueError("must be one of enum values ('TranscriptionMetadata')")
         return value
 
+    @field_validator('visibilities')
+    def visibilities_validate_enum(cls, value):
+        """Validates the enum"""
+        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(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/update_should_alert_model.py

@@ -26,7 +26,7 @@ class UpdateShouldAlertModel(BaseModel):
     """
     The model for updating the shouldAlert field of all rapids within a validation set.
     """ # noqa: E501
-    should_alert: StrictBool = Field(description="If the users should be alerted if answering wrong.", alias="shouldAlert")
+    should_alert: StrictBool = Field(description="A flag indicating whether the users should be alerted if they answer incorrectly.", alias="shouldAlert")
     __properties: ClassVar[List[str]] = ["shouldAlert"]
 
     model_config = ConfigDict(

rapidata/api_client/models/validation_set_model.py

@@ -29,9 +29,9 @@ class ValidationSetModel(BaseModel):
     """ # noqa: E501
     id: StrictStr
     name: StrictStr
-    asset_type: Optional[StrictStr] = Field(default=None, alias="assetType")
-    modality: Optional[StrictStr] = None
-    prompt_type: Optional[StrictStr] = Field(default=None, alias="promptType")
+    asset_type: Optional[List[StrictStr]] = Field(default=None, alias="assetType")
+    modality: Optional[List[StrictStr]] = None
+    prompt_type: Optional[List[StrictStr]] = Field(default=None, alias="promptType")
     owner_id: StrictStr = Field(alias="ownerId")
     owner_mail: StrictStr = Field(alias="ownerMail")
     created_at: datetime = Field(alias="createdAt")
@@ -43,8 +43,9 @@ class ValidationSetModel(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['None', 'Image', 'Video', 'Audio', 'Text']):
-            raise ValueError("must be one of enum values ('None', 'Image', 'Video', 'Audio', 'Text')")
+        for i in value:
+            if i not in set(['None', 'Image', 'Video', 'Audio', 'Text']):
+                raise ValueError("each list item must be one of ('None', 'Image', 'Video', 'Audio', 'Text')")
         return value
 
     @field_validator('modality')
@@ -53,8 +54,9 @@ class ValidationSetModel(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['None', 'BoundingBox', 'Classify', 'Compare', 'FreeText', 'Line', 'Locate', 'NamedEntity', 'Polygon', 'Scrub', 'Transcription']):
-            raise ValueError("must be one of enum values ('None', 'BoundingBox', 'Classify', 'Compare', 'FreeText', 'Line', 'Locate', 'NamedEntity', 'Polygon', 'Scrub', 'Transcription')")
+        for i in value:
+            if i not in set(['None', 'BoundingBox', 'Classify', 'Compare', 'FreeText', 'Line', 'Locate', 'NamedEntity', 'Polygon', 'Scrub', 'Transcription']):
+                raise ValueError("each list item must be one of ('None', 'BoundingBox', 'Classify', 'Compare', 'FreeText', 'Line', 'Locate', 'NamedEntity', 'Polygon', 'Scrub', 'Transcription')")
         return value
 
     @field_validator('prompt_type')
@@ -63,8 +65,9 @@ class ValidationSetModel(BaseModel):
         if value is None:
             return value
 
-        if value not in set(['None', 'Text', 'Asset']):
-            raise ValueError("must be one of enum values ('None', 'Text', 'Asset')")
+        for i in value:
+            if i not in set(['None', 'Text', 'Asset']):
+                raise ValueError("each list item must be one of ('None', 'Text', 'Asset')")
         return value
 
     model_config = ConfigDict(
@@ -106,21 +109,6 @@ class ValidationSetModel(BaseModel):
             exclude=excluded_fields,
             exclude_none=True,
         )
-        # set to None if asset_type (nullable) is None
-        # and model_fields_set contains the field
-        if self.asset_type is None and "asset_type" in self.model_fields_set:
-            _dict['assetType'] = None
-
-        # set to None if modality (nullable) is None
-        # and model_fields_set contains the field
-        if self.modality is None and "modality" in self.model_fields_set:
-            _dict['modality'] = None
-
-        # set to None if prompt_type (nullable) is None
-        # and model_fields_set contains the field
-        if self.prompt_type is None and "prompt_type" in self.model_fields_set:
-            _dict['promptType'] = None
-
         return _dict
 
     @classmethod

rapidata/api_client/models/video_duration_metadata.py

@@ -28,7 +28,7 @@ class VideoDurationMetadata(BaseModel):
     """ # noqa: E501
     t: StrictStr = Field(description="Discriminator value for VideoDurationMetadata", alias="_t")
     duration: StrictStr
-    visibilities: Optional[StrictStr] = None
+    visibilities: Optional[List[StrictStr]] = None
     __properties: ClassVar[List[str]] = ["_t", "duration", "visibilities"]
 
     @field_validator('t')
@@ -38,6 +38,17 @@ class VideoDurationMetadata(BaseModel):
             raise ValueError("must be one of enum values ('VideoDurationMetadata')")
         return value
 
+    @field_validator('visibilities')
+    def visibilities_validate_enum(cls, value):
+        """Validates the enum"""
+        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(
         populate_by_name=True,
         validate_assignment=True,

rapidata/api_client/models/workflow_aggregation_step_model.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.aggregator_type import AggregatorType
 from typing import Optional, Set
 from typing_extensions import Self
 
@@ -30,7 +31,7 @@ class WorkflowAggregationStepModel(BaseModel):
     campaign_artifact_id: StrictStr = Field(description="The campaign artifact id", alias="campaignArtifactId")
     workflow_artifact_id: StrictStr = Field(description="The workflow artifact id", alias="workflowArtifactId")
     file_artifact_id: StrictStr = Field(description="The file artifact id", alias="fileArtifactId")
-    aggregator_type: Optional[StrictStr] = Field(default='NonCommittal', description="The type of the aggregator", alias="aggregatorType")
+    aggregator_type: Optional[AggregatorType] = Field(default=None, alias="aggregatorType")
     __properties: ClassVar[List[str]] = ["_t", "campaignArtifactId", "workflowArtifactId", "fileArtifactId", "aggregatorType"]
 
     @field_validator('t')
@@ -40,16 +41,6 @@ class WorkflowAggregationStepModel(BaseModel):
             raise ValueError("must be one of enum values ('WorkflowAggregationStepModel')")
         return value
 
-    @field_validator('aggregator_type')
-    def aggregator_type_validate_enum(cls, value):
-        """Validates the enum"""
-        if value is None:
-            return value
-
-        if value not in set(['NonCommittal', 'MajorityVote', 'SimpleMatchup', 'LocateCluster', 'Classification', 'Locate', 'BoundingBox', 'Line', 'Transcription', 'SinglePointLocate', 'FreeText', 'Scrub']):
-            raise ValueError("must be one of enum values ('NonCommittal', 'MajorityVote', 'SimpleMatchup', 'LocateCluster', 'Classification', 'Locate', 'BoundingBox', 'Line', 'Transcription', 'SinglePointLocate', 'FreeText', 'Scrub')")
-        return value
-
     model_config = ConfigDict(
         populate_by_name=True,
         validate_assignment=True,
@@ -105,7 +96,7 @@ class WorkflowAggregationStepModel(BaseModel):
             "campaignArtifactId": obj.get("campaignArtifactId"),
             "workflowArtifactId": obj.get("workflowArtifactId"),
             "fileArtifactId": obj.get("fileArtifactId"),
-            "aggregatorType": obj.get("aggregatorType") if obj.get("aggregatorType") is not None else 'NonCommittal'
+            "aggregatorType": obj.get("aggregatorType")
         })
         return _obj
 

rapidata/api_client_README.md

@@ -234,7 +234,6 @@ Class | Method | HTTP request | Description
  - [AssetMetadata](rapidata/api_client/docs/AssetMetadata.md)
  - [AssetMetadataAsset](rapidata/api_client/docs/AssetMetadataAsset.md)
  - [AssetMetadataModel](rapidata/api_client/docs/AssetMetadataModel.md)
- - [AssetType](rapidata/api_client/docs/AssetType.md)
  - [AttachCategoryRapidBlueprint](rapidata/api_client/docs/AttachCategoryRapidBlueprint.md)
  - [AttachCategoryResult](rapidata/api_client/docs/AttachCategoryResult.md)
  - [AttachCategoryTruth](rapidata/api_client/docs/AttachCategoryTruth.md)
@@ -385,6 +384,7 @@ Class | Method | HTTP request | Description
  - [GetPublicOrdersResult](rapidata/api_client/docs/GetPublicOrdersResult.md)
  - [GetPublicResponsesResult](rapidata/api_client/docs/GetPublicResponsesResult.md)
  - [GetRapidResponsesResult](rapidata/api_client/docs/GetRapidResponsesResult.md)
+ - [GetRecommendedValidationSetResult](rapidata/api_client/docs/GetRecommendedValidationSetResult.md)
  - [GetResponsesResult](rapidata/api_client/docs/GetResponsesResult.md)
  - [GetSampleByIdResult](rapidata/api_client/docs/GetSampleByIdResult.md)
  - [GetSimpleWorkflowResultsModel](rapidata/api_client/docs/GetSimpleWorkflowResultsModel.md)
@@ -430,7 +430,6 @@ Class | Method | HTTP request | Description
  - [LocationMetadataExistsFilterConfig](rapidata/api_client/docs/LocationMetadataExistsFilterConfig.md)
  - [LocationMetadataModel](rapidata/api_client/docs/LocationMetadataModel.md)
  - [LogicOperator](rapidata/api_client/docs/LogicOperator.md)
- - [MetadataVisibilities](rapidata/api_client/docs/MetadataVisibilities.md)
  - [MultiAsset](rapidata/api_client/docs/MultiAsset.md)
  - [MultiAssetInput](rapidata/api_client/docs/MultiAssetInput.md)
  - [MultiAssetInput1](rapidata/api_client/docs/MultiAssetInput1.md)
@@ -487,6 +486,7 @@ Class | Method | HTTP request | Description
  - [PrivateTextMetadataInput](rapidata/api_client/docs/PrivateTextMetadataInput.md)
  - [ProbabilisticAttachCategoryRefereeConfig](rapidata/api_client/docs/ProbabilisticAttachCategoryRefereeConfig.md)
  - [ProbabilisticAttachCategoryRefereeInfo](rapidata/api_client/docs/ProbabilisticAttachCategoryRefereeInfo.md)
+ - [ProblemDetails](rapidata/api_client/docs/ProblemDetails.md)
  - [PromptAssetMetadataInput](rapidata/api_client/docs/PromptAssetMetadataInput.md)
  - [PromptAssetMetadataInputAsset](rapidata/api_client/docs/PromptAssetMetadataInputAsset.md)
  - [PromptByBenchmarkResult](rapidata/api_client/docs/PromptByBenchmarkResult.md)
@@ -494,7 +494,6 @@ Class | Method | HTTP request | Description
  - [PromptMetadata](rapidata/api_client/docs/PromptMetadata.md)
  - [PromptMetadataInput](rapidata/api_client/docs/PromptMetadataInput.md)
  - [PromptMetadataModel](rapidata/api_client/docs/PromptMetadataModel.md)
- - [PromptType](rapidata/api_client/docs/PromptType.md)
  - [ProxyFileWrapper](rapidata/api_client/docs/ProxyFileWrapper.md)
  - [PublicOrderModel](rapidata/api_client/docs/PublicOrderModel.md)
  - [PublicRapidResponse](rapidata/api_client/docs/PublicRapidResponse.md)
@@ -502,7 +501,6 @@ Class | Method | HTTP request | Description
  - [QueryModel](rapidata/api_client/docs/QueryModel.md)
  - [QueryValidationModel](rapidata/api_client/docs/QueryValidationModel.md)
  - [RapidIssue](rapidata/api_client/docs/RapidIssue.md)
- - [RapidModality](rapidata/api_client/docs/RapidModality.md)
  - [RapidModel](rapidata/api_client/docs/RapidModel.md)
  - [RapidModelPagedResult](rapidata/api_client/docs/RapidModelPagedResult.md)
  - [RapidModelReferee](rapidata/api_client/docs/RapidModelReferee.md)

rapidata/rapidata_client/__init__.py

@@ -6,7 +6,7 @@ from .selection import (
     ConditionalValidationSelection,
     CappedSelection,
     ShufflingSelection,
-    RetrievalMode,
+    RapidataRetrievalMode,
     EffortSelection,
 )
 from .datapoints import Datapoint

rapidata/rapidata_client/benchmark/participant/_participant.py

@@ -7,7 +7,7 @@ from rapidata.rapidata_client.logging import logger
 from rapidata.rapidata_client.logging.output_manager import RapidataOutputManager
 from rapidata.api_client.models.create_sample_model import CreateSampleModel
 from rapidata.service.openapi_service import OpenAPIService
-from rapidata.rapidata_client.config.config import rapidata_config
+from rapidata.rapidata_client.config.rapidata_config import rapidata_config
 from rapidata.rapidata_client.api.rapidata_exception import (
     suppress_rapidata_error_logging,
 )
@@ -42,7 +42,7 @@ class BenchmarkParticipant:
             urls = [asset.path]
 
         last_exception = None
-        for attempt in range(rapidata_config.upload_max_retries):
+        for attempt in range(rapidata_config.uploadMaxRetries):
             try:
                 with suppress_rapidata_error_logging():
                     self.__openapi_service.participant_api.participant_participant_id_sample_post(
@@ -56,7 +56,7 @@ class BenchmarkParticipant:
 
             except Exception as e:
                 last_exception = e
-                if attempt < rapidata_config.upload_max_retries - 1:
+                if attempt < rapidata_config.uploadMaxRetries - 1:
                     # Exponential backoff: wait 1s, then 2s, then 4s
                     retry_delay = 2**attempt
                     time.sleep(retry_delay)
@@ -64,7 +64,7 @@ class BenchmarkParticipant:
                     logger.debug(
                         "Retrying %s of %s...",
                         attempt + 1,
-                        rapidata_config.upload_max_retries,
+                        rapidata_config.uploadMaxRetries,
                     )
 
         logger.error(f"Upload failed for {identifier}. Error: {str(last_exception)}")
@@ -90,7 +90,7 @@ class BenchmarkParticipant:
         total_uploads = len(assets)
 
         with ThreadPoolExecutor(
-            max_workers=rapidata_config.max_upload_workers
+            max_workers=rapidata_config.maxUploadWorkers
         ) as executor:
             futures = [
                 executor.submit(

rapidata/rapidata_client/benchmark/rapidata_benchmark.py

@@ -18,6 +18,7 @@ from rapidata.api_client.models.source_url_metadata_model import SourceUrlMetada
 from rapidata.api_client.models.and_user_filter_model_filters_inner import (
     AndUserFilterModelFiltersInner,
 )
+from rapidata.api_client.models.filter_operator import FilterOperator
 
 from rapidata.rapidata_client.benchmark.participant._participant import (
     BenchmarkParticipant,
@@ -150,7 +151,7 @@ class RapidataBenchmark:
                                 filters=[
                                     Filter(
                                         field="BenchmarkId",
-                                        operator="Eq",
+                                        operator=FilterOperator.EQ,
                                         value=self.id,
                                     )
                                 ]

rapidata/rapidata_client/benchmark/rapidata_benchmark_manager.py

@@ -7,6 +7,8 @@ from rapidata.api_client.models.page_info import PageInfo
 from rapidata.api_client.models.root_filter import RootFilter
 from rapidata.api_client.models.filter import Filter
 from rapidata.api_client.models.sort_criterion import SortCriterion
+from rapidata.api_client.models.sort_direction import SortDirection
+from rapidata.api_client.models.filter_operator import FilterOperator
 
 
 class RapidataBenchmarkManager:
@@ -139,10 +141,16 @@ class RapidataBenchmarkManager:
             QueryModel(
                 page=PageInfo(index=1, size=amount),
                 filter=RootFilter(
-                    filters=[Filter(field="Name", operator="Contains", value=name)]
+                    filters=[
+                        Filter(
+                            field="Name", operator=FilterOperator.CONTAINS, value=name
+                        )
+                    ]
                 ),
                 sortCriteria=[
-                    SortCriterion(direction="Desc", propertyName="CreatedAt")
+                    SortCriterion(
+                        direction=SortDirection.DESC, propertyName="CreatedAt"
+                    )
                 ],
             )
         )

rapidata/rapidata_client/config/__init__.py

@@ -1 +1 @@
-from .config import rapidata_config
+from .rapidata_config import rapidata_config

rapidata/rapidata_client/config/rapidata_config.py

@@ -0,0 +1,31 @@
+from pydantic import BaseModel, Field
+
+
+class RapidataConfig(BaseModel):
+    """
+    Holds the configuration for the Rapidata client.
+
+    To adjust the configurations used, you can modify the `rapidata_config` object.
+
+    Args:
+        maxUploadWorkers (int): The maximum number of worker threads for processing media paths. Defaults to 10.
+        uploadMaxRetries (int): The maximum number of retries for failed uploads. Defaults to 3.
+        enableBetaFeatures (bool): Whether to enable beta features. Defaults to False.
+        minOrderDatapointsForValidation (int): The minimum number of datapoints required so that an automatic validationset gets created if no recommended was found. Defaults to 50.
+        autoValidationSetSize (int): The maximum size of the auto-generated validation set. Defaults to 20.
+
+    Example:
+        ```python
+        from rapidata import rapidata_config
+        rapidata_config.maxUploadWorkers = 20
+        ```
+    """
+
+    maxUploadWorkers: int = Field(default=10)
+    uploadMaxRetries: int = Field(default=3)
+    enableBetaFeatures: bool = False
+    minOrderDatapointsForValidation: int = Field(default=50)
+    autoValidationSetSize: int = Field(default=20)
+
+
+rapidata_config = RapidataConfig()