.

Author: alexander-alderman-webb

sentry_sdk/integrations/openai_agents/patches/agent_run.py

@@ -66,13 +66,13 @@ def _maybe_start_agent_span(
         current_agent = _get_current_agent(context_wrapper)
         if current_agent and current_agent != agent:
             span = getattr(context_wrapper, "_sentry_agent_span", None)
-            if not span:
-                return
+            if span:
+                update_invoke_agent_span(
+                    span=span, context=context_wrapper, agent=agent
+                )
 
-            update_invoke_agent_span(span=span, context=context_wrapper, agent=agent)
-
-            span.__exit__(None, None, None)
-            delattr(context_wrapper, "_sentry_agent_span")
+                span.__exit__(None, None, None)
+                delattr(context_wrapper, "_sentry_agent_span")
 
     # Store the agent on the context wrapper so we can access it later
     context_wrapper._sentry_current_agent = agent

sentry_sdk/integrations/openai_agents/patches/error_tracing.py

@@ -2,7 +2,7 @@
 from typing import TYPE_CHECKING
 
 import sentry_sdk
-from sentry_sdk.tracing_utils import set_span_errored
+from sentry_sdk.consts import SPANSTATUS
 
 if TYPE_CHECKING:
     from typing import Any
@@ -56,7 +56,7 @@ def sentry_attach_error_to_current_span(
         # Set the current Sentry span to errored
         current_span = sentry_sdk.get_current_span()
         if current_span is not None:
-            set_span_errored(current_span)
+            current_span.set_status(SPANSTATUS.INTERNAL_ERROR)
 
         # Call the original function
         return original_attach_error(error, *args, **kwargs)