From f13dd00b62750fb71a70fda692dd26ecb9376d37 Mon Sep 17 00:00:00 2001
From: Nuno Campos <nuno@langchain.dev>
Date: Thu, 5 Sep 2024 17:04:56 -0700
Subject: [PATCH] Fix import

---
 libs/checkpoint/langgraph/checkpoint/base/__init__.py | 2 +-
 libs/checkpoint/langgraph/checkpoint/serde/types.py   | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/libs/checkpoint/langgraph/checkpoint/base/__init__.py b/libs/checkpoint/langgraph/checkpoint/base/__init__.py
index db1467149..17caf46d8 100644
--- a/libs/checkpoint/langgraph/checkpoint/base/__init__.py
+++ b/libs/checkpoint/langgraph/checkpoint/base/__init__.py
@@ -22,10 +22,10 @@ from langgraph.checkpoint.serde.base import SerializerProtocol, maybe_add_typed_
 from langgraph.checkpoint.serde.jsonplus import JsonPlusSerializer
 from langgraph.checkpoint.serde.types import (
     ERROR,
+    SCHEDULED,
     ChannelProtocol,
     SendProtocol,
 )
-from langgraph.constants import SCHEDULED
 
 V = TypeVar("V", int, float, str)
 PendingWrite = Tuple[str, str, Any]
diff --git a/libs/checkpoint/langgraph/checkpoint/serde/types.py b/libs/checkpoint/langgraph/checkpoint/serde/types.py
index 67f59ae89..3fc82b68d 100644
--- a/libs/checkpoint/langgraph/checkpoint/serde/types.py
+++ b/libs/checkpoint/langgraph/checkpoint/serde/types.py
@@ -13,6 +13,7 @@ from langchain_core.runnables import RunnableConfig
 from typing_extensions import Self
 
 ERROR = "__error__"
+SCHEDULED = "__scheduled__"
 TASKS = "__pregel_tasks"
 
 Value = TypeVar("Value")