feat(openai-agents): Support span streaming#6404

ericapisani · 2026-06-11T14:39:48Z

Taking a look at get_start_span_function, it returns one of:
sentry_sdk.traces.start_span
sentry_sdk.start_span
sentry_sdk.start_transaction
depending on if span streaming is enabled or not or if there's a transaction that's currently active.
What we can do here in order to leverage the existing streamed span awareness within that function is to, instead of invoking start_span above, change how we're invoking the get_start_span_function based on whether span streaming is enabled or not.
So the function body would look something like the following:
Suggested change
span_streaming=has_span_streaming_enabled(sentry_sdk.get_client().options)
ifspan_streaming:
span=sentry_sdk.traces.start_span(
name=f"{agent.name} workflow", attributes={"sentry.origin": SPAN_ORIGIN}
)
returnspan
span_streaming=has_span_streaming_enabled(sentry_sdk.get_client().options)
span_func=get_start_span_function()
ifspan_streaming:
returnspan_func(
name=f"{agent.name} workflow",
attributes={"sentry.origin": SPAN_ORIGIN}
)
else:
returnspan_func(
name=f"{agent.name} workflow",
origin=SPAN_ORIGIN
)

We don't need get_start_span_function() in span first anymore, since spans are automatically promoted to segment spans if there is no active segment span.
I've opened a PR to stop get_start_span_function() from returning sentry_sdk.traces.start_span:
#6560

ericapisani · 2026-05-26T13:19:34Z

I'm not sure how useful this comment is anymore - any chance we can remove it?

I think this is still a bug 😬.
Created #6417 to track.

-Original file line number
+Diff line change
@@ Expand Up / @@ -3,6 +3,7 @@ @@
 from sentry_sdk.consts import SPANDATA
 from sentry_sdk.integrations import DidNotEnable
+from sentry_sdk.traces import StreamedSpan
 from sentry_sdk.utils import capture_internal_exceptions, reraise
 from ..spans import (
@@ Expand All / @@ -12,7 +13,7 @@ @@
 )
 if TYPE_CHECKING:
- from typing import Any, Awaitable, Callable, Optional
+ from typing import Any, Awaitable, Callable, Optional, Union
  from agents.run_internal.run_steps import SingleStepResult
@@ Expand DownExpand Up / @@ -50,7 +51,7 @@ def _maybe_start_agent_span( @@
  should_run_agent_start_hooks: bool,
  span_kwargs: "dict[str, Any]",
  is_streaming: bool = False,
-) -> "Optional[Span]":
+) -> "Optional[Union[Span, StreamedSpan]]":
  """
  Start an agent invocation span if conditions are met.
  Handles ending any existing span for a different agent.
@@ Expand DownExpand Up / @@ -78,7 +79,12 @@ def _maybe_start_agent_span( @@
  context_wrapper._sentry_agent_span = span
  agent._sentry_agent_span = span
- if is_streaming:
+ if not is_streaming:
+ return span
+ if isinstance(span, StreamedSpan):
+ span.set_attribute(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
+ else:
  span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
  return span
@@ Expand DownExpand Up / @@ -108,7 +114,11 @@ async def _run_single_turn( @@
  context_wrapper, agent, should_run_agent_start_hooks, kwargs
  )
- if span is None or span.timestamp is not None:
+ if (
+ span is None
+ or (isinstance(span, StreamedSpan) and span.end_timestamp is not None)
+ or (not isinstance(span, StreamedSpan) and span.timestamp is not None)
+ ):
  return await original_run_single_turn(*args, **kwargs)
  try:
@@ Expand DownExpand Up / @@ -188,7 +198,11 @@ async def _run_single_turn_streamed( @@
  is_streaming=True,
  )
- if span is None or span.timestamp is not None:
+ if (
+ span is None
+ or (isinstance(span, StreamedSpan) and span.end_timestamp is not None)
+ or (not isinstance(span, StreamedSpan) and span.timestamp is not None)
+ ):
  return await original_run_single_turn_streamed(*args, **kwargs)
  try:
@@ Expand Down @@

-Original file line number
+Diff line change
@@ Expand Up / @@ -3,6 +3,8 @@ @@
 import sentry_sdk
 from sentry_sdk.consts import SPANSTATUS
+from sentry_sdk.traces import SpanStatus
+from sentry_sdk.tracing_utils import has_span_streaming_enabled
 if TYPE_CHECKING:
  from typing import Any
@@ Expand DownExpand Up / @@ -54,9 +56,15 @@ def sentry_attach_error_to_current_span( @@
  the agents library swallows exceptions.
  """
  # Set the current Sentry span to errored
- current_span = sentry_sdk.get_current_span()
- if current_span is not None:
- current_span.set_status(SPANSTATUS.INTERNAL_ERROR)
+ span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
+ if span_streaming:
+ current_span = sentry_sdk.get_current_scope().streamed_span
+ if current_span is not None:
+ current_span.status = SpanStatus.ERROR
+ else:
+ current_span = sentry_sdk.get_current_span()
+ if current_span is not None:
+ current_span.set_status(SPANSTATUS.INTERNAL_ERROR)
  # Call the original function
  return original_attach_error(error, *args, **kwargs)
@@ Expand Down @@

-Original file line number
+Diff line change
@@ Expand Up / @@ -6,6 +6,7 @@ @@
 import sentry_sdk
 from sentry_sdk.consts import SPANDATA
 from sentry_sdk.integrations import DidNotEnable
+from sentry_sdk.traces import StreamedSpan
 from sentry_sdk.tracing import BAGGAGE_HEADER_NAME
 from sentry_sdk.tracing_utils import (
  add_sentry_baggage_to_headers,
@@ Expand All / @@ -16,7 +17,7 @@ @@
 from ..spans import ai_client_span, update_ai_client_span
 if TYPE_CHECKING:
- from typing import Any, Callable, Optional
+ from typing import Any, Callable, Optional, Union
  from sentry_sdk.tracing import Span
@@ Expand All / @@ -34,11 +35,14 @@ def _set_response_model_on_agent_span( @@
  if response_model:
  agent_span = getattr(agent, "_sentry_agent_span", None)
  if agent_span:
- agent_span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, response_model)
+ if isinstance(agent_span, StreamedSpan):
+ agent_span.set_attribute(SPANDATA.GEN_AI_RESPONSE_MODEL, response_model)
+ else:
+ agent_span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, response_model)
 def _inject_trace_propagation_headers(
- hosted_tool: "HostedMCPTool", span: "Span"
+ hosted_tool: "HostedMCPTool", span: "Union[Span, StreamedSpan]"
 ) -> None:
  headers = hosted_tool.tool_config.get("headers")
  if headers is None:
@@ Expand DownExpand Up @@
  for hosted_tool in hosted_tools:
  _inject_trace_propagation_headers(hosted_tool, span=span)
- span.set_data(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
+ set_on_span = (
+ span.set_attribute
+ if isinstance(span, StreamedSpan)
+ else span.set_data
+ )
+ set_on_span(SPANDATA.GEN_AI_RESPONSE_STREAMING, True)
  streaming_response = None
  ttft_recorded = False
@@ Expand All @@
  # Detect first content token (text delta event)
  if not ttft_recorded and hasattr(event, "delta"):
  ttft = time.perf_counter() - start_time
- span.set_data(
- SPANDATA.GEN_AI_RESPONSE_TIME_TO_FIRST_TOKEN, ttft
- )
+ set_on_span(SPANDATA.GEN_AI_RESPONSE_TIME_TO_FIRST_TOKEN, ttft)
  ttft_recorded = True
  # Capture the full response from ResponseCompletedEvent
@@ Expand Down @@

-Original file line number
+Diff line change
@@ Expand Up / @@ -4,6 +4,7 @@ @@
 import sentry_sdk
 from sentry_sdk.consts import SPANDATA
 from sentry_sdk.integrations import DidNotEnable
+from sentry_sdk.traces import StreamedSpan
 from sentry_sdk.utils import capture_internal_exceptions, reraise
 from ..spans import agent_workflow_span, update_invoke_agent_span
@@ Expand DownExpand Up / @@ -43,9 +44,15 @@ async def wrapper(*args: "Any", **kwargs: "Any") -> "Any": @@
  conversation_id = kwargs.get("conversation_id")
  if conversation_id:
  agent._sentry_conversation_id = conversation_id
- workflow_span.set_data(
- SPANDATA.GEN_AI_CONVERSATION_ID, conversation_id
- )
+ if isinstance(workflow_span, StreamedSpan):
+ workflow_span.set_attribute(
+ SPANDATA.GEN_AI_CONVERSATION_ID, conversation_id
+ )
+ else:
+ workflow_span.set_data(
+ SPANDATA.GEN_AI_CONVERSATION_ID, conversation_id
+ )
  args = (agent, *args[1:])
  try:
@@ Expand All / @@ -61,9 +68,15 @@ async def wrapper(*args: "Any", **kwargs: "Any") -> "Any": @@
  context_wrapper, "_sentry_agent_span", None
  )
- if (
- invoke_agent_span is not None
- and invoke_agent_span.timestamp is None
+ if invoke_agent_span is not None and (
+ (
+ isinstance(invoke_agent_span, StreamedSpan)
+ and invoke_agent_span.end_timestamp is None
+ )
+ or (
+ not isinstance(invoke_agent_span, StreamedSpan)
+ and invoke_agent_span.timestamp is None
+ )
  ):
  update_invoke_agent_span(
  span=invoke_agent_span,
@@ Expand DownExpand Up / @@ -135,7 +148,12 @@ def wrapper(*args: "Any", **kwargs: "Any") -> "Any": @@
  # Set conversation ID on workflow span early so it's captured even on errors
  if conversation_id:
- workflow_span.set_data(SPANDATA.GEN_AI_CONVERSATION_ID, conversation_id)
+ if isinstance(workflow_span, StreamedSpan):
+ workflow_span.set_attribute(
+ SPANDATA.GEN_AI_CONVERSATION_ID, conversation_id
+ )
+ else:
+ workflow_span.set_data(SPANDATA.GEN_AI_CONVERSATION_ID, conversation_id)
  # Store span on agent for cleanup
  agent._sentry_workflow_span = workflow_span
@@ Expand Down @@

-Original file line number
+Diff line change
@@ Expand Up / @@ -2,6 +2,8 @@ @@
 import sentry_sdk
 from sentry_sdk.consts import OP, SPANDATA
+from sentry_sdk.traces import StreamedSpan
+from sentry_sdk.tracing_utils import has_span_streaming_enabled
 from ..consts import SPAN_ORIGIN
 from ..utils import (
@@ Expand All / @@ -12,14 +14,14 @@ @@
 )
 if TYPE_CHECKING:
- from typing import Any, Optional
+ from typing import Any, Optional, Union
  from agents import Agent
 def ai_client_span(
  agent: "Agent", get_response_kwargs: "dict[str, Any]"
-) -> "sentry_sdk.tracing.Span":
+) -> "Union[sentry_sdk.tracing.Span, StreamedSpan]":
  # TODO-anton: implement other types of operations. Now "chat" is hardcoded.
  # Get model name from agent.model or fall back to request model (for when agent.model is None/default)
  model_name = None
@@ Expand All / @@ -28,13 +30,24 @@ def ai_client_span( @@
  elif hasattr(agent, "_sentry_request_model"):
  model_name = agent._sentry_request_model
- span = sentry_sdk.start_span(
- op=OP.GEN_AI_CHAT,
- name=f"chat {model_name}",
- origin=SPAN_ORIGIN,
- )
- # TODO-anton: remove hardcoded stuff and replace something that also works for embedding and so on
- span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "chat")
+ span_streaming = has_span_streaming_enabled(sentry_sdk.get_client().options)
+ if span_streaming:
+ span = sentry_sdk.traces.start_span(
+ name=f"chat {model_name}",
+ attributes={
+ "sentry.op": OP.GEN_AI_CHAT,
+ "sentry.origin": SPAN_ORIGIN,
+ SPANDATA.GEN_AI_OPERATION_NAME: "chat",
+ },
+ )
+ else:
+ span = sentry_sdk.start_span(
+ op=OP.GEN_AI_CHAT,
+ name=f"chat {model_name}",
+ origin=SPAN_ORIGIN,
+ )
+ # TODO-anton: remove hardcoded stuff and replace something that also works for embedding and so on
+ span.set_data(SPANDATA.GEN_AI_OPERATION_NAME, "chat")
  _set_agent_data(span, agent)
  _set_input_data(span, get_response_kwargs)
@@ Expand All / @@ -43,7 +56,7 @@ def ai_client_span( @@
 def update_ai_client_span(
- span: "sentry_sdk.tracing.Span",
+ span: "Union[sentry_sdk.tracing.Span, StreamedSpan]",
  response: "Any",
  response_model: "Optional[str]" = None,
  agent: "Optional[Agent]" = None,
@@ Expand All / @@ -55,13 +68,17 @@ def update_ai_client_span( @@
  if hasattr(response, "output") and response.output:
  _set_output_data(span, response)
+ set_on_span = (
+ span.set_attribute if isinstance(span, StreamedSpan) else span.set_data
+ )
  if response_model is not None:
-span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, response_model)
+set_on_span(SPANDATA.GEN_AI_RESPONSE_MODEL, response_model)
  elif hasattr(response, "model") and response.model:
-span.set_data(SPANDATA.GEN_AI_RESPONSE_MODEL, str(response.model))
+set_on_span(SPANDATA.GEN_AI_RESPONSE_MODEL, str(response.model))
  # Set conversation ID from agent if available
  if agent:
  conv_id = getattr(agent, "_sentry_conversation_id", None)
  if conv_id:
-span.set_data(SPANDATA.GEN_AI_CONVERSATION_ID, conv_id)
+set_on_span(SPANDATA.GEN_AI_CONVERSATION_ID, conv_id)

Navigation Menu

Uh oh!

feat(openai-agents): Support span streaming#6404

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!

ericapisaniJun 11, 2026

Uh oh!

alexander-alderman-webbJun 12, 2026•
edited
Loading

Uh oh!

Uh oh!

ericapisaniMay 26, 2026

Uh oh!

alexander-alderman-webbMay 26, 2026

Uh oh!

Uh oh!

Uh oh!

Uh oh!

feat(openai-agents): Support span streaming#6404

Uh oh!

feat(openai-agents): Support span streaming #6404

Uh oh!

Uh oh!

Diff view

Diff view

There are no files selected for viewing

Uh oh!

ericapisaniJun 11, 2026

Choose a reason for hiding this comment

Uh oh!

alexander-alderman-webbJun 12, 2026• edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Uh oh!

ericapisaniMay 26, 2026

Choose a reason for hiding this comment

Uh oh!

alexander-alderman-webbMay 26, 2026

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

alexander-alderman-webbJun 12, 2026•
edited
Loading