@langchain/langgraph-checkpoint 0.0.0 → 0.0.1

package/dist/base.cjs

@@ -1,8 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
-const base_js_1 = require("./serde/base.cjs");
 const id_js_1 = require("./id.cjs");
+const jsonplus_js_1 = require("./serde/jsonplus.cjs");
 function deepCopy(obj) {
     if (typeof obj !== "object" || obj === null) {
         return obj;
@@ -46,7 +46,7 @@ class BaseCheckpointSaver {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: base_js_1.DefaultSerializer
+            value: new jsonplus_js_1.JsonPlusSerializer()
         });
         this.serde = serde || this.serde;
     }

package/dist/base.d.ts

@@ -55,8 +55,8 @@ export type CheckpointListOptions = {
     filter?: Record<string, any>;
 };
 export declare abstract class BaseCheckpointSaver<V = number> {
-    serde: SerializerProtocol<unknown>;
-    constructor(serde?: SerializerProtocol<unknown>);
+    serde: SerializerProtocol;
+    constructor(serde?: SerializerProtocol);
     get(config: RunnableConfig): Promise<Checkpoint | undefined>;
     abstract getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;
     abstract list(config: RunnableConfig, options?: CheckpointListOptions): AsyncGenerator<CheckpointTuple>;

package/dist/base.js

@@ -1,5 +1,5 @@
-import { DefaultSerializer } from "./serde/base.js";
 import { uuid6 } from "./id.js";
+import { JsonPlusSerializer } from "./serde/jsonplus.js";
 export function deepCopy(obj) {
     if (typeof obj !== "object" || obj === null) {
         return obj;
@@ -40,7 +40,7 @@ export class BaseCheckpointSaver {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: DefaultSerializer
+            value: new JsonPlusSerializer()
         });
         this.serde = serde || this.serde;
     }

package/dist/memory.cjs

@@ -33,7 +33,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
                     [];
                 const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [taskId, channel, await this.serde.parse(value)];
+                    return [
+                        taskId,
+                        channel,
+                        await this.serde.loadsTyped("json", value),
+                    ];
                 }));
                 const parentConfig = parentCheckpointId !== undefined
                     ? {
@@ -46,8 +50,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                     : undefined;
                 return {
                     config,
-                    checkpoint: (await this.serde.parse(checkpoint)),
-                    metadata: (await this.serde.parse(metadata)),
+                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                     parentConfig,
                 };
@@ -63,7 +67,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
                     [];
                 const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [taskId, channel, await this.serde.parse(value)];
+                    return [
+                        taskId,
+                        channel,
+                        await this.serde.loadsTyped("json", value),
+                    ];
                 }));
                 const parentConfig = parentCheckpointId !== undefined
                     ? {
@@ -82,8 +90,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                             checkpoint_ns,
                         },
                     },
-                    checkpoint: (await this.serde.parse(checkpoint)),
-                    metadata: (await this.serde.parse(metadata)),
+                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                     parentConfig,
                 };
@@ -109,7 +117,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                     continue;
                 }
                 // Parse metadata
-                const metadata = (await this.serde.parse(metadataStr));
+                const metadata = (await this.serde.loadsTyped("json", metadataStr));
                 // Limit search results
                 if (limit !== undefined) {
                     if (limit <= 0)
@@ -119,7 +127,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 }
                 const writes = this.writes[_generateKey(threadId, checkpointNamespace, checkpointId)] ?? [];
                 const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [taskId, channel, await this.serde.parse(value)];
+                    return [
+                        taskId,
+                        channel,
+                        await this.serde.loadsTyped("json", value),
+                    ];
                 }));
                 yield {
                     config: {
@@ -129,7 +141,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                             checkpoint_id: checkpointId,
                         },
                     },
-                    checkpoint: (await this.serde.parse(checkpoint)),
+                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
                     metadata,
                     pendingWrites,
                     parentConfig: parentCheckpointId
@@ -160,9 +172,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
         if (!this.storage[threadId][checkpointNamespace]) {
             this.storage[threadId][checkpointNamespace] = {};
         }
+        const [, serializedCheckpoint] = this.serde.dumpsTyped(checkpoint);
+        const [, serializedMetadata] = this.serde.dumpsTyped(metadata);
         this.storage[threadId][checkpointNamespace][checkpoint.id] = [
-            this.serde.stringify(checkpoint),
-            this.serde.stringify(metadata),
+            serializedCheckpoint,
+            serializedMetadata,
             config.configurable?.checkpoint_id, // parent
         ];
         return {
@@ -188,7 +202,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
             this.writes[key] = [];
         }
         const pendingWrites = writes.map(([channel, value]) => {
-            return [taskId, channel, this.serde.stringify(value)];
+            const [, serializedValue] = this.serde.dumpsTyped(value);
+            return [taskId, channel, serializedValue];
         });
         this.writes[key].push(...pendingWrites);
     }

package/dist/memory.d.ts

@@ -3,9 +3,9 @@ import { BaseCheckpointSaver, Checkpoint, CheckpointListOptions, CheckpointTuple
 import { SerializerProtocol } from "./serde/base.js";
 import { CheckpointMetadata, CheckpointPendingWrite, PendingWrite } from "./types.js";
 export declare class MemorySaver extends BaseCheckpointSaver {
-    storage: Record<string, Record<string, Record<string, [string, string, string | undefined]>>>;
+    storage: Record<string, Record<string, Record<string, [Uint8Array, Uint8Array, string | undefined]>>>;
     writes: Record<string, CheckpointPendingWrite[]>;
-    constructor(serde?: SerializerProtocol<unknown>);
+    constructor(serde?: SerializerProtocol);
     getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;
     list(config: RunnableConfig, options?: CheckpointListOptions): AsyncGenerator<CheckpointTuple>;
     put(config: RunnableConfig, checkpoint: Checkpoint, metadata: CheckpointMetadata): Promise<RunnableConfig>;

package/dist/memory.js

@@ -30,7 +30,11 @@ export class MemorySaver extends BaseCheckpointSaver {
                 const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
                     [];
                 const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [taskId, channel, await this.serde.parse(value)];
+                    return [
+                        taskId,
+                        channel,
+                        await this.serde.loadsTyped("json", value),
+                    ];
                 }));
                 const parentConfig = parentCheckpointId !== undefined
                     ? {
@@ -43,8 +47,8 @@ export class MemorySaver extends BaseCheckpointSaver {
                     : undefined;
                 return {
                     config,
-                    checkpoint: (await this.serde.parse(checkpoint)),
-                    metadata: (await this.serde.parse(metadata)),
+                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                     parentConfig,
                 };
@@ -60,7 +64,11 @@ export class MemorySaver extends BaseCheckpointSaver {
                 const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
                     [];
                 const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [taskId, channel, await this.serde.parse(value)];
+                    return [
+                        taskId,
+                        channel,
+                        await this.serde.loadsTyped("json", value),
+                    ];
                 }));
                 const parentConfig = parentCheckpointId !== undefined
                     ? {
@@ -79,8 +87,8 @@ export class MemorySaver extends BaseCheckpointSaver {
                             checkpoint_ns,
                         },
                     },
-                    checkpoint: (await this.serde.parse(checkpoint)),
-                    metadata: (await this.serde.parse(metadata)),
+                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                     parentConfig,
                 };
@@ -106,7 +114,7 @@ export class MemorySaver extends BaseCheckpointSaver {
                     continue;
                 }
                 // Parse metadata
-                const metadata = (await this.serde.parse(metadataStr));
+                const metadata = (await this.serde.loadsTyped("json", metadataStr));
                 // Limit search results
                 if (limit !== undefined) {
                     if (limit <= 0)
@@ -116,7 +124,11 @@ export class MemorySaver extends BaseCheckpointSaver {
                 }
                 const writes = this.writes[_generateKey(threadId, checkpointNamespace, checkpointId)] ?? [];
                 const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
-                    return [taskId, channel, await this.serde.parse(value)];
+                    return [
+                        taskId,
+                        channel,
+                        await this.serde.loadsTyped("json", value),
+                    ];
                 }));
                 yield {
                     config: {
@@ -126,7 +138,7 @@ export class MemorySaver extends BaseCheckpointSaver {
                             checkpoint_id: checkpointId,
                         },
                     },
-                    checkpoint: (await this.serde.parse(checkpoint)),
+                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
                     metadata,
                     pendingWrites,
                     parentConfig: parentCheckpointId
@@ -157,9 +169,11 @@ export class MemorySaver extends BaseCheckpointSaver {
         if (!this.storage[threadId][checkpointNamespace]) {
             this.storage[threadId][checkpointNamespace] = {};
         }
+        const [, serializedCheckpoint] = this.serde.dumpsTyped(checkpoint);
+        const [, serializedMetadata] = this.serde.dumpsTyped(metadata);
         this.storage[threadId][checkpointNamespace][checkpoint.id] = [
-            this.serde.stringify(checkpoint),
-            this.serde.stringify(metadata),
+            serializedCheckpoint,
+            serializedMetadata,
             config.configurable?.checkpoint_id, // parent
         ];
         return {
@@ -185,7 +199,8 @@ export class MemorySaver extends BaseCheckpointSaver {
             this.writes[key] = [];
         }
         const pendingWrites = writes.map(([channel, value]) => {
-            return [taskId, channel, this.serde.stringify(value)];
+            const [, serializedValue] = this.serde.dumpsTyped(value);
+            return [taskId, channel, serializedValue];
         });
         this.writes[key].push(...pendingWrites);
     }

package/dist/serde/base.cjs

@@ -1,8 +1,2 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.DefaultSerializer = void 0;
-const load_1 = require("@langchain/core/load");
-exports.DefaultSerializer = {
-    stringify: JSON.stringify,
-    parse: load_1.load,
-};

package/dist/serde/base.d.ts

@@ -1,12 +1,4 @@
-import { load } from "@langchain/core/load";
-export interface SerializerProtocol<D> {
-    stringify(obj: D): string;
-    parse(data: string): Promise<D>;
+export interface SerializerProtocol {
+    dumpsTyped(data: any): [string, Uint8Array];
+    loadsTyped(type: string, data: Uint8Array | string): any;
 }
-export declare const DefaultSerializer: {
-    stringify: {
-        (value: any, replacer?: ((this: any, key: string, value: any) => any) | undefined, space?: string | number | undefined): string;
-        (value: any, replacer?: (string | number)[] | null | undefined, space?: string | number | undefined): string;
-    };
-    parse: typeof load;
-};

package/dist/serde/base.js

@@ -1,5 +1 @@
-import { load } from "@langchain/core/load";
-export const DefaultSerializer = {
-    stringify: JSON.stringify,
-    parse: load,
-};
+export {};

package/dist/serde/jsonplus.cjs

@@ -0,0 +1,131 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.JsonPlusSerializer = void 0;
+/* eslint-disable @typescript-eslint/no-explicit-any */
+/* eslint-disable no-instanceof/no-instanceof */
+const load_1 = require("@langchain/core/load");
+async function _reviver(value) {
+    if (value && typeof value === "object") {
+        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 (value.lc === 1) {
+            return (0, load_1.load)(JSON.stringify(value));
+        }
+        else if (Array.isArray(value)) {
+            return Promise.all(value.map((item) => _reviver(item)));
+        }
+        else {
+            const revivedObj = {};
+            for (const [k, v] of Object.entries(value)) {
+                revivedObj[k] = await _reviver(v);
+            }
+            return revivedObj;
+        }
+    }
+    return value;
+}
+function _encodeConstructorArgs(
+// eslint-disable-next-line @typescript-eslint/ban-types
+constructor, method, args, kwargs) {
+    return {
+        lc: 2,
+        type: "constructor",
+        id: [constructor.name],
+        method,
+        args: args ?? [],
+        kwargs: kwargs ?? {},
+    };
+}
+function _default(_key, obj) {
+    if (obj instanceof Set || obj instanceof Map) {
+        return _encodeConstructorArgs(obj.constructor, undefined, [
+            Array.from(obj),
+        ]);
+    }
+    else if (obj instanceof RegExp) {
+        return _encodeConstructorArgs(RegExp, undefined, [obj.source, obj.flags]);
+    }
+    else if (obj instanceof Error) {
+        return _encodeConstructorArgs(obj.constructor, undefined, [obj.message]);
+    }
+    else {
+        return 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);
+    }
+    dumpsTyped(obj) {
+        if (obj instanceof Uint8Array) {
+            return ["bytes", obj];
+        }
+        else {
+            return ["json", this._dumps(obj)];
+        }
+    }
+    async _loads(data) {
+        const parsed = JSON.parse(data);
+        return _reviver(parsed);
+    }
+    async loadsTyped(type, data) {
+        if (type === "bytes") {
+            return typeof data === "string" ? new TextEncoder().encode(data) : data;
+        }
+        else if (type === "json") {
+            return this._loads(typeof data === "string" ? data : new TextDecoder().decode(data));
+        }
+        else {
+            throw new Error(`Unknown serialization type: ${type}`);
+        }
+    }
+}
+exports.JsonPlusSerializer = JsonPlusSerializer;

package/dist/serde/jsonplus.d.ts

@@ -0,0 +1,7 @@
+import { SerializerProtocol } from "./base.js";
+export declare class JsonPlusSerializer implements SerializerProtocol {
+    protected _dumps(obj: any): Uint8Array;
+    dumpsTyped(obj: any): [string, Uint8Array];
+    protected _loads(data: string): Promise<any>;
+    loadsTyped(type: string, data: Uint8Array | string): Promise<any>;
+}

package/dist/serde/jsonplus.js

@@ -0,0 +1,127 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
+/* eslint-disable no-instanceof/no-instanceof */
+import { load } from "@langchain/core/load";
+async function _reviver(value) {
+    if (value && typeof value === "object") {
+        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 (value.lc === 1) {
+            return load(JSON.stringify(value));
+        }
+        else if (Array.isArray(value)) {
+            return Promise.all(value.map((item) => _reviver(item)));
+        }
+        else {
+            const revivedObj = {};
+            for (const [k, v] of Object.entries(value)) {
+                revivedObj[k] = await _reviver(v);
+            }
+            return revivedObj;
+        }
+    }
+    return value;
+}
+function _encodeConstructorArgs(
+// eslint-disable-next-line @typescript-eslint/ban-types
+constructor, method, args, kwargs) {
+    return {
+        lc: 2,
+        type: "constructor",
+        id: [constructor.name],
+        method,
+        args: args ?? [],
+        kwargs: kwargs ?? {},
+    };
+}
+function _default(_key, obj) {
+    if (obj instanceof Set || obj instanceof Map) {
+        return _encodeConstructorArgs(obj.constructor, undefined, [
+            Array.from(obj),
+        ]);
+    }
+    else if (obj instanceof RegExp) {
+        return _encodeConstructorArgs(RegExp, undefined, [obj.source, obj.flags]);
+    }
+    else if (obj instanceof Error) {
+        return _encodeConstructorArgs(obj.constructor, undefined, [obj.message]);
+    }
+    else {
+        return 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);
+    }
+    dumpsTyped(obj) {
+        if (obj instanceof Uint8Array) {
+            return ["bytes", obj];
+        }
+        else {
+            return ["json", this._dumps(obj)];
+        }
+    }
+    async _loads(data) {
+        const parsed = JSON.parse(data);
+        return _reviver(parsed);
+    }
+    async loadsTyped(type, data) {
+        if (type === "bytes") {
+            return typeof data === "string" ? new TextEncoder().encode(data) : data;
+        }
+        else if (type === "json") {
+            return this._loads(typeof data === "string" ? data : new TextDecoder().decode(data));
+        }
+        else {
+            throw new Error(`Unknown serialization type: ${type}`);
+        }
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-checkpoint",
-  "version": "0.0.0",
+  "version": "0.0.1",
   "description": "Library with base interfaces for LangGraph checkpoint savers.",
   "type": "module",
   "engines": {
@@ -14,7 +14,7 @@
   },
   "scripts": {
     "build": "yarn turbo:command build:internal --filter=@langchain/langgraph-checkpoint",
-    "build:internal": "yarn lc_build_v2 --create-entrypoints --pre --tree-shaking",
+    "build:internal": "yarn lc_build --create-entrypoints --pre --tree-shaking",
     "lint:eslint": "NODE_OPTIONS=--max-old-space-size=4096 eslint --cache --ext .ts,.js src/",
     "lint:dpdm": "dpdm --exit-code circular:1 --no-warning --no-tree src/*.ts src/**/*.ts",
     "lint": "yarn lint:eslint && yarn lint:dpdm",
@@ -37,7 +37,7 @@
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
-    "@langchain/scripts": "^0.0.22",
+    "@langchain/scripts": ">=0.1.0 <0.2.0",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",
     "@tsconfig/recommended": "^1.0.3",