Fix

langchain-ai · Jan 23, 2025 · 3f2557c · 3f2557c
1 parent cbad17f
commit 3f2557c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/libs/langgraph/langgraph/pregel/loop.py b/libs/langgraph/langgraph/pregel/loop.py
@@ -567,7 +567,10 @@ def _first(self, *, input_keys: Union[str, Sequence[str]]) -> None:
         if scratchpad := cast(
             Optional[PregelScratchpad], configurable.get(CONFIG_KEY_SCRATCHPAD)
         ):
-            if scratchpad.null_resume is not None:
+            if (
+                isinstance(scratchpad, PregelScratchpad)
+                and scratchpad.null_resume is not None
+            ):
                 self.put_writes(NULL_TASK_ID, [(RESUME, scratchpad.null_resume)])
         # map command to writes
         if isinstance(self.input, Command):