@langchain/langgraph-checkpoint 0.0.13 → 0.0.14

package/dist/base.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.WRITES_IDX_MAP = exports.maxChannelVersion = exports.compareChannelVersions = exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
+exports.getCheckpointId = exports.WRITES_IDX_MAP = exports.maxChannelVersion = exports.compareChannelVersions = exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
 const id_js_1 = require("./id.cjs");
 const types_js_1 = require("./serde/types.cjs");
 const jsonplus_js_1 = require("./serde/jsonplus.cjs");
@@ -99,3 +99,7 @@ exports.WRITES_IDX_MAP = {
     [types_js_1.INTERRUPT]: -3,
     [types_js_1.RESUME]: -4,
 };
+function getCheckpointId(config) {
+    return (config.configurable?.checkpoint_id || config.configurable?.thread_ts || "");
+}
+exports.getCheckpointId = getCheckpointId;

package/dist/base.d.ts

@@ -86,4 +86,5 @@ export declare function maxChannelVersion(...versions: ChannelVersion[]): Channe
  * Each Checkpointer implementation should use this mapping in put_writes.
  */
 export declare const WRITES_IDX_MAP: Record<string, number>;
+export declare function getCheckpointId(config: RunnableConfig): string;
 export {};

package/dist/base.js

@@ -90,3 +90,6 @@ export const WRITES_IDX_MAP = {
     [INTERRUPT]: -3,
     [RESUME]: -4,
 };
+export function getCheckpointId(config) {
+    return (config.configurable?.checkpoint_id || config.configurable?.thread_ts || "");
+}

package/dist/memory.cjs

@@ -26,7 +26,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
     async _getPendingSends(threadId, checkpointNs, parentCheckpointId) {
         let pendingSends = [];
         if (parentCheckpointId !== undefined) {
-            pendingSends = await Promise.all(this.writes[_generateKey(threadId, checkpointNs, parentCheckpointId)]
+            const key = _generateKey(threadId, checkpointNs, parentCheckpointId);
+            pendingSends = await Promise.all(Object.values(this.writes[key] || {})
                 ?.filter(([_taskId, channel]) => {
                 return channel === types_js_1.TASKS;
             })
@@ -39,19 +40,18 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
     async getTuple(config) {
         const thread_id = config.configurable?.thread_id;
         const checkpoint_ns = config.configurable?.checkpoint_ns ?? "";
-        let checkpoint_id = config.configurable?.checkpoint_id;
+        let checkpoint_id = (0, base_js_1.getCheckpointId)(config);
         if (checkpoint_id) {
             const saved = this.storage[thread_id]?.[checkpoint_ns]?.[checkpoint_id];
             if (saved !== undefined) {
                 const [checkpoint, metadata, parentCheckpointId] = saved;
-                const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
-                    [];
+                const key = _generateKey(thread_id, checkpoint_ns, checkpoint_id);
                 const pending_sends = await this._getPendingSends(thread_id, checkpoint_ns, parentCheckpointId);
                 const deserializedCheckpoint = {
                     ...(await this.serde.loadsTyped("json", checkpoint)),
                     pending_sends,
                 };
-                const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
+                const pendingWrites = await Promise.all(Object.values(this.writes[key] || {}).map(async ([taskId, channel, value]) => {
                     return [
                         taskId,
                         channel,
@@ -83,14 +83,13 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 checkpoint_id = Object.keys(checkpoints).sort((a, b) => b.localeCompare(a))[0];
                 const saved = checkpoints[checkpoint_id];
                 const [checkpoint, metadata, parentCheckpointId] = saved;
-                const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
-                    [];
+                const key = _generateKey(thread_id, checkpoint_ns, checkpoint_id);
                 const pending_sends = await this._getPendingSends(thread_id, checkpoint_ns, parentCheckpointId);
                 const deserializedCheckpoint = {
                     ...(await this.serde.loadsTyped("json", checkpoint)),
                     pending_sends,
                 };
-                const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
+                const pendingWrites = await Promise.all(Object.values(this.writes[key] || {}).map(async ([taskId, channel, value]) => {
                     return [
                         taskId,
                         channel,
@@ -130,6 +129,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
             ? [config.configurable?.thread_id]
             : Object.keys(this.storage);
         const configCheckpointNamespace = config.configurable?.checkpoint_ns;
+        const configCheckpointId = config.configurable?.checkpoint_id;
         for (const threadId of threadIds) {
             for (const checkpointNamespace of Object.keys(this.storage[threadId] ?? {})) {
                 if (configCheckpointNamespace !== undefined &&
@@ -139,7 +139,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 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
+                    // Filter by checkpoint ID from config
+                    if (configCheckpointId && checkpointId !== configCheckpointId) {
+                        continue;
+                    }
+                    // Filter by checkpoint ID from before config
                     if (before &&
                         before.configurable?.checkpoint_id &&
                         checkpointId >= before.configurable.checkpoint_id) {
@@ -155,10 +159,10 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                     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 key = _generateKey(threadId, checkpointNamespace, checkpointId);
+                    const writes = Object.values(this.writes[key] || {});
                     const pending_sends = await this._getPendingSends(threadId, checkpointNamespace, parentCheckpointId);
                     const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
                         return [
@@ -236,15 +240,23 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
         if (checkpointId === undefined) {
             throw new Error(`Failed to put writes. The passed RunnableConfig is missing a required "checkpoint_id" field in its "configurable" property.`);
         }
-        const key = _generateKey(threadId, checkpointNamespace, checkpointId);
-        if (this.writes[key] === undefined) {
-            this.writes[key] = [];
+        const outerKey = _generateKey(threadId, checkpointNamespace, checkpointId);
+        const outerWrites_ = this.writes[outerKey];
+        if (this.writes[outerKey] === undefined) {
+            this.writes[outerKey] = {};
         }
-        const pendingWrites = writes.map(([channel, value]) => {
+        writes.forEach(([channel, value], idx) => {
             const [, serializedValue] = this.serde.dumpsTyped(value);
-            return [taskId, channel, serializedValue];
+            const innerKey = [
+                taskId,
+                base_js_1.WRITES_IDX_MAP[channel] || idx,
+            ];
+            const innerKeyStr = `${innerKey[0]},${innerKey[1]}`;
+            if (innerKey[1] >= 0 && outerWrites_ && innerKeyStr in outerWrites_) {
+                return;
+            }
+            this.writes[outerKey][innerKeyStr] = [taskId, channel, serializedValue];
         });
-        this.writes[key].push(...pendingWrites);
     }
 }
 exports.MemorySaver = MemorySaver;

package/dist/memory.d.ts

@@ -1,11 +1,11 @@
 import type { RunnableConfig } from "@langchain/core/runnables";
 import { BaseCheckpointSaver, Checkpoint, CheckpointListOptions, CheckpointTuple } from "./base.js";
 import { SerializerProtocol } from "./serde/base.js";
-import { CheckpointMetadata, CheckpointPendingWrite, PendingWrite } from "./types.js";
+import { CheckpointMetadata, PendingWrite } from "./types.js";
 import { SendProtocol } from "./serde/types.js";
 export declare class MemorySaver extends BaseCheckpointSaver {
     storage: Record<string, Record<string, Record<string, [Uint8Array, Uint8Array, string | undefined]>>>;
-    writes: Record<string, CheckpointPendingWrite[]>;
+    writes: Record<string, Record<string, [string, string, Uint8Array]>>;
     constructor(serde?: SerializerProtocol);
     _getPendingSends(threadId: string, checkpointNs: string, parentCheckpointId?: string): Promise<SendProtocol[]>;
     getTuple(config: RunnableConfig): Promise<CheckpointTuple | undefined>;

package/dist/memory.js

@@ -1,4 +1,4 @@
-import { BaseCheckpointSaver, copyCheckpoint, } from "./base.js";
+import { BaseCheckpointSaver, copyCheckpoint, getCheckpointId, WRITES_IDX_MAP, } from "./base.js";
 import { TASKS } from "./serde/types.js";
 function _generateKey(threadId, checkpointNamespace, checkpointId) {
     return JSON.stringify([threadId, checkpointNamespace, checkpointId]);
@@ -23,7 +23,8 @@ export class MemorySaver extends BaseCheckpointSaver {
     async _getPendingSends(threadId, checkpointNs, parentCheckpointId) {
         let pendingSends = [];
         if (parentCheckpointId !== undefined) {
-            pendingSends = await Promise.all(this.writes[_generateKey(threadId, checkpointNs, parentCheckpointId)]
+            const key = _generateKey(threadId, checkpointNs, parentCheckpointId);
+            pendingSends = await Promise.all(Object.values(this.writes[key] || {})
                 ?.filter(([_taskId, channel]) => {
                 return channel === TASKS;
             })
@@ -36,19 +37,18 @@ export class MemorySaver extends BaseCheckpointSaver {
     async getTuple(config) {
         const thread_id = config.configurable?.thread_id;
         const checkpoint_ns = config.configurable?.checkpoint_ns ?? "";
-        let checkpoint_id = config.configurable?.checkpoint_id;
+        let checkpoint_id = getCheckpointId(config);
         if (checkpoint_id) {
             const saved = this.storage[thread_id]?.[checkpoint_ns]?.[checkpoint_id];
             if (saved !== undefined) {
                 const [checkpoint, metadata, parentCheckpointId] = saved;
-                const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
-                    [];
+                const key = _generateKey(thread_id, checkpoint_ns, checkpoint_id);
                 const pending_sends = await this._getPendingSends(thread_id, checkpoint_ns, parentCheckpointId);
                 const deserializedCheckpoint = {
                     ...(await this.serde.loadsTyped("json", checkpoint)),
                     pending_sends,
                 };
-                const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
+                const pendingWrites = await Promise.all(Object.values(this.writes[key] || {}).map(async ([taskId, channel, value]) => {
                     return [
                         taskId,
                         channel,
@@ -80,14 +80,13 @@ export class MemorySaver extends BaseCheckpointSaver {
                 checkpoint_id = Object.keys(checkpoints).sort((a, b) => b.localeCompare(a))[0];
                 const saved = checkpoints[checkpoint_id];
                 const [checkpoint, metadata, parentCheckpointId] = saved;
-                const writes = this.writes[_generateKey(thread_id, checkpoint_ns, checkpoint_id)] ??
-                    [];
+                const key = _generateKey(thread_id, checkpoint_ns, checkpoint_id);
                 const pending_sends = await this._getPendingSends(thread_id, checkpoint_ns, parentCheckpointId);
                 const deserializedCheckpoint = {
                     ...(await this.serde.loadsTyped("json", checkpoint)),
                     pending_sends,
                 };
-                const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
+                const pendingWrites = await Promise.all(Object.values(this.writes[key] || {}).map(async ([taskId, channel, value]) => {
                     return [
                         taskId,
                         channel,
@@ -127,6 +126,7 @@ export class MemorySaver extends BaseCheckpointSaver {
             ? [config.configurable?.thread_id]
             : Object.keys(this.storage);
         const configCheckpointNamespace = config.configurable?.checkpoint_ns;
+        const configCheckpointId = config.configurable?.checkpoint_id;
         for (const threadId of threadIds) {
             for (const checkpointNamespace of Object.keys(this.storage[threadId] ?? {})) {
                 if (configCheckpointNamespace !== undefined &&
@@ -136,7 +136,11 @@ export class MemorySaver extends BaseCheckpointSaver {
                 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
+                    // Filter by checkpoint ID from config
+                    if (configCheckpointId && checkpointId !== configCheckpointId) {
+                        continue;
+                    }
+                    // Filter by checkpoint ID from before config
                     if (before &&
                         before.configurable?.checkpoint_id &&
                         checkpointId >= before.configurable.checkpoint_id) {
@@ -152,10 +156,10 @@ export class MemorySaver extends BaseCheckpointSaver {
                     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 key = _generateKey(threadId, checkpointNamespace, checkpointId);
+                    const writes = Object.values(this.writes[key] || {});
                     const pending_sends = await this._getPendingSends(threadId, checkpointNamespace, parentCheckpointId);
                     const pendingWrites = await Promise.all(writes.map(async ([taskId, channel, value]) => {
                         return [
@@ -233,14 +237,22 @@ export class MemorySaver extends BaseCheckpointSaver {
         if (checkpointId === undefined) {
             throw new Error(`Failed to put writes. The passed RunnableConfig is missing a required "checkpoint_id" field in its "configurable" property.`);
         }
-        const key = _generateKey(threadId, checkpointNamespace, checkpointId);
-        if (this.writes[key] === undefined) {
-            this.writes[key] = [];
+        const outerKey = _generateKey(threadId, checkpointNamespace, checkpointId);
+        const outerWrites_ = this.writes[outerKey];
+        if (this.writes[outerKey] === undefined) {
+            this.writes[outerKey] = {};
         }
-        const pendingWrites = writes.map(([channel, value]) => {
+        writes.forEach(([channel, value], idx) => {
             const [, serializedValue] = this.serde.dumpsTyped(value);
-            return [taskId, channel, serializedValue];
+            const innerKey = [
+                taskId,
+                WRITES_IDX_MAP[channel] || idx,
+            ];
+            const innerKeyStr = `${innerKey[0]},${innerKey[1]}`;
+            if (innerKey[1] >= 0 && outerWrites_ && innerKeyStr in outerWrites_) {
+                return;
+            }
+            this.writes[outerKey][innerKeyStr] = [taskId, channel, serializedValue];
         });
-        this.writes[key].push(...pendingWrites);
     }
 }

package/dist/store/batch.cjs

@@ -117,5 +117,16 @@ class AsyncBatchedStore extends base_js_1.BaseStore {
             }
         }
     }
+    // AsyncBatchedStore is internal and gets passed as args into traced tasks
+    // some BaseStores contain circular references so just serialize without it
+    // as this causes warnings when tracing with LangSmith.
+    toJSON() {
+        return {
+            queue: this.queue,
+            nextKey: this.nextKey,
+            running: this.running,
+            store: "[LangGraphStore]",
+        };
+    }
 }
 exports.AsyncBatchedStore = AsyncBatchedStore;

package/dist/store/batch.d.ts

@@ -26,4 +26,14 @@ export declare class AsyncBatchedStore extends BaseStore {
     stop(): Promise<void>;
     private enqueueOperation;
     private processBatchQueue;
+    toJSON(): {
+        queue: Map<number, {
+            operation: Operation;
+            resolve: (value: any) => void;
+            reject: (reason?: any) => void;
+        }>;
+        nextKey: number;
+        running: boolean;
+        store: string;
+    };
 }

package/dist/store/batch.js

@@ -114,4 +114,15 @@ export class AsyncBatchedStore extends BaseStore {
             }
         }
     }
+    // AsyncBatchedStore is internal and gets passed as args into traced tasks
+    // some BaseStores contain circular references so just serialize without it
+    // as this causes warnings when tracing with LangSmith.
+    toJSON() {
+        return {
+            queue: this.queue,
+            nextKey: this.nextKey,
+            running: this.running,
+            store: "[LangGraphStore]",
+        };
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-checkpoint",
-  "version": "0.0.13",
+  "version": "0.0.14",
   "description": "Library with base interfaces for LangGraph checkpoint savers.",
   "type": "module",
   "engines": {
@@ -58,7 +58,7 @@
     "jest-environment-node": "^29.6.4",
     "prettier": "^2.8.3",
     "release-it": "^17.6.0",
-    "rollup": "^4.22.4",
+    "rollup": "^4.29.2",
     "ts-jest": "^29.1.0",
     "tsx": "^4.7.0",
     "typescript": "^4.9.5 || ^5.4.5"