From 3ddb6b1477bab5d788cb51a0866f2c9f6c441a03 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 25 Jun 2025 17:05:41 -0700 Subject: [PATCH] One more --- libs/langgraph/langgraph/pregel/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 40f0212cf..48b46c0d3 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -2126,7 +2126,11 @@ class Pregel(PregelProtocol[StateT, InputT, OutputT], Generic[StateT, InputT, Ou next_tasks[tid].writes.append((k, v)) if tasks := [t for t in next_tasks.values() if t.writes]: apply_writes( - checkpoint, channels, tasks, None, self.trigger_to_nodes + checkpoint, + channels, + tasks, + checkpointer.get_next_version, + self.trigger_to_nodes, ) valid_updates: list[tuple[str, dict[str, Any] | None, str | None]] = [] if len(updates) == 1: