From 973ad76a587f0831a2e936620934a8db59d73d2e Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Fri, 15 Nov 2024 17:05:18 -0800
Subject: [PATCH] Fix

---
 libs/langgraph/langgraph/pregel/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/libs/langgraph/langgraph/pregel/__init__.py b/libs/langgraph/langgraph/pregel/__init__.py
index b6bfa895a..e624d6b46 100644
--- a/libs/langgraph/langgraph/pregel/__init__.py
+++ b/libs/langgraph/langgraph/pregel/__init__.py
@@ -946,7 +946,7 @@ class Pregel(PregelProtocol):
                     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 @@ class Pregel(PregelProtocol):
                     next_checkpoint,
                     {
                         **checkpoint_metadata,
-                        "source": "update",
+                        "source": "fork",
                         "step": step + 1,
                         "writes": {},
                         "parents": saved.metadata.get("parents", {}) if saved else {},