rapidata 0.1.7__tar.gz → 0.1.9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rapidata
-Version: 0.1.7
+Version: 0.1.9
 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.7 → rapidata-0.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "rapidata"
-version = "0.1.7"
+version = "0.1.9"
 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-0.1.7 → rapidata-0.1.9}/rapidata/rapidata_client/feature_flags/feature_flags.py

@@ -7,16 +7,19 @@ class FeatureFlags:
 
     def to_list(self) -> list[FeatureFlagModel]:
         return [FeatureFlagModel(key=name, value=value) for name, value in self._flags.items()]
-    
+
     def alert_on_fast_response(self, value: int):
-        self._flags["alertOnFastResponse"] = str(value)
+        self._flags["aler_on_fast_response"] = str(value)
         return self
-    
-    def disable_translation(self, value: bool):
-        self._flags["disableTranslation"] = str(value)
+
+    def disable_translation(self, value: bool = True):
+        self._flags["disable_translation"] = str(value)
         return self
-    
+
     def free_text_minimum_characters(self, value: int):
-        self._flags["freeTextMinimumCharacters"] = str(value)
+        self._flags["free_text_minimum_characters"] = str(value)
+        return self
+
+    def no_shuffle(self, value: bool = True):
+        self._flags["no_shuffle"] = str(value)
         return self
-    

{rapidata-0.1.7 → rapidata-0.1.9}/rapidata/rapidata_client/metadata/base_metadata.py

@@ -4,8 +4,8 @@ from typing import Any
 
 class Metadata:
 
-    def __init__(self):
-        pass
+    def __init__(self, identifier: str):
+        self._identifier = identifier
 
     @abstractmethod
     def to_model(self) -> Any:

rapidata-0.1.9/rapidata/rapidata_client/metadata/private_text_metadata.py

@@ -0,0 +1,16 @@
+from rapidata.api_client.models.private_text_metadata_input import (
+    PrivateTextMetadataInput,
+)
+from rapidata.rapidata_client.metadata.base_metadata import Metadata
+
+
+class PrivateTextMetadata(Metadata):
+
+    def __init__(self, text: str, identifier: str = "private_text"):
+        super().__init__(identifier=identifier)
+        self._text = text
+
+    def to_model(self):
+        return PrivateTextMetadataInput(
+            _t="PrivateTextMetadataInput", identifier=self._identifier, text=self._text
+        )

rapidata-0.1.9/rapidata/rapidata_client/metadata/prompt_metadata.py

@@ -0,0 +1,13 @@
+from rapidata.rapidata_client.metadata.base_metadata import Metadata
+from rapidata.api_client.models.prompt_metadata_input import PromptMetadataInput
+
+
+class PromptMetadata(Metadata):
+
+    def __init__(self, prompt: str, identifier: str = "prompt"):
+        super().__init__(identifier=identifier)
+        self._prompt = prompt
+    
+
+    def to_model(self):
+        return PromptMetadataInput(_t="PromptMetadataInput", identifier=self._identifier, prompt=self._prompt)

{rapidata-0.1.7 → rapidata-0.1.9}/rapidata/rapidata_client/metadata/transcription_metadata.py

@@ -7,7 +7,7 @@ from rapidata.rapidata_client.metadata.base_metadata import Metadata
 class TranscriptionMetadata(Metadata):
 
     def __init__(self, transcription: str, identifier: str = "transcription"):
-        super().__init__()
+        super().__init__(identifier=identifier)
         self.identifier = identifier
         self.transcription = transcription
 

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

@@ -128,6 +128,9 @@ class ValidationSetBuilder:
         for media_path in media_paths:
             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-0.1.7 → rapidata-0.1.9}/rapidata/rapidata_client/referee/classify_early_stopping_referee.py

@@ -1,4 +1,6 @@
+from typing import Any
 from rapidata.rapidata_client.referee.base_referee import Referee
+from rapidata.api_client.models.probabilistic_attach_category_referee_config import ProbabilisticAttachCategoryRefereeConfig
 
 
 class ClassifyEarlyStoppingReferee(Referee):
@@ -18,3 +20,10 @@ class ClassifyEarlyStoppingReferee(Referee):
             "threshold": self.threshold,
             "maxVotes": self.max_vote_count,
         }
+    
+    def to_model(self) -> Any:
+        return ProbabilisticAttachCategoryRefereeConfig(
+            _t="ProbabilisticAttachCategoryRefereeConfig",
+            threshold=self.threshold,
+            maxVotes=self.max_vote_count,
+        )