@tanstack/ai-client 0.30.0 → 0.31.1

package/dist/esm/connection-adapters.js

@@ -474,6 +474,7 @@ function normalizeConnectionAdapter(connection) {
 	if (!hasConnect) throw new Error("Connection adapter must provide either connect or both subscribe and send");
 	let activeBuffer = [];
 	let activeWaiters = [];
+	let activeSubscriber;
 	function push(chunk, runId) {
 		if (runId) chunkRunIds.set(chunk, runId);
 		const waiter = activeWaiters.shift();
@@ -481,17 +482,14 @@ function normalizeConnectionAdapter(connection) {
 		else activeBuffer.push(chunk);
 	}
 	async function waitUntilSubscriberIdle(abortSignal) {
-		const idle = () => activeBuffer.length === 0 && (activeWaiters.length > 0 || abortSignal?.aborted);
+		const idle = () => activeSubscriber !== activeWaiters || activeBuffer.length === 0 && (activeWaiters.length > 0 || abortSignal?.aborted);
 		for (let i = 0; i < 16 && !abortSignal?.aborted; i++) {
 			if (idle()) return;
 			await Promise.resolve();
 		}
-		let macrotaskWaits = 0;
 		while (!abortSignal?.aborted) {
 			if (idle()) return;
 			await new Promise((resolve) => setTimeout(resolve, 0));
-			macrotaskWaits++;
-			if (activeWaiters.length === 0 && macrotaskWaits >= 32) return;
 		}
 	}
 	return {
@@ -501,19 +499,24 @@ function normalizeConnectionAdapter(connection) {
 			activeBuffer = myBuffer;
 			activeWaiters = myWaiters;
 			return (async function* () {
-				while (!abortSignal?.aborted) {
-					let chunk;
-					const buffered = myBuffer.shift();
-					if (buffered !== void 0) chunk = buffered;
-					else chunk = await new Promise((resolve) => {
-						const onAbort = () => resolve(null);
-						myWaiters.push((c) => {
-							abortSignal?.removeEventListener("abort", onAbort);
-							resolve(c);
+				activeSubscriber = myWaiters;
+				try {
+					while (!abortSignal?.aborted) {
+						let chunk;
+						const buffered = myBuffer.shift();
+						if (buffered !== void 0) chunk = buffered;
+						else chunk = await new Promise((resolve) => {
+							const onAbort = () => resolve(null);
+							myWaiters.push((c) => {
+								abortSignal?.removeEventListener("abort", onAbort);
+								resolve(c);
+							});
+							abortSignal?.addEventListener("abort", onAbort, { once: true });
 						});
-						abortSignal?.addEventListener("abort", onAbort, { once: true });
-					});
-					if (chunk !== null) yield chunk;
+						if (chunk !== null) yield chunk;
+					}
+				} finally {
+					if (activeSubscriber === myWaiters) activeSubscriber = void 0;
 				}
 			})();
 		},