rapidata 1.3.0__tar.gz → 1.3.1__tar.gz

{rapidata-1.3.0 → rapidata-1.3.1}/PKG-INFO

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

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

@@ -1,4 +1,10 @@
 from rapidata.api_client.models.feature_flag_model import FeatureFlagModel
+from enum import StrEnum
+
+class TranslationBehaviour(StrEnum):
+    BOTH = "both"
+    ONLY_ORIGINAL = "only original"
+    ONLY_TRANSLATED = "only translated"
 
 
 class FeatureFlags:
@@ -49,16 +55,22 @@ class FeatureFlags:
         self._flags["alert_on_fast_response"] = str(milliseconds)
         return self
 
-    def disable_translation(self, value: bool = True):
-        """Disable automatic translation of all texts in the UI.
+
+    def translation_behaviour(self, behaviour: TranslationBehaviour = TranslationBehaviour.BOTH):
+        """Defines what's the behaviour of the translation in the UI.
+
+        The behaviour can be set to:
+            - TranslationBehaviour.BOTH: Show both the original and the translated text.
+            - TranslationBehaviour.ONLY_ORIGINAL: Show only the original text.
+            - TranslationBehaviour.ONLY_TRANSLATED: Show only the translated text.
 
         Args:
-            value (bool, optional): Whether to disable translation. Defaults to True.
+            behaviour (TranslationBehaviour): The translation behaviour. Defaults to TranslationBehaviour.BOTH.
 
         Returns:
             FeatureFlags: The current FeatureFlags instance for method chaining.
         """
-        self._flags["disable_translation"] = str(value)
+        self._flags["translation_behaviour"] = behaviour
         return self
 
     def free_text_minimum_characters(self, value: int):
@@ -108,4 +120,4 @@ class FeatureFlags:
             FeatureFlags: The current FeatureFlags instance for method chaining.
         """
         self._flags[key] = value
-        return self
+        return self