From 4eae5cbeeabbb02a078ec46ab62f524eab89aa63 Mon Sep 17 00:00:00 2001 From: Robert Shaw Date: Sun, 20 Jul 2025 16:50:36 +0000 Subject: [PATCH] updated Signed-off-by: Robert Shaw --- vllm/v1/engine/async_llm.py | 7 +++---- vllm/v1/metrics/loggers.py | 11 ----------- 2 files changed, 3 insertions(+), 15 deletions(-) diff --git a/vllm/v1/engine/async_llm.py b/vllm/v1/engine/async_llm.py index d6ffbba323423..b8a40f77df314 100644 --- a/vllm/v1/engine/async_llm.py +++ b/vllm/v1/engine/async_llm.py @@ -36,7 +36,7 @@ from vllm.v1.engine.output_processor import (OutputProcessor, from vllm.v1.engine.parallel_sampling import ParentRequest from vllm.v1.engine.processor import Processor from vllm.v1.executor.abstract import Executor -from vllm.v1.metrics.loggers import StatLoggerFactory, setup_default_loggers +from vllm.v1.metrics.loggers import StatLoggerFactory, StatLoggerManager from vllm.v1.metrics.prometheus import shutdown_prometheus from vllm.v1.metrics.stats import IterationStats @@ -102,12 +102,11 @@ class AsyncLLM(EngineClient): engine_idxs = [ idx for idx in range(start_idx, start_idx + local_engines) ] - self.logger_manager = setup_default_loggers( + self.logger_manager = StatLoggerManager( vllm_config=vllm_config, - log_stats=self.log_stats, engine_idxs=engine_idxs, custom_stat_loggers=stat_loggers, - ) + ) if self.log_stats else None # Tokenizer (+ ensure liveness if running in another process). self.tokenizer = init_tokenizer_from_configs( diff --git a/vllm/v1/metrics/loggers.py b/vllm/v1/metrics/loggers.py index 6de65280364d8..b6b8b6d0a4d86 100644 --- a/vllm/v1/metrics/loggers.py +++ b/vllm/v1/metrics/loggers.py @@ -600,17 +600,6 @@ def build_1_2_5_buckets(max_value: int) -> list[int]: return build_buckets([1, 2, 5], max_value) -def setup_default_loggers( - vllm_config: VllmConfig, - log_stats: bool, - engine_idxs: list[int], - custom_stat_loggers: Optional[list[StatLoggerFactory]] = None, -) -> Optional["StatLoggerManager"]: - """Setup logging and prometheus metrics.""" - return (None if not log_stats else StatLoggerManager( - vllm_config, engine_idxs, custom_stat_loggers)) - - class StatLoggerManager: """ StatLoggerManager: