@langchain/langgraph-checkpoint 0.0.17 → 0.1.0

package/CHANGELOG.md

@@ -0,0 +1,13 @@
+# @langchain/langgraph-checkpoint
+
+## 0.1.0
+
+### Minor Changes
+
+- 10f292a: Remove pending_sends from parent checkpoints in favour of TASKS channel
+- 3fd7f73: Allow asynchronous serialization and deserialization
+- 773ec0d: Remove Checkpoint.writes
+
+### Patch Changes
+
+- ccbcbc1: Add delete thread method to checkpointers

package/dist/base.cjs

@@ -1,6 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getCheckpointId = exports.WRITES_IDX_MAP = exports.maxChannelVersion = exports.compareChannelVersions = exports.BaseCheckpointSaver = exports.copyCheckpoint = exports.emptyCheckpoint = exports.deepCopy = void 0;
+exports.WRITES_IDX_MAP = exports.BaseCheckpointSaver = void 0;
+exports.deepCopy = deepCopy;
+exports.emptyCheckpoint = emptyCheckpoint;
+exports.copyCheckpoint = copyCheckpoint;
+exports.compareChannelVersions = compareChannelVersions;
+exports.maxChannelVersion = maxChannelVersion;
+exports.getCheckpointId = getCheckpointId;
 const id_js_1 = require("./id.cjs");
 const types_js_1 = require("./serde/types.cjs");
 const jsonplus_js_1 = require("./serde/jsonplus.cjs");
@@ -16,20 +22,17 @@ function deepCopy(obj) {
     }
     return newObj;
 }
-exports.deepCopy = deepCopy;
 /** @hidden */
 function emptyCheckpoint() {
     return {
-        v: 1,
+        v: 4,
         id: (0, id_js_1.uuid6)(-2),
         ts: new Date().toISOString(),
         channel_values: {},
         channel_versions: {},
         versions_seen: {},
-        pending_sends: [],
     };
 }
-exports.emptyCheckpoint = emptyCheckpoint;
 /** @hidden */
 function copyCheckpoint(checkpoint) {
     return {
@@ -39,10 +42,8 @@ function copyCheckpoint(checkpoint) {
         channel_values: { ...checkpoint.channel_values },
         channel_versions: { ...checkpoint.channel_versions },
         versions_seen: deepCopy(checkpoint.versions_seen),
-        pending_sends: [...checkpoint.pending_sends],
     };
 }
-exports.copyCheckpoint = copyCheckpoint;
 class BaseCheckpointSaver {
     constructor(serde) {
         Object.defineProperty(this, "serde", {
@@ -63,7 +64,7 @@ class BaseCheckpointSaver {
      * 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, _channel) {
+    getNextVersion(current) {
         if (typeof current === "string") {
             throw new Error("Please override this method to use string versions.");
         }
@@ -77,7 +78,6 @@ function compareChannelVersions(a, b) {
     }
     return String(a).localeCompare(String(b));
 }
-exports.compareChannelVersions = compareChannelVersions;
 function maxChannelVersion(...versions) {
     return versions.reduce((max, version, idx) => {
         if (idx === 0)
@@ -85,7 +85,6 @@ function maxChannelVersion(...versions) {
         return compareChannelVersions(max, version) >= 0 ? max : version;
     });
 }
-exports.maxChannelVersion = maxChannelVersion;
 /**
  * Mapping from error type to error index.
  * Regular writes just map to their index in the list of writes being saved.
@@ -102,5 +101,4 @@ exports.WRITES_IDX_MAP = {
 function getCheckpointId(config) {
     return (config.configurable?.checkpoint_id || config.configurable?.thread_ts || "");
 }
-exports.getCheckpointId = getCheckpointId;
 //# sourceMappingURL=base.js.map

package/dist/base.d.ts

@@ -1,12 +1,11 @@
 import type { RunnableConfig } from "@langchain/core/runnables";
 import { SerializerProtocol } from "./serde/base.js";
 import type { PendingWrite, CheckpointPendingWrite, CheckpointMetadata } from "./types.js";
-import { type ChannelProtocol, type SendProtocol } from "./serde/types.js";
 type ChannelVersion = number | string;
 export type ChannelVersions = Record<string, ChannelVersion>;
 export interface Checkpoint<N extends string = string, C extends string = string> {
     /**
-     * The version of the checkpoint format. Currently 1
+     * The version of the checkpoint format. Currently 4
      */
     v: number;
     /**
@@ -29,11 +28,6 @@ export interface Checkpoint<N extends string = string, C extends string = string
      * @default {}
      */
     versions_seen: Record<N, Record<C, ChannelVersion>>;
-    /**
-     * List of packets sent to nodes but not yet processed.
-     * Cleared by the next checkpoint.
-     */
-    pending_sends: SendProtocol[];
 }
 export interface ReadonlyCheckpoint extends Readonly<Checkpoint> {
     readonly channel_values: Readonly<Record<string, unknown>>;
@@ -68,13 +62,18 @@ export declare abstract class BaseCheckpointSaver<V extends string | number = nu
      * Store intermediate writes linked to a checkpoint.
      */
     abstract putWrites(config: RunnableConfig, writes: PendingWrite[], taskId: string): Promise<void>;
+    /**
+     * Delete all checkpoints and writes associated with a specific thread ID.
+     * @param threadId The thread ID whose checkpoints should be deleted.
+     */
+    abstract deleteThread(threadId: string): Promise<void>;
     /**
      * Generate the next version ID for a channel.
      *
      * 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): V;
+    getNextVersion(current: V | undefined): V;
 }
 export declare function compareChannelVersions(a: ChannelVersion, b: ChannelVersion): number;
 export declare function maxChannelVersion(...versions: ChannelVersion[]): ChannelVersion;

package/dist/base.js

@@ -1,5 +1,5 @@
 import { uuid6 } from "./id.js";
-import { ERROR, INTERRUPT, RESUME, SCHEDULED, } from "./serde/types.js";
+import { ERROR, INTERRUPT, RESUME, SCHEDULED } from "./serde/types.js";
 import { JsonPlusSerializer } from "./serde/jsonplus.js";
 export function deepCopy(obj) {
     if (typeof obj !== "object" || obj === null) {
@@ -16,13 +16,12 @@ export function deepCopy(obj) {
 /** @hidden */
 export function emptyCheckpoint() {
     return {
-        v: 1,
+        v: 4,
         id: uuid6(-2),
         ts: new Date().toISOString(),
         channel_values: {},
         channel_versions: {},
         versions_seen: {},
-        pending_sends: [],
     };
 }
 /** @hidden */
@@ -34,7 +33,6 @@ export function copyCheckpoint(checkpoint) {
         channel_values: { ...checkpoint.channel_values },
         channel_versions: { ...checkpoint.channel_versions },
         versions_seen: deepCopy(checkpoint.versions_seen),
-        pending_sends: [...checkpoint.pending_sends],
     };
 }
 export class BaseCheckpointSaver {
@@ -57,7 +55,7 @@ export class BaseCheckpointSaver {
      * 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, _channel) {
+    getNextVersion(current) {
         if (typeof current === "string") {
             throw new Error("Please override this method to use string versions.");
         }

package/dist/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../src/base.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAMhC,OAAO,EACL,KAAK,EACL,SAAS,EACT,MAAM,EACN,SAAS,GAGV,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AAiDzD,MAAM,UAAU,QAAQ,CAAI,GAAM;IAChC,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE;QAC3C,OAAO,GAAG,CAAC;KACZ;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAE5C,KAAK,MAAM,GAAG,IAAI,GAAG,EAAE;QACrB,IAAI,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE;YACjD,MAAuC,CAAC,GAAG,CAAC,GAAG,QAAQ,CACrD,GAA+B,CAAC,GAAG,CAAC,CACtC,CAAC;SACH;KACF;IAED,OAAO,MAAW,CAAC;AACrB,CAAC;AAED,cAAc;AACd,MAAM,UAAU,eAAe;IAC7B,OAAO;QACL,CAAC,EAAE,CAAC;QACJ,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;QACb,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QAC5B,cAAc,EAAE,EAAE;QAClB,gBAAgB,EAAE,EAAE;QACpB,aAAa,EAAE,EAAE;QACjB,aAAa,EAAE,EAAE;KAClB,CAAC;AACJ,CAAC;AAED,cAAc;AACd,MAAM,UAAU,cAAc,CAAC,UAA8B;IAC3D,OAAO;QACL,CAAC,EAAE,UAAU,CAAC,CAAC;QACf,EAAE,EAAE,UAAU,CAAC,EAAE;QACjB,EAAE,EAAE,UAAU,CAAC,EAAE;QACjB,cAAc,EAAE,EAAE,GAAG,UAAU,CAAC,cAAc,EAAE;QAChD,gBAAgB,EAAE,EAAE,GAAG,UAAU,CAAC,gBAAgB,EAAE;QACpD,aAAa,EAAE,QAAQ,CAAC,UAAU,CAAC,aAAa,CAAC;QACjD,aAAa,EAAE,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;KAC7C,CAAC;AACJ,CAAC;AAiBD,MAAM,OAAgB,mBAAmB;IAGvC,YAAY,KAA0B;QAFtC;;;;mBAA4B,IAAI,kBAAkB,EAAE;WAAC;QAGnD,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAsB;QAC9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC1C,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9C,CAAC;IA2BD;;;;;OAKG;IACH,cAAc,CAAC,OAAsB,EAAE,QAAyB;QAC9D,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE;YAC/B,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;SACxE;QACD,OAAO,CACL,OAAO,KAAK,SAAS,IAAI,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAClE,CAAC;IACT,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CACpC,CAAiB,EACjB,CAAiB;IAEjB,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE;QAClD,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;KACzB;IAED,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,GAAG,QAA0B;IAE7B,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE;QAC3C,IAAI,GAAG,KAAK,CAAC;YAAE,OAAO,OAAO,CAAC;QAC9B,OAAO,sBAAsB,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;IACnE,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,cAAc,GAA2B;IACpD,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACX,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACf,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACf,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;CACb,CAAC;AAEF,MAAM,UAAU,eAAe,CAAC,MAAsB;IACpD,OAAO,CACL,MAAM,CAAC,YAAY,EAAE,aAAa,IAAI,MAAM,CAAC,YAAY,EAAE,SAAS,IAAI,EAAE,CAC3E,CAAC;AACJ,CAAC"}
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../src/base.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAMhC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AACvE,OAAO,EAAE,kBAAkB,EAAE,MAAM,qBAAqB,CAAC;AA4CzD,MAAM,UAAU,QAAQ,CAAI,GAAM;IAChC,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,EAAE,CAAC;QAC5C,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAE5C,KAAK,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;QACtB,IAAI,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;YAClD,MAAuC,CAAC,GAAG,CAAC,GAAG,QAAQ,CACrD,GAA+B,CAAC,GAAG,CAAC,CACtC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,MAAW,CAAC;AACrB,CAAC;AAED,cAAc;AACd,MAAM,UAAU,eAAe;IAC7B,OAAO;QACL,CAAC,EAAE,CAAC;QACJ,EAAE,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;QACb,EAAE,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QAC5B,cAAc,EAAE,EAAE;QAClB,gBAAgB,EAAE,EAAE;QACpB,aAAa,EAAE,EAAE;KAClB,CAAC;AACJ,CAAC;AAED,cAAc;AACd,MAAM,UAAU,cAAc,CAAC,UAA8B;IAC3D,OAAO;QACL,CAAC,EAAE,UAAU,CAAC,CAAC;QACf,EAAE,EAAE,UAAU,CAAC,EAAE;QACjB,EAAE,EAAE,UAAU,CAAC,EAAE;QACjB,cAAc,EAAE,EAAE,GAAG,UAAU,CAAC,cAAc,EAAE;QAChD,gBAAgB,EAAE,EAAE,GAAG,UAAU,CAAC,gBAAgB,EAAE;QACpD,aAAa,EAAE,QAAQ,CAAC,UAAU,CAAC,aAAa,CAAC;KAClD,CAAC;AACJ,CAAC;AAiBD,MAAM,OAAgB,mBAAmB;IAGvC,YAAY,KAA0B;QAFtC;;;;mBAA4B,IAAI,kBAAkB,EAAE;WAAC;QAGnD,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,MAAsB;QAC9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAC1C,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9C,CAAC;IAiCD;;;;;OAKG;IACH,cAAc,CAAC,OAAsB;QACnC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,MAAM,IAAI,KAAK,CAAC,qDAAqD,CAAC,CAAC;QACzE,CAAC;QACD,OAAO,CACL,OAAO,KAAK,SAAS,IAAI,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAClE,CAAC;IACT,CAAC;CACF;AAED,MAAM,UAAU,sBAAsB,CACpC,CAAiB,EACjB,CAAiB;IAEjB,IAAI,OAAO,CAAC,KAAK,QAAQ,IAAI,OAAO,CAAC,KAAK,QAAQ,EAAE,CAAC;QACnD,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IAC1B,CAAC;IAED,OAAO,MAAM,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAC5C,CAAC;AAED,MAAM,UAAU,iBAAiB,CAC/B,GAAG,QAA0B;IAE7B,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE;QAC3C,IAAI,GAAG,KAAK,CAAC;YAAE,OAAO,OAAO,CAAC;QAC9B,OAAO,sBAAsB,CAAC,GAAG,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC;IACnE,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,cAAc,GAA2B;IACpD,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IACX,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACf,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IACf,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;CACb,CAAC;AAEF,MAAM,UAAU,eAAe,CAAC,MAAsB;IACpD,OAAO,CACL,MAAM,CAAC,YAAY,EAAE,aAAa,IAAI,MAAM,CAAC,YAAY,EAAE,SAAS,IAAI,EAAE,CAC3E,CAAC;AACJ,CAAC"}

package/dist/cache/base.cjs

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseCache = void 0;
+const jsonplus_js_1 = require("../serde/jsonplus.cjs");
+class BaseCache {
+    /**
+     * Initialize the cache with a serializer.
+     *
+     * @param serde - The serializer to use.
+     */
+    constructor(serde) {
+        Object.defineProperty(this, "serde", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new jsonplus_js_1.JsonPlusSerializer()
+        });
+        this.serde = serde || this.serde;
+    }
+}
+exports.BaseCache = BaseCache;
+//# sourceMappingURL=base.js.map

package/dist/cache/base.d.ts

@@ -0,0 +1,38 @@
+import { SerializerProtocol } from "../serde/base.js";
+export type CacheNamespace = string[];
+export type CacheFullKey = [namespace: CacheNamespace, key: string];
+export declare abstract class BaseCache<V = unknown> {
+    serde: SerializerProtocol;
+    /**
+     * Initialize the cache with a serializer.
+     *
+     * @param serde - The serializer to use.
+     */
+    constructor(serde?: SerializerProtocol);
+    /**
+     * Get the cached values for the given keys.
+     *
+     * @param keys - The keys to get.
+     */
+    abstract get(keys: CacheFullKey[]): Promise<{
+        key: CacheFullKey;
+        value: V;
+    }[]>;
+    /**
+     * Set the cached values for the given keys and TTLs.
+     *
+     * @param pairs - The pairs to set.
+     */
+    abstract set(pairs: {
+        key: CacheFullKey;
+        value: V;
+        ttl?: number;
+    }[]): Promise<void>;
+    /**
+     * Delete the cached values for the given namespaces.
+     * If no namespaces are provided, clear all cached values.
+     *
+     * @param namespaces - The namespaces to clear.
+     */
+    abstract clear(namespaces: CacheNamespace[]): Promise<void>;
+}

package/dist/cache/base.js

@@ -0,0 +1,18 @@
+import { JsonPlusSerializer } from "../serde/jsonplus.js";
+export class BaseCache {
+    /**
+     * Initialize the cache with a serializer.
+     *
+     * @param serde - The serializer to use.
+     */
+    constructor(serde) {
+        Object.defineProperty(this, "serde", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new JsonPlusSerializer()
+        });
+        this.serde = serde || this.serde;
+    }
+}
+//# sourceMappingURL=base.js.map

package/dist/cache/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../src/cache/base.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAK1D,MAAM,OAAgB,SAAS;IAG7B;;;;OAIG;IACH,YAAY,KAA0B;QAPtC;;;;mBAA4B,IAAI,kBAAkB,EAAE;WAAC;QAQnD,IAAI,CAAC,KAAK,GAAG,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;IACnC,CAAC;CA2BF"}

package/dist/cache/index.cjs

@@ -0,0 +1,19 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./base.cjs"), exports);
+__exportStar(require("./memory.cjs"), exports);
+//# sourceMappingURL=index.js.map

package/dist/cache/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./base.js";
+export * from "./memory.js";

package/dist/cache/index.js

@@ -0,0 +1,3 @@
+export * from "./base.js";
+export * from "./memory.js";
+//# sourceMappingURL=index.js.map

package/dist/cache/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/cache/index.ts"],"names":[],"mappings":"AAAA,cAAc,WAAW,CAAC;AAC1B,cAAc,aAAa,CAAC"}

package/dist/cache/memory.cjs

@@ -0,0 +1,59 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryCache = void 0;
+const base_js_1 = require("./base.cjs");
+class InMemoryCache extends base_js_1.BaseCache {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "cache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: {}
+        });
+    }
+    async get(keys) {
+        if (!keys.length)
+            return [];
+        const now = Date.now();
+        return (await Promise.all(keys.map(async (fullKey) => {
+            const [namespace, key] = fullKey;
+            const strNamespace = namespace.join(",");
+            if (strNamespace in this.cache && key in this.cache[strNamespace]) {
+                const cached = this.cache[strNamespace][key];
+                if (cached.exp == null || now < cached.exp) {
+                    const value = await this.serde.loadsTyped(cached.enc, cached.val);
+                    return [{ key: fullKey, value }];
+                }
+                else {
+                    delete this.cache[strNamespace][key];
+                }
+            }
+            return [];
+        }))).flat();
+    }
+    async set(pairs) {
+        const now = Date.now();
+        for (const { key: fullKey, value, ttl } of pairs) {
+            const [namespace, key] = fullKey;
+            const strNamespace = namespace.join(",");
+            const [enc, val] = await this.serde.dumpsTyped(value);
+            const exp = ttl != null ? ttl * 1000 + now : null;
+            this.cache[strNamespace] ??= {};
+            this.cache[strNamespace][key] = { enc, val, exp };
+        }
+    }
+    async clear(namespaces) {
+        if (!namespaces.length) {
+            this.cache = {};
+            return;
+        }
+        for (const namespace of namespaces) {
+            const strNamespace = namespace.join(",");
+            if (strNamespace in this.cache)
+                delete this.cache[strNamespace];
+        }
+    }
+}
+exports.InMemoryCache = InMemoryCache;
+//# sourceMappingURL=memory.js.map

package/dist/cache/memory.d.ts

@@ -0,0 +1,14 @@
+import { BaseCache, type CacheFullKey, type CacheNamespace } from "./base.js";
+export declare class InMemoryCache<V = unknown> extends BaseCache<V> {
+    private cache;
+    get(keys: CacheFullKey[]): Promise<{
+        key: CacheFullKey;
+        value: V;
+    }[]>;
+    set(pairs: {
+        key: CacheFullKey;
+        value: V;
+        ttl?: number;
+    }[]): Promise<void>;
+    clear(namespaces: CacheNamespace[]): Promise<void>;
+}

package/dist/cache/memory.js

@@ -0,0 +1,55 @@
+import { BaseCache } from "./base.js";
+export class InMemoryCache extends BaseCache {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "cache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: {}
+        });
+    }
+    async get(keys) {
+        if (!keys.length)
+            return [];
+        const now = Date.now();
+        return (await Promise.all(keys.map(async (fullKey) => {
+            const [namespace, key] = fullKey;
+            const strNamespace = namespace.join(",");
+            if (strNamespace in this.cache && key in this.cache[strNamespace]) {
+                const cached = this.cache[strNamespace][key];
+                if (cached.exp == null || now < cached.exp) {
+                    const value = await this.serde.loadsTyped(cached.enc, cached.val);
+                    return [{ key: fullKey, value }];
+                }
+                else {
+                    delete this.cache[strNamespace][key];
+                }
+            }
+            return [];
+        }))).flat();
+    }
+    async set(pairs) {
+        const now = Date.now();
+        for (const { key: fullKey, value, ttl } of pairs) {
+            const [namespace, key] = fullKey;
+            const strNamespace = namespace.join(",");
+            const [enc, val] = await this.serde.dumpsTyped(value);
+            const exp = ttl != null ? ttl * 1000 + now : null;
+            this.cache[strNamespace] ??= {};
+            this.cache[strNamespace][key] = { enc, val, exp };
+        }
+    }
+    async clear(namespaces) {
+        if (!namespaces.length) {
+            this.cache = {};
+            return;
+        }
+        for (const namespace of namespaces) {
+            const strNamespace = namespace.join(",");
+            if (strNamespace in this.cache)
+                delete this.cache[strNamespace];
+        }
+    }
+}
+//# sourceMappingURL=memory.js.map

package/dist/cache/memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../src/cache/memory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAA0C,MAAM,WAAW,CAAC;AAE9E,MAAM,OAAO,aAA2B,SAAQ,SAAY;IAA5D;;QACU;;;;mBAQJ,EAAE;WAAC;IA0DT,CAAC;IAxDC,KAAK,CAAC,GAAG,CAAC,IAAoB;QAC5B,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,OAAO,CACL,MAAM,OAAO,CAAC,GAAG,CACf,IAAI,CAAC,GAAG,CACN,KAAK,EAAE,OAAO,EAA8C,EAAE;YAC5D,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC;YACjC,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEzC,IAAI,YAAY,IAAI,IAAI,CAAC,KAAK,IAAI,GAAG,IAAI,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,EAAE,CAAC;gBAClE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC;gBAC7C,IAAI,MAAM,CAAC,GAAG,IAAI,IAAI,IAAI,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC;oBAC3C,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CACvC,MAAM,CAAC,GAAG,EACV,MAAM,CAAC,GAAG,CACX,CAAC;oBACF,OAAO,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC,CAAC;gBACnC,CAAC;qBAAM,CAAC;oBACN,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC;gBACvC,CAAC;YACH,CAAC;YAED,OAAO,EAAE,CAAC;QACZ,CAAC,CACF,CACF,CACF,CAAC,IAAI,EAAE,CAAC;IACX,CAAC;IAED,KAAK,CAAC,GAAG,CACP,KAAsD;QAEtD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,KAAK,EAAE,CAAC;YACjD,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,GAAG,OAAO,CAAC;YACjC,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;YACtD,MAAM,GAAG,GAAG,GAAG,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,GAAG,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;YAElD,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAChC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC;QACpD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,UAA4B;QACtC,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACvB,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;YAChB,OAAO;QACT,CAAC;QAED,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;YACnC,MAAM,YAAY,GAAG,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACzC,IAAI,YAAY,IAAI,IAAI,CAAC,KAAK;gBAAE,OAAO,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;QAClE,CAAC;IACH,CAAC;CACF"}

package/dist/id.cjs

@@ -1,11 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.uuid5 = exports.uuid6 = void 0;
+exports.uuid6 = uuid6;
+exports.uuid5 = uuid5;
 const uuid_1 = require("uuid");
 function uuid6(clockseq) {
     return (0, uuid_1.v6)({ clockseq });
 }
-exports.uuid6 = uuid6;
 // Skip UUID validation check, since UUID6s
 // generated with negative clockseq are not
 // technically compliant, but still work.
@@ -17,5 +17,4 @@ function uuid5(name, namespace) {
         .map((byte) => parseInt(byte, 16));
     return (0, uuid_1.v5)(name, new Uint8Array(namespaceBytes));
 }
-exports.uuid5 = uuid5;
 //# sourceMappingURL=id.js.map

package/dist/index.cjs

@@ -23,4 +23,5 @@ __exportStar(require("./types.cjs"), exports);
 __exportStar(require("./serde/base.cjs"), exports);
 __exportStar(require("./serde/types.cjs"), exports);
 __exportStar(require("./store/index.cjs"), exports);
+__exportStar(require("./cache/index.cjs"), exports);
 //# sourceMappingURL=index.js.map

package/dist/index.d.ts

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

package/dist/index.js

@@ -5,4 +5,5 @@ export * from "./types.js";
 export * from "./serde/base.js";
 export * from "./serde/types.js";
 export * from "./store/index.js";
+export * from "./cache/index.js";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,cAAc,WAAW,CAAC;AAC1B,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AACjC,cAAc,kBAAkB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,cAAc,WAAW,CAAC;AAC1B,cAAc,SAAS,CAAC;AACxB,cAAc,YAAY,CAAC;AAC3B,cAAc,iBAAiB,CAAC;AAChC,cAAc,kBAAkB,CAAC;AACjC,cAAc,kBAAkB,CAAC;AACjC,cAAc,kBAAkB,CAAC"}

package/dist/memory.cjs

@@ -6,6 +6,10 @@ const types_js_1 = require("./serde/types.cjs");
 function _generateKey(threadId, checkpointNamespace, checkpointId) {
     return JSON.stringify([threadId, checkpointNamespace, checkpointId]);
 }
+function _parseKey(key) {
+    const [threadId, checkpointNamespace, checkpointId] = JSON.parse(key);
+    return { threadId, checkpointNamespace, checkpointId };
+}
 class MemorySaver extends base_js_1.BaseCheckpointSaver {
     constructor(serde) {
         super(serde);
@@ -23,19 +27,19 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
             value: {}
         });
     }
-    async _getPendingSends(threadId, checkpointNs, parentCheckpointId) {
-        let pendingSends = [];
-        if (parentCheckpointId !== undefined) {
-            const key = _generateKey(threadId, checkpointNs, parentCheckpointId);
-            pendingSends = await Promise.all(Object.values(this.writes[key] || {})
-                ?.filter(([_taskId, channel]) => {
-                return channel === types_js_1.TASKS;
-            })
-                .map(([_taskId, _channel, writes]) => {
-                return this.serde.loadsTyped("json", writes);
-            }) ?? []);
-        }
-        return pendingSends;
+    async _migratePendingSends(mutableCheckpoint, threadId, checkpointNs, parentCheckpointId) {
+        const deseriablizableCheckpoint = mutableCheckpoint;
+        const parentKey = _generateKey(threadId, checkpointNs, parentCheckpointId);
+        const pendingSends = await Promise.all(Object.values(this.writes[parentKey] ?? {})
+            .filter(([_taskId, channel]) => channel === types_js_1.TASKS)
+            .map(async ([_taskId, _channel, writes]) => await this.serde.loadsTyped("json", writes)));
+        deseriablizableCheckpoint.channel_values ??= {};
+        deseriablizableCheckpoint.channel_values[types_js_1.TASKS] = pendingSends;
+        deseriablizableCheckpoint.channel_versions ??= {};
+        deseriablizableCheckpoint.channel_versions[types_js_1.TASKS] =
+            Object.keys(deseriablizableCheckpoint.channel_versions).length > 0
+                ? (0, base_js_1.maxChannelVersion)(...Object.values(deseriablizableCheckpoint.channel_versions))
+                : this.getNextVersion(undefined);
     }
     async getTuple(config) {
         const thread_id = config.configurable?.thread_id;
@@ -46,11 +50,10 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
             if (saved !== undefined) {
                 const [checkpoint, metadata, parentCheckpointId] = saved;
                 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 deserializedCheckpoint = await this.serde.loadsTyped("json", checkpoint);
+                if (deserializedCheckpoint.v < 4 && parentCheckpointId !== undefined) {
+                    await this._migratePendingSends(deserializedCheckpoint, thread_id, checkpoint_ns, parentCheckpointId);
+                }
                 const pendingWrites = await Promise.all(Object.values(this.writes[key] || {}).map(async ([taskId, channel, value]) => {
                     return [
                         taskId,
@@ -84,11 +87,10 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 const saved = checkpoints[checkpoint_id];
                 const [checkpoint, metadata, parentCheckpointId] = saved;
                 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 deserializedCheckpoint = await this.serde.loadsTyped("json", checkpoint);
+                if (deserializedCheckpoint.v < 4 && parentCheckpointId !== undefined) {
+                    await this._migratePendingSends(deserializedCheckpoint, thread_id, checkpoint_ns, parentCheckpointId);
+                }
                 const pendingWrites = await Promise.all(Object.values(this.writes[key] || {}).map(async ([taskId, channel, value]) => {
                     return [
                         taskId,
@@ -163,7 +165,6 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                     }
                     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 [
                             taskId,
@@ -171,10 +172,11 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                             await this.serde.loadsTyped("json", value),
                         ];
                     }));
-                    const deserializedCheckpoint = {
-                        ...(await this.serde.loadsTyped("json", checkpoint)),
-                        pending_sends,
-                    };
+                    const deserializedCheckpoint = await this.serde.loadsTyped("json", checkpoint);
+                    if (deserializedCheckpoint.v < 4 &&
+                        parentCheckpointId !== undefined) {
+                        await this._migratePendingSends(deserializedCheckpoint, threadId, checkpointNamespace, parentCheckpointId);
+                    }
                     const checkpointTuple = {
                         config: {
                             configurable: {
@@ -203,7 +205,6 @@ 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) {
@@ -215,8 +216,10 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
         if (!this.storage[threadId][checkpointNamespace]) {
             this.storage[threadId][checkpointNamespace] = {};
         }
-        const [, serializedCheckpoint] = this.serde.dumpsTyped(preparedCheckpoint);
-        const [, serializedMetadata] = this.serde.dumpsTyped(metadata);
+        const [[, serializedCheckpoint], [, serializedMetadata]] = await Promise.all([
+            this.serde.dumpsTyped(preparedCheckpoint),
+            this.serde.dumpsTyped(metadata),
+        ]);
         this.storage[threadId][checkpointNamespace][checkpoint.id] = [
             serializedCheckpoint,
             serializedMetadata,
@@ -245,8 +248,8 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
         if (this.writes[outerKey] === undefined) {
             this.writes[outerKey] = {};
         }
-        writes.forEach(([channel, value], idx) => {
-            const [, serializedValue] = this.serde.dumpsTyped(value);
+        await Promise.all(writes.map(async ([channel, value], idx) => {
+            const [, serializedValue] = await this.serde.dumpsTyped(value);
             const innerKey = [
                 taskId,
                 base_js_1.WRITES_IDX_MAP[channel] || idx,
@@ -256,7 +259,14 @@ class MemorySaver extends base_js_1.BaseCheckpointSaver {
                 return;
             }
             this.writes[outerKey][innerKeyStr] = [taskId, channel, serializedValue];
-        });
+        }));
+    }
+    async deleteThread(threadId) {
+        delete this.storage[threadId];
+        for (const key of Object.keys(this.writes)) {
+            if (_parseKey(key).threadId === threadId)
+                delete this.writes[key];
+        }
     }
 }
 exports.MemorySaver = MemorySaver;