@langchain/langgraph-checkpoint 0.0.6 → 0.0.8-rc.0

package/dist/index.cjs

@@ -22,3 +22,4 @@ __exportStar(require("./id.cjs"), exports);
 __exportStar(require("./types.cjs"), exports);
 __exportStar(require("./serde/base.cjs"), exports);
 __exportStar(require("./serde/types.cjs"), exports);
+__exportStar(require("./store/index.cjs"), exports);

package/dist/index.d.ts

@@ -4,3 +4,4 @@ export * from "./id.js";
 export * from "./types.js";
 export * from "./serde/base.js";
 export * from "./serde/types.js";
+export * from "./store/index.js";

package/dist/index.js

@@ -4,3 +4,4 @@ export * from "./id.js";
 export * from "./types.js";
 export * from "./serde/base.js";
 export * from "./serde/types.js";
+export * from "./store/index.js";

package/dist/memory.cjs

@@ -129,61 +129,67 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
         const threadIds = config.configurable?.thread_id
             ? [config.configurable?.thread_id]
             : Object.keys(this.storage);
-        const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
+        const configCheckpointNamespace = config.configurable?.checkpoint_ns;
         for (const threadId of threadIds) {
-            const checkpoints = this.storage[threadId]?.[checkpointNamespace] ?? {};
-            const sortedCheckpoints = Object.entries(checkpoints).sort((a, b) => b[0].localeCompare(a[0]));
-            for (const [checkpointId, [checkpoint, metadataStr, parentCheckpointId],] of sortedCheckpoints) {
-                // Filter by checkpoint ID
-                if (before &&
-                    before.configurable?.checkpoint_id &&
-                    checkpointId >= before.configurable.checkpoint_id) {
+            for (const checkpointNamespace of Object.keys(this.storage[threadId])) {
+                if (configCheckpointNamespace !== undefined &&
+                    checkpointNamespace !== configCheckpointNamespace) {
                     continue;
                 }
-                // Parse metadata
-                const metadata = (await this.serde.loadsTyped("json", metadataStr));
-                // Limit search results
-                if (limit !== undefined) {
-                    if (limit <= 0)
-                        break;
-                    // eslint-disable-next-line no-param-reassign
-                    limit -= 1;
-                }
-                const writes = this.writes[_generateKey(threadId, checkpointNamespace, checkpointId)] ?? [];
-                const pending_sends = await this._getPendingSends(threadId, checkpointNamespace, parentCheckpointId);
-                const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [
-                        taskId,
-                        channel,
-                        await this.serde.loadsTyped("json", value),
-                    ];
-                }));
-                const deserializedCheckpoint = {
-                    ...(await this.serde.loadsTyped("json", checkpoint)),
-                    pending_sends,
-                };
-                const checkpointTuple = {
-                    config: {
-                        configurable: {
-                            thread_id: threadId,
-                            checkpoint_ns: checkpointNamespace,
-                            checkpoint_id: checkpointId,
-                        },
-                    },
-                    checkpoint: deserializedCheckpoint,
-                    metadata,
-                    pendingWrites,
-                };
-                if (parentCheckpointId !== undefined) {
-                    checkpointTuple.parentConfig = {
-                        configurable: {
-                            thread_id: threadId,
-                            checkpoint_ns: checkpointNamespace,
-                            checkpoint_id: parentCheckpointId,
+                const checkpoints = this.storage[threadId]?.[checkpointNamespace] ?? {};
+                const sortedCheckpoints = Object.entries(checkpoints).sort((a, b) => b[0].localeCompare(a[0]));
+                for (const [checkpointId, [checkpoint, metadataStr, parentCheckpointId],] of sortedCheckpoints) {
+                    // Filter by checkpoint ID
+                    if (before &&
+                        before.configurable?.checkpoint_id &&
+                        checkpointId >= before.configurable.checkpoint_id) {
+                        continue;
+                    }
+                    // Parse metadata
+                    const metadata = (await this.serde.loadsTyped("json", metadataStr));
+                    // Limit search results
+                    if (limit !== undefined) {
+                        if (limit <= 0)
+                            break;
+                        // eslint-disable-next-line no-param-reassign
+                        limit -= 1;
+                    }
+                    const writes = this.writes[_generateKey(threadId, checkpointNamespace, checkpointId)] ?? [];
+                    const pending_sends = await this._getPendingSends(threadId, checkpointNamespace, parentCheckpointId);
+                    const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
+                        return [
+                            taskId,
+                            channel,
+                            await this.serde.loadsTyped("json", value),
+                        ];
+                    }));
+                    const deserializedCheckpoint = {
+                        ...(await this.serde.loadsTyped("json", checkpoint)),
+                        pending_sends,
+                    };
+                    const checkpointTuple = {
+                        config: {
+                            configurable: {
+                                thread_id: threadId,
+                                checkpoint_ns: checkpointNamespace,
+                                checkpoint_id: checkpointId,
+                            },
                         },
+                        checkpoint: deserializedCheckpoint,
+                        metadata,
+                        pendingWrites,
                     };
+                    if (parentCheckpointId !== undefined) {
+                        checkpointTuple.parentConfig = {
+                            configurable: {
+                                thread_id: threadId,
+                                checkpoint_ns: checkpointNamespace,
+                                checkpoint_id: parentCheckpointId,
+                            },
+                        };
+                    }
+                    yield checkpointTuple;
                 }
-                yield checkpointTuple;
             }
         }
     }

package/dist/memory.js

@@ -126,61 +126,67 @@ export class MemorySaver extends BaseCheckpointSaver {
         const threadIds = config.configurable?.thread_id
             ? [config.configurable?.thread_id]
             : Object.keys(this.storage);
-        const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
+        const configCheckpointNamespace = config.configurable?.checkpoint_ns;
         for (const threadId of threadIds) {
-            const checkpoints = this.storage[threadId]?.[checkpointNamespace] ?? {};
-            const sortedCheckpoints = Object.entries(checkpoints).sort((a, b) => b[0].localeCompare(a[0]));
-            for (const [checkpointId, [checkpoint, metadataStr, parentCheckpointId],] of sortedCheckpoints) {
-                // Filter by checkpoint ID
-                if (before &&
-                    before.configurable?.checkpoint_id &&
-                    checkpointId >= before.configurable.checkpoint_id) {
+            for (const checkpointNamespace of Object.keys(this.storage[threadId])) {
+                if (configCheckpointNamespace !== undefined &&
+                    checkpointNamespace !== configCheckpointNamespace) {
                     continue;
                 }
-                // Parse metadata
-                const metadata = (await this.serde.loadsTyped("json", metadataStr));
-                // Limit search results
-                if (limit !== undefined) {
-                    if (limit <= 0)
-                        break;
-                    // eslint-disable-next-line no-param-reassign
-                    limit -= 1;
-                }
-                const writes = this.writes[_generateKey(threadId, checkpointNamespace, checkpointId)] ?? [];
-                const pending_sends = await this._getPendingSends(threadId, checkpointNamespace, parentCheckpointId);
-                const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [
-                        taskId,
-                        channel,
-                        await this.serde.loadsTyped("json", value),
-                    ];
-                }));
-                const deserializedCheckpoint = {
-                    ...(await this.serde.loadsTyped("json", checkpoint)),
-                    pending_sends,
-                };
-                const checkpointTuple = {
-                    config: {
-                        configurable: {
-                            thread_id: threadId,
-                            checkpoint_ns: checkpointNamespace,
-                            checkpoint_id: checkpointId,
-                        },
-                    },
-                    checkpoint: deserializedCheckpoint,
-                    metadata,
-                    pendingWrites,
-                };
-                if (parentCheckpointId !== undefined) {
-                    checkpointTuple.parentConfig = {
-                        configurable: {
-                            thread_id: threadId,
-                            checkpoint_ns: checkpointNamespace,
-                            checkpoint_id: parentCheckpointId,
+                const checkpoints = this.storage[threadId]?.[checkpointNamespace] ?? {};
+                const sortedCheckpoints = Object.entries(checkpoints).sort((a, b) => b[0].localeCompare(a[0]));
+                for (const [checkpointId, [checkpoint, metadataStr, parentCheckpointId],] of sortedCheckpoints) {
+                    // Filter by checkpoint ID
+                    if (before &&
+                        before.configurable?.checkpoint_id &&
+                        checkpointId >= before.configurable.checkpoint_id) {
+                        continue;
+                    }
+                    // Parse metadata
+                    const metadata = (await this.serde.loadsTyped("json", metadataStr));
+                    // Limit search results
+                    if (limit !== undefined) {
+                        if (limit <= 0)
+                            break;
+                        // eslint-disable-next-line no-param-reassign
+                        limit -= 1;
+                    }
+                    const writes = this.writes[_generateKey(threadId, checkpointNamespace, checkpointId)] ?? [];
+                    const pending_sends = await this._getPendingSends(threadId, checkpointNamespace, parentCheckpointId);
+                    const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
+                        return [
+                            taskId,
+                            channel,
+                            await this.serde.loadsTyped("json", value),
+                        ];
+                    }));
+                    const deserializedCheckpoint = {
+                        ...(await this.serde.loadsTyped("json", checkpoint)),
+                        pending_sends,
+                    };
+                    const checkpointTuple = {
+                        config: {
+                            configurable: {
+                                thread_id: threadId,
+                                checkpoint_ns: checkpointNamespace,
+                                checkpoint_id: checkpointId,
+                            },
                         },
+                        checkpoint: deserializedCheckpoint,
+                        metadata,
+                        pendingWrites,
                     };
+                    if (parentCheckpointId !== undefined) {
+                        checkpointTuple.parentConfig = {
+                            configurable: {
+                                thread_id: threadId,
+                                checkpoint_ns: checkpointNamespace,
+                                checkpoint_id: parentCheckpointId,
+                            },
+                        };
+                    }
+                    yield checkpointTuple;
                 }
-                yield checkpointTuple;
             }
         }
     }

package/dist/serde/jsonplus.cjs

@@ -4,101 +4,71 @@ 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));
-}
+const index_js_1 = require("./utils/fast-safe-stringify/index.cjs");
 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);
-};
+/**
+ * The replacer in stringify does not allow delegation to built-in LangChain
+ * serialization methods, and instead immediately calls `.toJSON()` and
+ * continues to stringify subfields.
+ *
+ * We therefore must start from the most nested elements in the input and
+ * deserialize upwards rather than top-down.
+ */
 async function _reviver(value) {
     if (value && typeof value === "object") {
-        if (value.lc === 2 && value.type === "undefined") {
-            return undefined;
-        }
-        else if (value.lc === 2 &&
-            value.type === "constructor" &&
-            Array.isArray(value.id)) {
-            try {
-                const constructorName = value.id[value.id.length - 1];
-                let constructor;
-                switch (constructorName) {
-                    case "Set":
-                        constructor = Set;
-                        break;
-                    case "Map":
-                        constructor = Map;
-                        break;
-                    case "RegExp":
-                        constructor = RegExp;
-                        break;
-                    case "Error":
-                        constructor = Error;
-                        break;
-                    default:
-                        return value;
-                }
-                if (value.method) {
-                    return constructor[value.method](...(value.args || []));
-                }
-                else {
-                    return new constructor(...(value.args || []));
-                }
-            }
-            catch (error) {
-                return value;
-            }
-        }
-        else if (isLangChainSerializedObject(value)) {
-            return (0, load_1.load)(JSON.stringify(value));
-        }
-        else if (Array.isArray(value)) {
-            return Promise.all(value.map((item) => _reviver(item)));
+        if (Array.isArray(value)) {
+            const revivedArray = await Promise.all(value.map((item) => _reviver(item)));
+            return revivedArray;
         }
         else {
             const revivedObj = {};
             for (const [k, v] of Object.entries(value)) {
                 revivedObj[k] = await _reviver(v);
             }
+            if (revivedObj.lc === 2 && revivedObj.type === "undefined") {
+                return undefined;
+            }
+            else if (revivedObj.lc === 2 &&
+                revivedObj.type === "constructor" &&
+                Array.isArray(revivedObj.id)) {
+                try {
+                    const constructorName = revivedObj.id[revivedObj.id.length - 1];
+                    let constructor;
+                    switch (constructorName) {
+                        case "Set":
+                            constructor = Set;
+                            break;
+                        case "Map":
+                            constructor = Map;
+                            break;
+                        case "RegExp":
+                            constructor = RegExp;
+                            break;
+                        case "Error":
+                            constructor = Error;
+                            break;
+                        default:
+                            return revivedObj;
+                    }
+                    if (revivedObj.method) {
+                        return constructor[revivedObj.method](...(revivedObj.args || []));
+                    }
+                    else {
+                        return new constructor(...(revivedObj.args || []));
+                    }
+                }
+                catch (error) {
+                    return revivedObj;
+                }
+            }
+            else if (isLangChainSerializedObject(revivedObj)) {
+                return (0, load_1.load)(JSON.stringify(revivedObj));
+            }
             return revivedObj;
         }
     }
@@ -116,7 +86,7 @@ constructor, method, args, kwargs) {
         kwargs: kwargs ?? {},
     };
 }
-function _default(_key, obj) {
+function _default(obj) {
     if (obj === undefined) {
         return {
             lc: 2,
@@ -148,7 +118,9 @@ function _default(_key, obj) {
 class JsonPlusSerializer {
     _dumps(obj) {
         const encoder = new TextEncoder();
-        return encoder.encode(_serialize(obj));
+        return encoder.encode((0, index_js_1.stringify)(obj, (_, value) => {
+            return _default(value);
+        }));
     }
     dumpsTyped(obj) {
         if (obj instanceof Uint8Array) {

package/dist/serde/jsonplus.js

@@ -1,101 +1,71 @@
 /* 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));
-}
+import { stringify } from "./utils/fast-safe-stringify/index.js";
 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);
-};
+/**
+ * The replacer in stringify does not allow delegation to built-in LangChain
+ * serialization methods, and instead immediately calls `.toJSON()` and
+ * continues to stringify subfields.
+ *
+ * We therefore must start from the most nested elements in the input and
+ * deserialize upwards rather than top-down.
+ */
 async function _reviver(value) {
     if (value && typeof value === "object") {
-        if (value.lc === 2 && value.type === "undefined") {
-            return undefined;
-        }
-        else if (value.lc === 2 &&
-            value.type === "constructor" &&
-            Array.isArray(value.id)) {
-            try {
-                const constructorName = value.id[value.id.length - 1];
-                let constructor;
-                switch (constructorName) {
-                    case "Set":
-                        constructor = Set;
-                        break;
-                    case "Map":
-                        constructor = Map;
-                        break;
-                    case "RegExp":
-                        constructor = RegExp;
-                        break;
-                    case "Error":
-                        constructor = Error;
-                        break;
-                    default:
-                        return value;
-                }
-                if (value.method) {
-                    return constructor[value.method](...(value.args || []));
-                }
-                else {
-                    return new constructor(...(value.args || []));
-                }
-            }
-            catch (error) {
-                return value;
-            }
-        }
-        else if (isLangChainSerializedObject(value)) {
-            return load(JSON.stringify(value));
-        }
-        else if (Array.isArray(value)) {
-            return Promise.all(value.map((item) => _reviver(item)));
+        if (Array.isArray(value)) {
+            const revivedArray = await Promise.all(value.map((item) => _reviver(item)));
+            return revivedArray;
         }
         else {
             const revivedObj = {};
             for (const [k, v] of Object.entries(value)) {
                 revivedObj[k] = await _reviver(v);
             }
+            if (revivedObj.lc === 2 && revivedObj.type === "undefined") {
+                return undefined;
+            }
+            else if (revivedObj.lc === 2 &&
+                revivedObj.type === "constructor" &&
+                Array.isArray(revivedObj.id)) {
+                try {
+                    const constructorName = revivedObj.id[revivedObj.id.length - 1];
+                    let constructor;
+                    switch (constructorName) {
+                        case "Set":
+                            constructor = Set;
+                            break;
+                        case "Map":
+                            constructor = Map;
+                            break;
+                        case "RegExp":
+                            constructor = RegExp;
+                            break;
+                        case "Error":
+                            constructor = Error;
+                            break;
+                        default:
+                            return revivedObj;
+                    }
+                    if (revivedObj.method) {
+                        return constructor[revivedObj.method](...(revivedObj.args || []));
+                    }
+                    else {
+                        return new constructor(...(revivedObj.args || []));
+                    }
+                }
+                catch (error) {
+                    return revivedObj;
+                }
+            }
+            else if (isLangChainSerializedObject(revivedObj)) {
+                return load(JSON.stringify(revivedObj));
+            }
             return revivedObj;
         }
     }
@@ -113,7 +83,7 @@ constructor, method, args, kwargs) {
         kwargs: kwargs ?? {},
     };
 }
-function _default(_key, obj) {
+function _default(obj) {
     if (obj === undefined) {
         return {
             lc: 2,
@@ -145,7 +115,9 @@ function _default(_key, obj) {
 export class JsonPlusSerializer {
     _dumps(obj) {
         const encoder = new TextEncoder();
-        return encoder.encode(_serialize(obj));
+        return encoder.encode(stringify(obj, (_, value) => {
+            return _default(value);
+        }));
     }
     dumpsTyped(obj) {
         if (obj instanceof Uint8Array) {