Remove redundant await call in file extraction pipeline
This commit is contained in:
parent
c9e73bb450
commit
c36afecba4
1 changed files with 2 additions and 3 deletions
|
|
@ -1835,7 +1835,7 @@ class LightRAG:
|
||||||
chunks, pipeline_status, pipeline_status_lock
|
chunks, pipeline_status, pipeline_status_lock
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
await entity_relation_task
|
chunk_results = await entity_relation_task
|
||||||
file_extraction_stage_ok = True
|
file_extraction_stage_ok = True
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|
@ -1917,8 +1917,7 @@ class LightRAG:
|
||||||
"User cancelled"
|
"User cancelled"
|
||||||
)
|
)
|
||||||
|
|
||||||
# Get chunk_results from entity_relation_task
|
# Use chunk_results from entity_relation_task
|
||||||
chunk_results = await entity_relation_task
|
|
||||||
await merge_nodes_and_edges(
|
await merge_nodes_and_edges(
|
||||||
chunk_results=chunk_results, # result collected from entity_relation_task
|
chunk_results=chunk_results, # result collected from entity_relation_task
|
||||||
knowledge_graph_inst=self.chunk_entity_relation_graph,
|
knowledge_graph_inst=self.chunk_entity_relation_graph,
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue