rapidata 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

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/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(

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/rapidata_client/types/__init__.py

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

{rapidata-0.1.10.dist-info → rapidata-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.1.10
+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

{rapidata-0.1.10.dist-info → rapidata-0.1.11.dist-info}/RECORD

@@ -202,15 +202,16 @@ rapidata/rapidata_client/metadata/prompt_metadata.py,sha256=_FypjKWrC3iKUO_G2CVw
 rapidata/rapidata_client/metadata/transcription_metadata.py,sha256=THtDEVCON4UlcXHmXrjilaOLHys4TrktUOPGWnXaCcc,631
 rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/order/dataset/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-rapidata/rapidata_client/order/dataset/rapidata_dataset.py,sha256=4fK7TF-jHu6KetATUStB_OwYVo2r5nX7gQTzTHu5pXc,2000
-rapidata/rapidata_client/order/dataset/validation_set_builder.py,sha256=RSH3qlmHfjNNnQ3VXvf_BQBXYs0hao9D-z6m3tapMv0,6944
+rapidata/rapidata_client/order/dataset/rapidata_dataset.py,sha256=mVx-YU3N6lvr_0NCg7yGodeBJh8A_Y0tFWRrxcFg9-w,2506
+rapidata/rapidata_client/order/dataset/validation_set_builder.py,sha256=bOdq4ERm0QeQFZ8bc1I82S5qYIqu5hzUsqmDBispMlY,6944
 rapidata/rapidata_client/order/rapidata_order.py,sha256=ve58PWrv396RxlpFo8OT3QeZvbJjLDRDkNxuCyZQkvg,3003
-rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=eawFAmGKw8OlUlN1uZCxORAEJuepzsZk-Mzc82tu73o,6331
+rapidata/rapidata_client/order/rapidata_order_builder.py,sha256=cgSj8yMVMbKDr_8yksTMjFpYqXet27U8QDXzxN7L4is,6491
 rapidata/rapidata_client/rapidata_client.py,sha256=cbn9d_kvOSLhgaGkUhQJ1lbjYfuw120Kw_x3Kyfzq4M,1607
 rapidata/rapidata_client/referee/__init__.py,sha256=x0AxGCsR6TlDjfqQ00lB9V7QVS9EZCJzweNEIzx42PI,207
 rapidata/rapidata_client/referee/base_referee.py,sha256=bMy7cw0a-pGNbFu6u_1_Jplu0A483Ubj4oDQzh8vu8k,493
 rapidata/rapidata_client/referee/classify_early_stopping_referee.py,sha256=-mPTCck81r2xn1GtMoizaOCftA3cOfj6YDSFbqsI3Ic,1140
 rapidata/rapidata_client/referee/naive_referee.py,sha256=9CSoZou4YBFc9PiRvA2ELVQqyk-S1nug1SqhTwO9Tvw,719
+rapidata/rapidata_client/types/__init__.py,sha256=Q9lzemtloA8e_XVVV1Y7ZZGsNg-xRqgj1GdJPrCzo5A,33
 rapidata/rapidata_client/workflow/__init__.py,sha256=CDG8bKOBhLV4A0uBlqOd__79fH9KW3-UlZsR1RANLf0,250
 rapidata/rapidata_client/workflow/base_workflow.py,sha256=335UyiNsIFix7ru_KZLeQHvfNg3GSId44ppPmWTXs_Y,1267
 rapidata/rapidata_client/workflow/classify_workflow.py,sha256=m_0z3y3BNnoOeJv27GpmKXfyjVhjeK-5gaVbtYLboVc,1221
@@ -222,7 +223,7 @@ rapidata/service/local_file_service.py,sha256=pgorvlWcx52Uh3cEG6VrdMK_t__7dacQ_5
 rapidata/service/openapi_service.py,sha256=xNlISI5NZByLy1Yeasc5Uafj1IA2mWJTzxUU1TtwyQU,1402
 rapidata/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/utils/image_utils.py,sha256=TldO3eJWG8IhfJjm5MfNGO0mEDm1mQTsRoA0HLU1Uxs,404
-rapidata-0.1.10.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-0.1.10.dist-info/METADATA,sha256=PzlMF-Z6gKSks4yLISLHT_EoJyOWhj6jwJhDv0KH5TE,768
-rapidata-0.1.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-rapidata-0.1.10.dist-info/RECORD,,
+rapidata-0.1.11.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-0.1.11.dist-info/METADATA,sha256=-c0TF8oPPvEJOfGlDhxNn13lj5dovqrgqbUqhhc70A4,768
+rapidata-0.1.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+rapidata-0.1.11.dist-info/RECORD,,