Simplify entity extraction prompt formatting logic

2025-04-23 18:55:03 +08:00
parent c48ead6d8e
commit a6f0c0b33b
1 changed files with 1 additions and 3 deletions
--- a/lightrag/operate.py
+++ b/lightrag/operate.py
@@ -601,9 +601,7 @@ async def extract_entities(
        file_path = chunk_dp.get("file_path", "unknown_source")

        # Get initial extraction
-        hint_prompt = entity_extract_prompt.format(
-            **context_base, input_text="{input_text}"
-        ).format(**context_base, input_text=content)
+        hint_prompt = entity_extract_prompt.format(**{**context_base, "input_text": content})

        final_result = await use_llm_func_with_cache(
            hint_prompt,