From 48df2d10847bd636d2f7ad6657cb3b406d523fb9 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 6 May 2024 09:56:16 -0700 Subject: [PATCH] Add assert --- langgraph/channels/base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/langgraph/channels/base.py b/langgraph/channels/base.py index c244e750c..368910523 100644 --- a/langgraph/channels/base.py +++ b/langgraph/channels/base.py @@ -125,6 +125,8 @@ def create_checkpoint( checkpoint: Checkpoint, channels: Mapping[str, BaseChannel] ) -> Checkpoint: """Create a checkpoint for the given channels.""" + ts = datetime.now(timezone.utc).isoformat() + assert ts > checkpoint["ts"], "Timestamps must be monotonically increasing" values: dict[str, Any] = {} for k, v in channels.items(): try: @@ -133,7 +135,7 @@ def create_checkpoint( pass return Checkpoint( v=1, - ts=datetime.now(timezone.utc).isoformat(), + ts=ts, channel_values=values, channel_versions=checkpoint["channel_versions"], versions_seen=checkpoint["versions_seen"],