Fix linting
This commit is contained in:
@@ -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():
|
||||||
|
Reference in New Issue
Block a user