From 683fac83dc8632e8c4462e2d1d1f2e9e161db222 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 30 May 2024 17:00:35 -0700 Subject: [PATCH] don't clear pending packets when just viewing --- langgraph/pregel/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py index aaf6662ac..9fd2d3b75 100644 --- a/langgraph/pregel/__init__.py +++ b/langgraph/pregel/__init__.py @@ -1632,7 +1632,8 @@ def _prepare_next_tasks( ) else: tasks.append(PregelTaskDescription(packet.node, val)) - checkpoint["pending_packets"].clear() + if for_execution: + checkpoint["pending_packets"].clear() # Check if any processes should be run in next step # If so, prepare the values to be passed to them for name, proc in processes.items():