set errors on streamed spans

Author: alexander-alderman-webb

sentry_sdk/integrations/openai_agents/patches/error_tracing.py

@@ -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
@@ -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)