gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/info_extraction.py
View File

@@ -4,7 +4,6 @@ import signal
from concurrent.futures import ThreadPoolExecutor, as_completed from concurrent.futures import ThreadPoolExecutor, as_completed
from threading import Lock, Event from threading import Lock, Event
import sys import sys
import glob
import datetime import datetime
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) 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.common.logger import get_logger
# from src.chat.knowledge.lpmmconfig import global_config # 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.ie_process import info_extract_from_str
from src.chat.knowledge.llm_client import LLMClient
from src.chat.knowledge.open_ie import OpenIE from src.chat.knowledge.open_ie import OpenIE
from rich.progress import ( from rich.progress import (
BarColumn, BarColumn,

4
src/chat/knowledge/ie_process.py
View File

@@ -4,9 +4,7 @@ from typing import List, Union
from .global_logger import logger from .global_logger import logger
from . import prompt_template from . import prompt_template
from .lpmmconfig import global_config, INVALID_ENTITY from .knowledge_lib import INVALID_ENTITY
from .llm_client import LLMClient
from src.chat.knowledge.utils.json_fix import new_fix_broken_generated_json
from src.llm_models.utils_model import LLMRequest from src.llm_models.utils_model import LLMRequest
from json_repair import repair_json from json_repair import repair_json
def _extract_json_from_text(text: str) -> dict: def _extract_json_from_text(text: str) -> dict: