From 3c50c7d3ac8a4002c30883aebd6ecbbc70a9cb05 Mon Sep 17 00:00:00 2001 From: Jin Hai Date: Wed, 3 Dec 2025 15:15:00 +0800 Subject: [PATCH] Refactor code (#11694) ### What problem does this PR solve? Rename function and refactor log message ### Type of change - [x] Refactoring Signed-off-by: Jin Hai --- common/misc_utils.py | 2 +- rag/svr/task_executor.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/common/misc_utils.py b/common/misc_utils.py index 694b8b494..08856fc56 100644 --- a/common/misc_utils.py +++ b/common/misc_utils.py @@ -153,7 +153,7 @@ def parse_mineru_paths() -> Dict[str, Path]: @once -def install_mineru() -> None: +def check_and_install_mineru() -> None: """ Ensure MinerU is installed. diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index 25dea2b24..7812740d7 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -68,7 +68,7 @@ from common.signal_utils import start_tracemalloc_and_snapshot, stop_tracemalloc from common.exceptions import TaskCanceledException from common import settings from common.constants import PAGERANK_FLD, TAG_FLD, SVR_CONSUMER_GROUP_NAME -from common.misc_utils import install_mineru +from common.misc_utils import check_and_install_mineru BATCH_SIZE = 64 @@ -1101,8 +1101,8 @@ async def main(): show_configs() settings.init_settings() settings.check_and_install_torch() - install_mineru() - logging.info(f'settings.EMBEDDING_CFG: {settings.EMBEDDING_CFG}') + check_and_install_mineru() + logging.info(f'default embedding config: {settings.EMBEDDING_CFG}') settings.print_rag_settings() if sys.platform != "win32": signal.signal(signal.SIGUSR1, start_tracemalloc_and_snapshot)