Update prompts

2025-03-09 01:21:39 +08:00
parent da59cc89d8
commit 548f9a8234
2 changed files with 82 additions and 44 deletions
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -384,8 +384,8 @@ async def extract_entities(
        language=language,
    )

-    continue_prompt = PROMPTS["entiti_continue_extraction"]
-    if_loop_prompt = PROMPTS["entiti_if_loop_extraction"]
+    continue_prompt = PROMPTS["entity_continue_extraction"]
+    if_loop_prompt = PROMPTS["entity_if_loop_extraction"]

    processed_chunks = 0
    total_chunks = len(ordered_chunks)