Ë
    tPêj}  ã                   óª   — d dl Z d dlZd dlmZ d dlmZmZmZ d dlm	Z	 d dl
mZmZ d dlmZ d dlmZ  G d„ d	e«      Z G d
„ dee«      Z G d„ d«      Zy)é    N)ÚEnum)ÚAnyÚFinalÚ
NamedTuple)ÚLiteralString)Úprint_verboseÚverbose_logger)Úuuid)Úredact_user_api_key_infoc                   ó,   — e Zd ZU eed<   eeef   ed<   y)Ú
SpanConfigÚmessage_templateÚ	span_dataN)Ú__name__Ú
__module__Ú__qualname__r   Ú__annotations__ÚdictÚstrr   © ó    úh/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/logfire_logger.pyr   r      s   … Ø#Ó#Ø�C˜�H‰~Ôr   r   c                   ó   — e Zd ZdZdZy)ÚLogfireLevelÚinfoÚerrorN)r   r   r   ÚINFOÚERRORr   r   r   r   r      s   „ Ø€DØ�Er   r   c                   ó6   — e Zd Zd„ Zdefd„Zdefd„Zdefd„Zy)ÚLogfireLoggerc                 ó  — 	 t        j                  d«       dd l}|j                  j                  j
                  r&|j                  t        j                  d«      ¬«       y y # t        $ r}t        d|› �«       |‚d }~ww xY w)Nzin init logfire loggerr   ÚLOGFIRE_TOKEN)Útokenz%Got exception on init logfire client )r	   ÚdebugÚlogfireÚDEFAULT_LOGFIRE_INSTANCEÚconfigÚsend_to_logfireÚ	configureÚosÚgetenvÚ	Exceptionr   )Úselfr%   Úes      r   Ú__init__zLogfireLogger.__init__   su   € ð
	Ü× Ñ Ð!9Ô:Ûð ×/Ñ/×6Ñ6×FÒFØ×!Ñ!¬¯	©	°/Ó(BÐ!ÕCð Gøäò 	ÜÐAÀ!ÀÐEÔFØˆGûð	ús   ‚AA" Á"	B Á+A;Á;B Úreturnc                 óÚ   — |d   dk(  s|d   dk(  rt        dd|i¬«      S |d   dk(  s|d   dk(  rt        d	d|i¬«      S |d   d
k(  s|d   dk(  rt        dd|i¬«      S t        dd|i¬«      S )NÚ	call_typeÚ
completionÚacompletionz,Chat Completion with {request_data[model]!r}Úrequest_data)r   r   Ú	embeddingÚ
aembeddingz/Embedding Creation with {request_data[model]!r}Úimage_generationÚaimage_generationz-Image Generation with {request_data[model]!r}z)Litellm Call with {request_data[model]!r})r   )r-   Úpayloads     r   Ú_get_span_configzLogfireLogger._get_span_config)   sµ   € Ø�;Ñ <Ò/°7¸;Ñ3GÈ=Ò3XÜØ!OØ)¨7Ð3ôð ð �[Ñ! [Ò0°G¸KÑ4HÈLÒ4XÜØ!RØ)¨7Ð3ôð ð �[Ñ!Ð%7Ò7¸7À;Ñ;OÐSfÒ;fÜØ!PØ)¨7Ð3ôð ô
 Ø!LØ)¨7Ð3ôð r   Úlevelc              ƒ   ó:   K  — | j                  ||||||¬«       y ­w)N)ÚkwargsÚresponse_objÚ
start_timeÚend_timer   r<   )Ú	log_event)r-   r>   r?   r@   rA   r   r<   s          r   Ú_async_log_eventzLogfireLogger._async_log_event?   s+   è ø€ ð 	�‰ØØ%Ø!ØØ'Øð 	õ 	
ùs   ‚c                 ó„  — 	 dd l }t        j                  d|«       |si }|j                  di «      }|j                  di «      xs i }	|j                  d«      }
|j                  di «      }|j                  dd«      }|j                  d	d
«      }|j                  di «      }|j                  dt	        t        j                  «       «      «      }	 ||z
  j                  «       }i }t        |	t        «      r"|	j                  «       D ]  \  }}|dv rŒ|||<   Œ t        |¬«      }|||||||j                  dd«      |j                  dd«      ||j                  dd«      |
|||dœ}|j                  d¬«      }| j                  |«      \  }}|t        j                   k(  r |j"                  |fi |¤Ž n*|t        j$                  k(  r |j&                  |fi |¤ddi¤Ž  |d|› �«        |d|› �«       y # t        $ r d }Y �Œw xY w# t        $ r3}t        j                  d|t)        j*                  «       «       Y d }~y d }~ww xY w)Nr   z6logfire Logging - Enters logging function for model %sÚlitellm_paramsÚmetadataÚmessagesÚoptional_paramsr2   r3   Ú	cache_hitFÚusageÚid)ÚendpointÚcaching_groupsÚprevious_models)rF   ÚmodelÚ ÚuserÚresponse_cost)rK   r2   rI   Ú	startTimeÚendTimezresponseTime (seconds)rO   rQ   ÚmodelParametersÚspendrG   ÚresponserJ   rF   Úopenai)Úcustom_scope_suffixÚ	_exc_infoTz
dd Logger - Logging payload = z/Logfire Layer Logging - final response object: zLogfire Layer Error - %s
%s)r%   r	   r$   Úgetr   r
   Úuuid4Útotal_secondsr,   Ú
isinstancer   Úitemsr   Úwith_settingsr;   r   r   r   r   r   Ú	tracebackÚ
format_exc)r-   r>   r@   rA   r   r<   r?   r%   rE   rF   rG   rH   r2   rI   rJ   rK   Úresponse_timeÚclean_metadataÚkeyÚvaluer:   Úlogfire_openair   r   r.   s                            r   rB   zLogfireLogger.log_eventQ   sp  € ðH	\Ûä× Ñ Ð!YÐ[aÔbáØ!�Ø$*§J¡JÐ/?ÀÓ$DˆNØ,×0Ñ0°¸RÓ@ÒFÀBˆHØ$Ÿj™j¨Ó4ˆHØ%+§Z¡ZÐ0AÀ2Ó%FˆOØ%Ÿz™z¨+°|ÓDˆIØ%Ÿz™z¨+°uÓ=ˆIØ'×+Ñ+¨G°RÓ8ˆEØ$×(Ñ(¨¬s´4·:±:³<Ó/@ÓAˆBð%Ø!)¨JÑ!6× EÑ EÓ G�ð  ˆNÜ˜(¤DÔ)Ø"*§.¡.Ó"2ò 	4‘J�C˜àð ñ ð
 !à.3˜ sÒ+ð	4ô 6¸~ÔNˆNð Ø&Ø&Ø'Ø#Ø*7ØŸ™ G¨RÓ0ØŸ
™
 6¨2Ó.Ø#2ØŸ™ O°QÓ7Ø$Ø(ØØ*ñˆGð  %,×$9Ñ$9ÈhÐ$9Ó$WˆNØ*.×*?Ñ*?ÀÓ*HÑ'Ð˜iØœ×)Ñ)Ò)Ø#�×#Ñ#Ø$ñàóð œ,×,Ñ,Ò,Ø$�×$Ñ$Ø$ñàñð #óñ
 Ð<¸W¸IÐFÔGáÐKÈLÈ>ÐZÕ[øôk ò %Ø $“ð%ûôl ò 	\Ü× Ñ Ð!?ÀÄI×DXÑDXÓDZ×[Ñ[ûð	\ús=   ‚CH ÃG1 Ã"DH Ç1H Ç<H Ç?H È H È	H?È)H:È:H?N)	r   r   r   r/   r   r;   r   rC   rB   r   r   r   r    r       s5   „ òð¨:ó ð,
ð ó
ð$Q\ð ôQ\r   r    )r*   ra   Úenumr   Útypingr   r   r   Útyping_extensionsr   Úlitellm._loggingr   r	   Úlitellm._uuidr
   Ú*litellm.litellm_core_utils.redact_messagesr   r   r   r   r    r   r   r   ú<module>rn      sJ   ðó 
Û Ý ß )Ñ )å +ç :Ý Ý Oô�ô ô
�3˜ô ÷
H\ò H\r   