@langchain/langgraph-sdk 1.9.19 → 1.9.21

package/dist/stream/controller.js

@@ -344,6 +344,17 @@ var StreamController = class {
 		this.#hydrationPromise = this.#createHydrationPromise();
 	}
 	/**
+	* Load thread state for hydration, preferring the active custom
+	* adapter's `getState()` when present.
+	*/
+	async #fetchHydrationState() {
+		const threadId = this.#currentThreadId;
+		if (threadId == null) return null;
+		const transport = this.#options.transport;
+		if (transport != null && typeof transport === "object" && typeof transport.getState === "function") return await transport.getState();
+		return this.#options.client.threads.getState(threadId);
+	}
+	/**
 	* Fetch the checkpointed thread state and seed the root snapshot.
 	* Re-calling with a different `threadId` swaps the underlying
 	* {@link ThreadStream}, rewires the registry to the new thread, and
@@ -416,7 +427,7 @@ var StreamController = class {
 		let threadExists = false;
 		let threadActive = true;
 		try {
-			const state = await this.#options.client.threads.getState(this.#currentThreadId);
+			const state = await this.#fetchHydrationState();
 			threadExists = state != null;
 			threadActive = isThreadStateActive(state);
 			if (state?.values != null) {
@@ -454,7 +465,25 @@ var StreamController = class {
 				* idempotent, so this is safe on re-hydrate.
 				*/
 				const seedMessages = state.values[this.#messagesKey];
-				if (Array.isArray(seedMessages)) this.#subagents.seedFromCheckpointMessages(seedMessages);
+				if (Array.isArray(seedMessages)) {
+					this.#subagents.seedFromCheckpointMessages(seedMessages);
+					/**
+					* An idle (finished) thread defers its root SSE pump; the first
+					* `submit()` brings it up and the transport replays the finished
+					* run from `seq=0`. Seal the seeded message ids so that replay's
+					* `messages` channel deltas can't downgrade the already-complete
+					* tail to empty partials (a visible "messages replay"). Only safe
+					* for idle threads, where every seeded message is final — an
+					* active thread's tail may still be streaming and must keep
+					* receiving deltas. New ids from the next run are never sealed,
+					* and the seal lifts once a newer checkpoint advances the
+					* timeline.
+					*/
+					if (!threadActive) {
+						const sealedIds = seedMessages.map((message) => message?.id).filter((id) => typeof id === "string");
+						if (sealedIds.length > 0) this.#rootMessages.sealMessageIds(sealedIds);
+					}
+				}
 			}
 			/**
 			* Converge to server truth: drop any optimistic messages the