
    ^h                     (    S SK r S SKr " S S5      rg)    Nc                        \ rS rSrS rS rSrg)AthinaLogger   c                     SS K nUR                  S5      U l        U R                  SS.U l        SU l        / SQU l        g )Nr   ATHINA_API_KEYzapplication/json)zathina-api-keyzContent-Typez*https://log.athina.ai/api/v1/log/inference)	environmentprompt_slugcustomer_idcustomer_user_id
session_idexternal_reference_idcontextexpected_response
user_query)osgetenvathina_api_keyheadersathina_logging_urladditional_keys)selfr   s     U/home/james-whalen/.local/lib/python3.13/site-packages/litellm/integrations/athina.py__init__AthinaLogger.__init__   sA     ii(89"11.
 #O
 
    c           	         SS K nSS Kn UR                  SS5      nU(       a&  SU;   a  US   n	U	(       a  U	R                  5       O0 n
Og U(       a  UR                  5       O0 n
UR                  S5      UU
U
R                  S0 5      R                  S5      U
R                  S0 5      R                  S5      U
R                  S0 5      R                  S	5      S
.n[	        U5      [
        R
                  L a=  [	        U5      [
        R
                  L a!  [        XC-
  R                  5       S-  5      US'   SU;   a  UR                  SS 5      US'   UR                  S0 5      nUR                  S UR                  5        5       5        UR                  S0 5      R                  S0 5      nU(       a   U R                   H  nX;   d  M
  X   X'   M     [        R                  R                  U R                  U R                  UR!                  U["        S9S9nUR$                  S:w  a#  U" SUR&                   SUR$                   35        g U" SUR&                   35        g ! [(         a&  nU" SU SUR+                  5        35         S nAg S nAff = f)Nr   streamFcomplete_streaming_responsemodelusageprompt_tokenscompletion_tokenstotal_tokens)language_model_idrequestresponser!   r"   r#   i  response_timemessagespromptoptional_paramsc              3   <   #    U  H  u  pUS ;   d  M  X4v   M     g7f))tools	functionsN ).0kvs      r   	<genexpr>)AthinaLogger.log_event.<locals>.<genexpr>G   s&      3DA.. 3s   
litellm_paramsmetadata)default)r   data   zAthina Logger Error - z, zAthina Logger Succeeded - z, Stack trace: )json	tracebackget
model_dumptypedatetimeinttotal_secondsupdateitemsr   litellmmodule_level_clientpostr   r   dumpsstrstatus_codetext	Exception
format_exc)r   kwargsresponse_obj
start_timeend_timeprint_verboser9   r:   	is_streamcompletion_responseresponse_jsonr7   r*   r5   keyr&   es                    r   	log_eventAthinaLogger.log_event   sp   C	

8U3I0F:*01N*O'<O+668UW "
  >J 7 7 9r%+ZZ%8!)!.!2!27B!?!C!CO!T%2%6%6w%C%G%G'& !. 1 1'2 > B B> R	D X("3"33$(9(99(+*99;dB)_% V#!'J!=X %jj):B?OKK +113  zz"2B7;;JKH//C$,M	 0 2277''ZZcZ2 8 H
 ##s*,X]]O2h>R>R=ST  :8==/JK 	(?9;O;O;Q:RS 		s+   >I 	E5I BI I 
J
$JJ
)r   r   r   r   N)__name__
__module____qualname____firstlineno__r   rV   __static_attributes__r.   r   r   r   r      s    
*Gr   r   )r>   rC   r   r.   r   r   <module>r]      s     ] ]r   