Fix

langchain-ai · Jan 23, 2025 · cbad17f · cbad17f
1 parent 17dacb8
commit cbad17f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py
@@ -228,7 +228,9 @@ def __init__(
         if self.stream is not None and CONFIG_KEY_STREAM in config[CONF]:
             self.stream = DuplexStream(self.stream, config[CONF][CONFIG_KEY_STREAM])
         scratchpad: Optional[PregelScratchpad] = config[CONF].get(CONFIG_KEY_SCRATCHPAD)
-        if not self.config[CONF].get(CONFIG_KEY_DELEGATE) and scratchpad is not None:
+        if not self.config[CONF].get(CONFIG_KEY_DELEGATE) and isinstance(
+            scratchpad, PregelScratchpad
+        ):
             if scratchpad.subgraph_counter:
                 self.config = patch_configurable(
                     self.config,