From b89041b5b38c5d4b2185fe9bc7b84d557d6ac981 Mon Sep 17 00:00:00 2001 From: Magic_yuan <317617749@qq.com> Date: Wed, 11 Dec 2024 13:53:05 +0800 Subject: [PATCH] =?UTF-8?q?feat(operate):=20=E6=B7=BB=E5=8A=A0=E5=AE=9E?= =?UTF-8?q?=E4=BD=93=E7=B1=BB=E5=9E=8B=E9=85=8D=E7=BD=AE=E5=B9=B6=E4=BC=98?= =?UTF-8?q?=E5=8C=96=E6=8F=90=E7=A4=BA=E7=94=9F=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在全局配置中添加 entity_types 参数，用于自定义实体类型 - 在生成实体提取和关系提取的提示时，使用配置的实体类型替代默认值 - 优化了提示生成逻辑，提高了代码的可配置性和灵活性 --- lightrag/operate.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/lightrag/operate.py b/lightrag/operate.py index bc5a9b13..8b8ad85b 100644 --- a/lightrag/operate.py +++ b/lightrag/operate.py @@ -260,6 +260,9 @@ async def extract_entities( language = global_config["addon_params"].get( "language", PROMPTS["DEFAULT_LANGUAGE"] ) + entity_types = global_config["addon_params"].get( + "entity_types", PROMPTS["DEFAULT_ENTITY_TYPES"] + ) example_number = global_config["addon_params"].get("example_number", None) if example_number and example_number < len(PROMPTS["entity_extraction_examples"]): examples = "\n".join( @@ -272,7 +275,7 @@ async def extract_entities( tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"], record_delimiter=PROMPTS["DEFAULT_RECORD_DELIMITER"], completion_delimiter=PROMPTS["DEFAULT_COMPLETION_DELIMITER"], - entity_types=",".join(PROMPTS["DEFAULT_ENTITY_TYPES"]), + entity_types=",".join(entity_types), language=language, ) # add example's format @@ -283,7 +286,7 @@ async def extract_entities( tuple_delimiter=PROMPTS["DEFAULT_TUPLE_DELIMITER"], record_delimiter=PROMPTS["DEFAULT_RECORD_DELIMITER"], completion_delimiter=PROMPTS["DEFAULT_COMPLETION_DELIMITER"], - entity_types=",".join(PROMPTS["DEFAULT_ENTITY_TYPES"]), + entity_types=",".join(entity_types), examples=examples, language=language, )