|
|
|
@@ -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 |