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

Fix: Initialize first_stage_tasks and entity_relation_task to prevent empty-task cancel errors

Browse source 
- Initialize first_stage_tasks = [] and entity_relation_task = None at coroutine start
- Ensure cancel block safely handles no-op when tasks lists are empty
This commit is contained in:
yangdx 2025-08-10 10:45:41 +08:00
parent ffb642a5ce
commit eb2320e556
1 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
lightrag/lightrag.py
View file

@ -1229,6 +1229,8 @@ class LightRAG:
async with semaphore:
nonlocal processed_count
current_file_number = 0
first_stage_tasks = []
entity_relation_task = None
try:
# Get file path from status document
file_path = getattr(
@ -1348,15 +1350,15 @@ class LightRAG:
)
pipeline_status["history_messages"].append(error_msg)
# Cancel tasks that are not yet completed
all_tasks = first_stage_tasks + (
[entity_relation_task]
if entity_relation_task
else []
)
for task in all_tasks:
if task and not task.done():
task.cancel()
# Cancel tasks that are not yet completed
all_tasks = first_stage_tasks + (
[entity_relation_task]
if entity_relation_task
else []
)
for task in all_tasks:
if task and not task.done():
task.cancel()
# Persistent llm cache
if self.llm_response_cache: