o BDi@sXddlmZddlZddlmZddlmZddlmZer&ddlm Z m Z dd Z dS) wrapsN)agent_workflow_span)_capture_exception) TYPE_CHECKING)AnyCallablecstfdd}|S)z Wraps the agents.Runner.run methods to create a root span for the agent workflow runs. Note agents.Runner.run_sync() is a wrapper around agents.Runner.run(), so it does not need to be wrapped separately. c s|d}t|?d}z|i|IdH}|WWdStyG}zt|t}|dur@|jdur@|ddd|dd}~ww1sKwYdS)Nr)r Exceptionr sentry_sdkget_current_span timestamp__exit__)argskwargsagentresultexc current_span original_funcm/var/www/Datamplify/venv/lib/python3.10/site-packages/sentry_sdk/integrations/openai_agents/patches/runner.pywrappers" z$_create_run_wrapper..wrapperr)rrrrr_create_run_wrappers r) functoolsrr spansrutilsrtypingrrr rrrrrs