feat: 移除LLMClient依赖,优化信息提取模块的导入逻辑

This commit is contained in:
墨梓柒
2025-07-15 16:55:24 +08:00
parent f15e074cca
commit eac2c17049
2 changed files with 2 additions and 6 deletions

View File

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

View File

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