diff --git a/scripts/import_openie.py b/scripts/import_openie.py index b7fc9f307..851cc8b31 100644 --- a/scripts/import_openie.py +++ b/scripts/import_openie.py @@ -30,7 +30,6 @@ OPENIE_DIR = ( logger = get_module_logger("OpenIE导入") - def hash_deduplicate( raw_paragraphs: dict[str, str], triple_list_data: dict[str, list[list[str]]], @@ -168,7 +167,6 @@ def main(): global_config["llm_providers"][key]["api_key"], ) - # 初始化Embedding库 embed_manager = EmbeddingManager(llm_client_list[global_config["embedding"]["provider"]]) logger.info("正在从文件加载Embedding库") diff --git a/src/plugins/knowledge/src/embedding_store.py b/src/plugins/knowledge/src/embedding_store.py index c68886fd0..5ee92a869 100644 --- a/src/plugins/knowledge/src/embedding_store.py +++ b/src/plugins/knowledge/src/embedding_store.py @@ -44,7 +44,7 @@ EMBEDDING_TEST_STRINGS = [ "来玩巧克甜恋谢谢喵", "水印", "我也在纠结晚饭,铁锅炒鸡听着就香!", - "test你妈喵" + "test你妈喵", ] EMBEDDING_TEST_FILE = os.path.join(ROOT_PATH, "data", "embedding_model_test.json") EMBEDDING_SIM_THRESHOLD = 0.99