@langchain/langgraph 1.4.18 → 1.4.20

package/dist/pregel/index.cjs

@@ -424,17 +424,62 @@ var Pregel = class extends PartialRunnable {
 		yield* this.getSubgraphs(namespace, recurse);
 	}
 	/**
+	* The checkpointer runs and state methods use: none for `checkpointer: false`,
+	* else the one a parent lends a subgraph through the config, else this
+	* graph's own.
+	*/
+	_resolveCheckpointer(config) {
+		if (this.checkpointer === false) return void 0;
+		const lent = config?.configurable?.[require_constants.CONFIG_KEY_CHECKPOINTER];
+		if (lent !== void 0) return lent;
+		if (this.checkpointer === true) throw new Error("checkpointer: true cannot be used for root graphs.");
+		return this.checkpointer;
+	}
+	_stateCheckpointer(config) {
+		const checkpointer = this._resolveCheckpointer(config);
+		if (typeof checkpointer !== "object" || checkpointer === null) throw new require_errors.GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
+		return checkpointer;
+	}
+	/**
+	* A `checkpointer: true` subgraph keeps one history per thread, stored
+	* under its namespace with the task ids removed.
+	*/
+	_ownCheckpointConfig(config) {
+		if (this.checkpointer !== true) return config;
+		const ns = config.configurable?.["checkpoint_ns"] ?? "";
+		return {
+			...config,
+			configurable: {
+				...config.configurable,
+				[require_constants.CONFIG_KEY_CHECKPOINT_NS]: ns.split("|").map((part) => part.split(":")[0]).join("|")
+			}
+		};
+	}
+	/**
+	* The subgraph a state method's namespaced config addresses, and the config
+	* to call it with, lending it `checkpointer`. `undefined` when the config is
+	* for this graph, comes from inside a running task, or names no declared
+	* subgraph (a dynamically created one, such as a tool-call subgraph).
+	*/
+	async _subgraphForNamespace(config, checkpointer) {
+		const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
+		if (checkpointNamespace === "" || config.configurable?.["__pregel_read"] !== void 0 || config.configurable?.["__pregel_checkpointer"] !== void 0) return;
+		const recastNamespace = require_config.recastCheckpointNamespace(checkpointNamespace);
+		for await (const [name, subgraph] of this.getSubgraphsAsync(recastNamespace, true)) if (name === recastNamespace) return [subgraph, require_utils.patchConfigurable(config, { [require_constants.CONFIG_KEY_CHECKPOINTER]: checkpointer })];
+	}
+	/**
 	* Prepares a state snapshot from saved checkpoint data.
 	* This is an internal method used by getState and getStateHistory.
 	*
 	* @param config - Configuration for preparing the snapshot
 	* @param saved - Optional saved checkpoint data
-	* @param subgraphCheckpointer - Optional checkpointer for subgraphs
+	* @param saver - Checkpointer the caller resolved; a subgraph's own `checkpointer` is unset or `true`
+	* @param recurse - Whether to include each subgraph task's state
 	* @param applyPendingWrites - Whether to apply pending writes to tasks and then to channels
 	* @returns A snapshot of the graph state
 	* @internal
 	*/
-	async _prepareStateSnapshot({ config, saved, subgraphCheckpointer, applyPendingWrites = false }) {
+	async _prepareStateSnapshot({ config, saved, saver, recurse = false, applyPendingWrites = false }) {
 		if (saved === void 0) return {
 			values: {},
 			next: [],
@@ -442,8 +487,9 @@ var Pregel = class extends PartialRunnable {
 			tasks: []
 		};
 		const channels = await require_base.channelsFromCheckpoint(this.channels, saved.checkpoint, {
-			saver: typeof this.checkpointer === "object" ? this.checkpointer : void 0,
-			config: saved.config ?? config
+			saver,
+			config: saved.config ?? config,
+			deltaWritesVersioned: require_base.isDeltaWritesVersioned(saved.metadata)
 		});
 		if (saved.pendingWrites?.length) {
 			const nullWrites = saved.pendingWrites.filter(([taskId, _]) => taskId === require_constants.NULL_TASK_ID).map(([_, channel, value]) => [String(channel), value]);
@@ -462,10 +508,10 @@ var Pregel = class extends PartialRunnable {
 		const taskStates = {};
 		for (const task of nextTasks) {
 			const matchingSubgraph = subgraphs.find(([name]) => name === task.name);
-			if (!matchingSubgraph) continue;
+			if (!matchingSubgraph || matchingSubgraph[1].checkpointer === false) continue;
 			let taskNs = `${String(task.name)}:${task.id}`;
 			if (parentNamespace) taskNs = `${parentNamespace}|${taskNs}`;
-			if (subgraphCheckpointer === void 0) {
+			if (!recurse) {
 				const config = { configurable: {
 					thread_id: saved.config.configurable?.thread_id,
 					checkpoint_ns: taskNs
@@ -473,7 +519,7 @@ var Pregel = class extends PartialRunnable {
 				taskStates[task.id] = config;
 			} else {
 				const subgraphConfig = { configurable: {
-					[require_constants.CONFIG_KEY_CHECKPOINTER]: subgraphCheckpointer,
+					[require_constants.CONFIG_KEY_CHECKPOINTER]: saver,
 					thread_id: saved.config.configurable?.thread_id,
 					checkpoint_ns: taskNs
 				} };
@@ -521,19 +567,20 @@ var Pregel = class extends PartialRunnable {
 	* @throws {GraphValueError} If no checkpointer is configured
 	*/
 	async getState(config, options) {
-		const checkpointer = config.configurable?.["__pregel_checkpointer"] ?? this.checkpointer;
-		if (!checkpointer) throw new require_errors.GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
-		const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
-		if (checkpointNamespace !== "" && config.configurable?.["__pregel_read"] === void 0 && config.configurable?.["__pregel_checkpointer"] === void 0) {
-			const recastNamespace = require_config.recastCheckpointNamespace(checkpointNamespace);
-			for await (const [name, subgraph] of this.getSubgraphsAsync(recastNamespace, true)) if (name === recastNamespace) return await subgraph.getState(require_utils.patchConfigurable(config, { [require_constants.CONFIG_KEY_CHECKPOINTER]: checkpointer }), { subgraphs: options?.subgraphs });
+		const checkpointer = this._stateCheckpointer(config);
+		const subgraph = await this._subgraphForNamespace(config, checkpointer);
+		if (subgraph !== void 0) {
+			const [pregel, subgraphConfig] = subgraph;
+			return await pregel.getState(subgraphConfig, { subgraphs: options?.subgraphs });
 		}
-		const mergedConfig = (0, _langchain_core_runnables.mergeConfigs)(this.config, config);
-		const saved = await checkpointer.getTuple(config);
+		const ownConfig = this._ownCheckpointConfig(config);
+		const mergedConfig = (0, _langchain_core_runnables.mergeConfigs)(this.config, ownConfig);
+		const saved = await checkpointer.getTuple(ownConfig);
 		return await this._prepareStateSnapshot({
 			config: mergedConfig,
 			saved,
-			subgraphCheckpointer: options?.subgraphs ? checkpointer : void 0,
+			saver: checkpointer,
+			recurse: options?.subgraphs,
 			applyPendingWrites: !config.configurable?.checkpoint_id
 		});
 	}
@@ -551,20 +598,18 @@ var Pregel = class extends PartialRunnable {
 	* @throws {Error} If no checkpointer is configured
 	*/
 	async *getStateHistory(config, options) {
-		const checkpointer = config.configurable?.["__pregel_checkpointer"] ?? this.checkpointer;
-		if (!checkpointer) throw new require_errors.GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
-		const checkpointNamespace = config.configurable?.checkpoint_ns ?? "";
-		if (checkpointNamespace !== "" && config.configurable?.["__pregel_checkpointer"] === void 0) {
-			const recastNamespace = require_config.recastCheckpointNamespace(checkpointNamespace);
-			for await (const [name, pregel] of this.getSubgraphsAsync(recastNamespace, true)) if (name === recastNamespace) {
-				yield* pregel.getStateHistory(require_utils.patchConfigurable(config, { [require_constants.CONFIG_KEY_CHECKPOINTER]: checkpointer }), options);
-				return;
-			}
+		const checkpointer = this._stateCheckpointer(config);
+		const subgraph = await this._subgraphForNamespace(config, checkpointer);
+		if (subgraph !== void 0) {
+			const [pregel, subgraphConfig] = subgraph;
+			yield* pregel.getStateHistory(subgraphConfig, options);
+			return;
 		}
-		const mergedConfig = (0, _langchain_core_runnables.mergeConfigs)(this.config, config, { configurable: { checkpoint_ns: checkpointNamespace } });
+		const mergedConfig = this._ownCheckpointConfig((0, _langchain_core_runnables.mergeConfigs)(this.config, config, { configurable: { checkpoint_ns: config.configurable?.checkpoint_ns ?? "" } }));
 		for await (const checkpointTuple of checkpointer.list(mergedConfig, options)) yield this._prepareStateSnapshot({
 			config: checkpointTuple.config,
-			saved: checkpointTuple
+			saved: checkpointTuple,
+			saver: checkpointer
 		});
 	}
 	/**
@@ -584,18 +629,18 @@ var Pregel = class extends PartialRunnable {
 	* @throws {InvalidUpdateError} If the update cannot be attributed to a node or an update can be only applied in sequence.
 	*/
 	async bulkUpdateState(startConfig, supersteps) {
-		const checkpointer = startConfig.configurable?.["__pregel_checkpointer"] ?? this.checkpointer;
-		if (!checkpointer) throw new require_errors.GraphValueError("No checkpointer set", { lc_error_code: "MISSING_CHECKPOINTER" });
+		const checkpointer = this._stateCheckpointer(startConfig);
 		if (supersteps.length === 0) throw new Error("No supersteps provided");
 		if (supersteps.some((s) => s.updates.length === 0)) throw new Error("No updates provided");
-		const checkpointNamespace = startConfig.configurable?.checkpoint_ns ?? "";
-		if (checkpointNamespace !== "" && startConfig.configurable?.["__pregel_checkpointer"] === void 0) {
-			const recastNamespace = require_config.recastCheckpointNamespace(checkpointNamespace);
-			for await (const [, pregel] of this.getSubgraphsAsync(recastNamespace, true)) return await pregel.bulkUpdateState(require_utils.patchConfigurable(startConfig, { [require_constants.CONFIG_KEY_CHECKPOINTER]: checkpointer }), supersteps);
-			throw new Error(`Subgraph "${recastNamespace}" not found`);
+		const subgraph = await this._subgraphForNamespace(startConfig, checkpointer);
+		if (subgraph !== void 0) {
+			const [pregel, subgraphConfig] = subgraph;
+			return await pregel.bulkUpdateState(subgraphConfig, supersteps);
 		}
+		const checkpointNamespace = startConfig.configurable?.checkpoint_ns ?? "";
+		if (checkpointNamespace !== "" && startConfig.configurable?.["__pregel_read"] === void 0 && startConfig.configurable?.["__pregel_checkpointer"] === void 0) throw new Error(`Subgraph "${require_config.recastCheckpointNamespace(checkpointNamespace)}" not found`);
 		const updateSuperStep = async (inputConfig, updates) => {
-			const config = this.config ? (0, _langchain_core_runnables.mergeConfigs)(this.config, inputConfig) : inputConfig;
+			const config = this._ownCheckpointConfig(this.config ? (0, _langchain_core_runnables.mergeConfigs)(this.config, inputConfig) : inputConfig);
 			const saved = await checkpointer.getTuple(config);
 			const checkpoint = saved !== void 0 ? (0, _langchain_langgraph_checkpoint.copyCheckpoint)(saved.checkpoint) : (0, _langchain_langgraph_checkpoint.emptyCheckpoint)();
 			const checkpointPreviousVersions = { ...saved?.checkpoint.channel_versions };
@@ -609,18 +654,22 @@ var Pregel = class extends PartialRunnable {
 					...checkpointMetadata
 				};
 			}
+			const deltaWritesVersioned = Object.values(this.channels).some(require_base.isDeltaChannel) && (saved === void 0 || require_base.isDeltaWritesVersioned(saved.metadata));
+			const versionedMetadata = deltaWritesVersioned ? { [require_base.DELTA_WRITES_VERSIONED]: true } : {};
 			const { values, asNode } = updates[0];
 			if (values == null && asNode === void 0) {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot create empty checkpoint with multiple updates`);
 				return require_index.patchCheckpointMap(await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, void 0, step), {
 					source: "update",
 					step: step + 1,
-					parents: saved?.metadata?.parents ?? {}
+					parents: saved?.metadata?.parents ?? {},
+					...versionedMetadata
 				}, {}), saved ? saved.metadata : void 0);
 			}
 			const channels = await require_base.channelsFromCheckpoint(this.channels, checkpoint, {
 				saver: checkpointer,
-				config: saved?.config ?? checkpointConfig
+				config: saved?.config ?? checkpointConfig,
+				deltaWritesVersioned
 			});
 			if (values === null && asNode === "__end__") {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot apply multiple updates when clearing state`);
@@ -647,11 +696,13 @@ var Pregel = class extends PartialRunnable {
 					}
 					require_algo._applyWrites(checkpoint, channels, Object.values(nextTasks), checkpointer.getNextVersion.bind(checkpointer), this.triggerToNodes);
 				}
+				const { [require_base.DELTA_WRITES_VERSIONED]: _carried, ...endMetadata } = checkpointMetadata;
 				return require_index.patchCheckpointMap(await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, channels, step), {
-					...checkpointMetadata,
+					...endMetadata,
 					source: "update",
 					step: step + 1,
-					parents: saved?.metadata?.parents ?? {}
+					parents: saved?.metadata?.parents ?? {},
+					...versionedMetadata
 				}, require_index.getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions)), saved ? saved.metadata : void 0);
 			}
 			if (asNode === "__copy__") {
@@ -666,7 +717,8 @@ var Pregel = class extends PartialRunnable {
 				const nextConfig = await checkpointer.put(saved.parentConfig ?? require_utils.patchConfigurable(saved.config, { checkpoint_id: void 0 }), nextCheckpoint, {
 					source: "fork",
 					step: step + 1,
-					parents: saved.metadata?.parents ?? {}
+					parents: saved.metadata?.parents ?? {},
+					...versionedMetadata
 				}, {});
 				if (isCopyWithUpdates(values)) {
 					const nextTasks = require_algo._prepareNextTasks(nextCheckpoint, saved.pendingWrites, this.nodes, channels, nextConfig, false, { step: step + 2 });
@@ -699,12 +751,13 @@ var Pregel = class extends PartialRunnable {
 					name: require_constants.INPUT,
 					writes: inputWrites,
 					triggers: []
-				}], checkpointer.getNextVersion.bind(this.checkpointer), this.triggerToNodes);
+				}], checkpointer.getNextVersion.bind(checkpointer), this.triggerToNodes);
 				const nextStep = saved?.metadata?.step != null ? saved.metadata.step + 1 : -1;
 				const nextConfig = await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, channels, nextStep), {
 					source: "input",
 					step: nextStep,
-					parents: saved?.metadata?.parents ?? {}
+					parents: saved?.metadata?.parents ?? {},
+					...versionedMetadata
 				}, require_index.getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions));
 				await checkpointer.putWrites(nextConfig, inputWrites, (0, _langchain_langgraph_checkpoint.uuid5)(require_constants.INPUT, checkpoint.id));
 				return require_index.patchCheckpointMap(nextConfig, saved ? saved.metadata : void 0);
@@ -716,11 +769,11 @@ var Pregel = class extends PartialRunnable {
 					step: (saved.metadata?.step ?? -1) + 1
 				});
 				const nullWrites = (saved.pendingWrites ?? []).filter((w) => w[0] === require_constants.NULL_TASK_ID).map((w) => w.slice(1));
-				if (nullWrites.length > 0) require_algo._applyWrites(saved.checkpoint, channels, [{
+				if (nullWrites.length > 0) require_algo._applyWrites(checkpoint, channels, [{
 					name: require_constants.INPUT,
 					writes: nullWrites,
 					triggers: []
-				}], void 0, this.triggerToNodes);
+				}], checkpointer.getNextVersion.bind(checkpointer), this.triggerToNodes);
 				for (const [tid, k, v] of saved.pendingWrites) {
 					if ([
 						"__error__",
@@ -732,7 +785,7 @@ var Pregel = class extends PartialRunnable {
 				const tasks = Object.values(nextTasks).filter((task) => {
 					return task.writes.length > 0;
 				});
-				if (tasks.length > 0) require_algo._applyWrites(checkpoint, channels, tasks, void 0, this.triggerToNodes);
+				if (tasks.length > 0) require_algo._applyWrites(checkpoint, channels, tasks, checkpointer.getNextVersion.bind(checkpointer), this.triggerToNodes);
 			}
 			const nonNullVersion = Object.values(checkpoint.versions_seen).map((seenVersions) => {
 				return Object.values(seenVersions);
@@ -769,7 +822,7 @@ var Pregel = class extends PartialRunnable {
 				});
 			}
 			const tasks = [];
-			for (const { asNode, values, taskId } of validUpdates) {
+			for (const [i, { asNode, values, taskId }] of validUpdates.entries()) {
 				if (this.nodes[asNode] === void 0) throw new require_errors.InvalidUpdateError(`Node "${asNode.toString()}" does not exist`);
 				const writers = this.nodes[asNode].getWriters();
 				if (!writers.length) throw new require_errors.InvalidUpdateError(`No writers found for node "${asNode.toString()}"`);
@@ -779,7 +832,7 @@ var Pregel = class extends PartialRunnable {
 					proc: writers.length > 1 ? _langchain_core_runnables.RunnableSequence.from(writers, { omitSequenceTags: true }) : writers[0],
 					writes: [],
 					triggers: [require_constants.INTERRUPT],
-					id: taskId ?? (0, _langchain_langgraph_checkpoint.uuid5)("__interrupt__", checkpoint.id),
+					id: taskId ?? (0, _langchain_langgraph_checkpoint.uuid5)(i === 0 ? "__interrupt__" : `__interrupt__:${i}`, checkpoint.id),
 					writers: []
 				});
 			}
@@ -797,12 +850,17 @@ var Pregel = class extends PartialRunnable {
 				const channelWrites = task.writes.filter((w) => w[0] !== require_constants.PUSH);
 				if (saved !== void 0 && channelWrites.length > 0) await checkpointer.putWrites(checkpointConfig, channelWrites, task.id);
 			}
-			require_algo._applyWrites(checkpoint, channels, tasks, checkpointer.getNextVersion.bind(this.checkpointer), this.triggerToNodes);
+			require_algo._applyWrites(checkpoint, channels, tasks, checkpointer.getNextVersion.bind(checkpointer), this.triggerToNodes);
+			const channelsToSnapshot = /* @__PURE__ */ new Set();
+			if (saved === void 0) {
+				for (const [k, ch] of Object.entries(channels)) if (require_base.isDeltaChannel(ch) && checkpoint.channel_versions[k] !== void 0) channelsToSnapshot.add(k);
+			}
 			const newVersions = require_index.getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions);
-			const nextConfig = await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, channels, step + 1), {
+			const nextConfig = await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, channels, step + 1, { channelsToSnapshot }), {
 				source: "update",
 				step: step + 1,
-				parents: saved?.metadata?.parents ?? {}
+				parents: saved?.metadata?.parents ?? {},
+				...versionedMetadata
 			}, newVersions);
 			for (const task of tasks) {
 				const pushWrites = task.writes.filter((w) => w[0] === require_constants.PUSH);
@@ -877,11 +935,7 @@ var Pregel = class extends PartialRunnable {
 			else defaultStreamMode = this.streamMode;
 			streamModeSingle = true;
 		}
-		let defaultCheckpointer;
-		if (this.checkpointer === false) defaultCheckpointer = void 0;
-		else if (config !== void 0 && config.configurable?.["__pregel_checkpointer"] !== void 0) defaultCheckpointer = config.configurable[require_constants.CONFIG_KEY_CHECKPOINTER];
-		else if (this.checkpointer === true) throw new Error("checkpointer: true cannot be used for root graphs.");
-		else defaultCheckpointer = this.checkpointer;
+		const defaultCheckpointer = this._resolveCheckpointer(config);
 		const defaultStore = config.store ?? this.store;
 		const defaultCache = config.cache ?? this.cache;
 		if (config.durability != null && config.checkpointDuring != null) throw new Error("Cannot use both `durability` and `checkpointDuring` at the same time.");
@@ -1021,11 +1075,7 @@ var Pregel = class extends PartialRunnable {
 		if (typeof config.context !== "undefined") config.context = await this._validateContext(config.context);
 		else config.configurable = await this._validateContext(config.configurable);
 		const stream = new require_stream.IterableReadableWritableStream({ modes: new Set(streamMode) });
-		if (this.checkpointer === true) {
-			config.configurable ??= {};
-			const ns = config.configurable["checkpoint_ns"] ?? "";
-			config.configurable[require_constants.CONFIG_KEY_CHECKPOINT_NS] = ns.split("|").map((part) => part.split(":")[0]).join("|");
-		}
+		config.configurable = this._ownCheckpointConfig(config).configurable;
 		if (streamMode.includes("messages")) {
 			const messageStreamer = isV3 ? new require_messages_v2.StreamProtocolMessagesHandler((chunk) => stream.push(chunk)) : new require_messages.StreamMessagesHandler((chunk) => stream.push(chunk));
 			const { callbacks } = config;