rapidata 2.39.1__py3-none-any.whl → 2.40.0__py3-none-any.whl

rapidata/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.39.1"
+__version__ = "2.40.0"
 
 from .rapidata_client import (
     RapidataClient,

rapidata/rapidata_client/config/order_config.py

@@ -10,5 +10,6 @@ class OrderConfig(BaseModel):
         autoValidationSetSize (int): The maximum size of the auto-generated validation set. Defaults to 20.
     """
 
+    autoValidationSetCreation: bool = Field(default=True)
     minOrderDatapointsForValidation: int = Field(default=50)
     autoValidationSetSize: int = Field(default=20)

rapidata/rapidata_client/order/_rapidata_order_builder.py

@@ -1,5 +1,9 @@
 from typing import Literal, Optional, Sequence
 import random
+import urllib.parse
+import webbrowser
+
+from colorama import Fore
 from rapidata.api_client.models.ab_test_selection_a_inner import AbTestSelectionAInner
 from rapidata.api_client.models.and_user_filter_model_filters_inner import (
     AndUserFilterModelFiltersInner,
@@ -38,7 +42,6 @@ from rapidata.rapidata_client.workflow import Workflow
 from rapidata.service.openapi_service import OpenAPIService
 from rapidata.rapidata_client.api.rapidata_api_client import (
     suppress_rapidata_error_logging,
-    RapidataApiClient,
 )
 
 
@@ -124,16 +127,19 @@ class RapidataOrderBuilder:
             ),
         )
 
-    def _set_validation_set_id(self) -> None:
+    def _set_validation_set_id(self) -> bool:
         """
         Get the validation set ID for the order.
+
+        Returns:
+            bool: True if a new validation set was created, False otherwise.
         """
         assert self.__workflow is not None
         if self.__validation_set_id:
             logger.debug(
                 "Using specified validation set with ID: %s", self.__validation_set_id
             )
-            return
+            return False
 
         try:
             with suppress_rapidata_error_logging():
@@ -163,14 +169,12 @@ class RapidataOrderBuilder:
             logger.debug(
                 "No recommended validation set found, dataset too small to create one."
             )
-            return
+            return False
 
         logger.info("No recommended validation set found, creating new one.")
 
         managed_print()
-        managed_print(
-            f"No recommended validation set found, new one will be created.\nWe recommend adding some truths to ensure the order is accurate."
-        )
+        managed_print(f"No recommended validation set found, new one will be created.")
         validation_set = self.__validation_set_manager._create_order_validation_set(
             workflow=self.__workflow,
             order_name=self._name,
@@ -185,6 +189,7 @@ class RapidataOrderBuilder:
 
         logger.debug("New validation set created for order: %s", validation_set)
         self.__validation_set_id = validation_set.id
+        return True
 
     def _create(self) -> RapidataOrder:
         """
@@ -197,8 +202,10 @@ class RapidataOrderBuilder:
         Returns:
             RapidataOrder: The created RapidataOrder instance.
         """
-        if rapidata_config.enableBetaFeatures:
-            self._set_validation_set_id()
+        if rapidata_config.order.autoValidationSetCreation:
+            new_validation_set = self._set_validation_set_id()
+        else:
+            new_validation_set = False
 
         order_model = self._to_model()
         logger.debug("Creating order with model: %s", order_model)
@@ -210,6 +217,24 @@ class RapidataOrderBuilder:
         self.order_id = str(result.order_id)
         logger.debug("Order created with ID: %s", self.order_id)
 
+        if rapidata_config.order.autoValidationSetCreation and new_validation_set:
+            required_amount = min(int(len(self.__datapoints) * 0.01) or 1, 10)
+            managed_print()
+            managed_print(
+                Fore.YELLOW
+                + f"A new validation set was created. Please annotate {required_amount} datapoint{('s' if required_amount != 1 else '')} so the order runs correctly."
+                + Fore.RESET
+            )
+            link = f"https://app.{self.__openapi_service.environment}/validation-set/detail/{self.__validation_set_id}/annotate?orderId={self.order_id}&required={required_amount}"
+            could_open_browser = webbrowser.open(link)
+            if not could_open_browser:
+                encoded_url = urllib.parse.quote(link, safe="%/:=&?~#+!$,;'@()*[]")
+                managed_print(
+                    Fore.RED
+                    + f"Please open this URL in your browser: '{encoded_url}'"
+                    + Fore.RESET
+                )
+
         self.__dataset = (
             RapidataDataset(result.dataset_id, self.__openapi_service)
             if result.dataset_id

rapidata/rapidata_client/rapidata_client.py

@@ -1,3 +1,5 @@
+import json
+from typing import Any
 import requests
 from packaging import version
 from rapidata import __version__
@@ -31,7 +33,7 @@ class RapidataClient:
         client_id: str | None = None,
         client_secret: str | None = None,
         environment: str = "rapidata.ai",
-        oauth_scope: str = "openid",
+        oauth_scope: str = "openid roles",
         cert_path: str | None = None,
         token: dict | None = None,
         leeway: int = 60,
@@ -51,10 +53,14 @@ class RapidataClient:
         Attributes:
             order (RapidataOrderManager): The RapidataOrderManager instance.
             validation (ValidationSetManager): The ValidationSetManager instance.
+            demographic (DemographicManager): The DemographicManager instance.
+            mri (RapidataBenchmarkManager): The RapidataBenchmarkManager instance.
         """
         with tracer.start_as_current_span("RapidataClient.__init__"):
             logger.debug("Checking version")
             self._check_version()
+            if environment != "rapidata.ai":
+                rapidata_config.logging.enable_otlp = False
 
             logger.debug("Initializing OpenAPIService")
             self._openapi_service = OpenAPIService(
@@ -83,13 +89,28 @@ class RapidataClient:
             logger.debug("Initializing RapidataBenchmarkManager")
             self.mri = RapidataBenchmarkManager(openapi_service=self._openapi_service)
 
+            self._check_beta_features()
+
     def reset_credentials(self):
         """Reset the credentials saved in the configuration file for the current environment."""
         self._openapi_service.reset_credentials()
 
-    def _enable_beta_features(self):
+    def _check_beta_features(self):
         """Enable beta features for the client."""
-        logger.debug("Enabling beta features")
+        result: dict[str, Any] = json.loads(
+            self._openapi_service.api_client.call_api(
+                "GET",
+                f"https://auth.{self._openapi_service.environment}/connect/userinfo",
+            )
+            .read()
+            .decode("utf-8")
+        )
+        logger.debug("Userinfo: %s", result)
+        if result.get("role") != ["Admin"]:
+            logger.debug("User is not an admin, not enabling beta features")
+            return
+
+        logger.debug("User is an admin, enabling beta features")
         rapidata_config.enableBetaFeatures = True
 
     def _check_version(self):

rapidata/rapidata_client/validation/validation_set_manager.py

@@ -641,6 +641,10 @@ class ValidationSetManager:
         if dimensions:
             validation_set.update_dimensions(dimensions)
 
+        self.__openapi_service.validation_api.validation_set_validation_set_id_update_labeling_hints_post(
+            validation_set_id=validation_set_id
+        )
+
         return validation_set
 
     def get_validation_set_by_id(self, validation_set_id: str) -> RapidataValidationSet:

{rapidata-2.39.1.dist-info → rapidata-2.40.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rapidata
-Version: 2.39.1
+Version: 2.40.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

{rapidata-2.39.1.dist-info → rapidata-2.40.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-rapidata/__init__.py,sha256=TLHvNFDmLEPs_wyQ0_qxuK8W9j-bcSy2_P6vtYQww9I,917
+rapidata/__init__.py,sha256=BZT4UfVlILJYOBc-BPU5ojwqvWzcOz_lKsc5HZmFrPw,917
 rapidata/api_client/__init__.py,sha256=utY2iWepKJQO_iGz6aIg_qSoqoDkV9pBMAA58pIFE4M,36016
 rapidata/api_client/api/__init__.py,sha256=07qqwzQiBYt5V2BtnzbXhZL2cmVHATyZmCSGshIXLck,1603
 rapidata/api_client/api/benchmark_api.py,sha256=Mlx2qDDJcgPjWvaBnps9dxvVd0re1knG0SyoLUiHKSc,119756
@@ -584,7 +584,7 @@ rapidata/rapidata_client/config/__init__.py,sha256=AdL9mKlOa_xm3spFKPBBdTYzhioQh
 rapidata/rapidata_client/config/logger.py,sha256=nQl9w6wLdvKGJFqtEQWoUBAlDtvYmo3w4ekIx4sNynI,5518
 rapidata/rapidata_client/config/logging_config.py,sha256=ePCh7X6v0Yzkgq1MhFNsHwLdcPEI4mYCCvBDxmWT7sU,2167
 rapidata/rapidata_client/config/managed_print.py,sha256=2T6dwgR1EZzFAdOEyPp_BBUsa-qrEuhOXgFhsSQKvRo,185
-rapidata/rapidata_client/config/order_config.py,sha256=eQxKifGN_eJfX8PpyP0IkW_HelLuEqXEImmXXXgk2tQ,557
+rapidata/rapidata_client/config/order_config.py,sha256=XxRZERzUUA9md6-PVlV__eCw8DD2kPbT_UmMwG1mAS4,615
 rapidata/rapidata_client/config/rapidata_config.py,sha256=mURnKdl5-2sE4e_IYY9-aBkix6a12t47otEErGE_q0c,1507
 rapidata/rapidata_client/config/tracer.py,sha256=h3GXzaX79HPcip4fBhLaLW0mRlXttR7D3KA78ZT0KVw,4736
 rapidata/rapidata_client/config/upload_config.py,sha256=AYba-Nw9fddLFyfGB4ar2G8zZIVOHrCL_HZjTmvrKGQ,434
@@ -632,11 +632,11 @@ rapidata/rapidata_client/filter/response_count_filter.py,sha256=i2u2YQD3_RLQRZyq
 rapidata/rapidata_client/filter/user_score_filter.py,sha256=4B3Zzp7aosDFmte3nLPTlXMN4zatT6Wcq5QLIoXqhgI,1910
 rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rapidata/rapidata_client/order/_rapidata_dataset.py,sha256=ftJD0czFX79AG9It7q2qdrQmUIGoyGq713AoUoC6nfU,18976
-rapidata/rapidata_client/order/_rapidata_order_builder.py,sha256=Ww98ZrCPW8AAs7ifGQ4cFl0h9FOa_um0DaPzLu2z1aw,15681
+rapidata/rapidata_client/order/_rapidata_order_builder.py,sha256=jLGdjjAKtZf5ZyLssx7nI3kyxIa9sg7N6RrMANnUN7o,16853
 rapidata/rapidata_client/order/rapidata_order.py,sha256=FvZi3t4dARRNsKWvYiNxVvM50AzPwQYR3AzI4utD6OI,14497
 rapidata/rapidata_client/order/rapidata_order_manager.py,sha256=XiV_BpJxG6d8o0rFDYhnB3_mb576CQG5hY-qVXlJZKY,42592
 rapidata/rapidata_client/order/rapidata_results.py,sha256=weL4S14fzug3ZOJbQk9Oj-4tv2jx5aZAMp7VJ-a6Qq4,8437
-rapidata/rapidata_client/rapidata_client.py,sha256=_RsGD_8zoneYpoOeaStTC2pmTiXqqu6shMSd6A7HsfI,4894
+rapidata/rapidata_client/rapidata_client.py,sha256=Dm7ZW2tv0QNL1pMk8NHY79vpFXyPeKU6s0WMFHHIH20,5728
 rapidata/rapidata_client/referee/__init__.py,sha256=J8oZJNUduPr-Tmn8iJwR-qBiSv7owhUFcEzXTRETecw,155
 rapidata/rapidata_client/referee/_base_referee.py,sha256=aoH3Werw-AtQo2TncG69OmCVvnbXYJBfsRCQry_Pll0,497
 rapidata/rapidata_client/referee/_early_stopping_referee.py,sha256=_JyFcTpncI0PrLf7Ix78FtLto31sdyGDS-2QbVTuvr8,2079
@@ -672,7 +672,7 @@ rapidata/rapidata_client/validation/rapids/__init__.py,sha256=WU5PPwtTJlte6U90MD
 rapidata/rapidata_client/validation/rapids/box.py,sha256=Tm2_fVdOuj-eTT8hDMajlpgWQLNtKZqw5_ud2xvjR2w,587
 rapidata/rapidata_client/validation/rapids/rapids.py,sha256=QckFItbKggezXb2awkDlYJ3mOw0Z7Z7gHzkWMBQEGl4,4822
 rapidata/rapidata_client/validation/rapids/rapids_manager.py,sha256=H4pjPghUUQS8Bk2mqcnxVmononVb14erpCvZm6q-kOk,14137
-rapidata/rapidata_client/validation/validation_set_manager.py,sha256=gjZNZLSSkhqC_Ogau-4IBBhpBQPRhnYyyY5Fd-UW9fg,34698
+rapidata/rapidata_client/validation/validation_set_manager.py,sha256=XdXpa0zko0tpTgowGrE5DAjnIM9lM8WKemGYibZeAa8,34864
 rapidata/rapidata_client/workflow/__init__.py,sha256=6kfMN7TQVpiQNGjNHg3X7KdnCyYe3K2TIq7ZVEatQcU,476
 rapidata/rapidata_client/workflow/_base_workflow.py,sha256=iuWomsy1zIcIxLzYTBKmAHs4sQ_D4xQkMaNWqf57y3E,1461
 rapidata/rapidata_client/workflow/_classify_workflow.py,sha256=kw-otP_m1TY-mWSjYeMtrD2lLRtsNoocxQ00x50sKo8,2520
@@ -689,7 +689,7 @@ rapidata/service/credential_manager.py,sha256=T3yL4tXVnibRytxjQkOC-ex3kFGQR5KcKU
 rapidata/service/local_file_service.py,sha256=0Q4LdoEtPFKzgXK2oZ1cQ-X7FipakscjGnnBH8dRFRQ,855
 rapidata/service/openapi_service.py,sha256=k3V4eMNcAjBcxEv17lDivK8LV5TEjRTL9B_5KBlhcas,5482
 rapidata/types/__init__.py,sha256=gSGrmWV5gEA6pPfAR5vwSy_DvibO5IjCZDiB7LtlMOQ,6134
-rapidata-2.39.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-2.39.1.dist-info/METADATA,sha256=beVJdZCLq02GH9UroU1KXGZSWZaQaeKPofdqTX59e5U,1406
-rapidata-2.39.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-rapidata-2.39.1.dist-info/RECORD,,
+rapidata-2.40.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-2.40.0.dist-info/METADATA,sha256=sAlNWCA528U0dUY9ytrZR3ezJeC-kc835jDFTuvhn-I,1406
+rapidata-2.40.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+rapidata-2.40.0.dist-info/RECORD,,