Merge pull request #25 from UiPath/fix/add_span_processor

fix: add add_span_processor method to trace manager

Author: cristipufu

pyproject.toml

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

src/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
 
@@ -38,7 +40,7 @@ def add_span_exporter(
         self,
         span_exporter: SpanExporter,
         batch: bool = True,
-    ) -> "UiPathTraceManager":
+    ) -> UiPathTraceManager:
         """Add a span processor to the tracer provider."""
         span_processor: SpanProcessor
         if batch:
@@ -49,6 +51,15 @@ def add_span_exporter(
         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
+
     def get_execution_spans(
         self,
         execution_id: str,