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: async with semaphore:
nonlocal processed_count nonlocal processed_count
current_file_number = 0 current_file_number = 0
# Initialize to prevent UnboundLocalError in error handling
first_stage_tasks = []
entity_relation_task = None
try: try:
# Get file path from status document # Get file path from status document
file_path = getattr( file_path = getattr(
@ -1348,15 +1351,13 @@ class LightRAG:
) )
pipeline_status["history_messages"].append(error_msg) pipeline_status["history_messages"].append(error_msg)
# Cancel tasks that are not yet completed # Cancel tasks that are not yet completed
all_tasks = first_stage_tasks + ( all_tasks = first_stage_tasks + (
[entity_relation_task] [entity_relation_task] if entity_relation_task else []
if entity_relation_task )
else [] for task in all_tasks:
) if task and not task.done():
for task in all_tasks: task.cancel()
if task and not task.done():
task.cancel()
# Persistent llm cache # Persistent llm cache
if self.llm_response_cache: if self.llm_response_cache: