
    ^h                     <    S SK r S SKrS SKJr  S SKr " S S5      rg)    N)	BaseModelc                        \ rS rSrS rS rSrg)PromptLayerLogger   c                 :    [         R                  " S5      U l        g )NPROMPTLAYER_API_KEY)osgetenvkey)selfs    [/home/james-whalen/.local/lib/python3.13/site-packages/litellm/integrations/prompt_layer.py__init__PromptLayerLogger.__init__   s    9923    c                     0 nUS   US'   US   US'   US    H  nUS   U   Xg'   M     / n0 n	SUS   ;   aI  SUS   S   ;   a  US   S   S   nUS   S   R                  5        V
Vs0 s H  u  pU
S:w  d  M  X_M     n	n
nU" SU SU 35        [        U[        5      (       a  UR                  5       n[        R
                  R                  S	S
UU[        U5      [        UR                  5       5      [        UR                  5       5      U R                  S.S9nUR                  5       nUR                  5       R                  SS5      (       d  [        S5      eU" SUR                   35        SU;   aM  U	(       aE  [        R
                  R                  SUS   U R                  U	S.S9nU" SUR                   35        g g g s  snn
f ! [         a!    U" S[        R                   " 5        35         g f = f)Nmodelmessagesoptional_paramsmetadatalitellm_paramspl_tagszAPrompt Layer Logging - Enters logging function for model kwargs: z
, response: z.https://api.promptlayer.com/rest/track-requestzopenai.ChatCompletion.create)function_namekwargstagsrequest_responserequest_start_timerequest_end_timeapi_key)jsonsuccessFz2Promptlayer did not successfully log the response!z7Prompt Layer Logging: success - final response object: 
request_idz/https://api.promptlayer.com/rest/track-metadata)r!   r   r   z?Prompt Layer Logging: success - metadata post response object: zerror: Prompt Layer Error - )items
isinstancer   
model_dumplitellmmodule_level_clientpostdictint	timestampr   r   get	Exceptiontext	traceback
format_exc)r   r   response_obj
start_timeend_timeprint_verbose
new_kwargsoptional_paramr   r   kvr   response_jsonresponses                  r   	log_eventPromptLayerLogger.log_event   s_   H	J"(/Jw%+J%7Jz" #)):";-34E-F~-V
* #< DHV$455'7 8 DD!"23J?	JD
 !''7 8 D J J L LI~ AD L   ST^S__mnzm{|
 ,	22+668&::??@%C( (,\(:*-j.B.B.D*E(+H,>,>,@(A#xx  @  " -113M#((*..y%@@ TUUIJZJ_J_I`a },&::??I*7*E'+xx(0  @  H "YZbZgZgYhi  -Oj  	89M9M9O8PQR	s+   A%G	 'G7G=EG	 G	 	(G43G4)r   N)__name__
__module____qualname____firstlineno__r   r:   __static_attributes__ r   r   r   r      s    4Jr   r   )r	   r.   pydanticr   r%   r   rA   r   r   <module>rC      s    
   P Pr   