.

Author: alexander-alderman-webb

sentry_sdk/integrations/openai_agents/patches/agent_run.py

@@ -114,7 +114,7 @@ async def _run_single_turn(
         return await original_run_single_turn(*args, **kwargs)
 
     try:
-        return await original_run_single_turn(*args, **kwargs)
+        result = await original_run_single_turn(*args, **kwargs)
     except Exception:
         exc_info = sys.exc_info()
         with capture_internal_exceptions():
@@ -124,6 +124,8 @@ async def _run_single_turn(
             delattr(context_wrapper, "_sentry_agent_span")
         reraise(*exc_info)
 
+    return result
+
 
 async def _run_single_turn_streamed(
     original_run_single_turn_streamed: "Callable[..., Awaitable[SingleStepResult]]",
@@ -193,13 +195,15 @@ async def _run_single_turn_streamed(
         agent=agent,
     ):
         try:
-            return await original_run_single_turn_streamed(*args, **kwargs)
+            result = await original_run_single_turn_streamed(*args, **kwargs)
         except Exception:
             exc_info = sys.exc_info()
             with capture_internal_exceptions():
                 _close_streaming_workflow_span(agent)
             reraise(*exc_info)
 
+    return result
+
 
 async def _execute_handoffs(
     original_execute_handoffs: "Callable[..., SingleStepResult]",