gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix linting

Browse source
This commit is contained in:
yangdx 2025-03-11 12:23:51 +08:00
parent 061350b2bf
commit 9d1dc2c9c3
1 changed files with 21 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
lightrag/operate.py
View file

@ -518,7 +518,9 @@ async def extract_entities(
history = pack_user_ass_to_openai_messages(hint_prompt, final_result) history = pack_user_ass_to_openai_messages(hint_prompt, final_result)
# Process initial extraction # Process initial extraction
maybe_nodes, maybe_edges = await _process_extraction_result(final_result, chunk_key) maybe_nodes, maybe_edges = await _process_extraction_result(
final_result, chunk_key
)
# Process additional gleaning results # Process additional gleaning results
for now_glean_index in range(entity_extract_max_gleaning): for now_glean_index in range(entity_extract_max_gleaning):
@ -529,7 +531,9 @@ async def extract_entities(
history += pack_user_ass_to_openai_messages(continue_prompt, glean_result) history += pack_user_ass_to_openai_messages(continue_prompt, glean_result)
# Process gleaning result separately # Process gleaning result separately
glean_nodes, glean_edges = await _process_extraction_result(glean_result, chunk_key) glean_nodes, glean_edges = await _process_extraction_result(
glean_result, chunk_key
)
# Merge results # Merge results
for entity_name, entities in glean_nodes.items(): for entity_name, entities in glean_nodes.items():