rapidata 1.0.0__tar.gz → 1.1.0__tar.gz

{rapidata-1.0.0 → rapidata-1.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 1.0.0
+Version: 1.1.0
 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.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: pillow (>=10.4.0,<11.0.0)
 Requires-Dist: pydantic (>=2.8.2,<3.0.0)
 Requires-Dist: pyjwt (>=2.9.0,<3.0.0)

{rapidata-1.0.0 → rapidata-1.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "1.0.0"
+version = "1.1.0"
 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"

{rapidata-1.0.0 → rapidata-1.1.0}/src/rapidata/rapidata_client/dataset/validation_set_builder.py

@@ -160,8 +160,8 @@ class ValidationSetBuilder:
         self,
         asset: MediaAsset | TextAsset,
         question: str,
-        transcription: list[str],
-        correct_words: list[str],
+        transcription: str,
+        truths: list[int],
         strict_grading: bool | None = None,
         metadata: list[Metadata] = [],
     ):
@@ -171,9 +171,9 @@ class ValidationSetBuilder:
             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 (bool | None, optional): The strict grading flag for the rapid. Defaults to None.
-            metadata (list[Metadata], optional): The metadata for the rapid. Defaults to an empty list.
+            truths (list[int]): The list of indices of the true word selections.
+            strict_grading (bool | None, optional): The strict grading for the rapid. Defaults to None.
+            metadata (list[Metadata], optional): The metadata for the rapid.
 
         Returns:
             ValidationSetBuilder: The ValidationSetBuilder instance.
@@ -183,16 +183,14 @@ class ValidationSetBuilder:
         """
         transcription_words = [
             TranscriptionWord(word=word, wordIndex=i)
-            for i, word in enumerate(transcription)
+            for i, word in enumerate(transcription.split())
         ]
 
-        correct_transcription_words = []
-        for word in correct_words:
-            if word not in transcription:
-                raise ValueError(f"Correct word '{word}' not found in transcription")
-            correct_transcription_words.append(
-                TranscriptionWord(word=word, wordIndex=transcription.index(word))
-            )
+        true_words = []
+        for idx in truths:
+            if idx > len(transcription_words) - 1:
+                raise ValueError(f"Index {idx} is out of bounds")
+            true_words.append(transcription_words[idx])
 
         payload = TranscriptionPayload(
             _t="TranscriptionPayload", title=question, transcription=transcription_words
@@ -200,7 +198,7 @@ class ValidationSetBuilder:
 
         model_truth = TranscriptionTruth(
             _t="TranscriptionTruth",
-            correctWords=correct_transcription_words,
+            correctWords=true_words,
             strictGrading=strict_grading,
         )
 

{rapidata-1.0.0 → rapidata-1.1.0}/src/rapidata/rapidata_client/order/rapidata_order_builder.py

@@ -13,6 +13,9 @@ from rapidata.api_client.models.create_order_model_workflow import (
     CreateOrderModelWorkflow,
 )
 from rapidata.api_client.models.country_user_filter_model import CountryUserFilterModel
+from rapidata.api_client.models.language_user_filter_model import (
+    LanguageUserFilterModel,
+)
 from rapidata.rapidata_client.feature_flags import FeatureFlags
 from rapidata.rapidata_client.metadata.base_metadata import Metadata
 from rapidata.rapidata_client.dataset.rapidata_dataset import RapidataDataset
@@ -58,6 +61,7 @@ class RapidataOrderBuilder:
         self._validation_set_id: str | None = None
         self._feature_flags: FeatureFlags | None = None
         self._country_codes: list[str] | None = None
+        self._language_codes: list[str] | None = None
         self._selections: list[Selection] = []
         self._rapids_per_bag: int = 2
         self._priority: int = 50
@@ -80,22 +84,32 @@ class RapidataOrderBuilder:
         if self._referee is None:
             print("No referee provided, using default NaiveReferee.")
             self._referee = NaiveReferee()
-        if self._country_codes is None:
-            country_filter = None
-        else:
-            country_filter = CountryUserFilterModel(
-                _t="CountryFilter", countries=self._country_codes
+
+        user_filters = []
+
+        if self._country_codes is not None:
+            user_filters.append(
+                CreateOrderModelUserFiltersInner(
+                    CountryUserFilterModel(
+                        _t="CountryFilter", countries=self._country_codes
+                    )
+                )
+            )
+
+        if self._language_codes is not None:
+            user_filters.append(
+                CreateOrderModelUserFiltersInner(
+                    LanguageUserFilterModel(
+                        _t="LanguageFilter", languages=self._language_codes
+                    )
+                )
             )
 
         return CreateOrderModel(
             _t="CreateOrderModel",
             orderName=self._name,
             workflow=CreateOrderModelWorkflow(self._workflow.to_model()),
-            userFilters=(
-                [CreateOrderModelUserFiltersInner(country_filter)]
-                if country_filter
-                else []
-            ),
+            userFilters=user_filters,
             referee=CreateOrderModelReferee(self._referee.to_model()),
             validationSetId=self._validation_set_id,
             featureFlags=(
@@ -241,7 +255,7 @@ class RapidataOrderBuilder:
 
     def country_filter(self, country_codes: list[str]) -> "RapidataOrderBuilder":
         """
-        Set the target country codes for the order.
+        Set the target country codes for the order. E.g. `country_codes=["DE", "CH", "AT"]` for Germany, Switzerland, and Austria.
 
         Args:
             country_codes (list[str]): The country codes to be set.
@@ -252,6 +266,19 @@ class RapidataOrderBuilder:
         self._country_codes = country_codes
         return self
 
+    def language_filter(self, language_codes: list[str]) -> "RapidataOrderBuilder":
+        """
+        Set the target language codes for the order. E.g. `language_codes=["de", "fr", "it"]` for German, French, and Italian.
+
+        Args:
+            language_codes (list[str]): The language codes to be set.
+
+        Returns:
+            RapidataOrderBuilder: The updated RapidataOrderBuilder instance.
+        """
+        self._language_codes = language_codes
+        return self
+
     def aggregator(self, aggregator: AggregatorType) -> "RapidataOrderBuilder":
         """
         Set the aggregator for the order.