@langchain/langgraph 1.4.20 → 1.5.0

package/dist/pregel/index.cjs

@@ -7,6 +7,7 @@ const require_utils = require("../utils.cjs");
 const require_write = require("./write.cjs");
 const require_read = require("./read.cjs");
 const require_callbacks = require("./callbacks.cjs");
+const require_checkpoint = require("./checkpoint.cjs");
 const require_runtime = require("./runtime.cjs");
 const require_io = require("./io.cjs");
 const require_index = require("./utils/index.cjs");
@@ -639,7 +640,7 @@ var Pregel = class extends PartialRunnable {
 		}
 		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 updateSuperStep = async (inputConfig, updates, isFirstSuperstep) => {
 			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)();
@@ -656,21 +657,46 @@ var Pregel = class extends PartialRunnable {
 			}
 			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 getNextVersion = (current) => checkpointer.getNextVersion(current);
+			const writtenChannels = () => new Set(Object.keys(require_index.getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions)));
+			const forkPending = isFirstSuperstep ? require_base.deltaChannelsWithPendingWrites(this.channels, saved?.pendingWrites) : /* @__PURE__ */ new Set();
+			const editsOlderCheckpoint = async () => isFirstSuperstep && saved !== void 0 && await require_checkpoint.checkpointSuperseded(checkpointer, config, saved);
+			const specs = this.channels;
+			const sealDeltaWrites = (writes) => {
+				for (const [ch] of require_base.deltaChannelWrites(specs, writes)) forkPending.add(String(ch));
+			};
+			const sealOverwrites = (writes) => {
+				for (const [ch, value] of require_base.deltaChannelWrites(specs, writes)) if (require_constants._isOverwriteValue(value)) forkPending.add(String(ch));
+			};
+			const loadChannels = () => require_base.channelsFromCheckpoint(this.channels, checkpoint, {
+				saver: checkpointer,
+				config: saved?.config ?? checkpointConfig,
+				deltaWritesVersioned
+			});
 			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), {
+				const counters = saved === void 0 ? {} : require_checkpoint.advanceDeltaCounters(this.channels, /* @__PURE__ */ new Set(), saved.metadata?.counters_since_delta_snapshot);
+				const channels = forkPending.size > 0 || Object.entries(counters).some(([name, count]) => require_base.reachedSnapshotBound(this.channels[name], count)) ? await loadChannels() : void 0;
+				const { channelsToSnapshot, metadata: countersMetadata } = channels === void 0 ? {
+					channelsToSnapshot: /* @__PURE__ */ new Set(),
+					metadata: require_checkpoint.deltaCountersMetadata(counters)
+				} : require_checkpoint.updateStateDeltaPlan(channels, /* @__PURE__ */ new Set(), saved, checkpoint.channel_versions, deltaWritesVersioned, forkPending);
+				const nextCheckpoint = require_base.createCheckpoint(checkpoint, channels, step, {
+					channelsToSnapshot,
+					storedVersions: checkpointPreviousVersions,
+					getNextVersion,
+					triggerToNodes: this.triggerToNodes
+				});
+				return require_index.patchCheckpointMap(await checkpointer.put(checkpointConfig, nextCheckpoint, {
 					source: "update",
 					step: step + 1,
 					parents: saved?.metadata?.parents ?? {},
-					...versionedMetadata
-				}, {}), saved ? saved.metadata : void 0);
+					...versionedMetadata,
+					...countersMetadata
+				}, require_index.getNewChannelVersions(checkpointPreviousVersions, nextCheckpoint.channel_versions)), saved ? saved.metadata : void 0);
 			}
-			const channels = await require_base.channelsFromCheckpoint(this.channels, checkpoint, {
-				saver: checkpointer,
-				config: saved?.config ?? checkpointConfig,
-				deltaWritesVersioned
-			});
+			const channels = await loadChannels();
 			if (values === null && asNode === "__end__") {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot apply multiple updates when clearing state`);
 				if (saved) {
@@ -696,14 +722,22 @@ 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), {
+				const { channelsToSnapshot, metadata: countersMetadata } = require_checkpoint.updateStateDeltaPlan(channels, writtenChannels(), saved, checkpoint.channel_versions, deltaWritesVersioned, forkPending);
+				const nextCheckpoint = require_base.createCheckpoint(checkpoint, channels, step, {
+					channelsToSnapshot,
+					storedVersions: checkpointPreviousVersions,
+					getNextVersion,
+					triggerToNodes: this.triggerToNodes
+				});
+				const { [require_base.DELTA_WRITES_VERSIONED]: _carried, counters_since_delta_snapshot: _counters, ...endMetadata } = checkpointMetadata;
+				return require_index.patchCheckpointMap(await checkpointer.put(checkpointConfig, nextCheckpoint, {
 					...endMetadata,
 					source: "update",
 					step: step + 1,
 					parents: saved?.metadata?.parents ?? {},
-					...versionedMetadata
-				}, require_index.getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions)), saved ? saved.metadata : void 0);
+					...versionedMetadata,
+					...countersMetadata
+				}, require_index.getNewChannelVersions(checkpointPreviousVersions, nextCheckpoint.channel_versions)), saved ? saved.metadata : void 0);
 			}
 			if (asNode === "__copy__") {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot copy checkpoint with multiple updates`);
@@ -718,7 +752,8 @@ var Pregel = class extends PartialRunnable {
 					source: "fork",
 					step: step + 1,
 					parents: saved.metadata?.parents ?? {},
-					...versionedMetadata
+					...versionedMetadata,
+					...require_checkpoint.deltaCountersMetadata(saved.metadata?.counters_since_delta_snapshot ?? {})
 				}, {});
 				if (isCopyWithUpdates(values)) {
 					const nextTasks = require_algo._prepareNextTasks(nextCheckpoint, saved.pendingWrites, this.nodes, channels, nextConfig, false, { step: step + 2 });
@@ -739,7 +774,7 @@ var Pregel = class extends PartialRunnable {
 						});
 						return acc;
 					}, {});
-					return updateSuperStep(require_index.patchCheckpointMap(nextConfig, saved.metadata), Object.values(userGroupBy).flat());
+					return updateSuperStep(require_index.patchCheckpointMap(nextConfig, saved.metadata), Object.values(userGroupBy).flat(), false);
 				}
 				return require_index.patchCheckpointMap(nextConfig, saved.metadata);
 			}
@@ -747,20 +782,30 @@ var Pregel = class extends PartialRunnable {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot apply multiple updates when updating as input`);
 				const inputWrites = await require_utils.gatherIterator(require_io.mapInput(this.inputChannels, values));
 				if (inputWrites.length === 0) throw new require_errors.InvalidUpdateError(`Received no input writes for ${JSON.stringify(this.inputChannels, null, 2)}`);
+				require_loop.ensureDeltaMessageIds(specs, inputWrites);
+				if (await editsOlderCheckpoint()) sealDeltaWrites(inputWrites);
+				else if (saved !== void 0) await checkpointer.putWrites(checkpointConfig, inputWrites, (0, _langchain_langgraph_checkpoint.uuid5)(require_constants.INPUT, checkpoint.id));
 				require_algo._applyWrites(checkpoint, channels, [{
 					name: require_constants.INPUT,
 					writes: inputWrites,
 					triggers: []
 				}], checkpointer.getNextVersion.bind(checkpointer), this.triggerToNodes);
+				sealOverwrites(inputWrites);
 				const nextStep = saved?.metadata?.step != null ? saved.metadata.step + 1 : -1;
-				const nextConfig = await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, channels, nextStep), {
+				const { channelsToSnapshot, metadata: countersMetadata } = require_checkpoint.updateStateDeltaPlan(channels, writtenChannels(), saved, checkpoint.channel_versions, deltaWritesVersioned, forkPending);
+				const nextCheckpoint = require_base.createCheckpoint(checkpoint, channels, nextStep, {
+					channelsToSnapshot,
+					storedVersions: checkpointPreviousVersions,
+					getNextVersion,
+					triggerToNodes: this.triggerToNodes
+				});
+				return require_index.patchCheckpointMap(await checkpointer.put(checkpointConfig, nextCheckpoint, {
 					source: "input",
 					step: nextStep,
 					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);
+					...versionedMetadata,
+					...countersMetadata
+				}, require_index.getNewChannelVersions(checkpointPreviousVersions, nextCheckpoint.channel_versions)), saved ? saved.metadata : void 0);
 			}
 			if (config.configurable?.checkpoint_id === void 0 && saved?.pendingWrites !== void 0 && saved.pendingWrites.length > 0) {
 				const nextTasks = require_algo._prepareNextTasks(checkpoint, saved.pendingWrites, this.nodes, channels, saved.config, true, {
@@ -787,7 +832,8 @@ var Pregel = class extends PartialRunnable {
 				});
 				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) => {
+			const versionsSeen = require_base.versionsSeenWithoutBumps(checkpoint.versions_seen);
+			const nonNullVersion = Object.values(versionsSeen).map((seenVersions) => {
 				return Object.values(seenVersions);
 			}).flat().find((v) => !!v);
 			const validUpdates = [];
@@ -797,7 +843,7 @@ var Pregel = class extends PartialRunnable {
 				else if (asNode === void 0 && nonNullVersion === void 0) {
 					if (typeof this.inputChannels === "string" && this.nodes[this.inputChannels] !== void 0) asNode = this.inputChannels;
 				} else if (asNode === void 0) {
-					const lastSeenByNode = Object.entries(checkpoint.versions_seen).map(([n, seen]) => {
+					const lastSeenByNode = Object.entries(versionsSeen).map(([n, seen]) => {
 						return Object.values(seen).map((v) => {
 							return [v, n];
 						});
@@ -832,6 +878,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],
+					path: [require_constants.INTERRUPT, i],
 					id: taskId ?? (0, _langchain_langgraph_checkpoint.uuid5)(i === 0 ? "__interrupt__" : `__interrupt__:${i}`, checkpoint.id),
 					writers: []
 				});
@@ -846,22 +893,28 @@ var Pregel = class extends PartialRunnable {
 					[require_constants.CONFIG_KEY_READ]: (select_, fresh_ = false) => require_algo._localRead(checkpoint, channels, task, select_, fresh_)
 				}
 			}));
-			for (const task of tasks) {
+			for (const task of tasks) require_loop.ensureDeltaMessageIds(specs, task.writes);
+			if (await editsOlderCheckpoint()) for (const task of tasks) sealDeltaWrites(task.writes);
+			else for (const task of tasks) {
 				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);
+				if (saved !== void 0 && channelWrites.length > 0) await checkpointer.putWrites(checkpointConfig, channelWrites, task.id, require_algo.taskPathStr(task.path));
 			}
 			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, { channelsToSnapshot }), {
+			for (const task of tasks) sealOverwrites(task.writes);
+			const { channelsToSnapshot, metadata: countersMetadata } = require_checkpoint.updateStateDeltaPlan(channels, writtenChannels(), saved, checkpoint.channel_versions, deltaWritesVersioned, forkPending);
+			const nextCheckpoint = require_base.createCheckpoint(checkpoint, channels, step + 1, {
+				channelsToSnapshot,
+				storedVersions: checkpointPreviousVersions,
+				getNextVersion,
+				triggerToNodes: this.triggerToNodes
+			});
+			const nextConfig = await checkpointer.put(checkpointConfig, nextCheckpoint, {
 				source: "update",
 				step: step + 1,
 				parents: saved?.metadata?.parents ?? {},
-				...versionedMetadata
-			}, newVersions);
+				...versionedMetadata,
+				...countersMetadata
+			}, require_index.getNewChannelVersions(checkpointPreviousVersions, nextCheckpoint.channel_versions));
 			for (const task of tasks) {
 				const pushWrites = task.writes.filter((w) => w[0] === require_constants.PUSH);
 				if (pushWrites.length > 0) await checkpointer.putWrites(nextConfig, pushWrites, task.id);
@@ -869,7 +922,7 @@ var Pregel = class extends PartialRunnable {
 			return require_index.patchCheckpointMap(nextConfig, saved ? saved.metadata : void 0);
 		};
 		let currentConfig = startConfig;
-		for (const { updates } of supersteps) currentConfig = await updateSuperStep(currentConfig, updates);
+		for (const [i, { updates }] of supersteps.entries()) currentConfig = await updateSuperStep(currentConfig, updates, i === 0);
 		return currentConfig;
 	}
 	/**
@@ -1258,7 +1311,7 @@ var Pregel = class extends PartialRunnable {
 		try {
 			while (await loop.tick({ inputKeys: this.inputChannels })) {
 				if (emitLifecycleEvents) await emitLifecycleEvents(loop.lifecycleEvents);
-				for (const { task } of await loop._matchCachedWrites()) loop._outputWrites(task.id, task.writes, true);
+				for (const { task } of await loop._matchCachedWrites()) loop.putWrites(task.id, task.writes, true);
 				if (debug) require_debug.printStepCheckpoint(loop.checkpointMetadata.step, loop.channels, this.streamChannelsList);
 				if (debug) require_debug.printStepTasks(loop.step, Object.values(loop.tasks));
 				await runner.tick({