BruceSun
/
MaiBot
mirror of https://github.com/Mai-with-u/MaiBot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

🤖 自动格式化代码 [skip ci]

pull/927/head
github-actions[bot] 2025-05-03 16:32:25 +00:00
parent b8d14add91
commit cea176d63d
2 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/import_openie.py
View File

@ -30,7 +30,6 @@ OPENIE_DIR = (
logger = get_module_logger("OpenIE导入") logger = get_module_logger("OpenIE导入")
def hash_deduplicate( def hash_deduplicate(
raw_paragraphs: dict[str, str], raw_paragraphs: dict[str, str],
triple_list_data: dict[str, list[list[str]]], triple_list_data: dict[str, list[list[str]]],
@ -168,7 +167,6 @@ def main():
global_config["llm_providers"][key]["api_key"], global_config["llm_providers"][key]["api_key"],
) )
# 初始化Embedding库 # 初始化Embedding库
embed_manager = EmbeddingManager(llm_client_list[global_config["embedding"]["provider"]]) embed_manager = EmbeddingManager(llm_client_list[global_config["embedding"]["provider"]])
logger.info("正在从文件加载Embedding库") logger.info("正在从文件加载Embedding库")

2
src/plugins/knowledge/src/embedding_store.py
View File

@ -44,7 +44,7 @@ EMBEDDING_TEST_STRINGS = [
"来玩巧克甜恋谢谢喵", "来玩巧克甜恋谢谢喵",
"水印", "水印",
"我也在纠结晚饭，铁锅炒鸡听着就香！", "我也在纠结晚饭，铁锅炒鸡听着就香！",
"test你妈喵" "test你妈喵",
] ]
EMBEDDING_TEST_FILE = os.path.join(ROOT_PATH, "data", "embedding_model_test.json") EMBEDDING_TEST_FILE = os.path.join(ROOT_PATH, "data", "embedding_model_test.json")
EMBEDDING_SIM_THRESHOLD = 0.99 EMBEDDING_SIM_THRESHOLD = 0.99