
    ^h                         S SK r S SKrS SKr\R                  R	                  S \ R                  R                  S5      5        S SKrS SKrS SKJ	r	  S r
 " S S\	5      r\" 5       rg)    Nz../..)CustomLoggerc                 F    [         R                  (       a  [        U 5        g g )N)litellmset_verboseprint)print_statements    l/home/james-whalen/.local/lib/python3.13/site-packages/litellm/proxy/example_config_yaml/custom_callbacks.pyprint_verboser
      s    o     c                   >    \ rS rSrS rS rS rS rS rS r	S r
S	rg
)MyCustomHandler   c           	      2   SnSn[        U S35         [        S5        [        U 5       Vs/ s H+  n[        R                  " [	        X5      5      (       d  M)  UPM-     nnU H  n[        SU 35        M     [        U 5        g s  snf ! [
         a     g f = f)Nz[94mz[0mz!Initialized LiteLLM custom loggerz*Logger Initialized with following methods:z - )r
   dirinspectismethodgetattr	Exception)selfblue_color_codereset_color_codemethodmethodss        r	   __init__MyCustomHandler.__init__   s    $$))JKL	FG "$i'F##GD$9: '   "F8n- "-.0  		s(   B	 (BB%B	 B	 	
BBc                     [        S5        g )NzPre-API Callr
   )r   modelmessageskwargss       r	   log_pre_api_call MyCustomHandler.log_pre_api_call,   s
    n%r   c                     [        S5        g )NzPost-API Callr   r   r    response_obj
start_timeend_times        r	   log_post_api_call!MyCustomHandler.log_post_api_call/   s
    o&r   c                     [        S5        g )Nz	On Streamr   r$   s        r	   log_stream_event MyCustomHandler.log_stream_event2   s
    k"r   c                     [        S5        g )NzOn Success!r   r$   s        r	   log_success_event!MyCustomHandler.log_success_event5   s
    m$r   c                 Z   #    [        S5        [        R                  " US9nUS:  d   eg 7f)NzOn Async Success!)completion_responseg        )r
   r   completion_cost)r   r    r%   r&   r'   response_costs         r	   async_log_success_event'MyCustomHandler.async_log_success_event8   s/     )*//LQs"""s   )+c                 n   #     [        S5        g ! [         a  n[        SU 35         S nAg S nAff = f7f)NzOn Async Failure !zException: )r
   r   )r   r    r%   r&   r'   es         r	   async_log_failure_event'MyCustomHandler.async_log_failure_event>   s4     	-./ 	-Ks+,,	-s    5 5
2-525 N)__name__
__module____qualname____firstlineno__r   r!   r(   r+   r.   r4   r8   __static_attributes__r:   r   r	   r   r      s%    &&'#%-r   r   )ossys	tracebackpathinsertabspathr   r   "litellm.integrations.custom_loggerr   r
   r   proxy_handler_instancer:   r   r	   <module>rH      sZ    	 
  rwww   ;

*-l *-Z )* r   