
    lh}                     ~    d dl mZ d dlmZ ddlmZmZ d dlmZ erd dlm	Z	m
Z
 	 d dlZd	 Zy# e$ r	  ed      w xY w)
    )wraps)DidNotEnable   ai_client_spanupdate_ai_client_span)TYPE_CHECKING)AnyCallableNzOpenAI Agents not installedc                 ^     t        t         d      r j                  n        fd       }|S )z|
    Wraps the agents.Runner._get_model method to wrap the get_response method of the model to create a AI client span.
    __func__c                 l     |      }|j                   t              fd       }||_         |S )Nc                     K   t        |      5 } | i | d {   }t        |||       d d d        |S 7 # 1 sw Y   S xY ww)Nr   )argskwargsspanresultagentoriginal_get_responses       /var/www/pru.catia.catastroantioquia-mas.com/tasa/lib/python3.12/site-packages/sentry_sdk/integrations/openai_agents/patches/models.pywrapped_get_responsezR_create_get_model_wrapper.<locals>.wrapped_get_model.<locals>.wrapped_get_response$   s\       v. C$4dEfEE%dE66BC
 M	 FC
 Ms%   A;9;
A;A A)get_responser   )clsr   
run_configmodelr   r   original_get_models    `   @r   wrapped_get_modelz4_create_get_model_wrapper.<locals>.wrapped_get_model   sD     #5*5 % 2 2	$	%	 
&	 2    )r   hasattrr   )r   r   s   ` r   _create_get_model_wrapperr       s=     %z2 	##

( r   )	functoolsr   sentry_sdk.integrationsr   spansr   r   typingr	   r
   r   agentsImportErrorr     r   r   <module>r(      sA     0 9  $6
	  6
4
556s   . <