From 251bd9744de670ced70dc93190ebd9ed750498b0 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 6 May 2024 11:08:13 -0700 Subject: [PATCH] Fix lineage of maual state updates --- langgraph/pregel/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py index 4e5cf13a0..2d50e4496 100644 --- a/langgraph/pregel/__init__.py +++ b/langgraph/pregel/__init__.py @@ -482,7 +482,7 @@ class Pregel( # apply to checkpoint and save _apply_writes(checkpoint, channels, task.writes) return self.checkpointer.put( - config, + saved.config if saved else config, create_checkpoint(checkpoint, channels), { "source": "update", @@ -551,7 +551,7 @@ class Pregel( # apply to checkpoint and save _apply_writes(checkpoint, channels, task.writes) return await self.checkpointer.aput( - config, + saved.config if saved else config, create_checkpoint(checkpoint, channels), { "source": "update",