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)