feat: propagate reference hierarchy context from LangGraph runtime#901
Open
jepadil23 wants to merge 13 commits into
Open
feat: propagate reference hierarchy context from LangGraph runtime#901jepadil23 wants to merge 13 commits into
jepadil23 wants to merge 13 commits into
Conversation
There was a problem hiding this comment.
Pull request overview
This PR adds propagation of an ambient ReferenceContext for UiPathLangGraphRuntime runs so that downstream tracing can attribute spans to a langgraph service entry layered on top of any parent reference hierarchy.
Changes:
- Add
UiPathLangGraphRuntime._push_reference_context()and wrapexecute()/stream()with push/reset intry/finally. - Introduce a new test module covering context stacking behavior and ensuring the context is reset after
execute()/stream()(including error paths).
Reviewed changes
Copilot reviewed 2 out of 2 changed files in this pull request and generated 9 comments.
| File | Description |
|---|---|
src/uipath_langchain/runtime/runtime.py |
Pushes a langgraph entry into ReferenceContextAccessor during execute()/stream() and resets it in finally. |
tests/runtime/test_reference_context_wiring.py |
Adds tests for _push_reference_context() and for context reset after runtime execution/streaming. |
danflorescu96
approved these changes
Jun 10, 2026
Contributor
|
Please make sure to address the review comments by Copilot and address these CI requirements |
Contributor
|
you'll need to bump up version here |
Author
… runtime UiPathLangGraphRuntime._push_reference_context() reads any parent ReferenceContext set by upstream (e.g. agents-python start_agent_run), appends a "langgraph" entry from UIPATH_AGENT_ID / UIPATH_PROCESS_VERSION, and activates it via ReferenceContextAccessor. Both execute() and stream() call this at entry and reset in finally, so all child spans emitted during the run carry the full hierarchy in their Context.referenceHierarchy wire payload. Co-Authored-By: Claude Sonnet 4.6 (1M context) <noreply@anthropic.com>
jepadil23
force-pushed
the
feat/reference-hierarchy-context
branch
from
b08070a to
65fdb0c
Compare
jepadil23
force-pushed
the
feat/reference-hierarchy-context
branch
from
7015ffe to
7e1f7d8
Compare
|
danflorescu96
approved these changes
Jul 6, 2026
saksharthakkar
approved these changes
Jul 6, 2026
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
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
UiPathLangGraphRuntime._push_reference_context()pushes a"langgraph"entry intoReferenceContextAccessorduringexecute()andstream(), resets infinallymaestro → agent → langgraph) in theirContext.referenceHierarchywire payloadDependencies
Requires
uipath-pythonPR #1708 (ReferenceContext/ReferenceContextAccessortypes).Test plan
tests/runtime/test_reference_context_wiring.py— verifiesreferenceHierarchyis present in LangGraph runtime spans with correctserviceType="langgraph"entry🤖 Generated with Claude Code