From 273ee08fb2f10068eff5bca7c7b5aa5b7206b63e 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 17:00:19 +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=E7=9B=B4=E6=8E=A5=E5=88=9D=E5=A7=8B=E5=8C=96?= =?UTF-8?q?EmbeddingManager=E5=B9=B6=E7=AE=80=E5=8C=96OpenIE=E6=95=B0?= =?UTF-8?q?=E6=8D=AE=E7=9B=AE=E5=BD=95=E8=B7=AF=E5=BE=84=E8=8E=B7=E5=8F=96?= =?UTF-8?q?=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/import_openie.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/scripts/import_openie.py b/scripts/import_openie.py index 144b1c014..791c64672 100644 --- a/scripts/import_openie.py +++ b/scripts/import_openie.py @@ -9,10 +9,7 @@ import os from time import sleep sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) - -from src.chat.knowledge.lpmmconfig import global_config from src.chat.knowledge.embedding_store import EmbeddingManager -from src.chat.knowledge.llm_client import LLMClient from src.chat.knowledge.open_ie import OpenIE from src.chat.knowledge.kg_manager import KGManager from src.common.logger import get_logger @@ -22,7 +19,7 @@ from src.manager.local_store_manager import local_storage # 添加项目根目录到 sys.path ROOT_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__), "..")) -OPENIE_DIR = global_config["persistence"]["openie_data_path"] or os.path.join(ROOT_PATH, "data", "openie") +OPENIE_DIR = os.path.join(ROOT_PATH, "data", "openie") logger = get_logger("OpenIE导入") @@ -194,15 +191,9 @@ def main(): # sourcery skip: dict-comprehension logger.info("----开始导入openie数据----\n") logger.info("创建LLM客户端") - llm_client_list = {} - for key in global_config["llm_providers"]: - llm_client_list[key] = LLMClient( - global_config["llm_providers"][key]["base_url"], - global_config["llm_providers"][key]["api_key"], - ) # 初始化Embedding库 - embed_manager = EmbeddingManager(llm_client_list[global_config["embedding"]["provider"]]) + embed_manager = EmbeddingManager() logger.info("正在从文件加载Embedding库") try: embed_manager.load_from_file()