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

Merge pull request #1931 from danielaskdd/fix-first-stage-tasks-missing

Browse source 
Fix: Initialize first_stage_tasks and entity_relation_task to prevent empty-task cancel errors
This commit is contained in:
Daniel.y 2025-08-12 19:19:04 +08:00 • committed by GitHub
commit 203e420b51
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
lightrag/lightrag.py
View file

@ -1229,6 +1229,9 @@ class LightRAG:
async with semaphore:
nonlocal processed_count
current_file_number = 0
# Initialize to prevent UnboundLocalError in error handling
first_stage_tasks = []
entity_relation_task = None
try:
# Get file path from status document
file_path = getattr(
@ -1348,15 +1351,13 @@ 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: