.

Author: sentrivana

sentry_sdk/integrations/opentelemetry/sampler.py

@@ -216,8 +216,8 @@ def should_sample(
             # it in PropagationContext
             sample_rand = parent_sample_rand
         else:
-            # There is no sample_rand yet, generate a new one
-            sample_rand = _generate_sample_rand(str(trace_id), (0, 1))
+            # We are the head SDK and we need to generate a new sample_rand
+            sample_rand = cast(Decimal, _generate_sample_rand(str(trace_id), (0, 1)))
 
         # Explicit sampled value provided at start_span
         custom_sampled = cast(

sentry_sdk/tracing_utils.py

@@ -764,7 +764,7 @@ def _generate_sample_rand(
     trace_id,  # type: Optional[str]
     interval=(0.0, 1.0),  # type: tuple[float, float]
 ):
-    # type: (...) -> decimal.Decimal
+    # type: (...) -> Optional[decimal.Decimal]
     """Generate a sample_rand value from a trace ID.
 
     The generated value will be pseudorandomly chosen from the provided
@@ -802,7 +802,7 @@ def _sample_rand_range(parent_sampled, sample_rate):
 
 
 def _round_sample_rand(sample_rand):
-    # type: (Decimal) -> Optional[Decimal]
+    # type: (Union[Decimal, float, str]) -> Optional[Decimal]
     # Round down to exactly six decimal-digit precision.
     # Setting the context is needed to avoid an InvalidOperation exception
     # in case the user has changed the default precision.