From dfe8ea941c380a14daadd3f2ff2f3c5db6f90a83 Mon Sep 17 00:00:00 2001 From: xiaolei373 Date: Tue, 23 Dec 2025 10:05:45 +0800 Subject: [PATCH] [log]console log to llm log (#5680) --- fastdeploy/metrics/prometheus_multiprocess_setup.py | 6 +++--- tests/metrics/test_prometheus_multiprocess_setup.py | 6 +++--- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/fastdeploy/metrics/prometheus_multiprocess_setup.py b/fastdeploy/metrics/prometheus_multiprocess_setup.py index 3a826c55a6..509fb7d158 100644 --- a/fastdeploy/metrics/prometheus_multiprocess_setup.py +++ b/fastdeploy/metrics/prometheus_multiprocess_setup.py @@ -18,7 +18,7 @@ import os import shutil import uuid -from fastdeploy.utils import console_logger +from fastdeploy.utils import llm_logger def setup_multiprocess_prometheus(): @@ -33,12 +33,12 @@ def setup_multiprocess_prometheus(): if os.path.exists(prom_dir): shutil.rmtree(prom_dir, ignore_errors=True) os.makedirs(prom_dir, exist_ok=True) - console_logger.info(f"PROMETHEUS_MULTIPROC_DIR is set to be {prom_dir}") + llm_logger.info(f"PROMETHEUS_MULTIPROC_DIR is set to be {prom_dir}") os.environ["PROMETHEUS_MULTIPROC_DIR"] = prom_dir return prom_dir else: prom_dir = os.environ["PROMETHEUS_MULTIPROC_DIR"] - console_logger.warning( + llm_logger.warning( f"Found PROMETHEUS_MULTIPROC_DIR:{prom_dir} was set by user. " "you will find inaccurate metrics. Unset the variable " "will properly handle cleanup." diff --git a/tests/metrics/test_prometheus_multiprocess_setup.py b/tests/metrics/test_prometheus_multiprocess_setup.py index f2341a68f0..5ccc960d52 100644 --- a/tests/metrics/test_prometheus_multiprocess_setup.py +++ b/tests/metrics/test_prometheus_multiprocess_setup.py @@ -37,7 +37,7 @@ class TestSetupMultiprocessPrometheus: with ( patch("uuid.uuid4", return_value="test-uuid"), - patch("fastdeploy.utils.console_logger.info") as mock_logger, + patch("fastdeploy.utils.llm_logger.info") as mock_logger, ): result = setup_multiprocess_prometheus() @@ -51,7 +51,7 @@ class TestSetupMultiprocessPrometheus: test_dir = "/tmp/existing_dir" os.environ["PROMETHEUS_MULTIPROC_DIR"] = test_dir - with patch("fastdeploy.utils.console_logger.warning") as mock_logger: + with patch("fastdeploy.utils.llm_logger.warning") as mock_logger: result = setup_multiprocess_prometheus() assert result == test_dir @@ -69,7 +69,7 @@ class TestSetupMultiprocessPrometheus: with ( patch("os.path.exists", return_value=True), patch("uuid.uuid4", return_value="test-uuid"), - patch("fastdeploy.utils.console_logger.info") as mock_logger, + patch("fastdeploy.utils.llm_logger.info") as mock_logger, ): # 模拟 rmtree 但确保 ignore_errors=True 生效