From 9a306ce9319d6ae599a99d7046a310579c91e41e Mon Sep 17 00:00:00 2001 From: vbarda Date: Thu, 22 Aug 2024 11:47:36 -0400 Subject: [PATCH] langgraph: check for empty checkpoint when creating snapshot tasks --- 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 563f097f8..270a3c7ba 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -371,7 +371,7 @@ class Pregel( saved.metadata if saved else None, saved.checkpoint["ts"] if saved else None, saved.parent_config if saved else None, - tasks_w_writes(next_tasks, saved.pending_writes), + tasks_w_writes(next_tasks, saved.pending_writes if saved else None), ) async def aget_state(self, config: RunnableConfig) -> StateSnapshot: @@ -402,7 +402,7 @@ class Pregel( saved.metadata if saved else None, saved.checkpoint["ts"] if saved else None, saved.parent_config if saved else None, - tasks_w_writes(next_tasks, saved.pending_writes), + tasks_w_writes(next_tasks, saved.pending_writes if saved else None), ) def get_state_history(