remove _SingleTurnException

Author: alexander-alderman-webb

sentry_sdk/integrations/openai_agents/patches/agent_run.py

@@ -1,14 +1,16 @@
+import sys
 from functools import wraps
 
 from sentry_sdk.integrations import DidNotEnable
 from sentry_sdk.tracing_utils import set_span_errored
+from sentry_sdk.utils import reraise
 from ..spans import (
     invoke_agent_span,
     update_invoke_agent_span,
     end_invoke_agent_span,
     handoff_span,
 )
-from ..utils import _capture_exception, _record_exception_on_span, _SingleTurnException
+from ..utils import _capture_exception, _record_exception_on_span
 
 from typing import TYPE_CHECKING
 
@@ -97,7 +99,7 @@ async def patched_run_single_turn(cls, *args, **kwargs):
                 _record_exception_on_span(span, exc)
                 end_invoke_agent_span(context_wrapper, agent)
 
-            raise _SingleTurnException(exc)
+            reraise(*sys.exc_info())
 
         return result
 

sentry_sdk/integrations/openai_agents/patches/runner.py

@@ -4,7 +4,7 @@
 from sentry_sdk.integrations import DidNotEnable
 
 from ..spans import agent_workflow_span, end_invoke_agent_span
-from ..utils import _capture_exception, _record_exception_on_span, _SingleTurnException
+from ..utils import _capture_exception, _record_exception_on_span
 
 try:
     from agents.exceptions import AgentsException
@@ -38,9 +38,6 @@ async def wrapper(*args, **kwargs):
                 args = (agent, *args[1:])
                 try:
                     run_result = await original_func(*args, **kwargs)
-                except _SingleTurnException as exc:
-                    _capture_exception(exc.original)
-                    raise exc.original from None
                 except AgentsException as exc:
                     _capture_exception(exc)
 

sentry_sdk/integrations/openai_agents/utils.py

@@ -27,12 +27,6 @@
     raise DidNotEnable("OpenAI Agents not installed")
 
 
-class _SingleTurnException(Exception):
-    def __init__(self, original):
-        # type: (Exception) -> None
-        self.original = original
-
-
 def _capture_exception(exc):
     # type: (Any) -> None
     set_span_errored()