@langchain/langgraph 0.2.9 → 0.2.10-rc.0

package/dist/constants.cjs

@@ -1,12 +1,11 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports._isSend = exports.Send = exports._isSendInterface = exports.CHECKPOINT_NAMESPACE_END = exports.CHECKPOINT_NAMESPACE_SEPARATOR = exports.RESERVED = exports.TASK_NAMESPACE = exports.PULL = exports.PUSH = exports.TASKS = exports.TAG_HIDDEN = exports.RECURSION_LIMIT_DEFAULT = exports.RUNTIME_PLACEHOLDER = exports.INTERRUPT = exports.CONFIG_KEY_CHECKPOINT_MAP = exports.CONFIG_KEY_STREAM = exports.CONFIG_KEY_TASK_ID = exports.CONFIG_KEY_RESUMING = exports.CONFIG_KEY_STORE = exports.CONFIG_KEY_CHECKPOINTER = exports.CONFIG_KEY_READ = exports.CONFIG_KEY_SEND = exports.ERROR = exports.INPUT = void 0;
+exports._isSend = exports.Send = exports._isSendInterface = exports.CHECKPOINT_NAMESPACE_END = exports.CHECKPOINT_NAMESPACE_SEPARATOR = exports.RESERVED = exports.TASK_NAMESPACE = exports.PULL = exports.PUSH = exports.TASKS = exports.TAG_HIDDEN = exports.RECURSION_LIMIT_DEFAULT = exports.RUNTIME_PLACEHOLDER = exports.INTERRUPT = exports.CONFIG_KEY_CHECKPOINT_MAP = exports.CONFIG_KEY_STREAM = exports.CONFIG_KEY_TASK_ID = exports.CONFIG_KEY_RESUMING = exports.CONFIG_KEY_CHECKPOINTER = exports.CONFIG_KEY_READ = exports.CONFIG_KEY_SEND = exports.ERROR = exports.INPUT = void 0;
 exports.INPUT = "__input__";
 exports.ERROR = "__error__";
 exports.CONFIG_KEY_SEND = "__pregel_send";
 exports.CONFIG_KEY_READ = "__pregel_read";
 exports.CONFIG_KEY_CHECKPOINTER = "__pregel_checkpointer";
-exports.CONFIG_KEY_STORE = "__pregel_store";
 exports.CONFIG_KEY_RESUMING = "__pregel_resuming";
 exports.CONFIG_KEY_TASK_ID = "__pregel_task_id";
 exports.CONFIG_KEY_STREAM = "__pregel_stream";

package/dist/constants.d.ts

@@ -3,7 +3,6 @@ export declare const ERROR = "__error__";
 export declare const CONFIG_KEY_SEND = "__pregel_send";
 export declare const CONFIG_KEY_READ = "__pregel_read";
 export declare const CONFIG_KEY_CHECKPOINTER = "__pregel_checkpointer";
-export declare const CONFIG_KEY_STORE = "__pregel_store";
 export declare const CONFIG_KEY_RESUMING = "__pregel_resuming";
 export declare const CONFIG_KEY_TASK_ID = "__pregel_task_id";
 export declare const CONFIG_KEY_STREAM = "__pregel_stream";

package/dist/constants.js

@@ -3,7 +3,6 @@ export const ERROR = "__error__";
 export const CONFIG_KEY_SEND = "__pregel_send";
 export const CONFIG_KEY_READ = "__pregel_read";
 export const CONFIG_KEY_CHECKPOINTER = "__pregel_checkpointer";
-export const CONFIG_KEY_STORE = "__pregel_store";
 export const CONFIG_KEY_RESUMING = "__pregel_resuming";
 export const CONFIG_KEY_TASK_ID = "__pregel_task_id";
 export const CONFIG_KEY_STREAM = "__pregel_stream";

package/dist/graph/state.d.ts

@@ -1,10 +1,9 @@
 import { RunnableLike } from "@langchain/core/runnables";
-import { All, BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
+import { All, BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { END, CompiledGraph, Graph, START, Branch, AddNodeOptions, NodeSpec } from "./graph.js";
 import { AnnotationRoot, SingleReducer, StateDefinition, StateType, UpdateType } from "./annotation.js";
 import type { RetryPolicy } from "../pregel/utils/index.js";
-import { BaseStore } from "../store/base.js";
 import { ManagedValueSpec } from "../managed/base.js";
 export type ChannelReducers<Channels extends object> = {
     [K in keyof Channels]: SingleReducer<Channels[K], any>;

package/dist/managed/shared_value.cjs

@@ -1,8 +1,8 @@
 "use strict";
+/* eslint-disable @typescript-eslint/no-explicit-any */
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SharedValue = void 0;
 const base_js_1 = require("./base.cjs");
-const constants_js_1 = require("../constants.cjs");
 const errors_js_1 = require("../errors.cjs");
 class SharedValue extends base_js_1.WritableManagedValue {
     constructor(config, params) {
@@ -32,7 +32,7 @@ class SharedValue extends base_js_1.WritableManagedValue {
             value: {}
         });
         this.scope = params.scope;
-        this.store = config.configurable?.[constants_js_1.CONFIG_KEY_STORE] || null;
+        this.store = config.store || null;
         if (!this.store) {
             this.ns = null;
         }
@@ -41,13 +41,12 @@ class SharedValue extends base_js_1.WritableManagedValue {
             const scopedValueString = typeof scopeValue === "string"
                 ? scopeValue
                 : JSON.stringify(scopeValue);
-            this.ns = `scoped:${this.scope}:${params.key}:${scopedValueString}`;
+            this.ns = ["scoped", this.scope, params.key, scopedValueString];
         }
         else {
             throw new Error(`Required scope "${this.scope}" for shared state key was not passed in "config.configurable".`);
         }
     }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static async initialize(config, args) {
         const instance = new this(config, args);
         await instance.loadStore();
@@ -73,7 +72,7 @@ class SharedValue extends base_js_1.WritableManagedValue {
                     if (k in this.value) {
                         delete this.value[k];
                         if (this.ns) {
-                            writes.push([this.ns, k, null]);
+                            writes.push({ namespace: this.ns, id: k, value: null });
                         }
                     }
                 }
@@ -83,7 +82,7 @@ class SharedValue extends base_js_1.WritableManagedValue {
                 else {
                     this.value[k] = v;
                     if (this.ns) {
-                        writes.push([this.ns, k, v]);
+                        writes.push({ namespace: this.ns, id: k, value: v });
                     }
                 }
             }
@@ -95,13 +94,16 @@ class SharedValue extends base_js_1.WritableManagedValue {
             this.processUpdate(values);
         }
         else {
-            await this.store.put(this.processUpdate(values));
+            await this.store.batch(this.processUpdate(values));
         }
     }
     async loadStore() {
         if (this.store && this.ns) {
-            const saved = await this.store.list([this.ns]);
-            this.value = saved[this.ns] || {};
+            const saved = await this.store.search(this.ns);
+            this.value = saved.reduce((acc, item) => {
+                acc[item.id] = item.value;
+                return acc;
+            }, {});
         }
         return false;
     }

package/dist/managed/shared_value.d.ts

@@ -1,8 +1,9 @@
 import { RunnableConfig } from "@langchain/core/runnables";
-import { BaseStore, type Values } from "../store/base.js";
+import { BaseStore } from "@langchain/langgraph-checkpoint";
 import { ConfiguredManagedValue, ManagedValue, ManagedValueParams, WritableManagedValue } from "./base.js";
-type Value = Record<string, Values>;
-type Update = Record<string, Values | null>;
+import { LangGraphRunnableConfig } from "../pregel/runnable_types.js";
+type Value = Record<string, Record<string, any>>;
+type Update = Record<string, Record<string, any> | null>;
 export interface SharedValueParams extends ManagedValueParams {
     scope: string;
     key: string;
@@ -10,9 +11,9 @@ export interface SharedValueParams extends ManagedValueParams {
 export declare class SharedValue extends WritableManagedValue<Value, Update> {
     scope: string;
     store: BaseStore | null;
-    ns: string | null;
+    ns: ["scoped", string, string, any] | null;
     value: Value;
-    constructor(config: RunnableConfig, params: SharedValueParams);
+    constructor(config: LangGraphRunnableConfig, params: SharedValueParams);
     static initialize<Value = any>(config: RunnableConfig, args: SharedValueParams): Promise<ManagedValue<Value>>;
     static on(scope: string): ConfiguredManagedValue<Value>;
     call(_step: number): Value;

package/dist/managed/shared_value.js

@@ -1,5 +1,5 @@
+/* eslint-disable @typescript-eslint/no-explicit-any */
 import { ChannelKeyPlaceholder, WritableManagedValue, } from "./base.js";
-import { CONFIG_KEY_STORE } from "../constants.js";
 import { InvalidUpdateError } from "../errors.js";
 export class SharedValue extends WritableManagedValue {
     constructor(config, params) {
@@ -29,7 +29,7 @@ export class SharedValue extends WritableManagedValue {
             value: {}
         });
         this.scope = params.scope;
-        this.store = config.configurable?.[CONFIG_KEY_STORE] || null;
+        this.store = config.store || null;
         if (!this.store) {
             this.ns = null;
         }
@@ -38,13 +38,12 @@ export class SharedValue extends WritableManagedValue {
             const scopedValueString = typeof scopeValue === "string"
                 ? scopeValue
                 : JSON.stringify(scopeValue);
-            this.ns = `scoped:${this.scope}:${params.key}:${scopedValueString}`;
+            this.ns = ["scoped", this.scope, params.key, scopedValueString];
         }
         else {
             throw new Error(`Required scope "${this.scope}" for shared state key was not passed in "config.configurable".`);
         }
     }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
     static async initialize(config, args) {
         const instance = new this(config, args);
         await instance.loadStore();
@@ -70,7 +69,7 @@ export class SharedValue extends WritableManagedValue {
                     if (k in this.value) {
                         delete this.value[k];
                         if (this.ns) {
-                            writes.push([this.ns, k, null]);
+                            writes.push({ namespace: this.ns, id: k, value: null });
                         }
                     }
                 }
@@ -80,7 +79,7 @@ export class SharedValue extends WritableManagedValue {
                 else {
                     this.value[k] = v;
                     if (this.ns) {
-                        writes.push([this.ns, k, v]);
+                        writes.push({ namespace: this.ns, id: k, value: v });
                     }
                 }
             }
@@ -92,13 +91,16 @@ export class SharedValue extends WritableManagedValue {
             this.processUpdate(values);
         }
         else {
-            await this.store.put(this.processUpdate(values));
+            await this.store.batch(this.processUpdate(values));
         }
     }
     async loadStore() {
         if (this.store && this.ns) {
-            const saved = await this.store.list([this.ns]);
-            this.value = saved[this.ns] || {};
+            const saved = await this.store.search(this.ns);
+            this.value = saved.reduce((acc, item) => {
+                acc[item.id] = item.value;
+                return acc;
+            }, {});
         }
         return false;
     }

package/dist/pregel/algo.cjs

@@ -277,6 +277,7 @@ function _prepareSingleTask(taskPath, checkpoint, processes, channels, managed,
                     config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, {
                         metadata,
                         tags: proc.tags,
+                        store: extra.store ?? config.store,
                     }), {
                         runName: packet.node,
                         callbacks: manager?.getChild(`graph:step:${step}`),
@@ -369,7 +370,11 @@ function _prepareSingleTask(taskPath, checkpoint, processes, channels, managed,
                         input: val,
                         proc: node,
                         writes,
-                        config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, { metadata, tags: proc.tags }), {
+                        config: (0, runnables_1.patchConfig)((0, runnables_1.mergeConfigs)(config, {
+                            metadata,
+                            tags: proc.tags,
+                            store: extra.store ?? config?.store,
+                        }), {
                             runName: name,
                             callbacks: manager?.getChild(`graph:step:${step}`),
                             configurable: {

package/dist/pregel/algo.d.ts

@@ -1,6 +1,6 @@
 import { RunnableConfig } from "@langchain/core/runnables";
 import { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-import { All, BaseCheckpointSaver, Checkpoint, ReadonlyCheckpoint, type PendingWrite, type PendingWriteValue } from "@langchain/langgraph-checkpoint";
+import { All, BaseCheckpointSaver, Checkpoint, ReadonlyCheckpoint, type PendingWrite, type PendingWriteValue, BaseStore } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { PregelNode } from "./read.js";
 import { PregelExecutableTask, PregelTaskDescription } from "./types.js";
@@ -26,9 +26,16 @@ export type NextTaskExtraFields = {
     isResuming?: boolean;
     checkpointer?: BaseCheckpointSaver;
     manager?: CallbackManagerForChainRun;
+    store?: BaseStore;
 };
-export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): Record<string, PregelTaskDescription>;
-export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): Record<string, PregelExecutableTask<keyof Nn, keyof Cc>>;
+export type NextTaskExtraFieldsWithStore = NextTaskExtraFields & {
+    store?: BaseStore;
+};
+export type NextTaskExtraFieldsWithoutStore = NextTaskExtraFields & {
+    store?: never;
+};
+export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFieldsWithoutStore): Record<string, PregelTaskDescription>;
+export declare function _prepareNextTasks<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFieldsWithStore): Record<string, PregelExecutableTask<keyof Nn, keyof Cc>>;
 export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: [string, string | number], checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: false, extra: NextTaskExtraFields): PregelTaskDescription | undefined;
 export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: [string, string | number], checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: true, extra: NextTaskExtraFields): PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
-export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: [string, string | number], checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: boolean, extra: NextTaskExtraFields): PregelTaskDescription | PregelExecutableTask<keyof Nn, keyof Cc> | undefined;
+export declare function _prepareSingleTask<Nn extends StrRecord<string, PregelNode>, Cc extends StrRecord<string, BaseChannel>>(taskPath: [string, string | number], checkpoint: ReadonlyCheckpoint, processes: Nn, channels: Cc, managed: ManagedValueMapping, config: RunnableConfig, forExecution: boolean, extra: NextTaskExtraFieldsWithStore): PregelTaskDescription | PregelExecutableTask<keyof Nn, keyof Cc> | undefined;

package/dist/pregel/algo.js

@@ -268,6 +268,7 @@ export function _prepareSingleTask(taskPath, checkpoint, processes, channels, ma
                     config: patchConfig(mergeConfigs(config, {
                         metadata,
                         tags: proc.tags,
+                        store: extra.store ?? config.store,
                     }), {
                         runName: packet.node,
                         callbacks: manager?.getChild(`graph:step:${step}`),
@@ -360,7 +361,11 @@ export function _prepareSingleTask(taskPath, checkpoint, processes, channels, ma
                         input: val,
                         proc: node,
                         writes,
-                        config: patchConfig(mergeConfigs(config, { metadata, tags: proc.tags }), {
+                        config: patchConfig(mergeConfigs(config, {
+                            metadata,
+                            tags: proc.tags,
+                            store: extra.store ?? config?.store,
+                        }), {
                             runName: name,
                             callbacks: manager?.getChild(`graph:step:${step}`),
                             configurable: {

package/dist/pregel/index.cjs

@@ -550,7 +550,10 @@ class Pregel extends runnables_1.Runnable {
             id: (0, langgraph_checkpoint_1.uuid5)(constants_js_1.INTERRUPT, checkpoint.id),
         };
         // execute task
-        await task.proc.invoke(task.input, (0, runnables_1.patchConfig)(config, {
+        await task.proc.invoke(task.input, (0, runnables_1.patchConfig)({
+            ...config,
+            store: config?.store ?? this.store,
+        }, {
             runName: config.runName ?? `${this.getName()}UpdateState`,
             configurable: {
                 [constants_js_1.CONFIG_KEY_SEND]: (items) => task.writes.push(...items),
@@ -613,6 +616,7 @@ class Pregel extends runnables_1.Runnable {
         else {
             defaultCheckpointer = this.checkpointer;
         }
+        const defaultStore = config.store ?? this.store;
         return [
             defaultDebug,
             defaultStreamMode,
@@ -622,6 +626,7 @@ class Pregel extends runnables_1.Runnable {
             defaultInterruptBefore,
             defaultInterruptAfter,
             defaultCheckpointer,
+            defaultStore,
         ];
     }
     /**
@@ -643,9 +648,10 @@ class Pregel extends runnables_1.Runnable {
         return super.stream(input, options);
     }
     async prepareSpecs(config, options) {
-        const configForManaged = (0, utils_js_1.patchConfigurable)(config, {
-            [constants_js_1.CONFIG_KEY_STORE]: this.store,
-        });
+        const configForManaged = {
+            ...config,
+            store: this.store,
+        };
         const channelSpecs = {};
         const managedSpecs = {};
         for (const [name, spec] of Object.entries(this.channels)) {
@@ -700,7 +706,7 @@ class Pregel extends runnables_1.Runnable {
         const runManager = await callbackManager?.handleChainStart(this.toJSON(), (0, index_js_1._coerceToDict)(input, "input"), inputConfig.runId, undefined, undefined, undefined, inputConfig?.runName ?? this.getName());
         delete inputConfig.runId;
         // assign defaults
-        const [debug, streamMode, , outputKeys, config, interruptBefore, interruptAfter, checkpointer,] = this._defaults(inputConfig);
+        const [debug, streamMode, , outputKeys, config, interruptBefore, interruptAfter, checkpointer, store,] = this._defaults(inputConfig);
         const { channelSpecs, managed } = await this.prepareSpecs(config);
         let loop;
         const stream = new double_ended_queue_1.default();
@@ -737,7 +743,7 @@ class Pregel extends runnables_1.Runnable {
                 managed,
                 outputKeys,
                 streamKeys: this.streamChannelsAsIs,
-                store: this.store,
+                store,
                 stream: new loop_js_1.StreamProtocol((chunk) => stream.push(chunk), new Set(streamMode)),
             });
             if (options?.subgraphs) {
@@ -816,7 +822,7 @@ class Pregel extends runnables_1.Runnable {
         finally {
             // Call `.stop()` again incase it was not called in the loop, e.g due to an error.
             if (loop) {
-                loop.store?.stop();
+                await loop.store?.stop();
             }
             await Promise.all([
                 loop?.checkpointerPromises ?? [],

package/dist/pregel/index.d.ts

@@ -1,14 +1,14 @@
 import { Runnable, RunnableConfig, RunnableFunc } from "@langchain/core/runnables";
 import { IterableReadableStream } from "@langchain/core/utils/stream";
-import { All, BaseCheckpointSaver, CheckpointListOptions, CheckpointTuple } from "@langchain/langgraph-checkpoint";
+import { All, BaseCheckpointSaver, BaseStore, CheckpointListOptions, CheckpointTuple } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { PregelNode } from "./read.js";
 import { ChannelWrite } from "./write.js";
 import { PregelInterface, PregelParams, StateSnapshot, StreamMode } from "./types.js";
 import { StrRecord } from "./algo.js";
 import { RetryPolicy } from "./utils/index.js";
-import { BaseStore } from "../store/base.js";
 import { ManagedValueMapping, type ManagedValueSpec } from "../managed/base.js";
+import { LangGraphRunnableConfig } from "./runnable_types.js";
 type WriteValue = Runnable | RunnableFunc<unknown, unknown> | unknown;
 export declare class Channel {
     static subscribeTo(channels: string, options?: {
@@ -37,6 +37,8 @@ export interface PregelOptions<Nn extends StrRecord<string, PregelNode>, Cc exte
     debug?: boolean;
     /** Whether to stream subgraphs. */
     subgraphs?: boolean;
+    /** The shared value store */
+    store?: BaseStore;
 }
 export type PregelInputType = any;
 export type PregelOutputType = any;
@@ -56,7 +58,7 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
     debug: boolean;
     checkpointer?: BaseCheckpointSaver;
     retryPolicy?: RetryPolicy;
-    config?: RunnableConfig;
+    config?: LangGraphRunnableConfig;
     store?: BaseStore;
     constructor(fields: PregelParams<Nn, Cc>);
     withConfig(config: RunnableConfig): typeof this;
@@ -84,7 +86,7 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
      * node `as_node`. If `as_node` is not provided, it will be set to the last node
      * that updated the state, if not ambiguous.
      */
-    updateState(inputConfig: RunnableConfig, values: Record<string, unknown> | unknown, asNode?: keyof Nn | string): Promise<RunnableConfig>;
+    updateState(inputConfig: LangGraphRunnableConfig, values: Record<string, unknown> | unknown, asNode?: keyof Nn | string): Promise<RunnableConfig>;
     _defaults(config: PregelOptions<Nn, Cc>): [
         boolean,
         StreamMode[],
@@ -98,7 +100,8 @@ export declare class Pregel<Nn extends StrRecord<string, PregelNode>, Cc extends
         // interrupt before
         All | string[],
         // interrupt after
-        BaseCheckpointSaver | undefined
+        BaseCheckpointSaver | undefined,
+        BaseStore | undefined
     ];
     /**
      * Stream graph steps for a single input.

package/dist/pregel/index.js

@@ -1,5 +1,5 @@
 /* eslint-disable no-param-reassign */
-import { Runnable, RunnableSequence, _coerceToRunnable, getCallbackManagerForConfig, mergeConfigs, patchConfig, } from "@langchain/core/runnables";
+import { Runnable, RunnableSequence, getCallbackManagerForConfig, mergeConfigs, patchConfig, _coerceToRunnable, } from "@langchain/core/runnables";
 import { compareChannelVersions, copyCheckpoint, emptyCheckpoint, uuid5, } from "@langchain/langgraph-checkpoint";
 import Deque from "double-ended-queue";
 import { createCheckpoint, emptyChannels, isBaseChannel, } from "../channels/base.js";
@@ -8,7 +8,7 @@ import { validateGraph, validateKeys } from "./validate.js";
 import { readChannels } from "./io.js";
 import { printStepCheckpoint, printStepTasks, printStepWrites, tasksWithWrites, } from "./debug.js";
 import { ChannelWrite, PASSTHROUGH } from "./write.js";
-import { CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_READ, CONFIG_KEY_SEND, CONFIG_KEY_STORE, ERROR, INTERRUPT, CHECKPOINT_NAMESPACE_SEPARATOR, CHECKPOINT_NAMESPACE_END, CONFIG_KEY_STREAM, CONFIG_KEY_TASK_ID, } from "../constants.js";
+import { CONFIG_KEY_CHECKPOINTER, CONFIG_KEY_READ, CONFIG_KEY_SEND, ERROR, INTERRUPT, CHECKPOINT_NAMESPACE_SEPARATOR, CHECKPOINT_NAMESPACE_END, CONFIG_KEY_STREAM, CONFIG_KEY_TASK_ID, } from "../constants.js";
 import { GraphRecursionError, GraphValueError, InvalidUpdateError, isGraphInterrupt, } from "../errors.js";
 import { _prepareNextTasks, _localRead, _applyWrites, } from "./algo.js";
 import { _coerceToDict, getNewChannelVersions, patchCheckpointMap, } from "./utils/index.js";
@@ -543,7 +543,10 @@ export class Pregel extends Runnable {
             id: uuid5(INTERRUPT, checkpoint.id),
         };
         // execute task
-        await task.proc.invoke(task.input, patchConfig(config, {
+        await task.proc.invoke(task.input, patchConfig({
+            ...config,
+            store: config?.store ?? this.store,
+        }, {
             runName: config.runName ?? `${this.getName()}UpdateState`,
             configurable: {
                 [CONFIG_KEY_SEND]: (items) => task.writes.push(...items),
@@ -606,6 +609,7 @@ export class Pregel extends Runnable {
         else {
             defaultCheckpointer = this.checkpointer;
         }
+        const defaultStore = config.store ?? this.store;
         return [
             defaultDebug,
             defaultStreamMode,
@@ -615,6 +619,7 @@ export class Pregel extends Runnable {
             defaultInterruptBefore,
             defaultInterruptAfter,
             defaultCheckpointer,
+            defaultStore,
         ];
     }
     /**
@@ -636,9 +641,10 @@ export class Pregel extends Runnable {
         return super.stream(input, options);
     }
     async prepareSpecs(config, options) {
-        const configForManaged = patchConfigurable(config, {
-            [CONFIG_KEY_STORE]: this.store,
-        });
+        const configForManaged = {
+            ...config,
+            store: this.store,
+        };
         const channelSpecs = {};
         const managedSpecs = {};
         for (const [name, spec] of Object.entries(this.channels)) {
@@ -693,7 +699,7 @@ export class Pregel extends Runnable {
         const runManager = await callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), inputConfig.runId, undefined, undefined, undefined, inputConfig?.runName ?? this.getName());
         delete inputConfig.runId;
         // assign defaults
-        const [debug, streamMode, , outputKeys, config, interruptBefore, interruptAfter, checkpointer,] = this._defaults(inputConfig);
+        const [debug, streamMode, , outputKeys, config, interruptBefore, interruptAfter, checkpointer, store,] = this._defaults(inputConfig);
         const { channelSpecs, managed } = await this.prepareSpecs(config);
         let loop;
         const stream = new Deque();
@@ -730,7 +736,7 @@ export class Pregel extends Runnable {
                 managed,
                 outputKeys,
                 streamKeys: this.streamChannelsAsIs,
-                store: this.store,
+                store,
                 stream: new StreamProtocol((chunk) => stream.push(chunk), new Set(streamMode)),
             });
             if (options?.subgraphs) {
@@ -809,7 +815,7 @@ export class Pregel extends Runnable {
         finally {
             // Call `.stop()` again incase it was not called in the loop, e.g due to an error.
             if (loop) {
-                loop.store?.stop();
+                await loop.store?.stop();
             }
             await Promise.all([
                 loop?.checkpointerPromises ?? [],

package/dist/pregel/loop.cjs

@@ -10,7 +10,6 @@ const io_js_1 = require("./io.cjs");
 const errors_js_1 = require("../errors.cjs");
 const index_js_1 = require("./utils/index.cjs");
 const debug_js_1 = require("./debug.cjs");
-const batch_js_1 = require("../store/batch.cjs");
 const INPUT_DONE = Symbol.for("INPUT_DONE");
 const INPUT_RESUMING = Symbol.for("INPUT_RESUMING");
 const DEFAULT_LOOP_LIMIT = 25;
@@ -292,7 +291,7 @@ class PregelLoop {
         const stop = step + (config.recursionLimit ?? DEFAULT_LOOP_LIMIT) + 1;
         const checkpointPreviousVersions = { ...checkpoint.channel_versions };
         const store = params.store
-            ? new batch_js_1.AsyncBatchedStore(params.store)
+            ? new langgraph_checkpoint_1.AsyncBatchedStore(params.store)
             : undefined;
         if (store) {
             // Start the store. This is a batch store, so it will run continuously
@@ -447,6 +446,7 @@ class PregelLoop {
                 checkpointer: this.checkpointer,
                 isResuming: this.input === INPUT_RESUMING,
                 manager,
+                store: this.store,
             });
             this.tasks = nextTasks;
             this.taskWritesLeft = Object.values(this.tasks).length - 1;

package/dist/pregel/loop.d.ts

@@ -1,12 +1,11 @@
 import type { RunnableConfig } from "@langchain/core/runnables";
 import type { CallbackManagerForChainRun } from "@langchain/core/callbacks/manager";
-import { BaseCheckpointSaver, Checkpoint, PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All } from "@langchain/langgraph-checkpoint";
+import { BaseCheckpointSaver, Checkpoint, PendingWrite, CheckpointPendingWrite, CheckpointMetadata, All, BaseStore, AsyncBatchedStore } from "@langchain/langgraph-checkpoint";
 import { BaseChannel } from "../channels/base.js";
 import { PregelExecutableTask, StreamMode } from "./types.js";
 import { PregelNode } from "./read.js";
-import { BaseStore } from "../store/base.js";
-import { AsyncBatchedStore } from "../store/batch.js";
 import { ManagedValueMapping } from "../managed/base.js";
+import { LangGraphRunnableConfig } from "./runnable_types.js";
 export type StreamChunk = [string[], StreamMode, unknown];
 export type PregelLoopInitializeParams = {
     input?: any;
@@ -50,7 +49,7 @@ export declare class StreamProtocol {
 export declare class PregelLoop {
     protected input?: any;
     output: any;
-    config: RunnableConfig;
+    config: LangGraphRunnableConfig;
     protected checkpointer?: BaseCheckpointSaver;
     protected checkpointerGetNextVersion: (current: number | undefined, channel: BaseChannel) => number;
     channels: Record<string, BaseChannel>;

package/dist/pregel/loop.js

@@ -1,4 +1,4 @@
-import { copyCheckpoint, emptyCheckpoint, } from "@langchain/langgraph-checkpoint";
+import { copyCheckpoint, emptyCheckpoint, AsyncBatchedStore, } from "@langchain/langgraph-checkpoint";
 import { createCheckpoint, emptyChannels, } from "../channels/base.js";
 import { CHECKPOINT_NAMESPACE_SEPARATOR, CONFIG_KEY_CHECKPOINT_MAP, CONFIG_KEY_READ, CONFIG_KEY_RESUMING, CONFIG_KEY_STREAM, ERROR, INPUT, INTERRUPT, TAG_HIDDEN, TASKS, } from "../constants.js";
 import { _applyWrites, _prepareNextTasks, increment, shouldInterrupt, } from "./algo.js";
@@ -7,7 +7,6 @@ import { mapInput, mapOutputUpdates, mapOutputValues, readChannels, } from "./io
 import { EmptyInputError, GraphInterrupt, isGraphInterrupt, } from "../errors.js";
 import { getNewChannelVersions, patchConfigurable } from "./utils/index.js";
 import { mapDebugTasks, mapDebugCheckpoint, mapDebugTaskResults, } from "./debug.js";
-import { AsyncBatchedStore } from "../store/batch.js";
 const INPUT_DONE = Symbol.for("INPUT_DONE");
 const INPUT_RESUMING = Symbol.for("INPUT_RESUMING");
 const DEFAULT_LOOP_LIMIT = 25;
@@ -443,6 +442,7 @@ export class PregelLoop {
                 checkpointer: this.checkpointer,
                 isResuming: this.input === INPUT_RESUMING,
                 manager,
+                store: this.store,
             });
             this.tasks = nextTasks;
             this.taskWritesLeft = Object.values(this.tasks).length - 1;

package/dist/pregel/read.d.ts

@@ -1,4 +1,4 @@
-import { Runnable, RunnableBinding, RunnableBindingArgs, RunnableConfig, RunnableLike } from "@langchain/core/runnables";
+import { Runnable, RunnableBinding, RunnableBindingArgs, RunnableConfig, type RunnableLike } from "@langchain/core/runnables";
 import { RunnableCallable } from "../utils.js";
 import type { RetryPolicy } from "./utils/index.js";
 export declare class ChannelRead<RunInput = any> extends RunnableCallable {

package/dist/pregel/runnable_types.cjs

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

package/dist/pregel/runnable_types.d.ts

@@ -0,0 +1,5 @@
+import { RunnableConfig } from "@langchain/core/runnables";
+import { BaseStore } from "@langchain/langgraph-checkpoint";
+export interface LangGraphRunnableConfig extends RunnableConfig {
+    store?: BaseStore;
+}

package/dist/pregel/runnable_types.js

@@ -0,0 +1 @@
+export {};

package/dist/pregel/types.d.ts

@@ -1,11 +1,11 @@
 import type { Runnable, RunnableConfig } from "@langchain/core/runnables";
-import type { All, PendingWrite, CheckpointMetadata, BaseCheckpointSaver } from "@langchain/langgraph-checkpoint";
+import type { All, PendingWrite, CheckpointMetadata, BaseCheckpointSaver, BaseStore } from "@langchain/langgraph-checkpoint";
 import type { BaseChannel } from "../channels/base.js";
 import type { PregelNode } from "./read.js";
 import { RetryPolicy } from "./utils/index.js";
 import { Interrupt } from "../constants.js";
-import { BaseStore } from "../store/base.js";
 import { type ManagedValueSpec } from "../managed/base.js";
+import { LangGraphRunnableConfig } from "./runnable_types.js";
 export type StreamMode = "values" | "updates" | "debug";
 /**
  * Construct a type with a set of properties K of type T
@@ -46,7 +46,7 @@ export interface PregelInterface<Nn extends StrRecord<string, PregelNode>, Cc ex
     debug?: boolean;
     checkpointer?: BaseCheckpointSaver;
     retryPolicy?: RetryPolicy;
-    config?: RunnableConfig;
+    config?: LangGraphRunnableConfig;
     /**
      * Memory store to use for SharedValues.
      */
@@ -58,7 +58,7 @@ export interface PregelTaskDescription {
     readonly name: string;
     readonly error?: unknown;
     readonly interrupts: Interrupt[];
-    readonly state?: RunnableConfig | StateSnapshot;
+    readonly state?: LangGraphRunnableConfig | StateSnapshot;
     readonly path?: [string, ...(string | number)[]];
 }
 export interface PregelExecutableTask<N extends PropertyKey, C extends PropertyKey> {
@@ -66,7 +66,7 @@ export interface PregelExecutableTask<N extends PropertyKey, C extends PropertyK
     readonly input: unknown;
     readonly proc: Runnable;
     readonly writes: PendingWrite<C>[];
-    readonly config?: RunnableConfig;
+    readonly config?: LangGraphRunnableConfig;
     readonly triggers: Array<string>;
     readonly retry_policy?: RetryPolicy;
     readonly id: string;

package/dist/pregel/utils/config.cjs

@@ -14,6 +14,7 @@ const CONFIG_KEYS = [
     "runId",
     "outputKeys",
     "streamMode",
+    "store",
 ];
 const DEFAULT_RECURSION_LIMIT = 25;
 function ensureLangGraphConfig(...configs) {

package/dist/pregel/utils/config.d.ts

@@ -1,2 +1,3 @@
 import { RunnableConfig } from "@langchain/core/runnables";
-export declare function ensureLangGraphConfig(...configs: (RunnableConfig | undefined)[]): RunnableConfig;
+import { LangGraphRunnableConfig } from "../runnable_types.js";
+export declare function ensureLangGraphConfig(...configs: (LangGraphRunnableConfig | undefined)[]): RunnableConfig;

package/dist/pregel/utils/config.js

@@ -11,6 +11,7 @@ const CONFIG_KEYS = [
     "runId",
     "outputKeys",
     "streamMode",
+    "store",
 ];
 const DEFAULT_RECURSION_LIMIT = 25;
 export function ensureLangGraphConfig(...configs) {

package/dist/web.cjs

@@ -14,7 +14,7 @@ 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 });
-exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Send = exports.BinaryOperatorAggregate = exports.BaseChannel = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
+exports.MemoryStore = exports.AsyncBatchedStore = exports.BaseStore = exports.BaseCheckpointSaver = exports.emptyCheckpoint = exports.copyCheckpoint = exports.MemorySaver = exports.Send = exports.BinaryOperatorAggregate = exports.BaseChannel = exports.Annotation = exports.messagesStateReducer = exports.MessageGraph = exports.StateGraph = exports.START = exports.Graph = exports.END = void 0;
 var index_js_1 = require("./graph/index.cjs");
 Object.defineProperty(exports, "END", { enumerable: true, get: function () { return index_js_1.END; } });
 Object.defineProperty(exports, "Graph", { enumerable: true, get: function () { return index_js_1.Graph; } });
@@ -34,5 +34,7 @@ Object.defineProperty(exports, "MemorySaver", { enumerable: true, get: function
 Object.defineProperty(exports, "copyCheckpoint", { enumerable: true, get: function () { return langgraph_checkpoint_1.copyCheckpoint; } });
 Object.defineProperty(exports, "emptyCheckpoint", { enumerable: true, get: function () { return langgraph_checkpoint_1.emptyCheckpoint; } });
 Object.defineProperty(exports, "BaseCheckpointSaver", { enumerable: true, get: function () { return langgraph_checkpoint_1.BaseCheckpointSaver; } });
-__exportStar(require("./store/index.cjs"), exports);
+Object.defineProperty(exports, "BaseStore", { enumerable: true, get: function () { return langgraph_checkpoint_1.BaseStore; } });
+Object.defineProperty(exports, "AsyncBatchedStore", { enumerable: true, get: function () { return langgraph_checkpoint_1.AsyncBatchedStore; } });
+Object.defineProperty(exports, "MemoryStore", { enumerable: true, get: function () { return langgraph_checkpoint_1.MemoryStore; } });
 __exportStar(require("./managed/index.cjs"), exports);

package/dist/web.d.ts

@@ -5,6 +5,6 @@ export { BaseChannel, type BinaryOperator, BinaryOperatorAggregate, type AnyValu
 export { type AnnotationRoot as _INTERNAL_ANNOTATION_ROOT } from "./graph/index.js";
 export { type RetryPolicy } from "./pregel/utils/index.js";
 export { Send } from "./constants.js";
-export { MemorySaver, type Checkpoint, type CheckpointMetadata, type CheckpointTuple, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, } from "@langchain/langgraph-checkpoint";
-export * from "./store/index.js";
+export { MemorySaver, type Checkpoint, type CheckpointMetadata, type CheckpointTuple, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, type Item, type GetOperation, type SearchOperation, type PutOperation, type Operation, type OperationResults, BaseStore, AsyncBatchedStore, MemoryStore, type NameSpacePath, type NamespaceMatchType, type MatchCondition, type ListNamespacesOperation, } from "@langchain/langgraph-checkpoint";
 export * from "./managed/index.js";
+export { type LangGraphRunnableConfig } from "./pregel/runnable_types.js";

package/dist/web.js

@@ -2,6 +2,5 @@ export { END, Graph, START, StateGraph, MessageGraph, messagesStateReducer, Anno
 export * from "./errors.js";
 export { BaseChannel, BinaryOperatorAggregate, } from "./channels/index.js";
 export { Send } from "./constants.js";
-export { MemorySaver, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, } from "@langchain/langgraph-checkpoint";
-export * from "./store/index.js";
+export { MemorySaver, copyCheckpoint, emptyCheckpoint, BaseCheckpointSaver, BaseStore, AsyncBatchedStore, MemoryStore, } from "@langchain/langgraph-checkpoint";
 export * from "./managed/index.js";

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/langgraph",
-  "version": "0.2.9",
+  "version": "0.2.10-rc.0",
   "description": "LangGraph",
   "type": "module",
   "engines": {

package/dist/store/base.cjs

@@ -1,12 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.BaseStore = void 0;
-class BaseStore {
-    stop() {
-        // no-op if not implemented.
-    }
-    start() {
-        // no-op if not implemented.
-    }
-}
-exports.BaseStore = BaseStore;

package/dist/store/base.d.ts

@@ -1,7 +0,0 @@
-export type Values = Record<string, any>;
-export declare abstract class BaseStore {
-    abstract list(prefixes: string[]): Promise<Record<string, Record<string, Values>>>;
-    abstract put(writes: Array<[string, string, Values | null]>): Promise<void>;
-    stop(): void;
-    start(): void;
-}

package/dist/store/base.js

@@ -1,8 +0,0 @@
-export class BaseStore {
-    stop() {
-        // no-op if not implemented.
-    }
-    start() {
-        // no-op if not implemented.
-    }
-}

package/dist/store/batch.cjs

@@ -1,126 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.AsyncBatchedStore = void 0;
-const base_js_1 = require("./base.cjs");
-/**
- * AsyncBatchedStore extends BaseStore to provide batched operations for list and put methods.
- * It queues operations and processes them in batches for improved efficiency. This store is
- * designed to run for the full duration of the process, or until `stop()` is called.
- */
-class AsyncBatchedStore extends base_js_1.BaseStore {
-    get isRunning() {
-        return this.running;
-    }
-    constructor(store) {
-        super();
-        /**
-         * The store to batch operations for.
-         * @type {BaseStore}
-         */
-        Object.defineProperty(this, "store", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        /**
-         * A queue of operations to be processed in batch.
-         * @type {QueueItem[]}
-         */
-        Object.defineProperty(this, "queue", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: []
-        });
-        /**
-         * Whether or not the batched processing is currently running.
-         * @type {boolean}
-         * @default {false}
-         */
-        Object.defineProperty(this, "running", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        this.store = store;
-    }
-    /**
-     * Queues a list operation to be processed in batch.
-     * @param {string[]} prefixes An array of prefixes to list.
-     * @returns {Promise<Record<string, Record<string, Values>>>} A promise that resolves with the list results.
-     */
-    async list(prefixes) {
-        return new Promise((resolve, reject) => {
-            this.queue.push({ resolve, reject, op: { prefixes } });
-        });
-    }
-    /**
-     * Queues a put operation to be processed in batch.
-     * @param {Array<[string, string, Values | null]>} writes An array of write operations to be performed.
-     * @returns {Promise<void>} A promise that resolves when the put operation is complete.
-     */
-    async put(writes) {
-        return new Promise((resolve, reject) => {
-            this.queue.push({ resolve, reject, op: { writes } });
-        });
-    }
-    /**
-     * Start running the batched processing of operations.
-     * This process will run continuously until the store is stopped,
-     * which can be done by calling the `stop()` method.
-     */
-    start() {
-        this.running = true;
-        void this.processBatchQueue();
-    }
-    /**
-     * Stops the batched processing of operations.
-     */
-    stop() {
-        this.running = false;
-    }
-    /**
-     * Runs the task that processes queued operations in batches.
-     * This method runs continuously until the store is stopped,
-     * or the process is terminated.
-     * @returns {Promise<void>} A promise that resolves when the task is complete.
-     */
-    async processBatchQueue() {
-        while (this.running) {
-            await new Promise((resolve) => {
-                setTimeout(resolve, 0);
-            });
-            if (this.queue.length === 0)
-                continue;
-            const taken = this.queue.splice(0);
-            const lists = taken.filter((item) => "prefixes" in item.op);
-            if (lists.length > 0) {
-                try {
-                    const allPrefixes = lists.flatMap((item) => item.op.prefixes);
-                    const results = await this.store.list(allPrefixes);
-                    lists.forEach((item) => {
-                        const { prefixes } = item.op;
-                        item.resolve(Object.fromEntries(prefixes.map((p) => [p, results[p] || {}])));
-                    });
-                }
-                catch (e) {
-                    lists.forEach((item) => item.reject(e));
-                }
-            }
-            const puts = taken.filter((item) => "writes" in item.op);
-            if (puts.length > 0) {
-                try {
-                    const allWrites = puts.flatMap((item) => item.op.writes);
-                    await this.store.put(allWrites);
-                    puts.forEach((item) => item.resolve());
-                }
-                catch (e) {
-                    puts.forEach((item) => item.reject(e));
-                }
-            }
-        }
-    }
-}
-exports.AsyncBatchedStore = AsyncBatchedStore;

package/dist/store/batch.d.ts

@@ -1,55 +0,0 @@
-import { BaseStore, type Values } from "./base.js";
-/**
- * AsyncBatchedStore extends BaseStore to provide batched operations for list and put methods.
- * It queues operations and processes them in batches for improved efficiency. This store is
- * designed to run for the full duration of the process, or until `stop()` is called.
- */
-export declare class AsyncBatchedStore extends BaseStore {
-    /**
-     * The store to batch operations for.
-     * @type {BaseStore}
-     */
-    private store;
-    /**
-     * A queue of operations to be processed in batch.
-     * @type {QueueItem[]}
-     */
-    private queue;
-    /**
-     * Whether or not the batched processing is currently running.
-     * @type {boolean}
-     * @default {false}
-     */
-    private running;
-    get isRunning(): boolean;
-    constructor(store: BaseStore);
-    /**
-     * Queues a list operation to be processed in batch.
-     * @param {string[]} prefixes An array of prefixes to list.
-     * @returns {Promise<Record<string, Record<string, Values>>>} A promise that resolves with the list results.
-     */
-    list(prefixes: string[]): Promise<Record<string, Record<string, Values>>>;
-    /**
-     * Queues a put operation to be processed in batch.
-     * @param {Array<[string, string, Values | null]>} writes An array of write operations to be performed.
-     * @returns {Promise<void>} A promise that resolves when the put operation is complete.
-     */
-    put(writes: Array<[string, string, Values | null]>): Promise<void>;
-    /**
-     * Start running the batched processing of operations.
-     * This process will run continuously until the store is stopped,
-     * which can be done by calling the `stop()` method.
-     */
-    start(): void;
-    /**
-     * Stops the batched processing of operations.
-     */
-    stop(): void;
-    /**
-     * Runs the task that processes queued operations in batches.
-     * This method runs continuously until the store is stopped,
-     * or the process is terminated.
-     * @returns {Promise<void>} A promise that resolves when the task is complete.
-     */
-    private processBatchQueue;
-}

package/dist/store/batch.js

@@ -1,122 +0,0 @@
-import { BaseStore } from "./base.js";
-/**
- * AsyncBatchedStore extends BaseStore to provide batched operations for list and put methods.
- * It queues operations and processes them in batches for improved efficiency. This store is
- * designed to run for the full duration of the process, or until `stop()` is called.
- */
-export class AsyncBatchedStore extends BaseStore {
-    get isRunning() {
-        return this.running;
-    }
-    constructor(store) {
-        super();
-        /**
-         * The store to batch operations for.
-         * @type {BaseStore}
-         */
-        Object.defineProperty(this, "store", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        /**
-         * A queue of operations to be processed in batch.
-         * @type {QueueItem[]}
-         */
-        Object.defineProperty(this, "queue", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: []
-        });
-        /**
-         * Whether or not the batched processing is currently running.
-         * @type {boolean}
-         * @default {false}
-         */
-        Object.defineProperty(this, "running", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        this.store = store;
-    }
-    /**
-     * Queues a list operation to be processed in batch.
-     * @param {string[]} prefixes An array of prefixes to list.
-     * @returns {Promise<Record<string, Record<string, Values>>>} A promise that resolves with the list results.
-     */
-    async list(prefixes) {
-        return new Promise((resolve, reject) => {
-            this.queue.push({ resolve, reject, op: { prefixes } });
-        });
-    }
-    /**
-     * Queues a put operation to be processed in batch.
-     * @param {Array<[string, string, Values | null]>} writes An array of write operations to be performed.
-     * @returns {Promise<void>} A promise that resolves when the put operation is complete.
-     */
-    async put(writes) {
-        return new Promise((resolve, reject) => {
-            this.queue.push({ resolve, reject, op: { writes } });
-        });
-    }
-    /**
-     * Start running the batched processing of operations.
-     * This process will run continuously until the store is stopped,
-     * which can be done by calling the `stop()` method.
-     */
-    start() {
-        this.running = true;
-        void this.processBatchQueue();
-    }
-    /**
-     * Stops the batched processing of operations.
-     */
-    stop() {
-        this.running = false;
-    }
-    /**
-     * Runs the task that processes queued operations in batches.
-     * This method runs continuously until the store is stopped,
-     * or the process is terminated.
-     * @returns {Promise<void>} A promise that resolves when the task is complete.
-     */
-    async processBatchQueue() {
-        while (this.running) {
-            await new Promise((resolve) => {
-                setTimeout(resolve, 0);
-            });
-            if (this.queue.length === 0)
-                continue;
-            const taken = this.queue.splice(0);
-            const lists = taken.filter((item) => "prefixes" in item.op);
-            if (lists.length > 0) {
-                try {
-                    const allPrefixes = lists.flatMap((item) => item.op.prefixes);
-                    const results = await this.store.list(allPrefixes);
-                    lists.forEach((item) => {
-                        const { prefixes } = item.op;
-                        item.resolve(Object.fromEntries(prefixes.map((p) => [p, results[p] || {}])));
-                    });
-                }
-                catch (e) {
-                    lists.forEach((item) => item.reject(e));
-                }
-            }
-            const puts = taken.filter((item) => "writes" in item.op);
-            if (puts.length > 0) {
-                try {
-                    const allWrites = puts.flatMap((item) => item.op.writes);
-                    await this.store.put(allWrites);
-                    puts.forEach((item) => item.resolve());
-                }
-                catch (e) {
-                    puts.forEach((item) => item.reject(e));
-                }
-            }
-        }
-    }
-}

package/dist/store/index.cjs

@@ -1,19 +0,0 @@
-"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("./batch.cjs"), exports);
-__exportStar(require("./memory.cjs"), exports);

package/dist/store/index.d.ts

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

package/dist/store/index.js

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

package/dist/store/memory.cjs

@@ -1,43 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.MemoryStore = void 0;
-const base_js_1 = require("./base.cjs");
-class MemoryStore extends base_js_1.BaseStore {
-    constructor() {
-        super(...arguments);
-        Object.defineProperty(this, "data", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: new Map()
-        });
-    }
-    async list(prefixes) {
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        const result = {};
-        for (const prefix of prefixes) {
-            if (this.data.has(prefix)) {
-                result[prefix] = Object.fromEntries(this.data.get(prefix));
-            }
-            else {
-                result[prefix] = {};
-            }
-        }
-        return Promise.resolve(result);
-    }
-    async put(writes) {
-        for (const [namespace, key, value] of writes) {
-            if (!this.data.has(namespace)) {
-                this.data.set(namespace, new Map());
-            }
-            const namespaceMap = this.data.get(namespace);
-            if (value === null) {
-                namespaceMap.delete(key);
-            }
-            else {
-                namespaceMap.set(key, value);
-            }
-        }
-    }
-}
-exports.MemoryStore = MemoryStore;

package/dist/store/memory.d.ts

@@ -1,6 +0,0 @@
-import { BaseStore, type Values } from "./base.js";
-export declare class MemoryStore extends BaseStore {
-    private data;
-    list(prefixes: string[]): Promise<Record<string, Record<string, Values>>>;
-    put(writes: Array<[string, string, Values | null]>): Promise<void>;
-}

package/dist/store/memory.js

@@ -1,39 +0,0 @@
-import { BaseStore } from "./base.js";
-export class MemoryStore extends BaseStore {
-    constructor() {
-        super(...arguments);
-        Object.defineProperty(this, "data", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: new Map()
-        });
-    }
-    async list(prefixes) {
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        const result = {};
-        for (const prefix of prefixes) {
-            if (this.data.has(prefix)) {
-                result[prefix] = Object.fromEntries(this.data.get(prefix));
-            }
-            else {
-                result[prefix] = {};
-            }
-        }
-        return Promise.resolve(result);
-    }
-    async put(writes) {
-        for (const [namespace, key, value] of writes) {
-            if (!this.data.has(namespace)) {
-                this.data.set(namespace, new Map());
-            }
-            const namespaceMap = this.data.get(namespace);
-            if (value === null) {
-                namespaceMap.delete(key);
-            }
-            else {
-                namespaceMap.set(key, value);
-            }
-        }
-    }
-}