Ë
    tPêjÖ  ã                   ó^   — d dl Z d dlZd dlmZmZ d dlmZ d dlZd„ Zd„ Zd„ Z	 G d„ d«      Z
y)	é    N)ÚdatetimeÚtimezone)ÚFinalc                 ó4   — d| v r| d   ndd| v r| d   dœS ddœS )NÚcompletion_tokensr   Úprompt_tokens)Ú
completionÚprompt© )Úusages    ú`/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/lunary.pyÚparse_usager      s;   € à4GÈ5Ñ4P�eÐ/Ò0ÐVWØ,;¸uÑ,D�%˜Ñ(ñð àJKñð ó    c                 óx   — | €y d„ }| D �cg c]$  }t        |dd «      €t        |dd «      � ||«      ‘Œ& c}S c c}w )Nc                 óä   — t        | dd «      }|�|j                  |j                  }}n,| j                  j                  | j                  j                  }}| j
                  | j                  ||dœdœS )NÚcustom)ÚnameÚ	arguments)ÚtypeÚidÚfunction)Úgetattrr   Úinputr   r   r   r   )Ú	tool_callr   r   r   s       r   Úclean_tool_callz)parse_tool_calls.<locals>.clean_tool_call   sk   € Ü 	¨8°TÓ:ˆØÐØ$Ÿk™k¨6¯<©<�)‰Dà'×0Ñ0×5Ñ5°y×7IÑ7I×7SÑ7S�)ˆDà—N‘NØ—,‘,àØ&ññ
ð 	
r   r   r   )r   )Ú
tool_callsr   r   s      r   Úparse_tool_callsr      sV   € ØÐØò
ð" $öàÜ�9˜j¨$Ó/Ð;¼wÀyÐRZÐ\`Ó?aÐ?mñ 	˜	Õ"òð ùò s   ‹)7c                 ó¬   ‡— | €y ˆfd„Št        | t        «      r/t        | «      dk(  r ‰| d   «      S | D �cg c]
  } ‰|«      ‘Œ c}S  ‰| «      S c c}w )Nc                 óì   •— t        | t        «      r| S d| v r ‰| d   «      S | j                  d«      | j                  d«      dœ}| j                  d«      rt        | j                  d«      «      |d<   |S )NÚmessageÚroleÚcontent)r!   r"   r   )Ú
isinstanceÚstrÚgetr   )r    Ú
serializedÚclean_messages     €r   r'   z%parse_messages.<locals>.clean_message1   sw   ø€ ä�gœsÔ#ØˆNà˜ÑÙ  ¨Ñ!3Ó4Ð4ð —K‘K Ó'Ø—{‘{ 9Ó-ñ
ˆ
ð �;‰;�|Ô$Ü'7¸¿¹ÀLÓ8QÓ'RˆJ�|Ñ$àÐr   é   r   )r#   ÚlistÚlen)r   Úmsgr'   s     @r   Úparse_messagesr,   -   s\   ø€ Ø€}Øôô& �%œÔÜˆu‹:˜Š?Ù   q¡Ó*Ð*à27Ö8¨3‘M #Õ&Ò8Ð8á˜UÓ#Ð#ùò 9s   ·Ac                   ó�   — e Zd Zd„ Zi ddd ej
                  ej                  «       ej
                  ej                  «      dfd„Zy)ÚLunaryLoggerc                 ó6  — 	 dd l }t        j                  j                  d«      }t        j                  j                  |«      t        j                  j                  d«      k  rt        d«       t        ‚|| _        y # t        $ r t        d«       t        ‚w xY w)Nr   Úlunaryz0.1.43zXLunary version outdated. Required: >= 0.1.43. Upgrade via 'pip install lunary --upgrade'zBLunary not installed. Please install it using 'pip install lunary')	r0   Ú	importlibÚmetadataÚversionÚ	packagingÚVersionÚprintÚImportErrorÚlunary_client)Úselfr0   r3   s      r   Ú__init__zLunaryLogger.__init__O   s‡   € ð	Ûä&×/Ñ/×7Ñ7¸ÓAˆGä× Ñ ×(Ñ(¨Ó1´I×4EÑ4E×4MÑ4MÈhÓ4WÒWÜØnôô "Ð!à!'ˆDÕøÜò 	ÜØTôô Ðð		ús   ‚A:A= Á=BNc                 óP  — 	  |d|› �«       d }|j                  di «      }|j                  di «      }|j                  di «      xs i }|ri |¥|¥}|j                  dd «      }|rF|j                  dd «       |j                  dd «       |j                  di «      j                  d	d «      }|j                  «       D ];  \  }}t        |t        t
        t        t        f«      rŒ'|d
k7  sŒ-	 t	        |«      ||<   Œ= |
r d|
v rt        |
d   «      nd }d|
v r|
d   nd }nd }d }|rd|i}nd }| j                  j                  |d||j                  dd «      |	|t        |«      |j                  t        j                  «      j!                  «       ||d||¬«       | j                  j                  ||||j                  t        j                  «      j!                  «       d|t        |«      |¬«       y # t        $ r Y �Œ=w xY w# t        $ r   |dt#        j$                  «       › �«       Y y w xY w)Nz+Lunary Logging - Logging request for model Úlitellm_paramsÚoptional_paramsr2   ÚtagsÚ
extra_bodyÚuserÚextra_headerszTemplate-IdÚtoolsr   ÚchoicesÚstackÚstartÚparent_run_idÚlitellm)
rF   Úuser_idr   r   Ú	timestampÚtemplate_idr2   Úruntimer>   Úparams)rI   rK   ÚerrorÚoutputÚtoken_usagezLunary Logging Error - )r%   ÚpopÚitemsr#   r$   ÚintÚboolÚfloatÚ	Exceptionr   r8   Útrack_eventr,   Ú
astimezoner   ÚutcÚ	isoformatÚ	tracebackÚ
format_exc)r9   Úkwargsr   ÚeventÚrun_idÚmodelÚprint_verboseÚextrar   rH   Úresponse_objÚ
start_timeÚend_timerM   rJ   r<   r=   r2   r>   ÚparamÚvaluer   rN   Ú	error_objs                           r   Ú	log_eventzLunaryLogger.log_eventb   sG  € ð D	NÙÐGÈÀwÐOÔPàˆKØ$*§J¡JÐ/?ÀÓ$DˆNØ%+§Z¡ZÐ0AÀ2Ó%FˆOØ,×0Ñ0°¸RÓ@ÒFÀBˆHáØ4˜5Ð4 OÐ4�à"Ÿ,™, v¨tÓ4ˆDáØ—	‘	˜,¨Ô-Ø—	‘	˜& $Ô'Ø#Ÿi™i¨¸Ó<×@Ñ@ÀÐPTÓU�ð !&§¡£ò ‘��uÜ! %¬#¬s´D¼%Ð)@ÕAÀeÈwÓFVðÜ'*¨5£z˜˜ešðñ Ø>EÈÑ>Uœ L°Ñ$9Ô:Ð[_�à4=ÀÑ4M˜ iÒ0ÐSW‘ð �Ø�áØ$ eÐ,‘	à �	à×Ñ×*Ñ*ØØØØ&Ÿl™l¨?¸DÓAØØÜ$ UÓ+Ø$×/Ñ/´·±Ó=×GÑGÓIØ'Ø!Ø!ØØð +ô ð  ×Ñ×*Ñ*ØØØØ"×-Ñ-¬h¯l©lÓ;×EÑEÓGØ!ØÜ% fÓ-Ø!ð +õ 	øôC %ò ÚðûôX ò 	NÙÐ3´I×4HÑ4HÓ4JÐ3KÐLÖMð	Nús=   ‚CG< Ã"G< Ã)G,Ã7C4G< Ç,	G9Ç5G< Ç8G9Ç9G< Ç<&H%È$H%)	Ú__name__Ú
__module__Ú__qualname__r:   r   Únowr   rX   rh   r   r   r   r.   r.   M   sC   „ òð6 ØØØØ�8—<‘< §¡Ó-Ø�—‘˜hŸl™lÓ+ØôTNr   r.   )r1   rZ   r   r   Útypingr   r4   r   r   r,   r.   r   r   r   ú<module>rn      s4   ðó Û ß 'Ý ã òòò4$÷@iNò iNr   