Merge pull request #2325 from langchain-ai/nc/4nov/unset-skip-done-tasks

lib: Unset skip_done_tasks after each tick of the loop
This commit is contained in:
Nuno Campos
2024-11-04 11:30:08 -08:00
committed by GitHub
+3
View File
@@ -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()):