🤖 自动格式化代码 [skip ci]
This commit is contained in:
@@ -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库")
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user