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")