@langchain/langgraph-checkpoint 0.0.2 → 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/dist/serde/jsonplus.cjs

@@ -4,6 +4,52 @@ exports.JsonPlusSerializer = void 0;
 /* eslint-disable @typescript-eslint/no-explicit-any */
 /* eslint-disable no-instanceof/no-instanceof */
 const load_1 = require("@langchain/core/load");
+function isLangChainSerializable(value) {
+    return (typeof value.lc_serializable === "boolean" && Array.isArray(value.lc_id));
+}
+function isLangChainSerializedObject(value) {
+    return (value !== null &&
+        value.lc === 1 &&
+        value.type === "constructor" &&
+        Array.isArray(value.id));
+}
+const _serialize = (value, seen = new WeakSet()) => {
+    const defaultValue = _default("", value);
+    if (defaultValue === null) {
+        return "null";
+    }
+    else if (typeof defaultValue === "string") {
+        return JSON.stringify(defaultValue);
+    }
+    else if (typeof defaultValue === "number" ||
+        typeof defaultValue === "boolean") {
+        return defaultValue.toString();
+    }
+    else if (typeof defaultValue === "object") {
+        if (seen.has(defaultValue)) {
+            throw new TypeError("Circular reference detected");
+        }
+        seen.add(defaultValue);
+        if (Array.isArray(defaultValue)) {
+            const result = `[${defaultValue
+                .map((item) => _serialize(item, seen))
+                .join(",")}]`;
+            seen.delete(defaultValue);
+            return result;
+        }
+        else if (isLangChainSerializable(defaultValue)) {
+            return JSON.stringify(defaultValue);
+        }
+        else {
+            const entries = Object.entries(defaultValue).map(([k, v]) => `${JSON.stringify(k)}:${_serialize(v, seen)}`);
+            const result = `{${entries.join(",")}}`;
+            seen.delete(defaultValue);
+            return result;
+        }
+    }
+    // Only be reached for functions or symbols
+    return JSON.stringify(defaultValue);
+};
 async function _reviver(value) {
     if (value && typeof value === "object") {
         if (value.lc === 2 && value.type === "undefined") {
@@ -42,7 +88,7 @@ async function _reviver(value) {
                 return value;
             }
         }
-        else if (value.lc === 1) {
+        else if (isLangChainSerializedObject(value)) {
             return (0, load_1.load)(JSON.stringify(value));
         }
         else if (Array.isArray(value)) {
@@ -94,24 +140,8 @@ function _default(_key, obj) {
 }
 class JsonPlusSerializer {
     _dumps(obj) {
-        const jsonString = JSON.stringify(obj, (key, value) => {
-            if (value && typeof value === "object") {
-                if (Array.isArray(value)) {
-                    // Handle arrays
-                    return value.map((item) => _default(key, item));
-                }
-                else {
-                    // Handle objects
-                    const serialized = {};
-                    for (const [k, v] of Object.entries(value)) {
-                        serialized[k] = _default(k, v);
-                    }
-                    return serialized;
-                }
-            }
-            return _default(key, value);
-        });
-        return new TextEncoder().encode(jsonString);
+        const encoder = new TextEncoder();
+        return encoder.encode(_serialize(obj));
     }
     dumpsTyped(obj) {
         if (obj instanceof Uint8Array) {

package/dist/serde/jsonplus.js

@@ -1,6 +1,52 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 /* eslint-disable no-instanceof/no-instanceof */
 import { load } from "@langchain/core/load";
+function isLangChainSerializable(value) {
+    return (typeof value.lc_serializable === "boolean" && Array.isArray(value.lc_id));
+}
+function isLangChainSerializedObject(value) {
+    return (value !== null &&
+        value.lc === 1 &&
+        value.type === "constructor" &&
+        Array.isArray(value.id));
+}
+const _serialize = (value, seen = new WeakSet()) => {
+    const defaultValue = _default("", value);
+    if (defaultValue === null) {
+        return "null";
+    }
+    else if (typeof defaultValue === "string") {
+        return JSON.stringify(defaultValue);
+    }
+    else if (typeof defaultValue === "number" ||
+        typeof defaultValue === "boolean") {
+        return defaultValue.toString();
+    }
+    else if (typeof defaultValue === "object") {
+        if (seen.has(defaultValue)) {
+            throw new TypeError("Circular reference detected");
+        }
+        seen.add(defaultValue);
+        if (Array.isArray(defaultValue)) {
+            const result = `[${defaultValue
+                .map((item) => _serialize(item, seen))
+                .join(",")}]`;
+            seen.delete(defaultValue);
+            return result;
+        }
+        else if (isLangChainSerializable(defaultValue)) {
+            return JSON.stringify(defaultValue);
+        }
+        else {
+            const entries = Object.entries(defaultValue).map(([k, v]) => `${JSON.stringify(k)}:${_serialize(v, seen)}`);
+            const result = `{${entries.join(",")}}`;
+            seen.delete(defaultValue);
+            return result;
+        }
+    }
+    // Only be reached for functions or symbols
+    return JSON.stringify(defaultValue);
+};
 async function _reviver(value) {
     if (value && typeof value === "object") {
         if (value.lc === 2 && value.type === "undefined") {
@@ -39,7 +85,7 @@ async function _reviver(value) {
                 return value;
             }
         }
-        else if (value.lc === 1) {
+        else if (isLangChainSerializedObject(value)) {
             return load(JSON.stringify(value));
         }
         else if (Array.isArray(value)) {
@@ -91,24 +137,8 @@ function _default(_key, obj) {
 }
 export class JsonPlusSerializer {
     _dumps(obj) {
-        const jsonString = JSON.stringify(obj, (key, value) => {
-            if (value && typeof value === "object") {
-                if (Array.isArray(value)) {
-                    // Handle arrays
-                    return value.map((item) => _default(key, item));
-                }
-                else {
-                    // Handle objects
-                    const serialized = {};
-                    for (const [k, v] of Object.entries(value)) {
-                        serialized[k] = _default(k, v);
-                    }
-                    return serialized;
-                }
-            }
-            return _default(key, value);
-        });
-        return new TextEncoder().encode(jsonString);
+        const encoder = new TextEncoder();
+        return encoder.encode(_serialize(obj));
     }
     dumpsTyped(obj) {
         if (obj instanceof Uint8Array) {

package/package.json

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