Skip to content

feat(langgraph): Support span streaming #6406

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
alexander-alderman-webb merged 17 commits into from
May 27, 2026
+137 −24
Merged

feat(langgraph): Support span streaming #6406

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
a23db4b
feat(langchain): Support span streaming
alexander-alderman-webb May 26, 2026
6ce1144
feat(langgraph): Support span streaming
alexander-alderman-webb May 26, 2026
1d05771
.
alexander-alderman-webb May 26, 2026
0d87458
.
alexander-alderman-webb May 26, 2026
d69069d
ref(langchain): Remove WatchedSpan class
alexander-alderman-webb May 26, 2026
ea62340
.
alexander-alderman-webb May 26, 2026
0acc9c7
merge
alexander-alderman-webb May 26, 2026
41cceeb
rename variable
alexander-alderman-webb May 26, 2026
1931895
Merge branch 'webb/langchain/watched-span' into webb/langchain/span-f…
alexander-alderman-webb May 26, 2026
8e63081
merge
alexander-alderman-webb May 26, 2026
888806b
use should_truncate_gen_ai_inputs in langgraph
alexander-alderman-webb May 26, 2026
d24eda4
merge
alexander-alderman-webb May 26, 2026
2804959
merge master
alexander-alderman-webb May 26, 2026
00282e0
Merge branch 'webb/langchain/span-first' into webb/langgraph/span-first
alexander-alderman-webb May 26, 2026
41ed46b
merge master
alexander-alderman-webb May 27, 2026
c2353ff
unparametrize one test
alexander-alderman-webb May 27, 2026
ea62a46
empty commit
alexander-alderman-webb May 27, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
69 changes: 60 additions & 9 deletions sentry_sdk/integrations/langgraph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -112,9 +112,11 @@ def _parse_langgraph_messages(state: "Any") -> "Optional[List[Any]]":
def _wrap_state_graph_compile(f: "Callable[..., Any]") -> "Callable[..., Any]":
@wraps(f)
def new_compile(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
integration = sentry_sdk.get_client().get_integration(LanggraphIntegration)
if integration is None:
client = sentry_sdk.get_client()
integration = client.get_integration(LanggraphIntegration)
if integration is None or has_span_streaming_enabled(client.options):
Comment thread
alexander-alderman-webb marked this conversation as resolved.
return f(self, *args, **kwargs)
Comment thread
alexander-alderman-webb marked this conversation as resolved.

with sentry_sdk.start_span(
op=OP.GEN_AI_CREATE_AGENT,
origin=LanggraphIntegration.origin,
Expand Down Expand Up @@ -242,11 +244,11 @@ def new_invoke(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
client = sentry_sdk.get_client()
scope = sentry_sdk.get_current_scope()
messages_data = (
normalized_input_messages
if client.options.get("stream_gen_ai_spans", False)
else truncate_and_annotate_messages(
truncate_and_annotate_messages(
normalized_input_messages, span, scope
)
if should_truncate_gen_ai_input(client.options)
else normalized_input_messages
)
if messages_data is not None:
set_data_normalized(
Expand All @@ -268,7 +270,8 @@ def new_invoke(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
def _wrap_pregel_ainvoke(f: "Callable[..., Any]") -> "Callable[..., Any]":
@wraps(f)
async def new_ainvoke(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
integration = sentry_sdk.get_client().get_integration(LanggraphIntegration)
client = sentry_sdk.get_client()
integration = client.get_integration(LanggraphIntegration)
if integration is None:
return await f(self, *args, **kwargs)

Expand All @@ -277,6 +280,54 @@ async def new_ainvoke(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
f"invoke_agent {graph_name}".strip() if graph_name else "invoke_agent"
Comment thread
alexander-alderman-webb marked this conversation as resolved.
)

if has_span_streaming_enabled(client.options):
with sentry_sdk.traces.start_span(
name=span_name,
attributes={
"sentry.op": OP.GEN_AI_INVOKE_AGENT,
"sentry.origin": LanggraphIntegration.origin,
SPANDATA.GEN_AI_OPERATION_NAME: "invoke_agent",
},
) as span:
if graph_name:
span.set_attribute(SPANDATA.GEN_AI_PIPELINE_NAME, graph_name)
span.set_attribute(SPANDATA.GEN_AI_AGENT_NAME, graph_name)

input_messages = None
if (
len(args) > 0
and should_send_default_pii()
and integration.include_prompts
):
input_messages = _parse_langgraph_messages(args[0])
if input_messages:
normalized_input_messages = normalize_message_roles(
input_messages
)

client = sentry_sdk.get_client()
scope = sentry_sdk.get_current_scope()
messages_data = (
truncate_and_annotate_messages(
normalized_input_messages, span, scope
)
if should_truncate_gen_ai_input(client.options)
else normalized_input_messages
)
if messages_data is not None:
set_data_normalized(
span,
SPANDATA.GEN_AI_REQUEST_MESSAGES,
messages_data,
unpack=False,
)

result = await f(self, *args, **kwargs)

_set_response_attributes(span, input_messages, result, integration)

return result

with get_start_span_function()(
op=OP.GEN_AI_INVOKE_AGENT,
name=span_name,
Expand All @@ -301,11 +352,11 @@ async def new_ainvoke(self: "Any", *args: "Any", **kwargs: "Any") -> "Any":
client = sentry_sdk.get_client()
scope = sentry_sdk.get_current_scope()
messages_data = (
normalized_input_messages
if client.options.get("stream_gen_ai_spans", False)
else truncate_and_annotate_messages(
truncate_and_annotate_messages(
normalized_input_messages, span, scope
)
if should_truncate_gen_ai_input(client.options)
else normalized_input_messages
)
if messages_data is not None:
set_data_normalized(
Expand Down
Loading
Loading