rapidata 0.5.1__py3-none-any.whl → 1.1.0__py3-none-any.whl

rapidata/api_client/models/capped_selection_selections_inner.py

@@ -0,0 +1,198 @@
+# coding: utf-8
+
+"""
+    Rapidata.Dataset
+
+    No description provided (generated by Openapi Generator https://github.com/openapitools/openapi-generator)
+
+    The version of the OpenAPI document: v1
+    Generated by OpenAPI Generator (https://openapi-generator.tech)
+
+    Do not edit the class manually.
+"""  # noqa: E501
+
+
+from __future__ import annotations
+import json
+import pprint
+from pydantic import BaseModel, ConfigDict, Field, StrictStr, ValidationError, field_validator
+from typing import Any, List, Optional
+from rapidata.api_client.models.conditional_validation_selection import ConditionalValidationSelection
+from rapidata.api_client.models.demographic_selection import DemographicSelection
+from rapidata.api_client.models.labeling_selection import LabelingSelection
+from rapidata.api_client.models.static_selection import StaticSelection
+from rapidata.api_client.models.validation_selection import ValidationSelection
+from pydantic import StrictStr, Field
+from typing import Union, List, Set, Optional, Dict
+from typing_extensions import Literal, Self
+
+CAPPEDSELECTIONSELECTIONSINNER_ONE_OF_SCHEMAS = ["CappedSelection", "ConditionalValidationSelection", "DemographicSelection", "LabelingSelection", "StaticSelection", "ValidationSelection"]
+
+class CappedSelectionSelectionsInner(BaseModel):
+    """
+    CappedSelectionSelectionsInner
+    """
+    # data type: CappedSelection
+    oneof_schema_1_validator: Optional[CappedSelection] = None
+    # data type: ConditionalValidationSelection
+    oneof_schema_2_validator: Optional[ConditionalValidationSelection] = None
+    # data type: DemographicSelection
+    oneof_schema_3_validator: Optional[DemographicSelection] = None
+    # data type: LabelingSelection
+    oneof_schema_4_validator: Optional[LabelingSelection] = None
+    # data type: StaticSelection
+    oneof_schema_5_validator: Optional[StaticSelection] = None
+    # data type: ValidationSelection
+    oneof_schema_6_validator: Optional[ValidationSelection] = None
+    actual_instance: Optional[Union[CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection]] = None
+    one_of_schemas: Set[str] = { "CappedSelection", "ConditionalValidationSelection", "DemographicSelection", "LabelingSelection", "StaticSelection", "ValidationSelection" }
+
+    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 = CappedSelectionSelectionsInner.model_construct()
+        error_messages = []
+        match = 0
+        # validate data type: CappedSelection
+        if not isinstance(v, CappedSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `CappedSelection`")
+        else:
+            match += 1
+        # validate data type: ConditionalValidationSelection
+        if not isinstance(v, ConditionalValidationSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ConditionalValidationSelection`")
+        else:
+            match += 1
+        # validate data type: DemographicSelection
+        if not isinstance(v, DemographicSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `DemographicSelection`")
+        else:
+            match += 1
+        # validate data type: LabelingSelection
+        if not isinstance(v, LabelingSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `LabelingSelection`")
+        else:
+            match += 1
+        # validate data type: StaticSelection
+        if not isinstance(v, StaticSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `StaticSelection`")
+        else:
+            match += 1
+        # validate data type: ValidationSelection
+        if not isinstance(v, ValidationSelection):
+            error_messages.append(f"Error! Input type `{type(v)}` is not `ValidationSelection`")
+        else:
+            match += 1
+        if match > 1:
+            # more than 1 match
+            raise ValueError("Multiple matches found when setting `actual_instance` in CappedSelectionSelectionsInner with oneOf schemas: CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when setting `actual_instance` in CappedSelectionSelectionsInner with oneOf schemas: CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. 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 CappedSelection
+        try:
+            instance.actual_instance = CappedSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into ConditionalValidationSelection
+        try:
+            instance.actual_instance = ConditionalValidationSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into DemographicSelection
+        try:
+            instance.actual_instance = DemographicSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into LabelingSelection
+        try:
+            instance.actual_instance = LabelingSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into StaticSelection
+        try:
+            instance.actual_instance = StaticSelection.from_json(json_str)
+            match += 1
+        except (ValidationError, ValueError) as e:
+            error_messages.append(str(e))
+        # deserialize data into ValidationSelection
+        try:
+            instance.actual_instance = ValidationSelection.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 CappedSelectionSelectionsInner with oneOf schemas: CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. Details: " + ", ".join(error_messages))
+        elif match == 0:
+            # no match
+            raise ValueError("No match found when deserializing the JSON string into CappedSelectionSelectionsInner with oneOf schemas: CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection. 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], CappedSelection, ConditionalValidationSelection, DemographicSelection, LabelingSelection, StaticSelection, ValidationSelection]]:
+        """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.capped_selection import CappedSelection
+# TODO: Rewrite to not use raise_errors
+CappedSelectionSelectionsInner.model_rebuild(raise_errors=False)
+

rapidata/api_client/models/create_order_model.py

@@ -19,8 +19,8 @@ import json
 
 from pydantic import BaseModel, ConfigDict, Field, StrictInt, StrictStr, field_validator
 from typing import Any, ClassVar, Dict, List, Optional
+from rapidata.api_client.models.capped_selection_selections_inner import CappedSelectionSelectionsInner
 from rapidata.api_client.models.create_order_model_referee import CreateOrderModelReferee
-from rapidata.api_client.models.create_order_model_selections_inner import CreateOrderModelSelectionsInner
 from rapidata.api_client.models.create_order_model_user_filters_inner import CreateOrderModelUserFiltersInner
 from rapidata.api_client.models.create_order_model_workflow import CreateOrderModelWorkflow
 from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
@@ -40,7 +40,7 @@ class CreateOrderModel(BaseModel):
     priority: Optional[StrictInt] = Field(default=None, description="The priority is used to prioritize over other orders.")
     user_filters: List[CreateOrderModelUserFiltersInner] = Field(description="The user filters are used to restrict the order to only collect votes from a specific demographic.", alias="userFilters")
     validation_set_id: Optional[StrictStr] = Field(default=None, description="The validation set id can be changed to point to a specific validation set. if not provided a sane default will be  used.", alias="validationSetId")
-    selections: Optional[List[CreateOrderModelSelectionsInner]] = Field(default=None, description="The selections are used to determine which tasks are shown to a user.")
+    selections: Optional[List[CappedSelectionSelectionsInner]] = Field(default=None, description="The selections are used to determine which tasks are shown to a user.")
     __properties: ClassVar[List[str]] = ["_t", "orderName", "workflow", "referee", "aggregator", "featureFlags", "priority", "userFilters", "validationSetId", "selections"]
 
     @field_validator('t')
@@ -172,7 +172,7 @@ class CreateOrderModel(BaseModel):
             "priority": obj.get("priority"),
             "userFilters": [CreateOrderModelUserFiltersInner.from_dict(_item) for _item in obj["userFilters"]] if obj.get("userFilters") is not None else None,
             "validationSetId": obj.get("validationSetId"),
-            "selections": [CreateOrderModelSelectionsInner.from_dict(_item) for _item in obj["selections"]] if obj.get("selections") is not None else None
+            "selections": [CappedSelectionSelectionsInner.from_dict(_item) for _item in obj["selections"]] if obj.get("selections") is not None else None
         })
         return _obj
 

rapidata/api_client_README.md

@@ -133,6 +133,7 @@ Class | Method | HTTP request | Description
 *RapidApi* | [**rapid_validate_current_rapid_bag_get**](rapidata/api_client/docs/RapidApi.md#rapid_validate_current_rapid_bag_get) | **GET** /Rapid/ValidateCurrentRapidBag | Validates that the rapids associated with the current user are active.
 *SimpleWorkflowApi* | [**simple_workflow_get_result_overview_get**](rapidata/api_client/docs/SimpleWorkflowApi.md#simple_workflow_get_result_overview_get) | **GET** /SimpleWorkflow/GetResultOverview | Get the result overview for a simple workflow.
 *ValidationApi* | [**validation_add_validation_rapid_post**](rapidata/api_client/docs/ValidationApi.md#validation_add_validation_rapid_post) | **POST** /Validation/AddValidationRapid | Adds a new validation rapid to the specified validation set.
+*ValidationApi* | [**validation_add_validation_text_rapid_post**](rapidata/api_client/docs/ValidationApi.md#validation_add_validation_text_rapid_post) | **POST** /Validation/AddValidationTextRapid | Adds a new validation rapid to the specified validation set.
 *ValidationApi* | [**validation_create_validation_set_post**](rapidata/api_client/docs/ValidationApi.md#validation_create_validation_set_post) | **POST** /Validation/CreateValidationSet | Creates a new empty validation set.
 *ValidationApi* | [**validation_get_available_validation_sets_get**](rapidata/api_client/docs/ValidationApi.md#validation_get_available_validation_sets_get) | **GET** /Validation/GetAvailableValidationSets | Gets the available validation sets for the current user.
 *ValidationApi* | [**validation_import_compare_post**](rapidata/api_client/docs/ValidationApi.md#validation_import_compare_post) | **POST** /Validation/ImportCompare | Imports a compare validation set from a zip file.
@@ -157,6 +158,7 @@ Class | Method | HTTP request | Description
  - [AddValidationRapidModelPayload](rapidata/api_client/docs/AddValidationRapidModelPayload.md)
  - [AddValidationRapidModelTruth](rapidata/api_client/docs/AddValidationRapidModelTruth.md)
  - [AddValidationRapidResult](rapidata/api_client/docs/AddValidationRapidResult.md)
+ - [AddValidationTextRapidModel](rapidata/api_client/docs/AddValidationTextRapidModel.md)
  - [AdminOrderModel](rapidata/api_client/docs/AdminOrderModel.md)
  - [AdminOrderModelPagedResult](rapidata/api_client/docs/AdminOrderModelPagedResult.md)
  - [AgeGroup](rapidata/api_client/docs/AgeGroup.md)
@@ -176,6 +178,8 @@ Class | Method | HTTP request | Description
  - [CampaignQueryModelPagedResult](rapidata/api_client/docs/CampaignQueryModelPagedResult.md)
  - [CampaignStatus](rapidata/api_client/docs/CampaignStatus.md)
  - [CampaignUserFilterModel](rapidata/api_client/docs/CampaignUserFilterModel.md)
+ - [CappedSelection](rapidata/api_client/docs/CappedSelection.md)
+ - [CappedSelectionSelectionsInner](rapidata/api_client/docs/CappedSelectionSelectionsInner.md)
  - [ClassificationMetadata](rapidata/api_client/docs/ClassificationMetadata.md)
  - [ClassificationMetadataFilterConfig](rapidata/api_client/docs/ClassificationMetadataFilterConfig.md)
  - [ClassifyPayload](rapidata/api_client/docs/ClassifyPayload.md)
@@ -211,7 +215,6 @@ Class | Method | HTTP request | Description
  - [CreateIndependentWorkflowResult](rapidata/api_client/docs/CreateIndependentWorkflowResult.md)
  - [CreateOrderModel](rapidata/api_client/docs/CreateOrderModel.md)
  - [CreateOrderModelReferee](rapidata/api_client/docs/CreateOrderModelReferee.md)
- - [CreateOrderModelSelectionsInner](rapidata/api_client/docs/CreateOrderModelSelectionsInner.md)
  - [CreateOrderModelUserFiltersInner](rapidata/api_client/docs/CreateOrderModelUserFiltersInner.md)
  - [CreateOrderModelWorkflow](rapidata/api_client/docs/CreateOrderModelWorkflow.md)
  - [CreateOrderResult](rapidata/api_client/docs/CreateOrderResult.md)

rapidata/rapidata_client/__init__.py

@@ -10,3 +10,4 @@ from .referee import NaiveReferee, ClassifyEarlyStoppingReferee
 from .metadata import PrivateTextMetadata, PublicTextMetadata, PromptMetadata, TranscriptionMetadata
 from .feature_flags import FeatureFlags
 from .country_codes import CountryCodes
+from .assets import MediaAsset, TextAsset, MultiAsset

rapidata/rapidata_client/assets/__init__.py

@@ -0,0 +1,8 @@
+"""Assets Package
+
+This package provides classes for different types of assets, including MediaAsset, TextAsset, and MultiAsset.
+"""
+
+from .media_asset import MediaAsset
+from .text_asset import TextAsset
+from .multi_asset import MultiAsset

rapidata/rapidata_client/assets/base_asset.py

@@ -0,0 +1,11 @@
+"""Base Asset Module
+
+Defines the BaseAsset class, which serves as the abstract base class for all asset types.
+"""
+
+class BaseAsset:
+    """BaseAsset Class
+
+    An abstract base class for different types of assets. This class is intended to be subclassed by specific asset types.
+    """
+    pass

rapidata/rapidata_client/assets/media_asset.py

@@ -0,0 +1,33 @@
+"""Media Asset Module
+
+Defines the MediaAsset class for handling media file paths within assets.
+"""
+
+import os
+from rapidata.rapidata_client.assets.base_asset import BaseAsset
+
+class MediaAsset(BaseAsset):
+    """MediaAsset Class
+
+    Represents a media asset by storing the file path.
+
+    Args:
+        path (str): The file system path to the media asset.
+
+    Raises:
+        FileNotFoundError: If the provided file path does not exist.
+    """
+
+    def __init__(self, path: str):
+        """
+        Initialize a MediaAsset instance.
+
+        Args:
+            path (str): The file system path to the media asset.
+
+        Raises:
+            FileNotFoundError: If the provided file path does not exist.
+        """
+        if not os.path.exists(path):
+            raise FileNotFoundError(f"File not found: {path}")
+        self.path = path

rapidata/rapidata_client/assets/multi_asset.py

@@ -0,0 +1,44 @@
+"""Multi Asset Module
+
+Defines the MultiAsset class for handling multiple BaseAsset instances.
+"""
+
+from rapidata.rapidata_client.assets.base_asset import BaseAsset
+from typing import Iterator, List
+
+
+class MultiAsset(BaseAsset):
+    """MultiAsset Class
+
+    Represents a collection of multiple BaseAsset instances.
+
+    Args:
+        assets (List[BaseAsset]): A list of BaseAsset instances to be managed together.
+    """
+
+    def __init__(self, assets: list[BaseAsset]):
+        """
+        Initialize a MultiAsset instance.
+
+        Args:
+            assets (List[BaseAsset]): A list of BaseAsset instances to be managed together.
+        """
+        self.assets = assets
+
+    def __len__(self) -> int:
+        """
+        Get the number of assets in the MultiAsset.
+
+        Returns:
+            int: The number of assets.
+        """
+        return len(self.assets)
+
+    def __iter__(self) -> Iterator[BaseAsset]:
+        """
+        Return an iterator over the assets in the MultiAsset.
+
+        Returns:
+            Iterator[BaseAsset]: An iterator over the assets.
+        """
+        return iter(self.assets)

rapidata/rapidata_client/assets/text_asset.py

@@ -0,0 +1,25 @@
+"""Text Asset Module
+
+Defines the TextAsset class for handling textual data within assets.
+"""
+
+from rapidata.rapidata_client.assets.base_asset import BaseAsset
+
+
+class TextAsset(BaseAsset):
+    """TextAsset Class
+
+    Represents a textual asset.
+
+    Args:
+        text (str): The text content of the asset.
+    """
+
+    def __init__(self, text: str):
+        """
+        Initialize a TextAsset instance.
+
+        Args:
+            text (str): The textual content of the asset.
+        """
+        self.text = text

rapidata/rapidata_client/dataset/rapidata_dataset.py

@@ -32,11 +32,11 @@ class RapidataDataset:
 
     def add_media_from_paths(
         self,
-        image_paths: list[str | list[str]],
+        media_paths: list[str | list[str]],
         metadata: list[Metadata] | None = None,
         max_workers: int = 10,
     ):
-        if metadata is not None and len(metadata) != len(image_paths):
+        if metadata is not None and len(metadata) != len(media_paths):
             raise ValueError(
                 "metadata must be None or have the same length as image_paths"
             )
@@ -66,11 +66,11 @@ class RapidataDataset:
                 files=media_paths_rapid if isinstance(media_paths_rapid, list) else [media_paths_rapid] # type: ignore
             )
 
-        total_uploads = len(image_paths)
+        total_uploads = len(media_paths)
         with ThreadPoolExecutor(max_workers=max_workers) as executor:
             futures = [
                 executor.submit(upload_datapoint, media_paths, meta)
-                for media_paths, meta in zip_longest(image_paths, metadata or [])
+                for media_paths, meta in zip_longest(media_paths, metadata or [])
             ]
 
             with tqdm(total=total_uploads, desc="Uploading datapoints") as pbar:

rapidata/rapidata_client/dataset/rapidata_validation_set.py

@@ -3,6 +3,9 @@ from typing import Any
 from rapidata.api_client.models.add_validation_rapid_model import (
     AddValidationRapidModel,
 )
+from rapidata.api_client.models.add_validation_text_rapid_model import (
+    AddValidationTextRapidModel,
+)
 from rapidata.api_client.models.add_validation_rapid_model_payload import (
     AddValidationRapidModelPayload,
 )
@@ -31,6 +34,9 @@ from rapidata.api_client.models.polygon_truth import PolygonTruth
 from rapidata.api_client.models.transcription_payload import TranscriptionPayload
 from rapidata.api_client.models.transcription_truth import TranscriptionTruth
 from rapidata.api_client.models.transcription_word import TranscriptionWord
+from rapidata.rapidata_client.assets.media_asset import MediaAsset
+from rapidata.rapidata_client.assets.multi_asset import MultiAsset
+from rapidata.rapidata_client.assets.text_asset import TextAsset
 from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.service.openapi_service import OpenAPIService
 
@@ -38,7 +44,7 @@ from rapidata.service.openapi_service import OpenAPIService
 class RapidataValidationSet:
     """A class for interacting with a Rapidata validation set.
 
-    Get a `ValidationSet` either by using `rapi.get_validation_set(id)` to get an exisitng validation set or by using `rapi.new_validation_set(name)` to create a new validation set.
+    Get a `ValidationSet` either by using `rapi.get_validation_set(id)` to get an existing validation set or by using `rapi.new_validation_set(name)` to create a new validation set.
     """
 
     def __init__(self, validation_set_id, openapi_service: OpenAPIService):
@@ -70,21 +76,25 @@ class RapidataValidationSet:
             | TranscriptionTruth
         ),
         metadata: list[Metadata],
-        media_paths: str | list[str],
+        asset: MediaAsset | TextAsset | MultiAsset,
         randomCorrectProbability: float,
     ) -> None:
         """Add a validation rapid to the validation set.
 
         Args:
-            payload (Union[BoundingBoxPayload, ClassifyPayload, ComparePayload, FreeTextPayload, LinePayload, LocatePayload, NamedEntityPayload, PolygonPayload, TranscriptionPayload]): The payload for the rapid.
-            truths (Union[AttachCategoryTruth, BoundingBoxTruth, CompareTruth, EmptyValidationTruth, LineTruth, LocateBoxTruth, NamedEntityTruth, PolygonTruth, TranscriptionTruth]): The truths for the rapid.
+            payload: The payload for the rapid.
+            truths: The truths for the rapid.
             metadata (list[Metadata]): The metadata for the rapid.
-            media_paths (Union[str, list[str]]): The media paths for the rapid.
+            asset: The asset(s) for the rapid.
             randomCorrectProbability (float): The random correct probability for the rapid.
 
         Returns:
             None
+
+        Raises:
+            ValueError: If an invalid asset type is provided.
         """
+
         model = AddValidationRapidModel(
             validationSetId=self.id,
             payload=AddValidationRapidModelPayload(payload),
@@ -95,14 +105,37 @@ class RapidataValidationSet:
             ],
             randomCorrectProbability=randomCorrectProbability,
         )
+        if isinstance(asset, MediaAsset):
+            self.openapi_service.validation_api.validation_add_validation_rapid_post(
+                model=model, files=[asset.path]
+            )
 
-        self.openapi_service.validation_api.validation_add_validation_rapid_post(
-            model=model, files=media_paths if isinstance(media_paths, list) else [media_paths]  # type: ignore
-        )
+        elif isinstance(asset, MultiAsset):
+            self.openapi_service.validation_api.validation_add_validation_rapid_post(
+                model=model, files=[a.path for a in asset if isinstance(a, MediaAsset)]
+            )
+
+        elif isinstance(asset, TextAsset):
+            model = AddValidationTextRapidModel(
+                validationSetId=self.id,
+                payload=AddValidationRapidModelPayload(payload),
+                truth=AddValidationRapidModelTruth(truths),
+                metadata=[
+                    DatapointMetadataModelMetadataInner(meta.to_model())
+                    for meta in metadata
+                ],
+                randomCorrectProbability=randomCorrectProbability,
+                text=asset.text,
+            )
+            self.openapi_service.validation_api.validation_add_validation_text_rapid_post(
+                add_validation_text_rapid_model=model
+            )
+        else:
+            raise ValueError("Invalid asset type")
 
     def add_classify_validation_rapid(
         self,
-        media_path: str,
+        asset: MediaAsset | TextAsset,
         question: str,
         categories: list[str],
         truths: list[str],
@@ -111,7 +144,7 @@ class RapidataValidationSet:
         """Add a classify rapid to the validation set.
 
         Args:
-            media_path (str): The path to the media file.
+            asset (MediaAsset | TextAsset): The asset for the rapid.
             question (str): The question for the rapid.
             categories (list[str]): The list of categories for the rapid.
             truths (list[str]): The list of truths for the rapid.
@@ -131,13 +164,13 @@ class RapidataValidationSet:
             payload=payload,
             truths=model_truth,
             metadata=metadata,
-            media_paths=media_path,
+            asset=asset,
             randomCorrectProbability=len(truths) / len(categories),
         )
 
     def add_compare_validation_rapid(
         self,
-        media_paths: list[str],
+        asset: MultiAsset,
         question: str,
         truth: str,
         metadata: list[Metadata] = [],
@@ -145,7 +178,7 @@ class RapidataValidationSet:
         """Add a compare rapid to the validation set.
 
         Args:
-            media_paths (list[str]): The list of media paths for the rapid.
+            asset (MultiAsset): The assets for the rapid.
             question (str): The question for the rapid.
             truth (str): The path to the truth file.
             metadata (list[Metadata], optional): The metadata for the rapid. Defaults to an empty list.
@@ -154,33 +187,27 @@ class RapidataValidationSet:
             None
 
         Raises:
-            ValueError: If the number of media paths is not exactly two.
-            FileNotFoundError: If any of the specified files are not found.
+            ValueError: If the number of assets is not exactly two.
         """
         payload = ComparePayload(_t="ComparePayload", criteria=question)
         # take only last part of truth path
         truth = os.path.basename(truth)
         model_truth = CompareTruth(_t="CompareTruth", winnerId=truth)
 
-        if len(media_paths) != 2:
+        if len(asset) != 2:
             raise ValueError("Compare rapid requires exactly two media paths")
 
-        # check that files exist
-        for media_path in media_paths:
-            if not os.path.exists(media_path):
-                raise FileNotFoundError(f"File not found: {media_path}")
-
         self.add_general_validation_rapid(
             payload=payload,
             truths=model_truth,
             metadata=metadata,
-            media_paths=media_paths,
-            randomCorrectProbability=1 / len(media_paths),
+            asset=asset,
+            randomCorrectProbability=1 / len(asset),
         )
 
     def add_transcription_validation_rapid(
         self,
-        media_path: str,
+        asset: MediaAsset | TextAsset,
         question: str,
         transcription: list[str],
         correct_words: list[str],
@@ -190,11 +217,11 @@ class RapidataValidationSet:
         """Add a transcription rapid to the validation set.
 
         Args:
-            media_path (str): The path to the media file.
+            asset (MediaAsset | TextAsset): The asset for the rapid.
             question (str): The question for the rapid.
             transcription (list[str]): The transcription for the rapid.
             correct_words (list[str]): The list of correct words for the rapid.
-            strict_grading (Optional[bool], optional): The strict grading for the rapid. Defaults to None.
+            strict_grading (bool | None, optional): The strict grading for the rapid. Defaults to None.
             metadata (list[Metadata], optional): The metadata for the rapid. Defaults to an empty list.
 
         Returns:
@@ -230,6 +257,6 @@ class RapidataValidationSet:
             payload=payload,
             truths=model_truth,
             metadata=metadata,
-            media_paths=media_path,
+            asset=asset,
             randomCorrectProbability=len(correct_words) / len(transcription),
         )

rapidata/rapidata_client/dataset/validation_rapid_parts.py

@@ -19,13 +19,16 @@ from rapidata.api_client.models.polygon_payload import PolygonPayload
 from rapidata.api_client.models.polygon_truth import PolygonTruth
 from rapidata.api_client.models.transcription_payload import TranscriptionPayload
 from rapidata.api_client.models.transcription_truth import TranscriptionTruth
+from rapidata.rapidata_client.assets.media_asset import MediaAsset
+from rapidata.rapidata_client.assets.multi_asset import MultiAsset
+from rapidata.rapidata_client.assets.text_asset import TextAsset
 from rapidata.rapidata_client.metadata.base_metadata import Metadata
 
 
 @dataclass
 class ValidatioRapidParts:
     question: str
-    media_paths: str | list[str]
+    asset: MediaAsset | TextAsset | MultiAsset
     payload: (
         BoundingBoxPayload
         | ClassifyPayload