feat(tracing): add current span ancestors provider

Author: ionmincu

src/uipath/core/tracing/manager.py

@@ -1,7 +1,7 @@
 """Tracing manager for handling tracer implementations and function registry."""
 
 import logging
-from typing import Any, Callable, Optional
+from typing import Any, Callable, List, Optional
 
 from opentelemetry import context, trace
 from opentelemetry.trace import set_span_in_context
@@ -13,6 +13,7 @@ class UiPathTracingManager:
     """Static utility class to manage tracing implementations and decorated functions."""
 
     _current_span_provider: Optional[Callable[[], Any]] = None
+    _current_span_ancestors_provider: Optional[Callable[[], List[Any]]] = None
 
     @classmethod
     def register_current_span_provider(
@@ -55,5 +56,23 @@ def get_parent_context():
 
         return ctx
 
+    @classmethod
+    def register_current_span_ancestors_provider(
+        cls, current_span_ancestors_provider: Optional[Callable[[], List[Any]]]
+    ):
+        """Register a custom current span ancestors provider function.
+
+        Args:
+            current_span_ancestors_provider: A function that returns a list of ancestor spans
+                                           from an external tracing framework. If None, no custom
+                                           span ancestor information will be used.
+        """
+        cls._current_span_ancestors_provider = current_span_ancestors_provider
+
+    @classmethod
+    def get_current_span_ancestors_provider(cls):
+        """Get the currently set custom span ancestors provider."""
+        return cls._current_span_ancestors_provider
+
 
 __all__ = ["UiPathTracingManager"]