fix(openai-agents): Patch execute_final_output functions following library refactor

Author: alexander-alderman-webb

sentry_sdk/integrations/openai_agents/__init__.py

@@ -11,7 +11,7 @@
     _execute_handoffs,
     _create_run_wrapper,
     _create_run_streamed_wrapper,
-    _patch_agent_run,
+    _execute_final_output,
     _patch_error_tracing,
 )
 
@@ -58,9 +58,6 @@ def _patch_runner() -> None:
         agents.run.DEFAULT_AGENT_RUNNER.run_streamed
     )
 
-    # Creating the actual spans for each agent run (works for both streaming and non-streaming).
-    _patch_agent_run()
-
 
 class OpenAIAgentsIntegration(Integration):
     """
@@ -151,6 +148,16 @@ async def new_wrapped_execute_handoffs(
                 new_wrapped_execute_handoffs
             )
 
+            @wraps(turn_resolution.execute_final_output)
+            async def new_wrapped_final_output(*args: "Any", **kwargs: "Any") -> "Any":
+                return await _execute_final_output(
+                    turn_resolution.execute_final_output, *args, **kwargs
+                )
+
+            agents.run_internal.turn_resolution.execute_final_output = (
+                new_wrapped_final_output
+            )
+
             return
 
         original_get_all_tools = AgentRunner._get_all_tools
@@ -212,3 +219,17 @@ async def old_wrapped_execute_handoffs(
         agents._run_impl.RunImpl.execute_handoffs = classmethod(
             old_wrapped_execute_handoffs
         )
+
+        original_execute_final_output = agents._run_impl.RunImpl.execute_final_output
+
+        @wraps(agents._run_impl.RunImpl.execute_final_output.__func__)
+        async def old_wrapped_final_output(
+            cls: "agents.Runner", *args: "Any", **kwargs: "Any"
+        ) -> "Any":
+            return await _execute_final_output(
+                original_execute_final_output, *args, **kwargs
+            )
+
+        agents._run_impl.RunImpl.execute_final_output = classmethod(
+            old_wrapped_final_output
+        )

sentry_sdk/integrations/openai_agents/patches/__init__.py

@@ -5,6 +5,6 @@
     _run_single_turn,
     _run_single_turn_streamed,
     _execute_handoffs,
-    _patch_agent_run,
+    _execute_final_output,
 )  # noqa: F401
 from .error_tracing import _patch_error_tracing  # noqa: F401

sentry_sdk/integrations/openai_agents/patches/agent_run.py

@@ -204,49 +204,28 @@ async def _execute_handoffs(
     return result
 
 
-def _patch_agent_run() -> None:
+async def _execute_final_output(
+    original_execute_final_output: "Callable[..., SingleStepResult]",
+    *args: "Any",
+    **kwargs: "Any",
+) -> "SingleStepResult":
     """
-    Patches AgentRunner methods to create agent invocation spans.
-    This directly patches the execution flow to track when agents start and stop.
+    Patched execute_final_output that
+    - ends the agent invocation span.
+    - ends the workflow span if the response is streamed.
     """
 
-    # Store original methods
-    original_execute_final_output = agents._run_impl.RunImpl.execute_final_output
+    agent = kwargs.get("agent")
+    context_wrapper = kwargs.get("context_wrapper")
+    final_output = kwargs.get("final_output")
 
-    @wraps(
-        original_execute_final_output.__func__
-        if hasattr(original_execute_final_output, "__func__")
-        else original_execute_final_output
-    )
-    async def patched_execute_final_output(
-        cls: "agents.Runner", *args: "Any", **kwargs: "Any"
-    ) -> "Any":
-        """
-        Patched execute_final_output that
-        - ends the agent invocation span.
-        - ends the workflow span if the response is streamed.
-        """
-
-        agent = kwargs.get("agent")
-        context_wrapper = kwargs.get("context_wrapper")
-        final_output = kwargs.get("final_output")
-
-        try:
-            result = await original_execute_final_output(*args, **kwargs)
-        finally:
-            with capture_internal_exceptions():
-                if (
-                    agent
-                    and context_wrapper
-                    and _has_active_agent_span(context_wrapper)
-                ):
-                    end_invoke_agent_span(context_wrapper, agent, final_output)
-                # For streaming, close the workflow span (non-streaming uses context manager in _create_run_wrapper)
-                _close_streaming_workflow_span(agent)
-
-        return result
-
-    # Apply patches
-    agents._run_impl.RunImpl.execute_final_output = classmethod(
-        patched_execute_final_output
-    )
+    try:
+        result = await original_execute_final_output(*args, **kwargs)
+    finally:
+        with capture_internal_exceptions():
+            if agent and context_wrapper and _has_active_agent_span(context_wrapper):
+                end_invoke_agent_span(context_wrapper, agent, final_output)
+            # For streaming, close the workflow span (non-streaming uses context manager in _create_run_wrapper)
+            _close_streaming_workflow_span(agent)
+
+    return result