readable span

Author: ionmincu

src/uipath/core/tracing/manager.py

@@ -4,6 +4,7 @@
 from typing import Any, Callable, Dict, List, Optional
 
 from opentelemetry import context, trace
+from opentelemetry.sdk.trace import ReadableSpan
 from opentelemetry.trace import Span, set_span_in_context
 
 logger = logging.getLogger(__name__)
@@ -18,7 +19,7 @@ def __init__(self):
         self._spans: Dict[int, Span] = {}  # span_id -> span
         self._parent_map: Dict[int, Optional[int]] = {}  # span_id -> parent_id
 
-    def register_span(self, span: Span) -> None:
+    def register_span(self, span: ReadableSpan) -> None:
         """Register a span and its parent relationship."""
         span_id = span.get_span_context().span_id
         parent_id = span.parent.span_id if span.parent else None
@@ -198,9 +199,9 @@ def get_parent_context() -> context.Context:
 
     @staticmethod
     def _get_bottom_most_span(
-        current_span: Span,
-        external_span: Span,
-    ) -> Span:
+        current_span: ReadableSpan,
+        external_span: ReadableSpan,
+    ) -> ReadableSpan:
         """Determine which span is deeper in the ancestor tree.
 
         Args:
@@ -257,21 +258,28 @@ def _get_bottom_most_span(
             return external_span
 
     @staticmethod
-    def get_external_current_span() -> Optional[Span]:
+    def get_external_current_span() -> Optional[ReadableSpan]:
         """Get the current span from the external provider, if any."""
         if UiPathTracingManager._current_span_provider is not None:
             try:
-                return UiPathTracingManager._current_span_provider()
+                result = UiPathTracingManager._current_span_provider()
+                return (
+                    result
+                    if result is None or isinstance(result, ReadableSpan)
+                    else None
+                )
             except Exception as e:
                 logger.warning(f"Error getting current span from provider: {e}")
         return None
 
     @staticmethod
-    def get_ancestor_spans() -> List[Span]:
+    def get_ancestor_spans() -> List[ReadableSpan]:
         """Get the ancestor spans from the registered provider, if any."""
         if UiPathTracingManager._current_span_ancestors_provider is not None:
             try:
-                return UiPathTracingManager._current_span_ancestors_provider()
+                result = UiPathTracingManager._current_span_ancestors_provider()
+
+                return [span for span in result if isinstance(span, ReadableSpan)]
             except Exception as e:
                 logger.warning(f"Error getting ancestor spans from provider: {e}")
         return []