feat(platform): add W3C trace context headers to LLM Gateway requests#1612
Merged
Conversation
ctiliescuuipath
force-pushed
the
feat/trace-context-headers-llm-gateway
branch
from
594f8f1 to
24cc085
Compare
github-actions
Bot
added
test:uipath-langchain
🚨 Heads up:
|
ctiliescuuipath
force-pushed
the
feat/trace-context-headers-llm-gateway
branch
from
08e43ed to
dd55382
Compare
radu-mocanu
reviewed
May 5, 2026
| span_id = format(ctx.span_id, "016x") | ||
| headers["x-uipath-traceparent-id"] = f"00-{trace_id}-{span_id}" | ||
|
|
||
| baggage_parts: list[str] = ["source=agents"] |
Collaborator
There was a problem hiding this comment.
this part should be moved to uipath-langchain and the other integration repos.
uipath-platform should remain framework/runtime agnostic (since function projects will also use it)
Suggested change
| baggage_parts: list[str] = ["source=agents"] | |
| baggage_parts: list[str] |
radu-mocanu
reviewed
May 5, 2026
| baggage_parts: list[str] = ["source=agents"] | ||
| if folder_key := os.getenv(ENV_FOLDER_KEY): | ||
| baggage_parts.append(f"folderKey={folder_key}") | ||
| if agent_id := os.getenv(ENV_UIPATH_PROCESS_UUID): |
Collaborator
There was a problem hiding this comment.
Suggested change
| if agent_id := os.getenv(ENV_UIPATH_PROCESS_UUID): | |
| if agent_id := UiPathConfig.process_uuid: |
let s use UiPathConfig here instead of using direct env vars accessors. we can even cache the data at that layer.
please note that some properties may not be available at the moment in UiPathConfig, we should add those
Add x-uipath-traceparent-id and x-uipath-tracebaggage headers to all LLM Gateway HTTP calls, gated behind the EnableTraceContextHeaders feature flag. Headers follow the W3C Trace Context spec and include source, folderKey, agentId, and processKey in the baggage. Bump uipath-platform to 0.1.41. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
ctiliescuuipath
force-pushed
the
feat/trace-context-headers-llm-gateway
branch
from
dd55382 to
4c3b5af
Compare
…baggage API Replace direct os.getenv() calls with UiPathConfig properties and remove hardcoded source=agents baggage to keep uipath-platform framework-agnostic. Add extra_baggage parameter so integration repos can inject their own entries. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
ctiliescuuipath
force-pushed
the
feat/trace-context-headers-llm-gateway
branch
from
4c3b5af to
090a043
Compare
radu-mocanu
approved these changes
May 5, 2026
ionmincu
reviewed
May 5, 2026
| return {} | ||
|
|
||
| headers: dict[str, str] = {} | ||
| span = trace.get_current_span() |
Contributor
There was a problem hiding this comment.
you should use:
from uipath.core.tracing.span_utils import UiPathSpanUtils
llmops_span = UiPathSpanUtils.get_external_current_span()
span = llmops_span or trace.get_current_span()
similar to https://github.com/UiPath/uipath-python/pull/1547/changes
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
x-uipath-traceparent-idandx-uipath-tracebaggageheaders to all LLM Gateway HTTP callsEnableTraceContextHeadersfeature flag (env var:UIPATH_FEATURE_EnableTraceContextHeaders=true)source=agents,folderKey,agentId,processKeybuild_trace_context_headers()defined once inllm_trace_context.py, consumed by bothUiPathLlmChatService/UiPathOpenAIServiceanduipath-langchainlegacy clientsuipath-platformversion to0.1.41Test plan
test_llm_trace_context.py(9 tests covering flag on/off, traceparent format, baggage content)UIPATH_FEATURE_EnableTraceContextHeaders=true🤖 Generated with Claude Code