diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py index ea2ad7a74..d14ebc51a 100644 --- a/rag/svr/task_executor.py +++ b/rag/svr/task_executor.py @@ -24,9 +24,6 @@ from graphrag.general.index import run_graphrag from graphrag.utils import get_llm_cache, set_llm_cache, get_tags_from_cache, set_tags_to_cache from rag.prompts import keyword_extraction, question_proposal, content_tagging -CONSUMER_NO = "0" if len(sys.argv) < 2 else sys.argv[1] -CONSUMER_NAME = "task_executor_" + CONSUMER_NO - import logging import os from datetime import datetime @@ -87,7 +84,9 @@ FACTORY = { } UNACKED_ITERATOR = None -CONSUMER_NAME = "task_consumer_" + CONSUMER_NO + +CONSUMER_NO = "0" if len(sys.argv) < 2 else sys.argv[1] +CONSUMER_NAME = "task_executor_" + CONSUMER_NO BOOT_AT = datetime.now().astimezone().isoformat(timespec="milliseconds") PENDING_TASKS = 0 LAG_TASKS = 0