@langchain/langgraph-checkpoint 0.1.0 → 1.0.0

package/dist/store/batch.js

@@ -1,147 +1,120 @@
-/* eslint-disable @typescript-eslint/no-explicit-any */
-import { BaseStore, } from "./base.js";
+import { BaseStore } from "./base.js";
+
+//#region src/store/batch.ts
 /**
- * Extracts and returns the underlying store from an `AsyncBatchedStore`,
- * or returns the input if it is not an `AsyncBatchedStore`.
- */
+* Extracts and returns the underlying store from an `AsyncBatchedStore`,
+* or returns the input if it is not an `AsyncBatchedStore`.
+*/
 const extractStore = (input) => {
-    if ("lg_name" in input && input.lg_name === "AsyncBatchedStore") {
-        // @ts-expect-error is a protected property
-        return input.store;
-    }
-    return input;
+	if ("lg_name" in input && input.lg_name === "AsyncBatchedStore") return input.store;
+	return input;
 };
-export class AsyncBatchedStore extends BaseStore {
-    constructor(store) {
-        super();
-        Object.defineProperty(this, "lg_name", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "AsyncBatchedStore"
-        });
-        Object.defineProperty(this, "store", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "queue", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: new Map()
-        });
-        Object.defineProperty(this, "nextKey", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-        Object.defineProperty(this, "running", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: false
-        });
-        Object.defineProperty(this, "processingTask", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: null
-        });
-        this.store = extractStore(store);
-    }
-    get isRunning() {
-        return this.running;
-    }
-    /**
-     * @ignore
-     * Batch is not implemented here as we're only extending `BaseStore`
-     * to allow it to be passed where `BaseStore` is expected, and implement
-     * the convenience methods (get, search, put, delete).
-     */
-    async batch(_operations) {
-        throw new Error("The `batch` method is not implemented on `AsyncBatchedStore`." +
-            "\n Instead, it calls the `batch` method on the wrapped store." +
-            "\n If you are seeing this error, something is wrong.");
-    }
-    async get(namespace, key) {
-        return this.enqueueOperation({ namespace, key });
-    }
-    async search(namespacePrefix, options) {
-        const { filter, limit = 10, offset = 0, query } = options || {};
-        return this.enqueueOperation({
-            namespacePrefix,
-            filter,
-            limit,
-            offset,
-            query,
-        });
-    }
-    async put(namespace, key, value) {
-        return this.enqueueOperation({ namespace, key, value });
-    }
-    async delete(namespace, key) {
-        return this.enqueueOperation({
-            namespace,
-            key,
-            value: null,
-        });
-    }
-    start() {
-        if (!this.running) {
-            this.running = true;
-            this.processingTask = this.processBatchQueue();
-        }
-    }
-    async stop() {
-        this.running = false;
-        if (this.processingTask) {
-            await this.processingTask;
-        }
-    }
-    enqueueOperation(operation) {
-        return new Promise((resolve, reject) => {
-            const key = this.nextKey;
-            this.nextKey += 1;
-            this.queue.set(key, { operation, resolve, reject });
-        });
-    }
-    async processBatchQueue() {
-        while (this.running) {
-            await new Promise((resolve) => {
-                setTimeout(resolve, 0);
-            });
-            if (this.queue.size === 0)
-                continue;
-            const batch = new Map(this.queue);
-            this.queue.clear();
-            try {
-                const operations = Array.from(batch.values()).map(({ operation }) => operation);
-                const results = await this.store.batch(operations);
-                batch.forEach(({ resolve }, key) => {
-                    const index = Array.from(batch.keys()).indexOf(key);
-                    resolve(results[index]);
-                });
-            }
-            catch (e) {
-                batch.forEach(({ reject }) => {
-                    reject(e);
-                });
-            }
-        }
-    }
-    // AsyncBatchedStore is internal and gets passed as args into traced tasks
-    // some BaseStores contain circular references so just serialize without it
-    // as this causes warnings when tracing with LangSmith.
-    toJSON() {
-        return {
-            queue: this.queue,
-            nextKey: this.nextKey,
-            running: this.running,
-            store: "[LangGraphStore]",
-        };
-    }
-}
+var AsyncBatchedStore = class extends BaseStore {
+	lg_name = "AsyncBatchedStore";
+	store;
+	queue = /* @__PURE__ */ new Map();
+	nextKey = 0;
+	running = false;
+	processingTask = null;
+	constructor(store) {
+		super();
+		this.store = extractStore(store);
+	}
+	get isRunning() {
+		return this.running;
+	}
+	/**
+	* @ignore
+	* Batch is not implemented here as we're only extending `BaseStore`
+	* to allow it to be passed where `BaseStore` is expected, and implement
+	* the convenience methods (get, search, put, delete).
+	*/
+	async batch(_operations) {
+		throw new Error("The `batch` method is not implemented on `AsyncBatchedStore`.\n Instead, it calls the `batch` method on the wrapped store.\n If you are seeing this error, something is wrong.");
+	}
+	async get(namespace, key) {
+		return this.enqueueOperation({
+			namespace,
+			key
+		});
+	}
+	async search(namespacePrefix, options) {
+		const { filter, limit = 10, offset = 0, query } = options || {};
+		return this.enqueueOperation({
+			namespacePrefix,
+			filter,
+			limit,
+			offset,
+			query
+		});
+	}
+	async put(namespace, key, value) {
+		return this.enqueueOperation({
+			namespace,
+			key,
+			value
+		});
+	}
+	async delete(namespace, key) {
+		return this.enqueueOperation({
+			namespace,
+			key,
+			value: null
+		});
+	}
+	start() {
+		if (!this.running) {
+			this.running = true;
+			this.processingTask = this.processBatchQueue();
+		}
+	}
+	async stop() {
+		this.running = false;
+		if (this.processingTask) await this.processingTask;
+	}
+	enqueueOperation(operation) {
+		return new Promise((resolve, reject) => {
+			const key = this.nextKey;
+			this.nextKey += 1;
+			this.queue.set(key, {
+				operation,
+				resolve,
+				reject
+			});
+		});
+	}
+	async processBatchQueue() {
+		while (this.running) {
+			await new Promise((resolve) => {
+				setTimeout(resolve, 0);
+			});
+			if (this.queue.size === 0) continue;
+			const batch = new Map(this.queue);
+			this.queue.clear();
+			try {
+				const operations = Array.from(batch.values()).map(({ operation }) => operation);
+				const results = await this.store.batch(operations);
+				batch.forEach(({ resolve }, key) => {
+					const index = Array.from(batch.keys()).indexOf(key);
+					resolve(results[index]);
+				});
+			} catch (e) {
+				batch.forEach(({ reject }) => {
+					reject(e);
+				});
+			}
+		}
+	}
+	toJSON() {
+		return {
+			queue: this.queue,
+			nextKey: this.nextKey,
+			running: this.running,
+			store: "[LangGraphStore]"
+		};
+	}
+};
+
+//#endregion
+export { AsyncBatchedStore };
 //# sourceMappingURL=batch.js.map

package/dist/store/batch.js.map

@@ -1 +1 @@
-{"version":3,"file":"batch.js","sourceRoot":"","sources":["../../src/store/batch.ts"],"names":[],"mappings":"AAAA,uDAAuD;AAEvD,OAAO,EACL,SAAS,GAOV,MAAM,WAAW,CAAC;AAEnB;;;GAGG;AACH,MAAM,YAAY,GAAG,CAAC,KAAoC,EAAa,EAAE;IACvE,IAAI,SAAS,IAAI,KAAK,IAAI,KAAK,CAAC,OAAO,KAAK,mBAAmB,EAAE,CAAC;QAChE,2CAA2C;QAC3C,OAAO,KAAK,CAAC,KAAK,CAAC;IACrB,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,OAAO,iBAAkB,SAAQ,SAAS;IAoB9C,YAAY,KAAgB;QAC1B,KAAK,EAAE,CAAC;QApBV;;;;mBAAU,mBAAmB;WAAC;QAEpB;;;;;WAAiB;QAEnB;;;;mBAOJ,IAAI,GAAG,EAAE;WAAC;QAEN;;;;mBAAkB,CAAC;WAAC;QAEpB;;;;mBAAU,KAAK;WAAC;QAEhB;;;;mBAAuC,IAAI;WAAC;QAIlD,IAAI,CAAC,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACnC,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,KAAK,CACT,WAAe;QAEf,MAAM,IAAI,KAAK,CACb,+DAA+D;YAC7D,+DAA+D;YAC/D,sDAAsD,CACzD,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,SAAmB,EAAE,GAAW;QACxC,OAAO,IAAI,CAAC,gBAAgB,CAAC,EAAE,SAAS,EAAE,GAAG,EAAkB,CAAC,CAAC;IACnE,CAAC;IAED,KAAK,CAAC,MAAM,CACV,eAAyB,EACzB,OAKC;QAED,MAAM,EAAE,MAAM,EAAE,KAAK,GAAG,EAAE,EAAE,MAAM,GAAG,CAAC,EAAE,KAAK,EAAE,GAAG,OAAO,IAAI,EAAE,CAAC;QAChE,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,eAAe;YACf,MAAM;YACN,KAAK;YACL,MAAM;YACN,KAAK;SACa,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,GAAG,CACP,SAAmB,EACnB,GAAW,EACX,KAA0B;QAE1B,OAAO,IAAI,CAAC,gBAAgB,CAAC,EAAE,SAAS,EAAE,GAAG,EAAE,KAAK,EAAkB,CAAC,CAAC;IAC1E,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAmB,EAAE,GAAW;QAC3C,OAAO,IAAI,CAAC,gBAAgB,CAAC;YAC3B,SAAS;YACT,GAAG;YACH,KAAK,EAAE,IAAI;SACI,CAAC,CAAC;IACrB,CAAC;IAED,KAAK;QACH,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACjD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,IAAI,CAAC,cAAc,CAAC;QAC5B,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAI,SAAoB;QAC9C,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC;YACzB,IAAI,CAAC,OAAO,IAAI,CAAC,CAAC;YAClB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QACtD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,iBAAiB;QAC7B,OAAO,IAAI,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;gBAC5B,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YACzB,CAAC,CAAC,CAAC;YACH,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC;gBAAE,SAAS;YAEpC,MAAM,KAAK,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAClC,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;YAEnB,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAC/C,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,SAAS,CAC7B,CAAC;gBACF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBAEnD,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,EAAE;oBACjC,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;oBACpD,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC1B,CAAC,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,KAAK,CAAC,OAAO,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE;oBAC3B,MAAM,CAAC,CAAC,CAAC,CAAC;gBACZ,CAAC,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAED,0EAA0E;IAC1E,2EAA2E;IAC3E,uDAAuD;IACvD,MAAM;QACJ,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,KAAK,EAAE,kBAAkB;SAC1B,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"batch.js","names":[],"sources":["../../src/store/batch.ts"],"sourcesContent":["/* eslint-disable @typescript-eslint/no-explicit-any */\n\nimport {\n  BaseStore,\n  type Item,\n  type SearchOperation,\n  type PutOperation,\n  type GetOperation,\n  type Operation,\n  OperationResults,\n} from \"./base.js\";\n\n/**\n * Extracts and returns the underlying store from an `AsyncBatchedStore`,\n * or returns the input if it is not an `AsyncBatchedStore`.\n */\nconst extractStore = (input: BaseStore | AsyncBatchedStore): BaseStore => {\n  if (\"lg_name\" in input && input.lg_name === \"AsyncBatchedStore\") {\n    // @ts-expect-error is a protected property\n    return input.store;\n  }\n  return input;\n};\n\nexport class AsyncBatchedStore extends BaseStore {\n  lg_name = \"AsyncBatchedStore\";\n\n  protected store: BaseStore;\n\n  private queue: Map<\n    number,\n    {\n      operation: Operation;\n      resolve: (value: any) => void;\n      reject: (reason?: any) => void;\n    }\n  > = new Map();\n\n  private nextKey: number = 0;\n\n  private running = false;\n\n  private processingTask: Promise<void> | null = null;\n\n  constructor(store: BaseStore) {\n    super();\n    this.store = extractStore(store);\n  }\n\n  get isRunning(): boolean {\n    return this.running;\n  }\n\n  /**\n   * @ignore\n   * Batch is not implemented here as we're only extending `BaseStore`\n   * to allow it to be passed where `BaseStore` is expected, and implement\n   * the convenience methods (get, search, put, delete).\n   */\n  async batch<Op extends Operation[]>(\n    _operations: Op\n  ): Promise<OperationResults<Op>> {\n    throw new Error(\n      \"The `batch` method is not implemented on `AsyncBatchedStore`.\" +\n        \"\\n Instead, it calls the `batch` method on the wrapped store.\" +\n        \"\\n If you are seeing this error, something is wrong.\"\n    );\n  }\n\n  async get(namespace: string[], key: string): Promise<Item | null> {\n    return this.enqueueOperation({ namespace, key } as GetOperation);\n  }\n\n  async search(\n    namespacePrefix: string[],\n    options?: {\n      filter?: Record<string, any>;\n      limit?: number;\n      offset?: number;\n      query?: string;\n    }\n  ): Promise<Item[]> {\n    const { filter, limit = 10, offset = 0, query } = options || {};\n    return this.enqueueOperation({\n      namespacePrefix,\n      filter,\n      limit,\n      offset,\n      query,\n    } as SearchOperation);\n  }\n\n  async put(\n    namespace: string[],\n    key: string,\n    value: Record<string, any>\n  ): Promise<void> {\n    return this.enqueueOperation({ namespace, key, value } as PutOperation);\n  }\n\n  async delete(namespace: string[], key: string): Promise<void> {\n    return this.enqueueOperation({\n      namespace,\n      key,\n      value: null,\n    } as PutOperation);\n  }\n\n  start(): void {\n    if (!this.running) {\n      this.running = true;\n      this.processingTask = this.processBatchQueue();\n    }\n  }\n\n  async stop(): Promise<void> {\n    this.running = false;\n    if (this.processingTask) {\n      await this.processingTask;\n    }\n  }\n\n  private enqueueOperation<T>(operation: Operation): Promise<T> {\n    return new Promise<T>((resolve, reject) => {\n      const key = this.nextKey;\n      this.nextKey += 1;\n      this.queue.set(key, { operation, resolve, reject });\n    });\n  }\n\n  private async processBatchQueue(): Promise<void> {\n    while (this.running) {\n      await new Promise((resolve) => {\n        setTimeout(resolve, 0);\n      });\n      if (this.queue.size === 0) continue;\n\n      const batch = new Map(this.queue);\n      this.queue.clear();\n\n      try {\n        const operations = Array.from(batch.values()).map(\n          ({ operation }) => operation\n        );\n        const results = await this.store.batch(operations);\n\n        batch.forEach(({ resolve }, key) => {\n          const index = Array.from(batch.keys()).indexOf(key);\n          resolve(results[index]);\n        });\n      } catch (e) {\n        batch.forEach(({ reject }) => {\n          reject(e);\n        });\n      }\n    }\n  }\n\n  // AsyncBatchedStore is internal and gets passed as args into traced tasks\n  // some BaseStores contain circular references so just serialize without it\n  // as this causes warnings when tracing with LangSmith.\n  toJSON() {\n    return {\n      queue: this.queue,\n      nextKey: this.nextKey,\n      running: this.running,\n      store: \"[LangGraphStore]\",\n    };\n  }\n}\n"],"mappings":";;;;;;;AAgBA,MAAM,gBAAgB,UAAoD;AACxE,KAAI,aAAa,SAAS,MAAM,YAAY,oBAE1C,QAAO,MAAM;AAEf,QAAO;;AAGT,IAAa,oBAAb,cAAuC,UAAU;CAC/C,UAAU;CAEV,AAAU;CAEV,AAAQ,wBAOJ,IAAI;CAER,AAAQ,UAAkB;CAE1B,AAAQ,UAAU;CAElB,AAAQ,iBAAuC;CAE/C,YAAY,OAAkB;AAC5B;AACA,OAAK,QAAQ,aAAa;;CAG5B,IAAI,YAAqB;AACvB,SAAO,KAAK;;;;;;;;CASd,MAAM,MACJ,aAC+B;AAC/B,QAAM,IAAI,MACR;;CAMJ,MAAM,IAAI,WAAqB,KAAmC;AAChE,SAAO,KAAK,iBAAiB;GAAE;GAAW;;;CAG5C,MAAM,OACJ,iBACA,SAMiB;EACjB,MAAM,EAAE,QAAQ,QAAQ,IAAI,SAAS,GAAG,UAAU,WAAW;AAC7D,SAAO,KAAK,iBAAiB;GAC3B;GACA;GACA;GACA;GACA;;;CAIJ,MAAM,IACJ,WACA,KACA,OACe;AACf,SAAO,KAAK,iBAAiB;GAAE;GAAW;GAAK;;;CAGjD,MAAM,OAAO,WAAqB,KAA4B;AAC5D,SAAO,KAAK,iBAAiB;GAC3B;GACA;GACA,OAAO;;;CAIX,QAAc;AACZ,MAAI,CAAC,KAAK,SAAS;AACjB,QAAK,UAAU;AACf,QAAK,iBAAiB,KAAK;;;CAI/B,MAAM,OAAsB;AAC1B,OAAK,UAAU;AACf,MAAI,KAAK,eACP,OAAM,KAAK;;CAIf,AAAQ,iBAAoB,WAAkC;AAC5D,SAAO,IAAI,SAAY,SAAS,WAAW;GACzC,MAAM,MAAM,KAAK;AACjB,QAAK,WAAW;AAChB,QAAK,MAAM,IAAI,KAAK;IAAE;IAAW;IAAS;;;;CAI9C,MAAc,oBAAmC;AAC/C,SAAO,KAAK,SAAS;AACnB,SAAM,IAAI,SAAS,YAAY;AAC7B,eAAW,SAAS;;AAEtB,OAAI,KAAK,MAAM,SAAS,EAAG;GAE3B,MAAM,QAAQ,IAAI,IAAI,KAAK;AAC3B,QAAK,MAAM;AAEX,OAAI;IACF,MAAM,aAAa,MAAM,KAAK,MAAM,UAAU,KAC3C,EAAE,gBAAgB;IAErB,MAAM,UAAU,MAAM,KAAK,MAAM,MAAM;AAEvC,UAAM,SAAS,EAAE,WAAW,QAAQ;KAClC,MAAM,QAAQ,MAAM,KAAK,MAAM,QAAQ,QAAQ;AAC/C,aAAQ,QAAQ;;YAEX,GAAG;AACV,UAAM,SAAS,EAAE,aAAa;AAC5B,YAAO;;;;;CASf,SAAS;AACP,SAAO;GACL,OAAO,KAAK;GACZ,SAAS,KAAK;GACd,SAAS,KAAK;GACd,OAAO"}