@langchain/langgraph-sdk 1.10.3-rc.2 → 1.11.0

package/dist/stream/controller.js

@@ -1217,7 +1217,10 @@ var StreamController = class {
 	*   still required.
 	*/
 	#ensureThread(threadId, deferRootPump = false) {
-		if (this.#thread != null) return this.#thread;
+		if (this.#thread != null) {
+			if (!deferRootPump && !this.#rootPumpDeferred && this.#rootPump == null) this.#startRootPump(this.#thread);
+			return this.#thread;
+		}
 		this.#thread = this.#options.client.threads.stream(threadId, {
 			assistantId: this.#options.assistantId,
 			transport: this.#options.transport,
@@ -1226,7 +1229,8 @@ var StreamController = class {
 			maxReconnectAttempts: this.#options.maxReconnectAttempts,
 			streamIdleReconnect: this.#options.streamIdleReconnect,
 			reconnectDelayMs: this.#options.reconnectDelayMs,
-			onReconnect: this.#options.onReconnect
+			onReconnect: this.#options.onReconnect,
+			onConnected: this.#options.onConnected
 		});
 		this.registry.bind(this.#thread);
 		if (deferRootPump) {
@@ -1369,6 +1373,7 @@ var StreamController = class {
 				error,
 				isLoading: false
 			}));
+			this.#rootSubscription?.close();
 		});
 		/**
 		* Persistent isLoading driver. Drives `isLoading` from
@@ -1380,7 +1385,7 @@ var StreamController = class {
 		*/
 		this.#rootEventListeners.add(this.#lifecycleLoading.listener);
 		this.#rootEventListeners.add(this.#runLifecycleListener);
-		this.#rootPump = (async () => {
+		const pump = (async () => {
 			try {
 				/**
 				* Root content pump: depth 1 is required because the controller
@@ -1471,7 +1476,15 @@ var StreamController = class {
 						} catch {}
 					}
 					if (this.#disposed || pumpGeneration !== this.#rootPumpGeneration) break;
-					if (!subscription.isPaused) break;
+					if (!subscription.isPaused) {
+						const snapshot = this.rootStore.getSnapshot();
+						if (snapshot.isLoading && snapshot.error == null) this.rootStore.setState((s) => ({
+							...s,
+							error: /* @__PURE__ */ new Error("Thread event stream closed before the active run completed."),
+							isLoading: false
+						}));
+						break;
+					}
 					await subscription.waitForResume();
 				}
 			} catch {
@@ -1479,6 +1492,21 @@ var StreamController = class {
 				resolveReady = void 0;
 			}
 		})();
+		this.#rootPump = pump;
+		pump.finally(() => {
+			if (pumpGeneration !== this.#rootPumpGeneration) return;
+			if (this.#rootPump !== pump) return;
+			this.#rootPumpGeneration += 1;
+			this.#rootSubscription = void 0;
+			this.#rootPump = void 0;
+			this.#rootPumpReady = void 0;
+			this.#threadEventUnsubscribe?.();
+			this.#threadEventUnsubscribe = void 0;
+			this.#threadErrorUnsubscribe?.();
+			this.#threadErrorUnsubscribe = void 0;
+			this.#rootEventListeners.delete(this.#lifecycleLoading.listener);
+			this.#rootEventListeners.delete(this.#runLifecycleListener);
+		});
 	}
 	/**
 	* Handle an event delivered via {@link ThreadStream.onEvent}.