@langchain/langgraph-checkpoint 0.0.4 → 0.0.6

package/dist/memory.cjs

@@ -2,6 +2,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.MemorySaver = void 0;
 const base_js_1 = require("./base.cjs");
+const types_js_1 = require("./serde/types.cjs");
 function _generateKey(threadId, checkpointNamespace, checkpointId) {
     return JSON.stringify([threadId, checkpointNamespace, checkpointId]);
 }
@@ -22,6 +23,19 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
             value: {}
         });
     }
+    async _getPendingSends(threadId, checkpointNs, parentCheckpointId) {
+        let pendingSends = [];
+        if (parentCheckpointId !== undefined) {
+            pendingSends = await Promise.all(this.writes[_generateKey(threadId, checkpointNs, parentCheckpointId)]
+                ?.filter(([_taskId, channel]) => {
+                return channel === types_js_1.TASKS;
+            })
+                .map(([_taskId, _channel, writes]) => {
+                return this.serde.loadsTyped("json", writes);
+            }) ?? []);
+        }
+        return pendingSends;
+    }
     async getTuple(config) {
         const thread_id = config.configurable?.thread_id;
         const checkpoint_ns = config.configurable?.checkpoint_ns ?? "";
@@ -32,6 +46,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 const [checkpoint, metadata, parentCheckpointId] = saved;
                 const writes = this.writes[_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]) => {
                     return [
                         taskId,
@@ -41,7 +60,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 }));
                 const checkpointTuple = {
                     config,
-                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    checkpoint: deserializedCheckpoint,
                     metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                 };
@@ -66,6 +85,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 const [checkpoint, metadata, parentCheckpointId] = saved;
                 const writes = this.writes[_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]) => {
                     return [
                         taskId,
@@ -81,7 +105,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                             checkpoint_ns,
                         },
                     },
-                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    checkpoint: deserializedCheckpoint,
                     metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                 };
@@ -126,6 +150,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                     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,
@@ -133,6 +158,10 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                         await this.serde.loadsTyped("json", value),
                     ];
                 }));
+                const deserializedCheckpoint = {
+                    ...(await this.serde.loadsTyped("json", checkpoint)),
+                    pending_sends,
+                };
                 const checkpointTuple = {
                     config: {
                         configurable: {
@@ -141,7 +170,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                             checkpoint_id: checkpointId,
                         },
                     },
-                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    checkpoint: deserializedCheckpoint,
                     metadata,
                     pendingWrites,
                 };
@@ -159,6 +188,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
         }
     }
     async put(config, checkpoint, metadata) {
+        const preparedCheckpoint = (0, base_js_1.copyCheckpoint)(checkpoint);
+        delete preparedCheckpoint.pending_sends;
         const threadId = config.configurable?.thread_id;
         const checkpointNamespace = config.configurable?.checkpoint_ns;
         if (threadId === undefined) {
@@ -173,7 +204,7 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
         if (!this.storage[threadId][checkpointNamespace]) {
             this.storage[threadId][checkpointNamespace] = {};
         }
-        const [, serializedCheckpoint] = this.serde.dumpsTyped(checkpoint);
+        const [, serializedCheckpoint] = this.serde.dumpsTyped(preparedCheckpoint);
         const [, serializedMetadata] = this.serde.dumpsTyped(metadata);
         this.storage[threadId][checkpointNamespace][checkpoint.id] = [
             serializedCheckpoint,

package/dist/memory.d.ts

@@ -2,10 +2,12 @@ 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 { 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[]>;
     constructor(serde?: SerializerProtocol);
+    _getPendingSends(threadId: string, checkpointNs: string, parentCheckpointId?: string): Promise<SendProtocol[]>;
     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

@@ -1,4 +1,5 @@
-import { BaseCheckpointSaver, } from "./base.js";
+import { BaseCheckpointSaver, copyCheckpoint, } from "./base.js";
+import { TASKS } from "./serde/types.js";
 function _generateKey(threadId, checkpointNamespace, checkpointId) {
     return JSON.stringify([threadId, checkpointNamespace, checkpointId]);
 }
@@ -19,6 +20,19 @@ export class MemorySaver extends BaseCheckpointSaver {
             value: {}
         });
     }
+    async _getPendingSends(threadId, checkpointNs, parentCheckpointId) {
+        let pendingSends = [];
+        if (parentCheckpointId !== undefined) {
+            pendingSends = await Promise.all(this.writes[_generateKey(threadId, checkpointNs, parentCheckpointId)]
+                ?.filter(([_taskId, channel]) => {
+                return channel === TASKS;
+            })
+                .map(([_taskId, _channel, writes]) => {
+                return this.serde.loadsTyped("json", writes);
+            }) ?? []);
+        }
+        return pendingSends;
+    }
     async getTuple(config) {
         const thread_id = config.configurable?.thread_id;
         const checkpoint_ns = config.configurable?.checkpoint_ns ?? "";
@@ -29,6 +43,11 @@ export class MemorySaver extends BaseCheckpointSaver {
                 const [checkpoint, metadata, parentCheckpointId] = saved;
                 const writes = this.writes[_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]) => {
                     return [
                         taskId,
@@ -38,7 +57,7 @@ export class MemorySaver extends BaseCheckpointSaver {
                 }));
                 const checkpointTuple = {
                     config,
-                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    checkpoint: deserializedCheckpoint,
                     metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                 };
@@ -63,6 +82,11 @@ export class MemorySaver extends BaseCheckpointSaver {
                 const [checkpoint, metadata, parentCheckpointId] = saved;
                 const writes = this.writes[_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]) => {
                     return [
                         taskId,
@@ -78,7 +102,7 @@ export class MemorySaver extends BaseCheckpointSaver {
                             checkpoint_ns,
                         },
                     },
-                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    checkpoint: deserializedCheckpoint,
                     metadata: (await this.serde.loadsTyped("json", metadata)),
                     pendingWrites,
                 };
@@ -123,6 +147,7 @@ export class MemorySaver extends BaseCheckpointSaver {
                     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,
@@ -130,6 +155,10 @@ export class MemorySaver extends BaseCheckpointSaver {
                         await this.serde.loadsTyped("json", value),
                     ];
                 }));
+                const deserializedCheckpoint = {
+                    ...(await this.serde.loadsTyped("json", checkpoint)),
+                    pending_sends,
+                };
                 const checkpointTuple = {
                     config: {
                         configurable: {
@@ -138,7 +167,7 @@ export class MemorySaver extends BaseCheckpointSaver {
                             checkpoint_id: checkpointId,
                         },
                     },
-                    checkpoint: (await this.serde.loadsTyped("json", checkpoint)),
+                    checkpoint: deserializedCheckpoint,
                     metadata,
                     pendingWrites,
                 };
@@ -156,6 +185,8 @@ export class MemorySaver extends BaseCheckpointSaver {
         }
     }
     async put(config, checkpoint, metadata) {
+        const preparedCheckpoint = copyCheckpoint(checkpoint);
+        delete preparedCheckpoint.pending_sends;
         const threadId = config.configurable?.thread_id;
         const checkpointNamespace = config.configurable?.checkpoint_ns;
         if (threadId === undefined) {
@@ -170,7 +201,7 @@ export class MemorySaver extends BaseCheckpointSaver {
         if (!this.storage[threadId][checkpointNamespace]) {
             this.storage[threadId][checkpointNamespace] = {};
         }
-        const [, serializedCheckpoint] = this.serde.dumpsTyped(checkpoint);
+        const [, serializedCheckpoint] = this.serde.dumpsTyped(preparedCheckpoint);
         const [, serializedMetadata] = this.serde.dumpsTyped(metadata);
         this.storage[threadId][checkpointNamespace][checkpoint.id] = [
             serializedCheckpoint,

package/dist/serde/jsonplus.cjs

@@ -133,6 +133,13 @@ function _default(_key, obj) {
     }
     else if (obj instanceof Error) {
         return _encodeConstructorArgs(obj.constructor, undefined, [obj.message]);
+        // TODO: Remove special case
+    }
+    else if (obj?.lg_name === "Send") {
+        return {
+            node: obj.node,
+            args: obj.args,
+        };
     }
     else {
         return obj;

package/dist/serde/jsonplus.js

@@ -130,6 +130,13 @@ function _default(_key, obj) {
     }
     else if (obj instanceof Error) {
         return _encodeConstructorArgs(obj.constructor, undefined, [obj.message]);
+        // TODO: Remove special case
+    }
+    else if (obj?.lg_name === "Send") {
+        return {
+            node: obj.node,
+            args: obj.args,
+        };
     }
     else {
         return obj;

package/dist/serde/types.cjs

@@ -1,2 +1,4 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.TASKS = void 0;
+exports.TASKS = "__pregel_tasks";

package/dist/serde/types.d.ts

@@ -1,3 +1,4 @@
+export declare const TASKS = "__pregel_tasks";
 export interface ChannelProtocol<ValueType = unknown, UpdateType = unknown, CheckpointType = unknown> {
     ValueType: ValueType;
     UpdateType: UpdateType;

package/dist/serde/types.js

@@ -1 +1 @@
-export {};
+export const TASKS = "__pregel_tasks";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph-checkpoint",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "Library with base interfaces for LangGraph checkpoint savers.",
   "type": "module",
   "engines": {
@@ -34,7 +34,7 @@
     "uuid": "^10.0.0"
   },
   "peerDependencies": {
-    "@langchain/core": ">=0.2.20 <0.3.0"
+    "@langchain/core": ">=0.2.31 <0.4.0"
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",