Ë
    tPêj©  ã                   ó4   — d dl Z d dlmZ d dlZ G d„ d«      Zy)é    N)ÚFinalc                   ó   — e Zd Zd„ Zd„ Zy)ÚAthinaLoggerc                 ó¦   — dd l }|j                  d«      | _        | j                  ddœ| _        |j                  dd«      dz   | _        g d¢| _        y )	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     ú`/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/athina.pyÚ__init__zAthinaLogger.__init__   sS   € Ûà Ÿi™iÐ(8Ó9ˆÔà"×1Ñ1Ø.ñ
ˆŒð #%§)¡)Ð,=Ð?VÓ"WÐZqÑ"qˆÔò 
ˆÕó    c           	      óì  — dd l }dd l}	 |j                  dd«      }|rd|v r|d   }	|	r|	j                  «       ni }
ny |r|j                  «       ni }
|j                  d«      ||
|
j                  di «      j                  d«      |
j                  di «      j                  d«      |
j                  di «      j                  d	«      d
œ}t	        |«      t
        j
                  u r=t	        |«      t
        j
                  u r"t        ||z
  j                  «       dz  «      |d<   d|v r|j                  dd «      |d<   |j                  di «      }|j                  d„ |j                  «       D «       «       |j                  di «      j                  di «      }|r| j                  D ]  }||v sŒ||   ||<   Œ t        j                  j                  | j                  | j                  |j!                  |t"        ¬«      ¬«      }|j$                  dk7  r# |d|j&                  › d|j$                  › �«       y  |d|j&                  › �«       y # t(        $ r&} |d|› d|j+                  «       › �«       Y d }~y d }~ww xY w)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   ó4   K  — | ]  \  }}|d v sŒ||f–— Œ y­w))ÚtoolsÚ	functionsN© )Ú.0ÚkÚvs      r   ú	<genexpr>z)AthinaLogger.log_event.<locals>.<genexpr>C   s    è ø€ Òb¡4 1 aÀaÐKaÒFa˜˜AœÑbùs   ‚�	Ú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.   r8   Úkeyr*   Úes                    r   Ú	log_eventzAthinaLogger.log_event!   sd  € ÛÛð0	_Ø%Ÿz™z¨(°EÓ:ˆIÙØ0°FÑ:à17Ð8UÑ1VÐ'ÙH[Ð$7×$BÑ$BÔ$DÐac‘Mð ñ >J × 7Ñ 7Ô 9Èr�à%+§Z¡Z°Ó%8Ø!Ø)Ø!.×!2Ñ!2°7¸BÓ!?×!CÑ!CÀOÓ!TØ%2×%6Ñ%6°wÀÓ%C×%GÑ%GÐH[Ó%\Ø -× 1Ñ 1°'¸2Ó >× BÑ BÀ>Ó RñˆDô �H‹~¤×!2Ñ!2Ñ2´t¸JÓ7GÌ8×K\ÑK\Ñ7\Ü(+¨X¸
Ñ-B×,QÑ,QÓ,SÐVZÑ,ZÓ([��_Ñ%à˜VÑ#Ø!'§¡¨J¸Ó!=��X‘ð &,§Z¡ZÐ0AÀ2Ó%FˆOØ�K‰KÑb¨?×+@Ñ+@Ó+BÔbÔbð %Ÿj™jÐ)9¸2Ó>×BÑBÀ:ÈrÓRˆHÙØ×/Ñ/ò 2�CØ˜h’Ø$,¨S¡M˜˜Sš	ð2ô &×9Ñ9×>Ñ>Ø×'Ñ'ØŸ™Ø—Z‘Z ¬c�ZÓ2ð ?ó ˆHð
 ×#Ñ# sÒ*ÙÐ 6°x·}±}°oÀRÈ×H\ÑH\ÐG]Ð^Õ_áÐ :¸8¿=¹=¸/ÐJÕKøÜò 	_ÙÐ2°1°#°_ÀY×EYÑEYÓE[ÐD\Ð]×^Ñ^ûð	_ús*   Š2I ½E)I Æ'BI È.I É	I3ÉI.É.I3N)Ú__name__Ú
__module__Ú__qualname__r   rY   r2   r   r   r   r      s   „ ò
ó24_r   r   )rA   Útypingr   rF   r   r2   r   r   ú<module>r^      s   ðÛ Ý ã ÷N_ò N_r   