gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Simplify entity extraction prompt formatting logic

Browse Source
This commit is contained in:
yangdx
2025-04-23 18:55:03 +08:00
parent c48ead6d8e
commit a6f0c0b33b
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lightrag/operate.py
View File

@@ -601,9 +601,7 @@ async def extract_entities(
file_path = chunk_dp.get("file_path", "unknown_source") file_path = chunk_dp.get("file_path", "unknown_source")
# Get initial extraction # Get initial extraction
hint_prompt = entity_extract_prompt.format( hint_prompt = entity_extract_prompt.format(**{**context_base, "input_text": content})
**context_base, input_text="{input_text}"
).format(**context_base, input_text=content)
final_result = await use_llm_func_with_cache( final_result = await use_llm_func_with_cache(
hint_prompt, hint_prompt,