uipath-core 0.1.9__py3-none-any.whl → 0.2.0__py3-none-any.whl

uipath/core/tracing/__init__.py

@@ -7,9 +7,11 @@ with OpenTelemetry tracing, including custom processors for UiPath execution tra
 from uipath.core.tracing.decorators import traced
 from uipath.core.tracing.span_utils import UiPathSpanUtils
 from uipath.core.tracing.trace_manager import UiPathTraceManager
+from uipath.core.tracing.types import UiPathTraceSettings
 
 __all__ = [
     "traced",
     "UiPathSpanUtils",
     "UiPathTraceManager",
+    "UiPathTraceSettings",
 ]

uipath/core/tracing/processors.py

@@ -1,22 +1,27 @@
 """Custom span processors for UiPath execution tracing."""
 
-from typing import Optional, cast
+from typing import cast
 
 from opentelemetry import context as context_api
 from opentelemetry import trace
-from opentelemetry.sdk.trace import Span
+from opentelemetry.sdk.trace import ReadableSpan, Span, SpanProcessor
 from opentelemetry.sdk.trace.export import (
     BatchSpanProcessor,
     SimpleSpanProcessor,
+    SpanExporter,
 )
 
+from uipath.core.tracing.types import UiPathTraceSettings
+
 
 class UiPathExecutionTraceProcessorMixin:
-    def on_start(
-        self, span: Span, parent_context: Optional[context_api.Context] = None
-    ):
+    """Mixin that propagates execution.id and optionally filters spans."""
+
+    _settings: UiPathTraceSettings | None = None
+
+    def on_start(self, span: Span, parent_context: context_api.Context | None = None):
         """Called when a span is started."""
-        parent_span: Optional[Span]
+        parent_span: Span | None
         if parent_context:
             parent_span = cast(Span, trace.get_current_span(parent_context))
         else:
@@ -27,17 +32,42 @@ class UiPathExecutionTraceProcessorMixin:
             if execution_id:
                 span.set_attribute("execution.id", execution_id)
 
+    def on_end(self, span: ReadableSpan):
+        """Called when a span ends. Filters before delegating to parent."""
+        span_filter = self._settings.span_filter if self._settings else None
+        if span_filter is None or span_filter(span):
+            parent = cast(SpanProcessor, super())
+            parent.on_end(span)
+
 
 class UiPathExecutionBatchTraceProcessor(
     UiPathExecutionTraceProcessorMixin, BatchSpanProcessor
 ):
-    """Batch span processor that propagates execution.id."""
+    """Batch span processor that propagates execution.id and optionally filters."""
+
+    def __init__(
+        self,
+        span_exporter: SpanExporter,
+        settings: UiPathTraceSettings | None = None,
+    ):
+        """Initialize the batch trace processor."""
+        super().__init__(span_exporter)
+        self._settings = settings
 
 
 class UiPathExecutionSimpleTraceProcessor(
     UiPathExecutionTraceProcessorMixin, SimpleSpanProcessor
 ):
-    """Simple span processor that propagates execution.id."""
+    """Simple span processor that propagates execution.id and optionally filters."""
+
+    def __init__(
+        self,
+        span_exporter: SpanExporter,
+        settings: UiPathTraceSettings | None = None,
+    ):
+        """Initialize the simple trace processor."""
+        super().__init__(span_exporter)
+        self._settings = settings
 
 
 __all__ = [

uipath/core/tracing/trace_manager.py

@@ -1,5 +1,7 @@
 """Tracing manager for handling tracer implementations and function registry."""
 
+from __future__ import annotations
+
 import contextlib
 from typing import Any, Generator, Optional
 
@@ -13,6 +15,7 @@ from uipath.core.tracing.processors import (
     UiPathExecutionBatchTraceProcessor,
     UiPathExecutionSimpleTraceProcessor,
 )
+from uipath.core.tracing.types import UiPathTraceSettings
 
 
 class UiPathTraceManager:
@@ -38,13 +41,31 @@ class UiPathTraceManager:
         self,
         span_exporter: SpanExporter,
         batch: bool = True,
-    ) -> "UiPathTraceManager":
-        """Add a span processor to the tracer provider."""
+        settings: UiPathTraceSettings | None = None,
+    ) -> UiPathTraceManager:
+        """Add a span exporter to the tracer provider.
+
+        Args:
+            span_exporter: The exporter to add.
+            batch: Whether to use batch processing (default: True).
+            settings: Optional trace settings for filtering, etc.
+        """
         span_processor: SpanProcessor
         if batch:
-            span_processor = UiPathExecutionBatchTraceProcessor(span_exporter)
+            span_processor = UiPathExecutionBatchTraceProcessor(span_exporter, settings)
         else:
-            span_processor = UiPathExecutionSimpleTraceProcessor(span_exporter)
+            span_processor = UiPathExecutionSimpleTraceProcessor(
+                span_exporter, settings
+            )
+        self.tracer_span_processors.append(span_processor)
+        self.tracer_provider.add_span_processor(span_processor)
+        return self
+
+    def add_span_processor(
+        self,
+        span_processor: SpanProcessor,
+    ) -> UiPathTraceManager:
+        """Add a span processor to the tracer provider."""
         self.tracer_span_processors.append(span_processor)
         self.tracer_provider.add_span_processor(span_processor)
         return self

uipath/core/tracing/types.py

@@ -0,0 +1,21 @@
+"""Tracing types for UiPath SDK."""
+
+from typing import Callable
+
+from opentelemetry.sdk.trace import ReadableSpan
+from pydantic import BaseModel, Field
+
+
+class UiPathTraceSettings(BaseModel):
+    """Trace settings for UiPath SDK."""
+
+    model_config = {"arbitrary_types_allowed": True}  # Needed for Callable
+
+    span_filter: Callable[[ReadableSpan], bool] | None = Field(
+        default=None,
+        description=(
+            "Optional filter to decide whether a span should be exported. "
+            "Called when a span ends with a ReadableSpan argument. "
+            "Return True to export, False to skip."
+        ),
+    )

{uipath_core-0.1.9.dist-info → uipath_core-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: uipath-core
-Version: 0.1.9
+Version: 0.2.0
 Summary: UiPath Core abstractions
 Project-URL: Homepage, https://uipath.com
 Project-URL: Repository, https://github.com/UiPath/uipath-core-python
@@ -13,8 +13,8 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Build Tools
 Requires-Python: >=3.11
-Requires-Dist: opentelemetry-instrumentation<1.0.0,>=0.60b0
-Requires-Dist: opentelemetry-sdk<2.0.0,>=1.39.0
+Requires-Dist: opentelemetry-instrumentation<1.0.0,>=0.60b1
+Requires-Dist: opentelemetry-sdk<2.0.0,>=1.39.1
 Requires-Dist: pydantic<3.0.0,>=2.12.5
 Description-Content-Type: text/markdown
 

{uipath_core-0.1.9.dist-info → uipath_core-0.2.0.dist-info}/RECORD

@@ -18,14 +18,15 @@ uipath/core/guardrails/__init__.py,sha256=hUCmD4y5te2iy01YnJlBuf2RWvqxmsNzoyOamX
 uipath/core/guardrails/_deterministic_guardrails_service.py,sha256=61ROXYmX3rjBfFUp7E83fm4Lk7h1DlJeukpZm7uzVqQ,6355
 uipath/core/guardrails/_evaluators.py,sha256=10tIRUufxoy9MkZPb-ytjsCSCIfIqQSOYXyEl4G7PSw,15649
 uipath/core/guardrails/guardrails.py,sha256=sTxsNHilEX908aF-WtQWVOQ4dCCwMgAepDebH35ngvo,5430
-uipath/core/tracing/__init__.py,sha256=1XNLYZ4J76XkRrizGO486mS6yxzVXUbrldpvxTyJe3E,483
+uipath/core/tracing/__init__.py,sha256=Y0wfRSGCVcDoFYTmTkVQElu5JE91h4dTNkKQnd86dBM,568
 uipath/core/tracing/_utils.py,sha256=FiCFGOFa4czruhlSF87Q5Q4jX9KKPHZiw8k14K7W5v4,6636
 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/processors.py,sha256=XlMKA_AWwrtC-0ytnAHxl4P9kXlQdsjcn8zwnSpZKts,2500
 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.9.dist-info/METADATA,sha256=wLk7EKaDR01IqA7RvbhRzw13dw2DRX6twknAm793_OY,938
-uipath_core-0.1.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-uipath_core-0.1.9.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
-uipath_core-0.1.9.dist-info/RECORD,,
+uipath/core/tracing/trace_manager.py,sha256=vmPup-C2pSgwpSdcCzAjzP_nTtRWZgqlxnNT50ygOfk,3843
+uipath/core/tracing/types.py,sha256=8A8fFuWqMGk0SzIrFbMajmY6LA3w57h-YA602OctCrI,634
+uipath_core-0.2.0.dist-info/METADATA,sha256=QaPAikNEVglQuX5U4GVytJ426EAsHleGxBRJ-yzOTm4,938
+uipath_core-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+uipath_core-0.2.0.dist-info/licenses/LICENSE,sha256=-KBavWXepyDjimmzH5fVAsi-6jNVpIKFc2kZs0Ri4ng,1058
+uipath_core-0.2.0.dist-info/RECORD,,