From 06f2eef74c8adf36c7421b075481888e2b12063a Mon Sep 17 00:00:00 2001 From: Tat Dat Duong Date: Mon, 10 Mar 2025 19:52:32 +0100 Subject: [PATCH] Fix bug with stale task_id --- libs/langgraph/langgraph/pregel/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 0a8213da4..964a860f7 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -1472,7 +1472,7 @@ class Pregel(PregelProtocol): ) # save task writes - for task in run_tasks: + for task_id, task in zip(run_task_ids, run_tasks): channel_writes = [w for w in task.writes if w[0] != PUSH] # channel writes are saved to current checkpoint @@ -1818,7 +1818,7 @@ class Pregel(PregelProtocol): ) # save task writes - for task in run_tasks: + for task_id, task in zip(run_task_ids, run_tasks): # channel writes are saved to current checkpoint channel_writes = [w for w in task.writes if w[0] != PUSH] if saved and channel_writes: