diff --git a/scripts/info_extraction.py b/scripts/info_extraction.py index 4a77fd5b3..92ec81a46 100644 --- a/scripts/info_extraction.py +++ b/scripts/info_extraction.py @@ -4,7 +4,6 @@ import signal from concurrent.futures import ThreadPoolExecutor, as_completed from threading import Lock, Event import sys -import glob import datetime sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) @@ -14,8 +13,7 @@ from rich.progress import Progress # 替换为 rich 进度条 from src.common.logger import get_logger # from src.chat.knowledge.lpmmconfig import global_config -from src.chat.knowledge.ie_process import _entity_extract, info_extract_from_str -from src.chat.knowledge.llm_client import LLMClient +from src.chat.knowledge.ie_process import info_extract_from_str from src.chat.knowledge.open_ie import OpenIE from rich.progress import ( BarColumn, diff --git a/src/chat/knowledge/ie_process.py b/src/chat/knowledge/ie_process.py index 4314ca5e6..bd0e17684 100644 --- a/src/chat/knowledge/ie_process.py +++ b/src/chat/knowledge/ie_process.py @@ -4,9 +4,7 @@ from typing import List, Union from .global_logger import logger from . import prompt_template -from .lpmmconfig import global_config, INVALID_ENTITY -from .llm_client import LLMClient -from src.chat.knowledge.utils.json_fix import new_fix_broken_generated_json +from .knowledge_lib import INVALID_ENTITY from src.llm_models.utils_model import LLMRequest from json_repair import repair_json def _extract_json_from_text(text: str) -> dict: