@langchain/langgraph 1.5.0 → 1.5.1

package/dist/pregel/loop.js

@@ -233,9 +233,11 @@ var PregelLoop = class PregelLoop {
 	_resumeLoadedWrites = [];
 	/** Tasks a resume handed loaded writes back to. */
 	_reappliedTaskIds = /* @__PURE__ */ new Set();
+	/** Tasks that wrote during this run, the only ones exit mode saves at exit. */
+	_writtenTaskIds = /* @__PURE__ */ new Set();
 	/**
-	* An update or fork checkpoint addressed by `checkpoint_id` that the thread
-	* has moved past, so it may already have children.
+	* A checkpoint addressed by `checkpoint_id` that the thread has moved past,
+	* so it may already have children.
 	*/
 	addressedCheckpointSuperseded;
 	/** Whether a real checkpoint was loaded from the saver at initialization. */
@@ -264,7 +266,9 @@ var PregelLoop = class PregelLoop {
 	outputKeys;
 	streamKeys;
 	nodes;
-	skipDoneTasks;
+	/** Whether the run points this loop at a checkpoint. */
+	addressed;
+	skipDoneTasks = true;
 	prevCheckpointConfig;
 	updatedChannels;
 	status = "pending";
@@ -365,7 +369,7 @@ var PregelLoop = class PregelLoop {
 		this.outputKeys = params.outputKeys;
 		this.streamKeys = params.streamKeys;
 		this.nodes = params.nodes;
-		this.skipDoneTasks = params.skipDoneTasks;
+		this.addressed = params.addressed;
 		this.store = params.store;
 		this.cache = params.cache ? new AsyncBatchedCache(params.cache) : void 0;
 		this.stream = params.stream;
@@ -387,7 +391,6 @@ var PregelLoop = class PregelLoop {
 	static async initialize(params) {
 		let { config, stream } = params;
 		if (stream !== void 0 && config.configurable?.["__pregel_stream"] !== void 0) stream = createDuplexStream(stream, config.configurable[CONFIG_KEY_STREAM]);
-		const skipDoneTasks = config.configurable ? !("checkpoint_id" in config.configurable) : true;
 		const scratchpad = config.configurable?.[CONFIG_KEY_SCRATCHPAD];
 		if (config.configurable && scratchpad) {
 			if (scratchpad.subgraphCounter > 0) config = patchConfigurable(config, { [CONFIG_KEY_CHECKPOINT_NS]: [config.configurable[CONFIG_KEY_CHECKPOINT_NS], scratchpad.subgraphCounter.toString()].join("|") });
@@ -400,6 +403,7 @@ var PregelLoop = class PregelLoop {
 		});
 		let checkpointConfig = config;
 		if (config.configurable?.checkpoint_id === void 0 && config.configurable?.["checkpoint_map"] !== void 0 && config.configurable?.["checkpoint_map"]?.[config.configurable?.checkpoint_ns]) checkpointConfig = patchConfigurable(config, { checkpoint_id: config.configurable[CONFIG_KEY_CHECKPOINT_MAP][config.configurable?.checkpoint_ns] });
+		const addressed = Boolean(checkpointConfig.configurable?.["checkpoint_id"] || config.configurable?.["__pregel_replay_state"]);
 		const checkpointNamespace = checkpointNamespaceFromNs(config.configurable?.checkpoint_ns);
 		let saved;
 		let loadedLatest = false;
@@ -446,7 +450,7 @@ var PregelLoop = class PregelLoop {
 			const superseded = await checkpointSuperseded(params.checkpointer, addressedConfig, saved);
 			const isUpdateOrFork = checkpointMetadata.source === "update" || checkpointMetadata.source === "fork";
 			resumeAtHead = addressedConfig.configurable?.checkpoint_id !== void 0 && !superseded && !isUpdateOrFork;
-			addressedCheckpointSuperseded = superseded && isUpdateOrFork;
+			addressedCheckpointSuperseded = superseded;
 		}
 		const deltaWritesVersioned = Object.values(params.channelSpecs).some(isDeltaChannel) && (!hasPersistedParent || isDeltaWritesVersioned(saved.metadata));
 		const channels = await channelsFromCheckpoint(params.channelSpecs, checkpoint, {
@@ -474,7 +478,7 @@ var PregelLoop = class PregelLoop {
 			addressedCheckpointSuperseded,
 			manager: params.manager,
 			hasGraphCallbacks: params.hasGraphCallbacks,
-			skipDoneTasks,
+			addressed,
 			step,
 			stop,
 			checkpointPreviousVersions,
@@ -513,6 +517,7 @@ var PregelLoop = class PregelLoop {
 	putWrites(taskId, writes, cached = false) {
 		let writesCopy = writes;
 		if (writesCopy.length === 0) return;
+		this._writtenTaskIds.add(taskId);
 		const deltaWrites = ensureDeltaMessageIds(this.channels, writes);
 		if (writesCopy.every(([key]) => key in WRITES_IDX_MAP)) writesCopy = Array.from(new Map(writesCopy.map((w) => [w[0], w])).values());
 		let hasUntrackedChannels = false;
@@ -835,7 +840,8 @@ var PregelLoop = class PregelLoop {
 	async _first(inputKeys) {
 		const { configurable } = this.config;
 		const inputIsCommand = isCommand(this.input);
-		const isTimeTraveling = this.isReplaying && (this.isNested && configurable?.["checkpoint_ns"] !== void 0 && configurable?.["checkpoint_ns"] !== "" && configurable?.["checkpoint_map"] !== void 0 && configurable["checkpoint_ns"] in configurable["checkpoint_map"] || !(inputIsCommand && this.input.resume != null || configurable?.["__pregel_resuming"] === true || this.resumeAtHead));
+		const isTimeTraveling = this.addressed && (this.isNested && configurable?.["checkpoint_ns"] !== void 0 && configurable?.["checkpoint_ns"] !== "" && configurable?.["checkpoint_map"] !== void 0 && configurable["checkpoint_ns"] in configurable["checkpoint_map"] || !(inputIsCommand && this.input.resume != null || configurable?.["__pregel_resuming"] === true || this.resumeAtHead));
+		this.skipDoneTasks = !(isTimeTraveling || this.addressedCheckpointSuperseded);
 		const isUpdateOrFork = this.checkpointMetadata.source === "update" || this.checkpointMetadata.source === "fork";
 		const forks = isTimeTraveling && (inputIsCommand || this.isResuming) && (!isUpdateOrFork || this.addressedCheckpointSuperseded);
 		const carried = [];
@@ -1162,6 +1168,7 @@ var PregelLoop = class PregelLoop {
 		});
 		const byTask = {};
 		for (const [tid, key, value] of this.checkpointPendingWrites) {
+			if (!this._writtenTaskIds.has(tid)) continue;
 			byTask[tid] ??= [];
 			byTask[tid].push([key, value]);
 		}