Fix

langchain-ai · Nov 16, 2024 · 973ad76 · 973ad76
1 parent 36e49eb
commit 973ad76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py
@@ -946,7 +946,7 @@ def update_state(
                     create_checkpoint(checkpoint, None, step),
                     {
                         **checkpoint_metadata,
-                        "source": "fork",
+                        "source": "update",
                         "step": step + 1,
                         "writes": {},
                         "parents": saved.metadata.get("parents", {}) if saved else {},
@@ -1226,7 +1226,7 @@ async def aupdate_state(
                     next_checkpoint,
                     {
                         **checkpoint_metadata,
-                        "source": "update",
+                        "source": "fork",
                         "step": step + 1,
                         "writes": {},
                         "parents": saved.metadata.get("parents", {}) if saved else {},