fix entity extract

HKUDS · Dec 4, 2024 · 7f8460c · 7f8460c
1 parent f3ae4fc
commit 7f8460c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lightrag/operate.py b/lightrag/operate.py
@@ -296,7 +296,9 @@ async def _process_single_content(chunk_key_dp: tuple[str, TextChunkSchema]):
         chunk_key = chunk_key_dp[0]
         chunk_dp = chunk_key_dp[1]
         content = chunk_dp["content"]
-        hint_prompt = entity_extract_prompt.format(**context_base, input_text=content)
+        # hint_prompt = entity_extract_prompt.format(**context_base, input_text=content)
+        hint_prompt = entity_extract_prompt.format(**context_base, input_text="{input_text}").format(**context_base, input_text=content)
+
         final_result = await use_llm_func(hint_prompt)
         history = pack_user_ass_to_openai_messages(hint_prompt, final_result)
         for now_glean_index in range(entity_extract_max_gleaning):