@langchain/langgraph-sdk 1.12.6 → 1.12.7

package/dist/stream/controller.cjs

@@ -186,6 +186,21 @@ var StreamController = class {
 	* from the newly-started run and must be accepted.
 	*/
 	#interruptReplayThroughSeq;
+	/**
+	* Command whose barrier is still to be found on the thread stream. The
+	* agent server answers HTTP commands from a fresh per-request manager,
+	* so `applied_through_seq` is 0 while the stream replays the whole
+	* history with positive seqs. The barrier is then the event before the
+	* `running` event of the run the command started: `run.start` names that
+	* run, and a resume starts the first root run after the answered
+	* interrupts were last requested. Unknown interrupts stay buffered until
+	* it is found.
+	*/
+	#awaitingRunBoundary;
+	/** Seq of each root run's `running` event, keyed by run id. */
+	#rootRunStartSeqs = /* @__PURE__ */ new Map();
+	/** Highest seq at which each interrupt was requested. */
+	#interruptRequestSeqs = /* @__PURE__ */ new Map();
 	/** Unknown inputs waiting to be classified as live or replayed history. */
 	#pendingHydrationInterruptEvents = [];
 	#settleParkedInterruptsPromise;
@@ -351,6 +366,7 @@ var StreamController = class {
 			awaitResumedRunTerminal: (signal) => this.#awaitResumedRunTerminal(signal),
 			onSubmitStart: () => {
 				this.#submitGeneration += 1;
+				this.#awaitingRunBoundary = void 0;
 				this.#interruptCommandPending = this.#hydratedActiveInterruptIds != null;
 				this.#pendingInterruptEvents = this.#pendingHydrationInterruptEvents;
 				this.#pendingHydrationInterruptEvents = [];
@@ -362,11 +378,12 @@ var StreamController = class {
 				for (const release of [...this.#heldLifecycleReleases]) release();
 			},
 			onRunCreated: (runId) => {
-				this.#advanceInterruptReplayBarrier();
+				this.#advanceInterruptReplayBarrier({ runId });
 				this.#notifyCreated(runId);
 			},
 			onRunCompleted: (reason, runId) => this.#notifyCompleted(reason, runId),
 			onRunEnd: () => {
+				this.#abandonRunBoundary();
 				this.#interruptCommandPending = false;
 				this.#markLocalRunEnd();
 			},
@@ -1100,7 +1117,7 @@ var StreamController = class {
 					config: options?.config,
 					metadata: options?.metadata
 				});
-				this.#advanceInterruptReplayBarrier();
+				this.#advanceInterruptReplayBarrier({ answeredInterruptIds: [resolved.interruptId] });
 				this.#markInterruptResolvedInRootStore(resolved.interruptId);
 			}, prepared?.handle);
 		} catch (error) {
@@ -1171,7 +1188,7 @@ var StreamController = class {
 					config: options?.config,
 					metadata: options?.metadata
 				});
-				this.#advanceInterruptReplayBarrier();
+				this.#advanceInterruptReplayBarrier({ answeredInterruptIds: responses.map((r) => r.interrupt_id) });
 				for (const { interrupt_id: interruptId } of responses) this.#markInterruptResolvedInRootStore(interruptId);
 			}, prepared?.handle);
 		} catch (error) {
@@ -1391,6 +1408,9 @@ var StreamController = class {
 		this.#hydratedActiveInterruptIds = null;
 		this.#hydratedCheckpointTime = void 0;
 		this.#interruptReplayThroughSeq = void 0;
+		this.#awaitingRunBoundary = void 0;
+		this.#rootRunStartSeqs.clear();
+		this.#interruptRequestSeqs.clear();
 		this.#pendingHydrationInterruptEvents = [];
 		this.#discardPendingInterruptEvents();
 		this.queueStore.setState(() => require_queue_adapter.EMPTY_QUEUE);
@@ -1607,10 +1627,70 @@ var StreamController = class {
 		* consumer. `ThreadStream.interrupts` is updated in parallel for
 		* resume targeting.
 		*/
+		this.#observeRunBoundary(event);
 		this.#recordInterrupt(event);
 		this.#settleParkedInterruptsOnTerminal(event);
 	}
 	/**
+	* Remember where each root run starts and where each interrupt was last
+	* requested, then try to place a pending command barrier. Both the
+	* content pump and the lifecycle watcher replay the same tape with the
+	* same seqs, so recording either copy is enough.
+	*/
+	#observeRunBoundary(event) {
+		const { seq } = event;
+		if (typeof seq !== "number") return;
+		if (event.method === "input.requested") {
+			const id = event.params.data?.interrupt_id;
+			if (typeof id !== "string") return;
+			if (seq <= (this.#interruptRequestSeqs.get(id) ?? -1)) return;
+			this.#interruptRequestSeqs.set(id, seq);
+		} else if (event.method === "lifecycle" && require_namespace.isRootNamespace(event.params.namespace) && event.params.data?.event === "running") {
+			const runId = synthesizedEventRunId(event.event_id);
+			if (runId == null || this.#rootRunStartSeqs.has(runId)) return;
+			this.#rootRunStartSeqs.set(runId, seq);
+		} else return;
+		this.#resolveRunBoundary();
+	}
+	/**
+	* Place the barrier of a command that reported `applied_through_seq: 0`
+	* once its run's `running` event has been seen, then classify the
+	* interrupts buffered while it was unknown.
+	*/
+	#resolveRunBoundary() {
+		const command = this.#awaitingRunBoundary;
+		if (command == null) return;
+		let runStartSeq;
+		if ("runId" in command) runStartSeq = this.#rootRunStartSeqs.get(command.runId);
+		else {
+			let lastRequested = -1;
+			for (const id of command.answeredInterruptIds) {
+				const seq = this.#interruptRequestSeqs.get(id);
+				if (seq == null) return;
+				lastRequested = Math.max(lastRequested, seq);
+			}
+			for (const seq of this.#rootRunStartSeqs.values()) if (seq > lastRequested && (runStartSeq == null || seq < runStartSeq)) runStartSeq = seq;
+		}
+		if (runStartSeq == null) return;
+		this.#awaitingRunBoundary = void 0;
+		this.#interruptReplayThroughSeq = runStartSeq - 1;
+		this.#interruptCommandPending = false;
+		this.#flushPendingInterruptEvents();
+	}
+	/**
+	* Give up on a barrier whose run never showed up on the stream: a run
+	* cancelled before it started, or a resume whose answered interrupts
+	* were trimmed from the tape. Stop buffering and let the parked
+	* interrupt state check decide what was held.
+	*/
+	#abandonRunBoundary() {
+		if (this.#awaitingRunBoundary == null) return;
+		this.#awaitingRunBoundary = void 0;
+		this.#interruptCommandPending = false;
+		this.#flushPendingInterruptEvents();
+		this.#settleParkedInterrupts();
+	}
+	/**
 	* Process one root-pump event and update all root projections.
 	*
 	* @param event - Event yielded by the root subscription.
@@ -1902,24 +1982,38 @@ var StreamController = class {
 	* Convert the hydrate-time allowlist into a replay barrier after a command
 	* is accepted. Built-in transports update `lastAppliedThroughSeq` from the
 	* command response before resolving `submitRun` / `respondInput`.
+	*
+	* @param command - The run the command started, used to find the barrier
+	*   on the thread stream when the response reports
+	*   `applied_through_seq: 0`.
 	*/
-	#advanceInterruptReplayBarrier() {
+	#advanceInterruptReplayBarrier(command) {
 		if (this.#hydratedActiveInterruptIds == null) {
 			this.#discardPendingInterruptEvents();
 			return;
 		}
 		const throughSeq = this.#thread?.ordering?.lastAppliedThroughSeq;
+		if (throughSeq === 0 && command != null) {
+			this.#awaitingRunBoundary = command;
+			this.#resolveRunBoundary();
+			return;
+		}
 		if (typeof throughSeq === "number") this.#interruptReplayThroughSeq = throughSeq;
 		else {
 			this.#hydratedActiveInterruptIds = null;
 			this.#interruptReplayThroughSeq = void 0;
 		}
 		this.#interruptCommandPending = false;
+		this.#flushPendingInterruptEvents();
+	}
+	/** Classify the interrupts buffered while a command was in flight. */
+	#flushPendingInterruptEvents() {
 		const pendingEvents = this.#pendingInterruptEvents;
 		this.#pendingInterruptEvents = [];
 		for (const event of pendingEvents) this.#recordInterrupt(event);
 	}
 	#discardPendingInterruptEvents() {
+		this.#awaitingRunBoundary = void 0;
 		this.#interruptCommandPending = false;
 		this.#pendingInterruptEvents = [];
 	}
@@ -2069,7 +2163,9 @@ var StreamController = class {
 	* `stream.interrupts` empty while the checkpoint still has siblings.
 	*/
 	async #awaitResumedRunTerminal(signal) {
+		const generation = this.#submitGeneration;
 		const terminal = await this.#awaitRootTerminal(signal, { skipInterruptedUntilRunning: true });
+		if (this.#submitGeneration === generation) this.#abandonRunBoundary();
 		if (!signal.aborted && !this.#disposed && (terminal.event === "interrupted" || terminal.event === "completed")) await this.#reconcilePendingInterruptsFromServer();
 		return terminal;
 	}