@langchain/langgraph-checkpoint 0.0.3 → 0.0.4

package/dist/base.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
+exports.maxChannelVersion = exports.compareChannelVersions = exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
 const id_js_1 = require("./id.cjs");
 const jsonplus_js_1 = require("./serde/jsonplus.cjs");
 function deepCopy(obj) {
@@ -63,7 +63,25 @@ class BaseCheckpointSaver {
      * as long as they are monotonically increasing.
      */
     getNextVersion(current, _channel) {
-        return current !== undefined ? current + 1 : 1;
+        if (typeof current === "string") {
+            throw new Error("Please override this method to use string versions.");
+        }
+        return (current !== undefined && typeof current === "number" ? current + 1 : 1);
     }
 }
 exports.BaseCheckpointSaver = BaseCheckpointSaver;
+function compareChannelVersions(a, b) {
+    if (typeof a === "number" && typeof b === "number") {
+        return Math.sign(a - b);
+    }
+    return String(a).localeCompare(String(b));
+}
+exports.compareChannelVersions = compareChannelVersions;
+function maxChannelVersion(...versions) {
+    return versions.reduce((max, version, idx) => {
+        if (idx === 0)
+            return version;
+        return compareChannelVersions(max, version) >= 0 ? max : version;
+    });
+}
+exports.maxChannelVersion = maxChannelVersion;

package/dist/base.d.ts

@@ -2,7 +2,8 @@ import type { RunnableConfig } from "@langchain/core/runnables";
 import { SerializerProtocol } from "./serde/base.js";
 import type { PendingWrite, CheckpointPendingWrite, CheckpointMetadata } from "./types.js";
 import type { ChannelProtocol, SendProtocol } from "./serde/types.js";
-export type ChannelVersions = Record<string, string | number>;
+type ChannelVersion = number | string;
+export type ChannelVersions = Record<string, ChannelVersion>;
 export interface Checkpoint<N extends string = string, C extends string = string> {
     /**
      * Version number
@@ -23,11 +24,11 @@ export interface Checkpoint<N extends string = string, C extends string = string
     /**
      * @default {}
      */
-    channel_versions: Record<C, number>;
+    channel_versions: Record<C, ChannelVersion>;
     /**
      * @default {}
      */
-    versions_seen: Record<N, Record<C, number>>;
+    versions_seen: Record<N, Record<C, ChannelVersion>>;
     /**
      * List of packets sent to nodes but not yet processed.
      * Cleared by the next checkpoint.
@@ -36,8 +37,8 @@ export interface Checkpoint<N extends string = string, C extends string = string
 }
 export interface ReadonlyCheckpoint extends Readonly<Checkpoint> {
     readonly channel_values: Readonly<Record<string, unknown>>;
-    readonly channel_versions: Readonly<Record<string, number>>;
-    readonly versions_seen: Readonly<Record<string, Readonly<Record<string, number>>>>;
+    readonly channel_versions: Readonly<Record<string, ChannelVersion>>;
+    readonly versions_seen: Readonly<Record<string, Readonly<Record<string, ChannelVersion>>>>;
 }
 export declare function deepCopy<T>(obj: T): T;
 /** @hidden */
@@ -56,7 +57,7 @@ export type CheckpointListOptions = {
     before?: RunnableConfig;
     filter?: Record<string, any>;
 };
-export declare abstract class BaseCheckpointSaver<V = number> {
+export declare abstract class BaseCheckpointSaver<V extends string | number = number> {
     serde: SerializerProtocol;
     constructor(serde?: SerializerProtocol);
     get(config: RunnableConfig): Promise<Checkpoint | undefined>;
@@ -73,5 +74,8 @@ export declare abstract class BaseCheckpointSaver<V = number> {
      * Default is to use integer versions, incrementing by 1. If you override, you can use str/int/float versions,
      * as long as they are monotonically increasing.
      */
-    getNextVersion(current: V | undefined, _channel: ChannelProtocol): number;
+    getNextVersion(current: V | undefined, _channel: ChannelProtocol): V;
 }
+export declare function compareChannelVersions(a: ChannelVersion, b: ChannelVersion): number;
+export declare function maxChannelVersion(...versions: ChannelVersion[]): ChannelVersion;
+export {};

package/dist/base.js

@@ -57,6 +57,22 @@ export class BaseCheckpointSaver {
      * as long as they are monotonically increasing.
      */
     getNextVersion(current, _channel) {
-        return current !== undefined ? current + 1 : 1;
+        if (typeof current === "string") {
+            throw new Error("Please override this method to use string versions.");
+        }
+        return (current !== undefined && typeof current === "number" ? current + 1 : 1);
+    }
+}
+export function compareChannelVersions(a, b) {
+    if (typeof a === "number" && typeof b === "number") {
+        return Math.sign(a - b);
     }
+    return String(a).localeCompare(String(b));
+}
+export function maxChannelVersion(...versions) {
+    return versions.reduce((max, version, idx) => {
+        if (idx === 0)
+            return version;
+        return compareChannelVersions(max, version) >= 0 ? max : version;
+    });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-checkpoint",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "description": "Library with base interfaces for LangGraph checkpoint savers.",
   "type": "module",
   "engines": {