uipath-core 0.1.6__py3-none-any.whl → 0.1.8__py3-none-any.whl

uipath/core/tracing/decorators.py

@@ -6,8 +6,10 @@ import random
 from functools import wraps
 from typing import Any, Callable, Optional
 
+from opentelemetry import context as context_api
 from opentelemetry import trace
-from opentelemetry.trace import NonRecordingSpan, SpanContext, TraceFlags
+from opentelemetry.context import _SUPPRESS_INSTRUMENTATION_KEY
+from opentelemetry.trace import SpanContext, TraceFlags
 from opentelemetry.trace.status import StatusCode
 
 from uipath.core.tracing._utils import (
@@ -15,7 +17,11 @@ from uipath.core.tracing._utils import (
     set_span_input_attributes,
     set_span_output_attributes,
 )
-from uipath.core.tracing.span_utils import UiPathSpanUtils
+from uipath.core.tracing.span_utils import (
+    ParentedNonRecordingSpan,
+    UiPathSpanUtils,
+    _span_registry,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -48,7 +54,10 @@ def _opentelemetry_traced(
         trace_name = name or func.__name__
 
         def get_span():
+            ctx = UiPathSpanUtils.get_parent_context()
             if not recording:
+                parent_context = trace.get_current_span(ctx).get_span_context()
+
                 # Create a valid but non-sampled trace context
                 # Generate a valid trace ID (not INVALID)
                 trace_id = random.getrandbits(128)
@@ -60,24 +69,30 @@ def _opentelemetry_traced(
                     is_remote=False,
                     trace_flags=TraceFlags(0x00),  # NOT sampled
                 )
-                non_recording = NonRecordingSpan(non_sampled_context)
+                non_recording = ParentedNonRecordingSpan(
+                    non_sampled_context, parent=parent_context
+                )
 
                 # Make it active so children see it
                 span_cm = trace.use_span(non_recording)
                 span_cm.__enter__()
-                return span_cm, non_recording
 
-            # Normal recording span
-            ctx = UiPathSpanUtils.get_parent_context()
-            span_cm = trace.get_tracer(__name__).start_as_current_span(
-                trace_name, context=ctx
-            )
-            span = span_cm.__enter__()
-            return span_cm, span
+                _span_registry.register_span(non_recording)
+
+                return span_cm, non_recording
+            else:
+                # Normal recording span
+                span_cm = trace.get_tracer(__name__).start_as_current_span(
+                    trace_name, context=ctx
+                )
+                span = span_cm.__enter__()
+                return span_cm, span
 
         # --------- Sync wrapper ---------
         @wraps(func)
         def sync_wrapper(*args: Any, **kwargs: Any) -> Any:
+            if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY):
+                return func(*args, **kwargs)
             span_cm, span = get_span()
             try:
                 # Set input attributes BEFORE execution
@@ -113,6 +128,8 @@ def _opentelemetry_traced(
         # --------- Async wrapper ---------
         @wraps(func)
         async def async_wrapper(*args: Any, **kwargs: Any) -> Any:
+            if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY):
+                return await func(*args, **kwargs)
             span_cm, span = get_span()
             try:
                 # Set input attributes BEFORE execution
@@ -148,6 +165,10 @@ def _opentelemetry_traced(
         # --------- Generator wrapper ---------
         @wraps(func)
         def generator_wrapper(*args: Any, **kwargs: Any) -> Any:
+            if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY):
+                for item in func(*args, **kwargs):
+                    yield item
+                return
             span_cm, span = get_span()
             try:
                 # Set input attributes BEFORE execution
@@ -186,6 +207,10 @@ def _opentelemetry_traced(
         # --------- Async generator wrapper ---------
         @wraps(func)
         async def async_generator_wrapper(*args: Any, **kwargs: Any) -> Any:
+            if context_api.get_value(_SUPPRESS_INSTRUMENTATION_KEY):
+                async for item in func(*args, **kwargs):
+                    yield item
+                return
             span_cm, span = get_span()
             try:
                 # Set input attributes BEFORE execution

uipath/core/tracing/span_utils.py

@@ -4,11 +4,33 @@ import logging
 from typing import Callable, Optional
 
 from opentelemetry import context, trace
-from opentelemetry.trace import Span, set_span_in_context
+from opentelemetry.trace import NonRecordingSpan, Span, set_span_in_context
 
 logger = logging.getLogger(__name__)
 
 
+class ParentedNonRecordingSpan(NonRecordingSpan):
+    """Non-recording span with explicit parent tracking.
+
+    Extends NonRecordingSpan to include a parent attribute, allowing the SpanRegistry
+    to properly track parent-child relationships for non-recording spans.
+    This is necessary because NonRecordingSpan instances created directly don't have
+    their parent automatically set like normal recording spans do.
+    """
+
+    def __init__(
+        self, context: trace.SpanContext, parent: Optional[trace.SpanContext] = None
+    ):
+        """Initialize a parented non-recording span.
+
+        Args:
+            context: The SpanContext for this span
+            parent: Optional parent SpanContext
+        """
+        super().__init__(context)
+        self.parent = parent
+
+
 class SpanRegistry:
     """Registry to track all spans and their parent relationships."""
 
@@ -309,4 +331,4 @@ class UiPathSpanUtils:
         return UiPathSpanUtils._current_span_ancestors_provider
 
 
-__all__ = ["UiPathSpanUtils"]
+__all__ = ["ParentedNonRecordingSpan", "UiPathSpanUtils"]

{uipath_core-0.1.6.dist-info → uipath_core-0.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-core
-Version: 0.1.6
+Version: 0.1.8
 Summary: UiPath Core abstractions
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-core-python

{uipath_core-0.1.6.dist-info → uipath_core-0.1.8.dist-info}/RECORD

@@ -20,12 +20,12 @@ uipath/core/guardrails/_evaluators.py,sha256=ovmVm-8iB8Pm9arjG7mHM9-GIRkrG3V6oHR
 uipath/core/guardrails/guardrails.py,sha256=DraeFkoDKVDnc0EKdFYYP29lQu7U2hneTsj1dxveHU4,4935
 uipath/core/tracing/__init__.py,sha256=1XNLYZ4J76XkRrizGO486mS6yxzVXUbrldpvxTyJe3E,483
 uipath/core/tracing/_utils.py,sha256=FiCFGOFa4czruhlSF87Q5Q4jX9KKPHZiw8k14K7W5v4,6636
-uipath/core/tracing/decorators.py,sha256=ag_MFwZ0TywrhbpLKqQwF1guvRA9sYiItxao5LN9_Iw,10942
+uipath/core/tracing/decorators.py,sha256=JDNULkUu-Ufg8pJslG4i6Q2pmqaGNDS8NFRPgFs29Dw,11937
 uipath/core/tracing/exporters.py,sha256=FClouEEQfk3F8J7G_NFoarDJM3R0-gA5jUxA5xRHx5s,1562
 uipath/core/tracing/processors.py,sha256=R_652rtjPmfpUtaXoIcmfZrRZylVXFRNwjOmJUUxOQw,1408
-uipath/core/tracing/span_utils.py,sha256=WYBrd6ZbawAs7r1Js-Zvo9_8GzkD9LhHNOls00bK_xI,12235
+uipath/core/tracing/span_utils.py,sha256=LZXNdnI0-fhKe49CLPsvMJIfh9zdzk8rK4g4YN5RfDU,13064
 uipath/core/tracing/trace_manager.py,sha256=51rscJcepkTK4bWoCZdE-DFc9wt2F-aSuFBaSXmkHl0,3130
-uipath_core-0.1.6.dist-info/METADATA,sha256=R2wS5I85yHd-uTvGgepwbhKiZA7cLb389XJ5R57ED7I,938
-uipath_core-0.1.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-uipath_core-0.1.6.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath_core-0.1.6.dist-info/RECORD,,
+uipath_core-0.1.8.dist-info/METADATA,sha256=WaZcOoWscnXrgIhPHw8YATSu8xZ3x64SmndWI5tndG4,938
+uipath_core-0.1.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+uipath_core-0.1.8.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath_core-0.1.8.dist-info/RECORD,,