
    Cj                     (    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        UR                  SS5      S-   U l        / SQU l        g )	Nr   ATHINA_API_KEYzapplication/json)zathina-api-keyzContent-TypeATHINA_BASE_URLzhttps://log.athina.aiz/api/v1/log/inference)environmentprompt_slugcustomer_idcustomer_user_id
session_idexternal_reference_idcontextexpected_response
user_querytagsuser_feedbackmodel_optionscustom_attributes)osgetenvathina_api_keyheadersathina_logging_urladditional_keys)selfr   s     m/home/rurouni/.local/share/pipx/venvs/strix-agent/lib/python3.13/site-packages/litellm/integrations/athina.py__init__AthinaLogger.__init__   sX     ii(89"11.

 II')@A%& 	 
    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>N   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_verboser>   r?   	is_streamcompletion_responseresponse_jsonr<   r/   r:   keyr+   es                    r   	log_eventAthinaLogger.log_event#   sp   B	

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   r[   __static_attributes__r3   r    r   r   r      s    
8Fr    r   )rC   rH   r   r3   r    r   <module>rb      s     c cr    