fix continue prompt format error

2025-03-17 16:58:04 +08:00
parent 3df20ae787
commit 60dd13f17e
1 changed files with 1 additions and 1 deletions
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -404,7 +404,7 @@ async def extract_entities(
        language=language,
    )

-    continue_prompt = PROMPTS["entity_continue_extraction"]
+    continue_prompt = PROMPTS["entity_continue_extraction"].format(**context_base)
    if_loop_prompt = PROMPTS["entity_if_loop_extraction"]

    processed_chunks = 0