Fix linting

This commit is contained in:
yangdx
2025-03-11 12:23:51 +08:00
parent 061350b2bf
commit 9d1dc2c9c3

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():