rapidata 2.42.5__py3-none-any.whl → 2.42.6__py3-none-any.whl

rapidata/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.42.5"
+__version__ = "2.42.6"
 
 from .rapidata_client import (
     RapidataClient,

rapidata/rapidata_client/config/tracer.py

@@ -1,5 +1,4 @@
 from typing import Protocol, runtime_checkable, Any
-from opentelemetry import trace
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor
@@ -15,6 +14,7 @@ class TracerProtocol(Protocol):
 
     def start_span(self, name: str, *args, **kwargs) -> Any: ...
     def start_as_current_span(self, name: str, *args, **kwargs) -> Any: ...
+    def set_session_id(self, session_id: str) -> None: ...
 
 
 class NoOpSpan:
@@ -52,11 +52,31 @@ class NoOpTracer:
     def start_as_current_span(self, name: str, *args, **kwargs) -> NoOpSpan:
         return NoOpSpan()
 
+    def set_session_id(self, session_id: str) -> None:
+        pass
+
     def __getattr__(self, name: str) -> Any:
         """Delegate to no-op behavior."""
         return lambda *args, **kwargs: NoOpSpan()
 
 
+class SpanContextManagerWrapper:
+    """Wrapper for span context managers to add session_id on enter."""
+
+    def __init__(self, context_manager: Any, session_id: str | None):
+        self._context_manager = context_manager
+        self.session_id = session_id
+
+    def __enter__(self):
+        span = self._context_manager.__enter__()
+        if self.session_id and hasattr(span, "set_attribute"):
+            span.set_attribute("SDK.session.id", self.session_id)
+        return span
+
+    def __exit__(self, *args):
+        return self._context_manager.__exit__(*args)
+
+
 class RapidataTracer:
     """Tracer implementation that updates when the configuration changes."""
 
@@ -67,6 +87,7 @@ class RapidataTracer:
         self._real_tracer = None
         self._no_op_tracer = NoOpTracer()
         self._enabled = True  # Default to enabled
+        self.session_id: str | None = None
 
         # Register this tracer to receive configuration updates
         register_config_handler(self._handle_config_update)
@@ -106,18 +127,31 @@ class RapidataTracer:
                 logger.warning(f"Failed to initialize tracing: {e}")
                 self._enabled = False
 
+    def _add_session_id_to_span(self, span: Any) -> Any:
+        """Add session_id attribute to a span if session_id is set."""
+        if self.session_id and hasattr(span, "set_attribute"):
+            span.set_attribute("SDK.session.id", self.session_id)
+        return span
+
     def start_span(self, name: str, *args, **kwargs) -> Any:
         """Start a span, or return a no-op span if tracing is disabled."""
         if self._enabled and self._real_tracer:
-            return self._real_tracer.start_span(name, *args, **kwargs)
+            span = self._real_tracer.start_span(name, *args, **kwargs)
+            return self._add_session_id_to_span(span)
         return self._no_op_tracer.start_span(name, *args, **kwargs)
 
     def start_as_current_span(self, name: str, *args, **kwargs) -> Any:
         """Start a span as current, or return a no-op span if tracing is disabled."""
         if self._enabled and self._real_tracer:
-            return self._real_tracer.start_as_current_span(name, *args, **kwargs)
+            context_manager = self._real_tracer.start_as_current_span(
+                name, *args, **kwargs
+            )
+            return SpanContextManagerWrapper(context_manager, self.session_id)
         return self._no_op_tracer.start_as_current_span(name, *args, **kwargs)
 
+    def set_session_id(self, session_id: str) -> None:
+        self.session_id = session_id
+
     def __getattr__(self, name: str) -> Any:
         """Delegate attribute access to the appropriate tracer."""
         if self._enabled and self._real_tracer:

rapidata/rapidata_client/order/rapidata_order_manager.py

@@ -321,8 +321,10 @@ class RapidataOrderManager:
             if any(not isinstance(datapoint, list) for datapoint in datapoints):
                 raise ValueError("Each datapoint must be a list of 2 paths/texts")
 
-            if any(len(datapoint) != 2 for datapoint in datapoints):
-                raise ValueError("Each datapoint must contain exactly two options")
+            if any(len(set(datapoint)) != 2 for datapoint in datapoints):
+                raise ValueError(
+                    "Each datapoint must contain exactly two unique options"
+                )
 
             if a_b_names is not None and len(a_b_names) != 2:
                 raise ValueError(
@@ -392,6 +394,9 @@ class RapidataOrderManager:
             if len(datapoints) < 2:
                 raise ValueError("At least two datapoints are required")
 
+            if len(set(datapoints)) != len(datapoints):
+                raise ValueError("Datapoints must be unique")
+
             return self._create_general_order(
                 name=name,
                 workflow=RankingWorkflow(

rapidata/rapidata_client/rapidata_client.py

@@ -3,7 +3,8 @@ from typing import Any
 import requests
 from packaging import version
 from rapidata import __version__
-
+import uuid
+import random
 from rapidata.service.openapi_service import OpenAPIService
 
 from rapidata.rapidata_client.order.rapidata_order_manager import RapidataOrderManager
@@ -56,6 +57,10 @@ class RapidataClient:
             demographic (DemographicManager): The DemographicManager instance.
             mri (RapidataBenchmarkManager): The RapidataBenchmarkManager instance.
         """
+        tracer.set_session_id(
+            uuid.UUID(int=random.Random().getrandbits(128), version=4).hex
+        )
+
         with tracer.start_as_current_span("RapidataClient.__init__"):
             logger.debug("Checking version")
             self._check_version()

rapidata/rapidata_client/validation/rapids/_validation_rapid_uploader.py

@@ -51,6 +51,8 @@ class ValidationRapidUploader:
                 metadata=metadata,
                 payload=self._get_payload(rapid),
                 truth=AddValidationRapidModelTruth(actual_instance=rapid.truth),
+                randomCorrectProbability=rapid.random_correct_probability,
+                explanation=rapid.explanation,
                 featureFlags=(
                     [setting._to_feature_flag() for setting in rapid.settings]
                     if rapid.settings

{rapidata-2.42.5.dist-info → rapidata-2.42.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rapidata
-Version: 2.42.5
+Version: 2.42.6
 Summary: Rapidata package containing the Rapidata Python Client to interact with the Rapidata Web API in an easy way.
 License: Apache-2.0
 License-File: LICENSE

{rapidata-2.42.5.dist-info → rapidata-2.42.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-rapidata/__init__.py,sha256=g_Z2skYSbi9A5QdchDqynk7kwq0xeSxH1ky3iGaz_oA,857
+rapidata/__init__.py,sha256=rsgNfH2hEqJNu89zai9F2RcA8MnUGP9XS6OX5HM3bKc,857
 rapidata/api_client/__init__.py,sha256=Rh2aAoa6nDv09Z7cpTNdrPotXixBcNN_2VevCDD1goc,37144
 rapidata/api_client/api/__init__.py,sha256=CaZJ54mxbAEzAFd0Cjy0mfiUabD-d_7tqVOgjU4RxZI,1749
 rapidata/api_client/api/asset_api.py,sha256=buYMBGvg5_SLkJ7aG07c7_1rT6TXTCqoLfoV8O2nJhM,34442
@@ -616,7 +616,7 @@ rapidata/rapidata_client/config/logging_config.py,sha256=ePCh7X6v0Yzkgq1MhFNsHwL
 rapidata/rapidata_client/config/managed_print.py,sha256=2T6dwgR1EZzFAdOEyPp_BBUsa-qrEuhOXgFhsSQKvRo,185
 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=9P27QybtkjJ0U5ocPkIu58DUQB-MgVksQOMhuwKHKcY,4687
+rapidata/rapidata_client/config/tracer.py,sha256=AkDcW89h5OrT1t0rk1ulidytKS4oJFG1kCLrHj8eBG4,5943
 rapidata/rapidata_client/config/upload_config.py,sha256=hjefl-w9WaCNeCEe6hdnrAQEMjgDy-r1zgUUIFR68wk,473
 rapidata/rapidata_client/datapoints/__init__.py,sha256=pRjH6YQLbIHOumykMgplKxFfhj6g7a2Dh2FPdFNyNsg,222
 rapidata/rapidata_client/datapoints/_asset_uploader.py,sha256=ie9MZfzhxa0ZrRveJ516cgwvA6QIeVIxN3g5ikwLOH0,2605
@@ -658,9 +658,9 @@ rapidata/rapidata_client/order/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5
 rapidata/rapidata_client/order/_rapidata_order_builder.py,sha256=FTJJRCW9KehIlqELOXmKeLhuGcLj9m9ImEU1agOIRa4,16915
 rapidata/rapidata_client/order/dataset/_rapidata_dataset.py,sha256=8EhoRVBxWMbp1mtS56Uogg2dwFC2vySw5hNqTPDTHzM,6345
 rapidata/rapidata_client/order/rapidata_order.py,sha256=w41HUWE6q9L2Stx0ScZdKb7qLjWFIUPMMDOe2fElLvM,14481
-rapidata/rapidata_client/order/rapidata_order_manager.py,sha256=Ya8s40hzigIIpo9WphL67sguuqY_3q-Z4AL8IHmX-ZI,41028
+rapidata/rapidata_client/order/rapidata_order_manager.py,sha256=3GafS6k815iq-s9wgS9RqUvh8OY1Kgu7LxSjLsKry5w,41197
 rapidata/rapidata_client/order/rapidata_results.py,sha256=weL4S14fzug3ZOJbQk9Oj-4tv2jx5aZAMp7VJ-a6Qq4,8437
-rapidata/rapidata_client/rapidata_client.py,sha256=5BJMstjJFmBwkflC_YnzlhoOF3SKP3u2wQ56G_hvB3Q,5862
+rapidata/rapidata_client/rapidata_client.py,sha256=9BgFbkPulmNNtKE7Cug7qcRQNuH-S8S8cof8NXnleuE,6004
 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
@@ -694,7 +694,7 @@ rapidata/rapidata_client/settings/translation_behaviour.py,sha256=uEke2WgzrjzRlS
 rapidata/rapidata_client/validation/__init__.py,sha256=s5wHVtcJkncXSFuL9I0zNwccNOKpWAqxqUjkeohzi2E,24
 rapidata/rapidata_client/validation/rapidata_validation_set.py,sha256=eJlv0XKUC75caJKu6_kpY0EL0ZtmHpt2c3SgFjlmqRU,4666
 rapidata/rapidata_client/validation/rapids/__init__.py,sha256=WU5PPwtTJlte6U90MDakzx4I8Y0laj7siw9teeXj5R0,21
-rapidata/rapidata_client/validation/rapids/_validation_rapid_uploader.py,sha256=ToHzpkRyyWt9_tcHrMPhR5-3GfeG3BAgMtjE9N_2cE4,3950
+rapidata/rapidata_client/validation/rapids/_validation_rapid_uploader.py,sha256=9O5kQNu9lom7x2X09sGfFFecXNO2dU4d2O1L-_n_hAw,4072
 rapidata/rapidata_client/validation/rapids/box.py,sha256=sJbxOpz86MplXmZl6gquzgQzpFIW8fw-F0eyQY4ClDo,1486
 rapidata/rapidata_client/validation/rapids/rapids.py,sha256=E4O4p2br0g9lgB3tOdfEBm07g8o6pCDQgQv7fi6QzF8,999
 rapidata/rapidata_client/validation/rapids/rapids_manager.py,sha256=YqGxnqJ2vYfAywXJDujQEBrYnlf0URdLxQ7hTR1JeOI,16225
@@ -715,7 +715,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=E2zVagI_ri15PK06ITO_VNKYDJ0VZQG1YQ1T6bEIVsY,5566
 rapidata/types/__init__.py,sha256=NLRuTGbTImRv3yIp8we_oqVvCuBd7TDNVlAzkVGs9oo,6056
-rapidata-2.42.5.dist-info/METADATA,sha256=cvSZF842ZjlBbblCSxm7XuDRhwbp1PLQqji1A6Pf9uY,1479
-rapidata-2.42.5.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-rapidata-2.42.5.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-rapidata-2.42.5.dist-info/RECORD,,
+rapidata-2.42.6.dist-info/METADATA,sha256=hGaKKniKHnrpgC3qBq6r1FQbzzk7af7zh9x2-1UY71g,1479
+rapidata-2.42.6.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+rapidata-2.42.6.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+rapidata-2.42.6.dist-info/RECORD,,