Bring back handoff span 2

Author: antonpirker

sentry_sdk/integrations/openai_agents/run_hooks.py

@@ -0,0 +1,19 @@
+from sentry_sdk.integrations import DidNotEnable
+
+from .spans import handoff_span
+
+
+try:
+    import agents
+except ImportError:
+    raise DidNotEnable("OpenAI Agents not installed")
+
+
+class SentryRunHooks(agents.RunHooks):  # type: ignore[misc]
+    async def on_handoff(
+        self,
+        context: agents.RunContextWrapper,
+        from_agent: agents.Agent,
+        to_agent: agents.Agent,
+    ) -> None:
+        handoff_span(context, from_agent, to_agent)

sentry_sdk/integrations/openai_agents/utils.py

@@ -176,9 +176,6 @@ def _wrap_hooks(hooks):
     enable our SentryRunHooks or if the users has given custom RunHooks wrap
     them so the Sentry hooks and the users hooks are both called
     """
-    import ipdb
-
-    ipdb.set_trace()
     from .run_hooks import SentryRunHooks
 
     sentry_hooks = SentryRunHooks()