uipath-core 0.1.6__tar.gz → 0.1.8__tar.gz

{uipath_core-0.1.6 → uipath_core-0.1.8}/.github/workflows/publish-dev.yml

@@ -60,9 +60,14 @@ jobs:
 
           Write-Output "Package version set to $DEV_VERSION"
 
+          $startMarker = "<!-- DEV_PACKAGE_START -->"
+          $endMarker = "<!-- DEV_PACKAGE_END -->"
+
           $dependencyMessage = @"
+          $startMarker
           ## Development Package
 
+          - Use ``uipath pack --nolock`` to get the latest dev build from this PR (requires version range).
           - Add this package as a dependency in your pyproject.toml:
 
           ``````toml
@@ -83,7 +88,13 @@ jobs:
 
           [tool.uv.sources]
           $PROJECT_NAME = { index = "testpypi" }
+
+          [tool.uv]
+          override-dependencies = [
+              "$PROJECT_NAME>=$MIN_VERSION,<$MAX_VERSION",
+          ]
           ``````
+          $endMarker
           "@
 
           # Get the owner and repo from the GitHub repository
@@ -101,10 +112,11 @@ jobs:
           $pr = Invoke-RestMethod -Uri $prUri -Method Get -Headers $headers
           $currentBody = $pr.body
 
-          # Check if there's already a development package section
-          if ($currentBody -match '## Development Package') {
-            # Replace the existing section with the new dependency message
-            $newBody = $currentBody -replace '## Development Package(\r?\n|.)*?(?=##|$)', $dependencyMessage
+          # Check if markers already exist in the PR description
+          $markerPattern = "(?s)$([regex]::Escape($startMarker)).*?$([regex]::Escape($endMarker))"
+          if ($currentBody -match $markerPattern) {
+            # Replace everything between markers (including markers)
+            $newBody = $currentBody -replace $markerPattern, $dependencyMessage
           } else {
             # Append the dependency message to the end of the description
             $newBody = if ($currentBody) { "$currentBody`n`n$dependencyMessage" } else { $dependencyMessage }

{uipath_core-0.1.6 → uipath_core-0.1.8}/PKG-INFO

@@ -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 → uipath_core-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-core"
-version = "0.1.6"
+version = "0.1.8"
 description = "UiPath Core abstractions"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

{uipath_core-0.1.6 → uipath_core-0.1.8}/src/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-0.1.6 → uipath_core-0.1.8}/src/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"]