pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 12:32:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor code (#11694)

Browse Source 
### What problem does this PR solve?

Rename function and refactor log message

### Type of change

- [x] Refactoring

Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2025-12-03 15:15:00 +08:00
committed by GitHub
parent b44e65a12e
commit 3c50c7d3ac
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
common/misc_utils.py
View File

@ -153,7 +153,7 @@ def parse_mineru_paths() -> Dict[str, Path]:
@once @once
def install_mineru() -> None: def check_and_install_mineru() -> None:
""" """
Ensure MinerU is installed. Ensure MinerU is installed.

6
rag/svr/task_executor.py
View File

@ -68,7 +68,7 @@ from common.signal_utils import start_tracemalloc_and_snapshot, stop_tracemalloc
from common.exceptions import TaskCanceledException from common.exceptions import TaskCanceledException
from common import settings from common import settings
from common.constants import PAGERANK_FLD, TAG_FLD, SVR_CONSUMER_GROUP_NAME 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 BATCH_SIZE = 64
@ -1101,8 +1101,8 @@ async def main():
show_configs() show_configs()
settings.init_settings() settings.init_settings()
settings.check_and_install_torch() settings.check_and_install_torch()
install_mineru() check_and_install_mineru()
logging.info(f'settings.EMBEDDING_CFG: {settings.EMBEDDING_CFG}') logging.info(f'default embedding config: {settings.EMBEDDING_CFG}')
settings.print_rag_settings() settings.print_rag_settings()
if sys.platform != "win32": if sys.platform != "win32":
signal.signal(signal.SIGUSR1, start_tracemalloc_and_snapshot) signal.signal(signal.SIGUSR1, start_tracemalloc_and_snapshot)