fix(platform): use LLMOps span ID in trace header (#1547)

Co-authored-by: Clement Fauchere <clement.fauchere@uipath.com>
Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: 3 people

packages/uipath-platform/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-platform"
-version = "0.1.21"
+version = "0.1.22"
 description = "HTTP client library for programmatic access to UiPath Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

packages/uipath-platform/src/uipath/platform/common/_base_service.py

@@ -67,8 +67,15 @@ def _get_caller_component() -> str:
 
 
 def _inject_trace_context(headers: dict[str, str]) -> None:
-    """Inject UiPath trace context header from the active OTEL span."""
-    span = trace.get_current_span()
+    """Inject UiPath trace context header from the active OTEL span.
+
+    Prefers the LLMOps tool span (from ContextVar) over the raw OTEL span,
+    so the header carries the span ID visible in the LLMOps trace.
+    """
+    from uipath.core.tracing.span_utils import UiPathSpanUtils
+
+    llmops_span = UiPathSpanUtils.get_external_current_span()
+    span = llmops_span or trace.get_current_span()
     ctx = span.get_span_context()
     if ctx.trace_id and ctx.span_id:
         headers[_TRACE_PARENT_HEADER] = (