rapidata 0.1.9__tar.gz → 0.1.11__tar.gz

{rapidata-0.1.9 → rapidata-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.1.9
+Version: 0.1.11
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 Author: Rapidata AG
@@ -11,6 +11,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: PyJWT (>=2.9.0,<3.0.0)
 Requires-Dist: pillow (>=10.4.0,<11.0.0)
+Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
 Description-Content-Type: text/markdown
 

{rapidata-0.1.9 → rapidata-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "0.1.9"
+version = "0.1.11"
 description = "Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way."
 authors = ["Rapidata AG <info@rapidata.ai>"]
 license = "Apache-2.0"
@@ -11,6 +11,7 @@ python = "^3.12"
 requests = "^2.32.3"
 PyJWT = "^2.9.0"
 pillow = "^10.4.0"
+pydantic = "^2.8.2"
 
 [tool.poetry.group.dev.dependencies]
 python-dotenv = "^1.0.1"

{rapidata-0.1.9 → rapidata-0.1.11}/rapidata/rapidata_client/feature_flags/feature_flags.py

@@ -23,3 +23,11 @@ class FeatureFlags:
     def no_shuffle(self, value: bool = True):
         self._flags["no_shuffle"] = str(value)
         return self
+    
+    def claire_design(self, value: bool = True):
+        self._flags["claire"] = str(value)
+        return self
+    
+    def key_value(self, key: str, value: str):
+        self._flags[key] = value
+        return self

{rapidata-0.1.9 → rapidata-0.1.11}/rapidata/rapidata_client/order/dataset/rapidata_dataset.py

@@ -1,6 +1,5 @@
 from itertools import zip_longest
 import os
-from typing import List
 
 from rapidata.api_client.models.datapoint_metadata_model import DatapointMetadataModel
 from rapidata.api_client.models.datapoint_metadata_model_metadata_inner import (
@@ -10,6 +9,7 @@ from rapidata.api_client.models.upload_text_sources_to_dataset_model import (
     UploadTextSourcesToDatasetModel,
 )
 from rapidata.rapidata_client.metadata.base_metadata import Metadata
+from rapidata.rapidata_client.types import RapidAsset
 from rapidata.service import LocalFileService
 from rapidata.service.openapi_service import OpenAPIService
 
@@ -31,22 +31,34 @@ class RapidataDataset:
 
     def add_media_from_paths(
         self,
-        image_paths: list[str],
-        metadata: List[Metadata] | None = None,
+        image_paths: list[RapidAsset],
+        metadata: list[Metadata] | None = None,
     ):
         if metadata is not None and len(metadata) != len(image_paths):
             raise ValueError(
                 "metadata must be None or have the same length as image_paths"
             )
 
-        for image_path, meta in zip_longest(image_paths, metadata or []):
-            if not os.path.exists(image_path):
-                raise FileNotFoundError(f"File not found: {image_path}")
-            
+        for media_paths_rapid, meta in zip_longest(image_paths, metadata or []):
+
+            if isinstance(media_paths_rapid, list) and not all(
+                os.path.exists(media_path) for media_path in media_paths_rapid
+            ):
+                raise FileNotFoundError(f"File not found: {media_paths_rapid}")
+
+            elif isinstance(media_paths_rapid, str) and not os.path.exists(
+                media_paths_rapid
+            ):
+                raise FileNotFoundError(f"File not found: {media_paths_rapid}")
+
             meta_model = meta.to_model() if meta else None
             model = DatapointMetadataModel(
                 datasetId=self.dataset_id,
-                metadata=[DatapointMetadataModelMetadataInner(meta_model)] if meta_model else [],
+                metadata=(
+                    [DatapointMetadataModelMetadataInner(meta_model)]
+                    if meta_model
+                    else []
+                ),
             )
 
-            self.openapi_service.dataset_api.dataset_create_datapoint_post(model=model, files=[image_path])  # type: ignore
+            self.openapi_service.dataset_api.dataset_create_datapoint_post(model=model, files=media_paths_rapid if isinstance(media_paths_rapid, list) else [media_paths_rapid])  # type: ignore

{rapidata-0.1.9 → rapidata-0.1.11}/rapidata/rapidata_client/order/rapidata_order_builder.py

@@ -1,18 +1,21 @@
-from rapidata.api_client.api_client import ApiClient
 from rapidata.api_client.models.aggregator_type import AggregatorType
 from rapidata.api_client.models.create_order_model import CreateOrderModel
-from rapidata.api_client.models.create_order_model_referee import CreateOrderModelReferee
-from rapidata.api_client.models.create_order_model_workflow import CreateOrderModelWorkflow
+from rapidata.api_client.models.create_order_model_referee import (
+    CreateOrderModelReferee,
+)
+from rapidata.api_client.models.create_order_model_workflow import (
+    CreateOrderModelWorkflow,
+)
 from rapidata.rapidata_client.feature_flags import FeatureFlags
 from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.rapidata_client.order.dataset.rapidata_dataset import RapidataDataset
 from rapidata.rapidata_client.referee.naive_referee import NaiveReferee
+from rapidata.rapidata_client.types import RapidAsset
 from rapidata.rapidata_client.workflow import Workflow
 from rapidata.rapidata_client.order.rapidata_order import RapidataOrder
 from rapidata.rapidata_client.referee import Referee
 from rapidata.service.openapi_service import OpenAPIService
 
-
 class RapidataOrderBuilder:
     """
     Builder object for creating Rapidata orders.
@@ -34,8 +37,8 @@ class RapidataOrderBuilder:
         self._openapi_service = openapi_service
         self._workflow: Workflow | None = None
         self._referee: Referee | None = None
-        self._media_paths: list[str] = []
-        self._metadata: list[Metadata] | None = None 
+        self._media_paths: list[RapidAsset] = []
+        self._metadata: list[Metadata] | None = None
         self._aggregator: AggregatorType | None = None
         self._validation_set_id: str | None = None
         self._feature_flags: FeatureFlags | None = None
@@ -61,7 +64,11 @@ class RapidataOrderBuilder:
             userFilters=[],
             referee=CreateOrderModelReferee(self._referee.to_model()),
             validationSetId=self._validation_set_id,
-            featureFlags=self._feature_flags.to_list() if self._feature_flags is not None else None,
+            featureFlags=(
+                self._feature_flags.to_list()
+                if self._feature_flags is not None
+                else None
+            ),
         )
 
         result = self._openapi_service.order_api.order_create_post(
@@ -70,7 +77,11 @@ class RapidataOrderBuilder:
 
         self.order_id = result.order_id
         self._dataset = RapidataDataset(result.dataset_id, self._openapi_service)
-        order = RapidataOrder(order_id=self.order_id, dataset=self._dataset, openapi_service=self._openapi_service)
+        order = RapidataOrder(
+            order_id=self.order_id,
+            dataset=self._dataset,
+            openapi_service=self._openapi_service,
+        )
 
         order.dataset.add_media_from_paths(self._media_paths, self._metadata)
 
@@ -104,7 +115,10 @@ class RapidataOrderBuilder:
         return self
 
     def media(
-        self, media_paths: list[str], metadata: list[Metadata] | None = None):
+        self,
+        media_paths: list[RapidAsset],
+        metadata: list[Metadata] | None = None,
+    ):
         """
         Set the media assets for the order.
 
@@ -157,7 +171,7 @@ class RapidataOrderBuilder:
         """
         self._aggregator = aggregator
         return self
-    
+
     def validation_set(self, validation_set_id: str):
         """
         Set the validation set for the order.

rapidata-0.1.11/rapidata/rapidata_client/types/__init__.py

@@ -0,0 +1 @@
+type RapidAsset = str | list[str]

{rapidata-0.1.9 → rapidata-0.1.11}/rapidata/rapidata_client/order/dataset/validation_set_builder.py

@@ -119,6 +119,8 @@ class ValidationSetBuilder:
 
     def add_compare_rapid(self, media_paths: list[str], question: str, truth: str):
         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:
@@ -129,8 +131,6 @@ class ValidationSetBuilder:
             if not os.path.exists(media_path):
                 raise FileNotFoundError(f"File not found: {media_path}")
             
-        # take only last part of truth path
-        truth = os.path.basename(truth)
 
         self._rapid_parts.append(
             ValidatioRapidParts(