
    gm!                       d dl mZ d dlZd dlZd dlmZmZ d dlmZm	Z	m
Z
mZmZ d dlmZ 	  ej                  d      Zeej"                  e<   y# e$ r ese
d   ZdZi Zd	 Z G d
 d      Z G d d      Z e       Zej6                  Zej8                  Zej:                  Zej<                  Zej>                  Zej@                  Z ejB                  Z!ejD                  Z"ejF                  Z#ejH                  Z$ejJ                  Z%ejL                  Z&ejN                  Z'ejP                  Z(ejR                  Z)ejT                  Z*ejV                  Z+ejX                  Z,ejZ                  Z-ej\                  Z.ej^                  Z/ej`                  Z0ejb                  Z1ejd                  Z2ejf                  Z3ejh                  Z4ejj                  Z5ejl                  Z6ejn                  Z7ejp                  Z8ejr                  Z9ejt                  Z:ejv                  Z;ejx                  Z<ejz                  Z=ej|                  Z>ej~                  Z?ej                  Z@ej                  ZAej                  ZBd'dZCd ZDed        ZE G d d      ZF G d d      ZG G d d      ZH G d d      ZI G d d      ZJ G d d      ZK G d d      ZL G d d       ZM G d! d"      ZN G d# d$      ZO G d% d&      ZPY yY yw xY w)(    )annotationsN)contextmanagernullcontext)AnyContextManagerLiteralTYPE_CHECKINGSequence	MagicMocklogfire)tracedebuginfonoticewarnwarningerrorfatalz0.0.0c                      y N )argskwargss     I/var/www/openai/venv/lib/python3.12/site-packages/logfire_api/__init__.py	configurer      s        c                      e Zd Zd Zd ZddZedd       Zedd       Zej                  dd       Zedd       Z
e
j                  dd       Z
dd	Zy
)LogfireSpanc                    t               S r   r   selfattrs     r   __getattr__zLogfireSpan.__getattr__   
     {"r   c                    | S r   r   r"   s    r   	__enter__zLogfireSpan.__enter__       r   c                     y r   r   r"   r   r   s      r   __exit__zLogfireSpan.__exit__       r   c                     yN r   r'   s    r   message_templatezLogfireSpan.message_template       r   c                     y)Nr   r   r'   s    r   tagszLogfireSpan.tags"   r2   r   c                     y r   r   )r"   new_tagss     r   r4   zLogfireSpan.tags&   s    =@r   c                     yr/   r   r'   s    r   messagezLogfireSpan.message)   r2   r   c                     y r   r   )r"   r8   s     r   r8   zLogfireSpan.message-   s    -0r   c                     y)NFr   r'   s    r   is_recordingzLogfireSpan.is_recording0   s    r   NreturnNone)r=   str)r=   ztuple[str, ...])r6   zSequence[str]r=   r>   )r8   r?   )r=   bool)__name__
__module____qualname__r$   r(   r,   propertyr1   r4   setterr8   r;   r   r   r   r   r      sv    # =    [[@ @  ^^0 0r   r   c                  D   e Zd Zd Zd*dZd+dZd*dZd*dZd*dZd*dZ	d*dZ
d*d	Zd*d
Zd*dZd*dZd*dZd,dZd,dZd*dZd*dZd*dZd Zd Zd Zd-dZd*dZd*dZd*dZd*dZd*dZd*dZd*dZd*dZ d*dZ!d*d Z"d*d!Z#d*d"Z$d*d#Z%d-d$Z&d-d%Z'd*d&Z(d*d'Z)d*d(Z*y)).Logfirec                    t               S r   r   r!   s     r   r$   zLogfire.__getattr__4   r%   r   c                     y r   r   r+   s      r   __init__zLogfire.__init__7   r-   r   c                    t               S r   )r   r+   s      r   spanzLogfire.span9   
    "}$r   c                     y r   r   r+   s      r   logzLogfire.log<   s    Cr   c                     y r   r   r+   s      r   r   zLogfire.trace>       cr   c                     y r   r   r+   s      r   r   zLogfire.debug@   rQ   r   c                     y r   r   r+   s      r   r   zLogfire.noticeB   s    sr   c                     y r   r   r+   s      r   r   zLogfire.infoD       Sr   c                     y r   r   r+   s      r   r   zLogfire.warnF   rU   r   c                     y r   r   r+   s      r   r   zLogfire.errorH   rQ   r   c                     y r   r   r+   s      r   	exceptionzLogfire.exceptionJ   s    #r   c                     y r   r   r+   s      r   r   zLogfire.fatalL   rQ   r   c                     y r   r   r+   s      r   suppress_scopeszLogfire.suppress_scopesN   s    r   c                    | S r   r   r+   s      r   	with_tagszLogfire.with_tagsP   r)   r   c                    | S r   r   r+   s      r   with_settingszLogfire.with_settingsS   r)   r   c                     y r   r   r+   s      r   force_flushzLogfire.force_flushV   s    Cr   c                    t               S r   r   r+   s      r   log_slow_async_callbacksz Logfire.log_slow_async_callbacksX   rM   r   c                     y r   r   r+   s      r   install_auto_tracingzLogfire.install_auto_tracing[       Sr   c                    d }|S )Nc                    | S r   r   )funcs    r   	decoratorz%Logfire.instrument.<locals>.decorator^   s    Kr   r   )r"   r   r   rl   s       r   
instrumentzLogfire.instrument]   s      ! r   c                    |S r   r   r"   appr   r   s       r   instrument_asgizLogfire.instrument_asgic       
r   c                    |S r   r   ro   s       r   instrument_wsgizLogfire.instrument_wsgif   rr   r   c                    t               S r   rd   r+   s      r   instrument_fastapizLogfire.instrument_fastapii   rM   r   c                     y r   r   r+   s      r   instrument_pydanticzLogfire.instrument_pydanticl       Cr   c                     y r   r   r+   s      r   instrument_pymongozLogfire.instrument_pymongon       3r   c                     y r   r   r+   s      r   instrument_sqlalchemyzLogfire.instrument_sqlalchemyp       cr   c                     y r   r   r+   s      r   instrument_sqlite3zLogfire.instrument_sqlite3r   r|   r   c                     y r   r   r+   s      r   instrument_aws_lambdazLogfire.instrument_aws_lambdat   r   r   c                     y r   r   r+   s      r   instrument_rediszLogfire.instrument_redisv       r   c                     y r   r   r+   s      r   instrument_flaskzLogfire.instrument_flaskx   r   r   c                     y r   r   r+   s      r   instrument_starlettezLogfire.instrument_starlettez   rh   r   c                     y r   r   r+   s      r   instrument_djangozLogfire.instrument_django|   s    #r   c                     y r   r   r+   s      r   instrument_psycopgzLogfire.instrument_psycopg~   r|   r   c                     y r   r   r+   s      r   instrument_requestszLogfire.instrument_requests   ry   r   c                     y r   r   r+   s      r   instrument_httpxzLogfire.instrument_httpx   r   r   c                     y r   r   r+   s      r   instrument_asyncpgzLogfire.instrument_asyncpg   r|   r   c                    t               S r   rd   r+   s      r   instrument_anthropiczLogfire.instrument_anthropic   rM   r   c                    t               S r   rd   r+   s      r   instrument_openaizLogfire.instrument_openai   rM   r   c                     y r   r   r+   s      r   instrument_aiohttp_clientz!Logfire.instrument_aiohttp_client       #r   c                     y r   r   r+   s      r   instrument_system_metricsz!Logfire.instrument_system_metrics   r   r   c                     y r   r   r+   s      r   shutdownzLogfire.shutdown   r-   r   Nr<   )r=   r   )r=   rG   )r=   zContextManager[None])+rA   rB   rC   r$   rJ   rL   rO   r   r   r   r   r   r   rY   r   r\   r^   r`   rb   re   rg   rm   rq   rt   rv   rx   r{   r~   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   rG   rG   3   s    # =% 899:889=9C @% I!% HFIFIDDHEFGDF%% NM<r   rG   c                      y r   r   r   r   r   loguru_handlerr      s    r   c                    | S r   r   )xs    r   no_auto_tracer      s    Hr   c               #     K   d  y wr   r   r   r   r   suppress_instrumentationr      s	     s   c                      e Zd ZddZy)ConsoleOptionsc                     y r   r   r+   s      r   rJ   zConsoleOptions.__init__   r-   r   Nr<   rA   rB   rC   rJ   r   r   r   r   r          <r   r   c                      e Zd ZddZy)SamplingOptionsc                     y r   r   r+   s      r   rJ   zSamplingOptions.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)
CodeSourcec                     y r   r   r+   s      r   rJ   zCodeSource.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)ScrubbingOptionsc                     y r   r   r+   s      r   rJ   zScrubbingOptions.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)AdvancedOptionsc                     y r   r   r+   s      r   rJ   zAdvancedOptions.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)MetricsOptionsc                     y r   r   r+   s      r   rJ   zMetricsOptions.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)PydanticPluginc                     y r   r   r+   s      r   rJ   zPydanticPlugin.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)
ScrubMatchc                     y r   r   r+   s      r   rJ   zScrubMatch.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)AutoTraceModulec                     y r   r   r+   s      r   rJ   zAutoTraceModule.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)StructlogProcessorc                     y r   r   r+   s      r   rJ   zStructlogProcessor.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   c                      e Zd ZddZy)LogfireLoggingHandlerc                     y r   r   r+   s      r   rJ   zLogfireLoggingHandler.__init__   r-   r   Nr<   r   r   r   r   r   r      r   r   r   )r=   zdict[str, Any])Q
__future__r   	importlibsys
contextlibr   r   typingr   r   r   r	   r
   unittest.mockr   import_modulelogfire_modulemodulesrA   ImportError	LevelNameVERSIONMETRICS_PREFERRED_TEMPORALITYr   r   rG   DEFAULT_LOGFIRE_INSTANCErL   rO   r   r   r   r   r   r   rY   r   r^   r`   rb   re   rg   rm   rq   rt   rx   rv   r   r   r   instrument_celeryr   r   r   r   r   r   r   r~   r   r   r   r{   instrument_mysqlr   r   r\   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   <module>r      s   "  
 2 H H #[=,Y,,Y7N*CKK W=cd	(*%+	 	<]	= ]	=~ $+9 ',,&**(..(..)00',,',,(..,66	(..,66	0>>.::#;#T#T 7LL-88
2BB2BB6JJ5HH4FF7LL5HH4FF3DD6JJ5HH4FF3DD7LL$<$V$V! 8 N N5HH 8 N N3DD5HH3DD$<$V$V!+442BB3	 
	 
		= 	=	= 	=	= 	=	= 	=	= 	=	= 	=	= 	=	= 	=	= 	=	= 	=	= 	=k W=s   !A JK K 