
    gh}                         S SK Jr  S SKJr  SSKJrJr  S SKJr  \(       a  S SKJ	r	J
r
   S SKrS	 rg! \ a	    \" S5      ef = f)
    )wraps)DidNotEnable   ai_client_spanupdate_ai_client_span)TYPE_CHECKING)AnyCallableNzOpenAI Agents not installedc                 j   ^  [        [        T S5      (       a  T R                  OT 5      U 4S j5       nU$ )zt
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   >^^ T" TU5      nUR                   m[        T5      UU4S j5       nXCl         U$ )Nc                     >#    [        TU5       nT" U 0 UD6I S h  vN n[        UTX5        S S S 5        U$  N! , (       d  f       W$ = f7f)Nr   )argskwargsspanresultagentoriginal_get_responses       w/home/kali/devsecops-assessor/venv/lib/python3.13/site-packages/sentry_sdk/integrations/openai_agents/patches/models.pywrapped_get_responseR_create_get_model_wrapper.<locals>.wrapped_get_model.<locals>.wrapped_get_response$   sS       v.$4dEfEE%dE6B /
 M	 F /.
 Ms%   A:8:
A:
A	A)get_responser   )clsr   
run_configmodelr   r   original_get_models    `   @r   wrapped_get_model4_create_get_model_wrapper.<locals>.wrapped_get_model   sB     #5*5 % 2 2	$	%	 
&	 2    )r   hasattrr   )r   r   s   ` r   _create_get_model_wrapperr"      s@     %z2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	   3 A