o á)i ã@s|UddlZddlZddlmZddlmZmZmZddlm Z e e ƒZ da eej ed<dd„Zdd „Zd d „Zd d „ZdS)éN)ÚOptional)ÚREGISTRYÚCollectorRegistryÚ multiprocess)Ú init_loggerÚ_prometheus_multiproc_dircCs>dtjvrt ¡atjtjd<t dtj¡dSt d¡dS)zPSet up prometheus multiprocessing directory if not already configured. ÚPROMETHEUS_MULTIPROC_DIRz&Created PROMETHEUS_MULTIPROC_DIR at %sz½Found PROMETHEUS_MULTIPROC_DIR was set by user. This directory must be wiped between vLLM runs or you will find inaccurate metrics. Unset the variable and vLLM will properly handle cleanup.N) ÚosÚenvironÚtempfileÚTemporaryDirectoryrÚnameÚloggerÚdebugÚwarning©rrúf/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/vllm/v1/metrics/prometheus.pyÚsetup_multiprocess_prometheuss  ÿrcCs0t d¡durt d¡tƒ}t |¡|StS)z“Get the appropriate prometheus registry based on multiprocessing configuration. Returns: Registry: A prometheus registry rNz2Using multiprocess registry for prometheus metrics)r ÚgetenvrrrrZMultiProcessCollectorr)ÚregistryrrrÚget_prometheus_registry's   rcCs6t}t|jƒD]}t|dƒrd|jvr| |¡qdS)a&Unregister any existing vLLM collectors from the prometheus registry. This is useful for testing and CI/CD where metrics may be registered multiple times across test runs. Also, in case of multiprocess, we need to unregister the metrics from the global registry. Ú_nameZvllmN)rÚlistZ_collector_to_namesÚhasattrrÚ unregister)rÚ collectorrrrÚunregister_vllm_metrics7s   €þrc Csnt}|durdSzt ¡}t ||¡t d|¡WdSty6}zt dt |ƒ¡WYd}~dSd}~ww)zShutdown prometheus metrics.Nz0Marked Prometheus metrics for process %d as deadz Error during metrics cleanup: %s) rr ÚgetpidrZmark_process_deadrrÚ ExceptionÚerrorÚstr)ÚpathÚpidÚerrrÚshutdown_prometheusGs €ÿr$)r r ÚtypingrZprometheus_clientrrrZ vllm.loggerrÚ__name__rrr Ú__annotations__rrrr$rrrrÚs