From 014f8485a6127a6b258c66065a4fd45ca14be2ae Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 4 Nov 2024 11:23:53 -0800 Subject: [PATCH] lib: Unset skip_done_tasks after each tick of the loop --- libs/langgraph/langgraph/pregel/loop.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py index 2ba8af3a6..b58ee4da7 100644 --- a/libs/langgraph/langgraph/pregel/loop.py +++ b/libs/langgraph/langgraph/pregel/loop.py @@ -399,6 +399,9 @@ class PregelLoop(LoopProtocol): for task in self.tasks.values(): if task.writes: self._output_writes(task.id, task.writes, cached=True) + elif not self.skip_done_tasks: + # "not skip_done_tasks" only applies to first tick after resuming + self.skip_done_tasks = True # if all tasks have finished, re-tick if all(task.writes for task in self.tasks.values()):