
    lh                     V    d dl mZ d dlZddlmZ ddlmZ d dlmZ erd dlm	Z	m
Z
 d Zy)	    wrapsN   )agent_workflow_span)_capture_exception)TYPE_CHECKING)AnyCallablec                 .     t                fd       }|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                    K   t        j                         5  | d   }t        |      5  d }	  | i | d {   }|cd d d        cd d d        S 7 # t        $ rH}t	        |       t        j
                         }||j                  |j                  d d d        |d d }~ww xY w# 1 sw Y   nxY w	 d d d        y # 1 sw Y   y xY ww)Nr   )
sentry_sdkisolation_scoper   	Exceptionr   get_current_span	timestamp__exit__)argskwargsagentresultexccurrent_spanoriginal_funcs         /var/www/pru.catia.catastroantioquia-mas.com/tasa/lib/python3.12/site-packages/sentry_sdk/integrations/openai_agents/patches/runner.pywrapperz$_create_run_wrapper.<locals>.wrapper   s     
 '') 	(GE$U+ ((#0$#A&#AAF!	( (	( 	(
 B  (&s+ $.#>#>#@L#/L4J4J4R$--dD$?4'(( ( (	( 	( 	(sf   C	B=B'AAA	B=
C	A	B$ABB$$B''B0	,B=4	C	=CC	r   )r   r   s   ` r   _create_run_wrapperr      s#     =( (* N    )	functoolsr   r   spansr   utilsr   typingr   r	   r
   r    r   r   <module>r#      s      ' &  $r   