@langchain/langgraph 1.1.0 → 1.1.2

package/dist/pregel/index.cjs

@@ -448,11 +448,11 @@ var Pregel = class extends PartialRunnable {
 			let taskNs = `${String(task.name)}${require_constants.CHECKPOINT_NAMESPACE_END}${task.id}`;
 			if (parentNamespace) taskNs = `${parentNamespace}${require_constants.CHECKPOINT_NAMESPACE_SEPARATOR}${taskNs}`;
 			if (subgraphCheckpointer === void 0) {
-				const config$1 = { configurable: {
+				const config = { configurable: {
 					thread_id: saved.config.configurable?.thread_id,
 					checkpoint_ns: taskNs
 				} };
-				taskStates[task.id] = config$1;
+				taskStates[task.id] = config;
 			} else {
 				const subgraphConfig = { configurable: {
 					[require_constants.CONFIG_KEY_CHECKPOINTER]: subgraphCheckpointer,
@@ -474,8 +474,8 @@ var Pregel = class extends PartialRunnable {
 				if (!(taskId in nextTaskById)) continue;
 				nextTaskById[taskId].writes.push([String(channel), value]);
 			}
-			const tasksWithWrites$1 = nextTasks.filter((task) => task.writes.length > 0);
-			if (tasksWithWrites$1.length > 0) require_algo._applyWrites(saved.checkpoint, channels, tasksWithWrites$1, void 0, this.triggerToNodes);
+			const tasksWithWrites = nextTasks.filter((task) => task.writes.length > 0);
+			if (tasksWithWrites.length > 0) require_algo._applyWrites(saved.checkpoint, channels, tasksWithWrites, void 0, this.triggerToNodes);
 		}
 		let metadata = saved?.metadata;
 		if (metadata && saved?.config?.configurable?.thread_id) metadata = {
@@ -638,39 +638,39 @@ var Pregel = class extends PartialRunnable {
 			if (asNode === require_constants.COPY) {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot copy checkpoint with multiple updates`);
 				if (saved == null) throw new require_errors.InvalidUpdateError(`Cannot copy a non-existent checkpoint`);
-				const isCopyWithUpdates = (values$1) => {
-					if (!Array.isArray(values$1)) return false;
-					if (values$1.length === 0) return false;
-					return values$1.every((v) => Array.isArray(v) && v.length === 2);
+				const isCopyWithUpdates = (values) => {
+					if (!Array.isArray(values)) return false;
+					if (values.length === 0) return false;
+					return values.every((v) => Array.isArray(v) && v.length === 2);
 				};
 				const nextCheckpoint = require_base.createCheckpoint(checkpoint, void 0, step);
-				const nextConfig$1 = await checkpointer.put(saved.parentConfig ?? require_utils.patchConfigurable(saved.config, { checkpoint_id: void 0 }), nextCheckpoint, {
+				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 ?? {}
 				}, {});
 				if (isCopyWithUpdates(values)) {
-					const nextTasks = require_algo._prepareNextTasks(nextCheckpoint, saved.pendingWrites, this.nodes, channels, nextConfig$1, false, { step: step + 2 });
+					const nextTasks = require_algo._prepareNextTasks(nextCheckpoint, saved.pendingWrites, this.nodes, channels, nextConfig, false, { step: step + 2 });
 					const tasksGroupBy = Object.values(nextTasks).reduce((acc, { name, id }) => {
 						acc[name] ??= [];
 						acc[name].push({ id });
 						return acc;
 					}, {});
 					const userGroupBy = values.reduce((acc, item) => {
-						const [values$1, asNode$1] = item;
-						acc[asNode$1] ??= [];
-						const targetIdx = acc[asNode$1].length;
-						const taskId = tasksGroupBy[asNode$1]?.[targetIdx]?.id;
-						acc[asNode$1].push({
-							values: values$1,
-							asNode: asNode$1,
+						const [values, asNode] = item;
+						acc[asNode] ??= [];
+						const targetIdx = acc[asNode].length;
+						const taskId = tasksGroupBy[asNode]?.[targetIdx]?.id;
+						acc[asNode].push({
+							values,
+							asNode,
 							taskId
 						});
 						return acc;
 					}, {});
-					return updateSuperStep(require_index.patchCheckpointMap(nextConfig$1, saved.metadata), Object.values(userGroupBy).flat());
+					return updateSuperStep(require_index.patchCheckpointMap(nextConfig, saved.metadata), Object.values(userGroupBy).flat());
 				}
-				return require_index.patchCheckpointMap(nextConfig$1, saved.metadata);
+				return require_index.patchCheckpointMap(nextConfig, saved.metadata);
 			}
 			if (asNode === require_constants.INPUT) {
 				if (updates.length > 1) throw new require_errors.InvalidUpdateError(`Cannot apply multiple updates when updating as input`);
@@ -682,13 +682,13 @@ var Pregel = class extends PartialRunnable {
 					triggers: []
 				}], checkpointer.getNextVersion.bind(this.checkpointer), this.triggerToNodes);
 				const nextStep = saved?.metadata?.step != null ? saved.metadata.step + 1 : -1;
-				const nextConfig$1 = await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, channels, nextStep), {
+				const nextConfig = await checkpointer.put(checkpointConfig, require_base.createCheckpoint(checkpoint, channels, nextStep), {
 					source: "input",
 					step: nextStep,
 					parents: saved?.metadata?.parents ?? {}
 				}, require_index.getNewChannelVersions(checkpointPreviousVersions, checkpoint.channel_versions));
-				await checkpointer.putWrites(nextConfig$1, inputWrites, (0, _langchain_langgraph_checkpoint.uuid5)(require_constants.INPUT, checkpoint.id));
-				return require_index.patchCheckpointMap(nextConfig$1, saved ? saved.metadata : void 0);
+				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);
 			}
 			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, {
@@ -710,53 +710,53 @@ var Pregel = class extends PartialRunnable {
 					].includes(k) || nextTasks[tid] === void 0) continue;
 					nextTasks[tid].writes.push([k, v]);
 				}
-				const tasks$1 = Object.values(nextTasks).filter((task) => {
+				const tasks = Object.values(nextTasks).filter((task) => {
 					return task.writes.length > 0;
 				});
-				if (tasks$1.length > 0) require_algo._applyWrites(checkpoint, channels, tasks$1, void 0, this.triggerToNodes);
+				if (tasks.length > 0) require_algo._applyWrites(checkpoint, channels, tasks, void 0, this.triggerToNodes);
 			}
 			const nonNullVersion = Object.values(checkpoint.versions_seen).map((seenVersions) => {
 				return Object.values(seenVersions);
 			}).flat().find((v) => !!v);
 			const validUpdates = [];
 			if (updates.length === 1) {
-				let { values: values$1, asNode: asNode$1, taskId } = updates[0];
-				if (asNode$1 === void 0 && Object.keys(this.nodes).length === 1) [asNode$1] = Object.keys(this.nodes);
-				else if (asNode$1 === void 0 && nonNullVersion === void 0) {
-					if (typeof this.inputChannels === "string" && this.nodes[this.inputChannels] !== void 0) asNode$1 = this.inputChannels;
-				} else if (asNode$1 === void 0) {
+				let { values, asNode, taskId } = updates[0];
+				if (asNode === void 0 && Object.keys(this.nodes).length === 1) [asNode] = Object.keys(this.nodes);
+				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]) => {
 						return Object.values(seen).map((v) => {
 							return [v, n];
 						});
 					}).flat().filter(([_, v]) => v !== require_constants.INTERRUPT).sort(([aNumber], [bNumber]) => (0, _langchain_langgraph_checkpoint.compareChannelVersions)(aNumber, bNumber));
 					if (lastSeenByNode) {
-						if (lastSeenByNode.length === 1) asNode$1 = lastSeenByNode[0][1];
-						else if (lastSeenByNode[lastSeenByNode.length - 1][0] !== lastSeenByNode[lastSeenByNode.length - 2][0]) asNode$1 = lastSeenByNode[lastSeenByNode.length - 1][1];
+						if (lastSeenByNode.length === 1) asNode = lastSeenByNode[0][1];
+						else if (lastSeenByNode[lastSeenByNode.length - 1][0] !== lastSeenByNode[lastSeenByNode.length - 2][0]) asNode = lastSeenByNode[lastSeenByNode.length - 1][1];
 					}
 				}
-				if (asNode$1 === void 0) throw new require_errors.InvalidUpdateError(`Ambiguous update, specify "asNode"`);
+				if (asNode === void 0) throw new require_errors.InvalidUpdateError(`Ambiguous update, specify "asNode"`);
 				validUpdates.push({
-					values: values$1,
-					asNode: asNode$1,
+					values,
+					asNode,
 					taskId
 				});
-			} else for (const { asNode: asNode$1, values: values$1, taskId } of updates) {
-				if (asNode$1 == null) throw new require_errors.InvalidUpdateError(`"asNode" is required when applying multiple updates`);
+			} else for (const { asNode, values, taskId } of updates) {
+				if (asNode == null) throw new require_errors.InvalidUpdateError(`"asNode" is required when applying multiple updates`);
 				validUpdates.push({
-					values: values$1,
-					asNode: asNode$1,
+					values,
+					asNode,
 					taskId
 				});
 			}
 			const tasks = [];
-			for (const { asNode: asNode$1, values: values$1, taskId } of validUpdates) {
-				if (this.nodes[asNode$1] === void 0) throw new require_errors.InvalidUpdateError(`Node "${asNode$1.toString()}" does not exist`);
-				const writers = this.nodes[asNode$1].getWriters();
-				if (!writers.length) throw new require_errors.InvalidUpdateError(`No writers found for node "${asNode$1.toString()}"`);
+			for (const { asNode, values, taskId } of validUpdates) {
+				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()}"`);
 				tasks.push({
-					name: asNode$1,
-					input: values$1,
+					name: asNode,
+					input: values,
 					proc: writers.length > 1 ? _langchain_core_runnables.RunnableSequence.from(writers, { omitSequenceTags: true }) : writers[0],
 					writes: [],
 					triggers: [require_constants.INTERRUPT],