o ưi@sDdZddlmZddlZddlZddlmZdd d Zdd dZdS)z{ Prometheus multiprocess directory cleanup utilities. Wipes all .db files on startup so workers start with a clean slate. ) annotationsN)verbose_proxy_logger directorystrreturnNonec Csttj|d}d}|D]*}z t||d7}Wqty8}ztd|d|WYd}~qd}~ww|rHtd|d|dSdS) zGDelete all .db files in the directory. Called once before workers fork.z*.dbrz'Failed to delete stale prometheus file z: NzPrometheus cleanup: wiped z stale .db files from ) globospathjoinremoveOSErrorrwarninginfo)rfilesZdeletedfilepatherW/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/prometheus_cleanup.pywipe_directorys"  r worker_pidintc Cs|tjdsdSzddlm}||td|dWdSty=}zt d|d|WYd}~dSd}~ww) zRRemove prometheus .db files for a dead worker. Called by gunicorn child_exit hook.ZPROMETHEUS_MULTIPROC_DIRNr) multiprocessz"Prometheus cleanup: marked worker z as deadz!Failed to mark prometheus worker z as dead: ) r environgetZprometheus_clientrZmark_process_deadrr Exceptionr)rrrrrrmark_worker_exit!s     r)rrrr)rrrr) __doc__ __future__rr r Zlitellm._loggingrrrrrrrs