From 0135c6f743742af1887dca841939c3f5c69c4d4e Mon Sep 17 00:00:00 2001 From: vbarda Date: Tue, 13 Aug 2024 10:07:52 -0400 Subject: [PATCH] correct check for using parent checkpointer --- libs/langgraph/langgraph/pregel/__init__.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py index 13aacf0e5..0b7dd76c9 100644 --- a/libs/langgraph/langgraph/pregel/__init__.py +++ b/libs/langgraph/langgraph/pregel/__init__.py @@ -989,6 +989,7 @@ class Pregel( if ( parent_checkpointer is not None + and self.checkpointer is not None and self.checkpointer != INHERIT_CHECKPOINTER ): raise ValueError( @@ -996,11 +997,14 @@ class Pregel( "Please compile the subgraph graph with checkpointer=INHERIT_CHECKPOINTER (from langgraph.pregel import INHERIT_CHECKPOINTER)." ) - checkpointer = ( - parent_checkpointer - if parent_checkpointer is not None - else self.checkpointer - ) + if ( + parent_checkpointer is not None + and self.checkpointer == INHERIT_CHECKPOINTER + ): + checkpointer = parent_checkpointer + else: + checkpointer = self.checkpointer + return ( debug, stream_mode,