From eac2c170497a4ff203ef09853c76bad421cffce8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com>
Date: Tue, 15 Jul 2025 16:55:24 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E7=A7=BB=E9=99=A4LLMClient=E4=BE=9D?=
 =?UTF-8?q?=E8=B5=96=EF=BC=8C=E4=BC=98=E5=8C=96=E4=BF=A1=E6=81=AF=E6=8F=90?=
 =?UTF-8?q?=E5=8F=96=E6=A8=A1=E5=9D=97=E7=9A=84=E5=AF=BC=E5=85=A5=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scripts/info_extraction.py       | 4 +---
 src/chat/knowledge/ie_process.py | 4 +---
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/scripts/info_extraction.py b/scripts/info_extraction.py
index 4a77fd5b..92ec81a4 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 4314ca5e..bd0e1768 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: