agents 0.20.0 → 0.21.0

package/dist/ws-chat-transport-CIoOBbO7.js

@@ -0,0 +1,561 @@
+import "./wire-types-CnMt6_HR.js";
+import { nanoid } from "nanoid";
+//#region src/chat/ws-chat-transport.ts
+/**
+* Short safety-net timeout for a resume probe when the server has said nothing.
+* Under normal operation the server answers a `STREAM_RESUME_REQUEST` with
+* `STREAM_RESUMING`, `STREAM_RESUME_NONE`, or `STREAM_PENDING` well before this.
+*/
+const RESUME_PROBE_TIMEOUT_MS = 5e3;
+/**
+* Extended backstop applied once the server says a turn is pending
+* (`STREAM_PENDING`, #1784). The pre-stream window (queueing, MCP setup,
+* debounce, model latency) can exceed the short probe timeout, and the server
+* guarantees a follow-up `STREAM_RESUMING` or `STREAM_RESUME_NONE` — so we wait
+* much longer (refreshed on every keep-waiting frame) but still cap it so a
+* dropped follow-up degrades to a null resolve instead of hanging forever.
+*/
+const RESUME_PENDING_TIMEOUT_MS = 6e4;
+/**
+* ChatTransport that sends messages over WebSocket and returns a
+* ReadableStream<UIMessageChunk> that the AI SDK's useChat consumes directly.
+*
+* This low-level transport handles new and regenerated request streams plus
+* cancellation. Higher-level protocol coordination such as automatic reconnect
+* resume, cross-tab transcript synchronization, and client-tool continuations
+* is provided by integrations such as `useAgentChat`.
+*/
+var WebSocketChatTransport = class {
+	constructor(options) {
+		this._resumeResolver = null;
+		this._resumeNoneResolver = null;
+		this._onStreamPending = null;
+		this._retryResumeProbe = null;
+		this._expectToolContinuation = false;
+		this._abortToolContinuation = null;
+		this._activeServerTurnId = null;
+		this._cancelAttachedStream = null;
+		this._detachResumeStream = null;
+		this.agent = options.agent;
+		this.prepareBody = options.prepareBody;
+		this.activeRequestIds = options.activeRequestIds;
+		this.cancelOnClientAbort = options.cancelOnClientAbort ?? false;
+	}
+	/**
+	* Point the singleton transport at a new Agent connection. A pending resolver
+	* belongs to the old Chat/socket generation and must settle before messages
+	* from the replacement connection can be consumed (#1914 review).
+	*/
+	setAgent(agent) {
+		if (this.agent === agent) return;
+		this.resetResumeState();
+		this.agent = agent;
+	}
+	setCancelOnClientAbort(cancelOnClientAbort) {
+		this.cancelOnClientAbort = cancelOnClientAbort;
+	}
+	/**
+	* Explicitly cancel the active server turn, if any.
+	* This is separate from generic client-side abort/cancel lifecycle so
+	* clients can detach locally without stopping server work.
+	*/
+	cancelActiveServerTurn() {
+		const requestId = this._activeServerTurnId;
+		let cancelledRequest = false;
+		if (requestId) {
+			this.sendCancelFrame(requestId);
+			this._cancelAttachedStream?.();
+			this.clearActiveServerTurn(requestId);
+			cancelledRequest = true;
+		}
+		const cancelledToolContinuation = this.abortActiveToolContinuation();
+		return cancelledRequest || cancelledToolContinuation;
+	}
+	sendCancelFrame(requestId) {
+		try {
+			this.agent.send(JSON.stringify({
+				id: requestId,
+				type: "cf_agent_chat_request_cancel"
+			}));
+		} catch {}
+	}
+	setActiveServerTurn(requestId, cancelAttachedStream) {
+		this._activeServerTurnId = requestId;
+		this._cancelAttachedStream = cancelAttachedStream;
+	}
+	clearActiveServerTurn(requestId) {
+		if (this._activeServerTurnId === requestId) {
+			this._activeServerTurnId = null;
+			this._cancelAttachedStream = null;
+		}
+	}
+	/**
+	* Mark that the next reconnectToStream() call should attach to a
+	* server-initiated tool continuation rather than a page-load resume.
+	*/
+	expectToolContinuation() {
+		this._expectToolContinuation = true;
+	}
+	/**
+	* Abort the active client-side tool continuation stream, if one is attached
+	* to a server request id.
+	*/
+	abortActiveToolContinuation() {
+		return this._abortToolContinuation?.() ?? false;
+	}
+	/**
+	* True when the transport is waiting for a resume handshake.
+	*/
+	isAwaitingResume() {
+		return this._resumeResolver !== null || this._resumeNoneResolver !== null;
+	}
+	/**
+	* Settle and detach the current handshake without interpreting it as a
+	* server-idle response. Used when the owning hook/agent generation changes.
+	*/
+	cancelPendingResume() {
+		const resolveNone = this._resumeNoneResolver;
+		if (!resolveNone) return false;
+		return resolveNone({});
+	}
+	/**
+	* Invalidate all client-side resume state for an obsolete hook/agent
+	* generation without cancelling its durable server turn.
+	*/
+	resetResumeState() {
+		this._expectToolContinuation = false;
+		this.cancelPendingResume();
+		this._detachResumeStream?.();
+	}
+	/**
+	* Re-send the active handshake request on the latest socket generation. This
+	* preserves one AI SDK resume operation while recovering a request/reply lost
+	* with the previous WebSocket.
+	*/
+	retryPendingResume() {
+		const retry = this._retryResumeProbe;
+		if (!retry) return false;
+		retry();
+		return true;
+	}
+	/**
+	* Called by onAgentMessage when it receives CF_AGENT_STREAM_RESUMING.
+	* If reconnectToStream is waiting, this handles the resume handshake
+	* (ACK + stream creation) and returns true. Otherwise returns false
+	* so the caller can use its own fallback path.
+	*/
+	handleStreamResuming(data) {
+		if (!this._resumeResolver) return false;
+		this._resumeResolver(data);
+		return true;
+	}
+	/**
+	* Called by onAgentMessage when it receives CF_AGENT_STREAM_RESUME_NONE.
+	* If reconnectToStream is waiting, resolves the promise with null
+	* immediately (no 5-second timeout). Returns true if handled.
+	*/
+	handleStreamResumeNone(data = {}) {
+		if (!this._resumeNoneResolver) return false;
+		return this._resumeNoneResolver(data);
+	}
+	/**
+	* Called by onAgentMessage when it receives CF_AGENT_STREAM_PENDING (#1784):
+	* the server accepted a turn but its stream has not started yet. If a resume
+	* path is awaiting, extend its probe timeout (so it keeps waiting for the
+	* eventual STREAM_RESUMING / STREAM_RESUME_NONE instead of resolving null
+	* after the short window). Returns true if a waiting path consumed it.
+	*/
+	handleStreamPending() {
+		if (!this._onStreamPending) return false;
+		this._onStreamPending();
+		return true;
+	}
+	/**
+	* Called by the hook's shared message handler when a server turn finishes
+	* outside the currently attached transport stream, such as after local-only
+	* client cleanup.
+	*/
+	handleServerTurnCompleted(requestId) {
+		this.clearActiveServerTurn(requestId);
+	}
+	/**
+	* Register a server turn that is being rendered outside a transport-owned
+	* stream, such as the hook's fallback cross-tab/resume observer path.
+	*/
+	observeServerTurn(requestId) {
+		this.setActiveServerTurn(requestId, null);
+	}
+	async sendMessages(options) {
+		const requestId = nanoid(8);
+		const abortController = new AbortController();
+		let completed = false;
+		let requestSent = false;
+		let extraBody = {};
+		if (this.prepareBody) extraBody = await this.prepareBody({
+			messages: options.messages,
+			trigger: options.trigger,
+			messageId: options.messageId
+		});
+		if (options.body) extraBody = {
+			...extraBody,
+			...options.body
+		};
+		const bodyPayload = JSON.stringify({
+			messages: options.messages,
+			trigger: options.trigger,
+			...extraBody
+		});
+		this.activeRequestIds?.add(requestId);
+		const agent = this.agent;
+		const activeIds = this.activeRequestIds;
+		const finish = (action, keepId = false, clearServerTurn = true) => {
+			if (completed) return;
+			completed = true;
+			if (clearServerTurn) this.clearActiveServerTurn(requestId);
+			try {
+				action();
+			} catch {}
+			if (!keepId) activeIds?.delete(requestId);
+			abortController.abort();
+		};
+		const abortError = /* @__PURE__ */ new Error("Aborted");
+		abortError.name = "AbortError";
+		const cancelActiveRequest = () => {
+			if (completed) return false;
+			finish(() => streamController.error(abortError), true);
+			return true;
+		};
+		this.setActiveServerTurn(requestId, cancelActiveRequest);
+		const onAbort = () => {
+			if (completed) return;
+			if (this.cancelOnClientAbort) {
+				if (requestSent) this.sendCancelFrame(requestId);
+				finish(() => streamController.error(abortError), requestSent);
+			} else finish(() => streamController.error(abortError), false, !requestSent);
+		};
+		let streamController;
+		const stream = new ReadableStream({
+			start(controller) {
+				streamController = controller;
+				const onMessage = (event) => {
+					try {
+						const data = JSON.parse(event.data);
+						if (data.type !== "cf_agent_use_chat_response") return;
+						if (data.id !== requestId) return;
+						if (data.error) {
+							finish(() => controller.error(new Error(data.body || "Stream error")));
+							return;
+						}
+						if (data.body?.trim()) try {
+							const chunk = JSON.parse(data.body);
+							controller.enqueue(chunk);
+						} catch {}
+						if (data.done) finish(() => controller.close());
+					} catch {}
+				};
+				const onClose = () => {
+					finish(() => controller.close(), false, false);
+				};
+				agent.addEventListener("message", onMessage, { signal: abortController.signal });
+				agent.addEventListener("close", onClose, { signal: abortController.signal });
+			},
+			cancel() {
+				onAbort();
+			}
+		});
+		if (options.abortSignal) {
+			options.abortSignal.addEventListener("abort", onAbort, { once: true });
+			if (options.abortSignal.aborted) onAbort();
+		}
+		if (completed) return stream;
+		requestSent = true;
+		agent.send(JSON.stringify({
+			id: requestId,
+			init: {
+				method: "POST",
+				body: bodyPayload
+			},
+			type: "cf_agent_use_chat_request"
+		}));
+		return stream;
+	}
+	async reconnectToStream(_options) {
+		if (this.isAwaitingResume()) return null;
+		if (this._expectToolContinuation) {
+			this._expectToolContinuation = false;
+			return this._createToolContinuationStream();
+		}
+		const activeIds = this.activeRequestIds;
+		const probeId = nanoid(8);
+		return new Promise((resolve) => {
+			let resolved = false;
+			let timeout;
+			let resumeResolver = null;
+			let resumeNoneResolver = null;
+			let onStreamPending = null;
+			let retryResumeProbe = null;
+			const clearOwnedCallbacks = () => {
+				if (resumeResolver && this._resumeResolver === resumeResolver) this._resumeResolver = null;
+				if (resumeNoneResolver && this._resumeNoneResolver === resumeNoneResolver) this._resumeNoneResolver = null;
+				if (onStreamPending && this._onStreamPending === onStreamPending) this._onStreamPending = null;
+				if (retryResumeProbe && this._retryResumeProbe === retryResumeProbe) this._retryResumeProbe = null;
+			};
+			const done = (value) => {
+				if (resolved) return;
+				resolved = true;
+				clearOwnedCallbacks();
+				if (timeout) clearTimeout(timeout);
+				resolve(value);
+			};
+			const armTimeout = (delay) => {
+				if (timeout) clearTimeout(timeout);
+				timeout = setTimeout(() => done(null), delay);
+			};
+			onStreamPending = () => {
+				if (resolved) return;
+				armTimeout(RESUME_PENDING_TIMEOUT_MS);
+			};
+			resumeNoneResolver = (data) => {
+				if (data.probeId && data.probeId !== probeId) return false;
+				done(null);
+				return true;
+			};
+			resumeResolver = (data) => {
+				const requestId = data.id;
+				activeIds?.add(requestId);
+				const stream = this._createResumeStream(requestId);
+				this.agent.send(JSON.stringify({
+					type: "cf_agent_stream_resume_ack",
+					id: requestId
+				}));
+				done(stream);
+			};
+			retryResumeProbe = () => {
+				if (resolved) return;
+				armTimeout(RESUME_PROBE_TIMEOUT_MS);
+				try {
+					this.agent.send(JSON.stringify({
+						type: "cf_agent_stream_resume_request",
+						probeId
+					}));
+				} catch {}
+			};
+			this._onStreamPending = onStreamPending;
+			this._resumeNoneResolver = resumeNoneResolver;
+			this._resumeResolver = resumeResolver;
+			this._retryResumeProbe = retryResumeProbe;
+			retryResumeProbe();
+		});
+	}
+	/**
+	* Creates a deferred ReadableStream for client-side tool continuations.
+	* The stream is returned immediately so AI SDK status becomes "submitted"
+	* right after addToolOutput()/addToolApprovalResponse(), then it waits for
+	* the server to announce the continuation via STREAM_RESUMING.
+	*/
+	_createToolContinuationStream() {
+		const agent = this.agent;
+		const activeIds = this.activeRequestIds;
+		const streamController = new AbortController();
+		const abortError = /* @__PURE__ */ new Error("Aborted");
+		abortError.name = "AbortError";
+		let completed = false;
+		let requestId = null;
+		let readerController = null;
+		const probeId = nanoid(8);
+		let onResumeRef = null;
+		let onResumeNoneRef = null;
+		let onStreamPendingRef = null;
+		let retryResumeProbeRef = null;
+		let abortToolContinuationRef = null;
+		let detachResumeStreamRef = null;
+		const clearOwnedHandshake = () => {
+			if (onResumeRef && this._resumeResolver === onResumeRef) this._resumeResolver = null;
+			if (onResumeNoneRef && this._resumeNoneResolver === onResumeNoneRef) this._resumeNoneResolver = null;
+			if (onStreamPendingRef && this._onStreamPending === onStreamPendingRef) this._onStreamPending = null;
+			if (retryResumeProbeRef && this._retryResumeProbe === retryResumeProbeRef) this._retryResumeProbe = null;
+		};
+		const finish = (action, keepRequestId = false) => {
+			if (completed) return;
+			completed = true;
+			if (this._abortToolContinuation === abortToolContinuationRef) this._abortToolContinuation = null;
+			if (this._detachResumeStream === detachResumeStreamRef) this._detachResumeStream = null;
+			clearOwnedHandshake();
+			try {
+				action();
+			} catch {}
+			if (requestId && !keepRequestId) activeIds?.delete(requestId);
+			streamController.abort();
+		};
+		const transport = this;
+		abortToolContinuationRef = () => {
+			if (completed) return false;
+			if (requestId === null) {
+				finish(() => readerController?.error(abortError));
+				return true;
+			}
+			try {
+				agent.send(JSON.stringify({
+					type: "cf_agent_chat_request_cancel",
+					id: requestId
+				}));
+			} catch {}
+			finish(() => readerController?.error(abortError), true);
+			return true;
+		};
+		this._abortToolContinuation = abortToolContinuationRef;
+		detachResumeStreamRef = () => {
+			if (completed) return false;
+			finish(() => readerController?.close());
+			return true;
+		};
+		this._detachResumeStream = detachResumeStreamRef;
+		return new ReadableStream({
+			start(controller) {
+				readerController = controller;
+				let timeout;
+				const armTimeout = (delay) => {
+					if (timeout) clearTimeout(timeout);
+					timeout = setTimeout(() => finish(() => controller.close()), delay);
+				};
+				const onResumeNone = (data) => {
+					if (data.probeId && data.probeId !== probeId) return false;
+					finish(() => controller.close());
+					return true;
+				};
+				const onResume = (data) => {
+					if (requestId) return;
+					requestId = data.id;
+					activeIds?.add(requestId);
+					clearOwnedHandshake();
+					if (timeout) clearTimeout(timeout);
+					agent.send(JSON.stringify({
+						type: "cf_agent_stream_resume_ack",
+						id: requestId
+					}));
+				};
+				const onStreamPending = () => {
+					if (completed) return;
+					armTimeout(RESUME_PENDING_TIMEOUT_MS);
+				};
+				const retryResumeProbe = () => {
+					if (completed || requestId !== null) return;
+					armTimeout(RESUME_PROBE_TIMEOUT_MS);
+					try {
+						transport.agent.send(JSON.stringify({
+							type: "cf_agent_stream_resume_request",
+							probeId
+						}));
+					} catch {}
+				};
+				onResumeRef = onResume;
+				onResumeNoneRef = onResumeNone;
+				onStreamPendingRef = onStreamPending;
+				retryResumeProbeRef = retryResumeProbe;
+				transport._resumeResolver = onResume;
+				transport._resumeNoneResolver = onResumeNone;
+				transport._onStreamPending = onStreamPending;
+				transport._retryResumeProbe = retryResumeProbe;
+				const onMessage = (event) => {
+					try {
+						const data = JSON.parse(event.data);
+						if (data.type !== "cf_agent_use_chat_response" || requestId == null || data.id !== requestId) return;
+						if (data.error) {
+							finish(() => controller.error(new Error(data.body || "Stream error")));
+							return;
+						}
+						if (data.body?.trim()) try {
+							const chunk = JSON.parse(data.body);
+							controller.enqueue(chunk);
+						} catch {}
+						if (data.done) finish(() => controller.close());
+					} catch {}
+				};
+				const onClose = () => finish(() => controller.close());
+				agent.addEventListener("message", onMessage, { signal: streamController.signal });
+				agent.addEventListener("close", onClose, { signal: streamController.signal });
+				retryResumeProbe();
+			},
+			cancel() {
+				if (requestId && transport.cancelOnClientAbort) {
+					transport.sendCancelFrame(requestId);
+					finish(() => {}, true);
+				} else finish(() => {});
+			}
+		});
+	}
+	/**
+	* Creates a ReadableStream that receives resumed stream chunks
+	* and forwards them to useChat as UIMessageChunk objects.
+	*/
+	_createResumeStream(requestId) {
+		const agent = this.agent;
+		const activeIds = this.activeRequestIds;
+		const chunkController = new AbortController();
+		const abortError = /* @__PURE__ */ new Error("Aborted");
+		abortError.name = "AbortError";
+		let completed = false;
+		let detachResumeStream = null;
+		const finish = (action, keepId = false, clearServerTurn = true) => {
+			if (completed) return;
+			completed = true;
+			if (clearServerTurn) this.clearActiveServerTurn(requestId);
+			if (this._detachResumeStream === detachResumeStream) this._detachResumeStream = null;
+			try {
+				action();
+			} catch {}
+			if (!keepId) activeIds?.delete(requestId);
+			chunkController.abort();
+		};
+		let streamController = null;
+		const cancelActiveRequest = () => {
+			if (completed) return false;
+			finish(() => streamController?.error(abortError), true);
+			return true;
+		};
+		this.setActiveServerTurn(requestId, cancelActiveRequest);
+		const transport = this;
+		detachResumeStream = () => {
+			if (completed) return false;
+			finish(() => streamController?.close());
+			return true;
+		};
+		this._detachResumeStream = detachResumeStream;
+		return new ReadableStream({
+			start(controller) {
+				streamController = controller;
+				const onMessage = (event) => {
+					try {
+						const data = JSON.parse(event.data);
+						if (data.type !== "cf_agent_use_chat_response") return;
+						if (data.id !== requestId) return;
+						if (data.error) {
+							finish(() => controller.error(new Error(data.body || "Stream error")));
+							return;
+						}
+						if (data.body?.trim()) try {
+							const chunk = JSON.parse(data.body);
+							controller.enqueue(chunk);
+						} catch {}
+						if (data.done) finish(() => controller.close());
+					} catch {}
+				};
+				const onClose = () => {
+					finish(() => controller.close(), false, false);
+				};
+				agent.addEventListener("message", onMessage, { signal: chunkController.signal });
+				agent.addEventListener("close", onClose, { signal: chunkController.signal });
+			},
+			cancel() {
+				if (transport.cancelOnClientAbort) {
+					transport.sendCancelFrame(requestId);
+					finish(() => {}, true);
+				} else finish(() => {}, false, false);
+			}
+		});
+	}
+};
+//#endregion
+export { WebSocketChatTransport as t };
+
+//# sourceMappingURL=ws-chat-transport-CIoOBbO7.js.map

package/dist/ws-chat-transport-CIoOBbO7.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-chat-transport-CIoOBbO7.js","names":[],"sources":["../src/chat/ws-chat-transport.ts"],"sourcesContent":["/**\n * Framework-neutral WebSocket transport for AI SDK chat clients.\n *\n * Speaks the CF_AGENT protocol directly and returns AI SDK UI message chunks\n * without reconstructing an HTTP response or parsing an SSE stream.\n */\n\nimport type { ChatTransport, UIMessage, UIMessageChunk } from \"ai\";\nimport { nanoid } from \"nanoid\";\nimport { MessageType, type OutgoingMessage } from \"./wire-types\";\n\n/**\n * Short safety-net timeout for a resume probe when the server has said nothing.\n * Under normal operation the server answers a `STREAM_RESUME_REQUEST` with\n * `STREAM_RESUMING`, `STREAM_RESUME_NONE`, or `STREAM_PENDING` well before this.\n */\nconst RESUME_PROBE_TIMEOUT_MS = 5000;\n\n/**\n * Extended backstop applied once the server says a turn is pending\n * (`STREAM_PENDING`, #1784). The pre-stream window (queueing, MCP setup,\n * debounce, model latency) can exceed the short probe timeout, and the server\n * guarantees a follow-up `STREAM_RESUMING` or `STREAM_RESUME_NONE` — so we wait\n * much longer (refreshed on every keep-waiting frame) but still cap it so a\n * dropped follow-up degrades to a null resolve instead of hanging forever.\n */\nconst RESUME_PENDING_TIMEOUT_MS = 60000;\n\n/**\n * Agent-like interface for sending/receiving WebSocket messages.\n * Matches `AgentClient` from `agents/client` and the connection returned by\n * `useAgent` from `agents/react`.\n */\nexport interface AgentConnection {\n  send: (data: string) => void;\n  addEventListener: (\n    type: string,\n    listener: (event: MessageEvent) => void,\n    options?: { signal?: AbortSignal }\n  ) => void;\n  removeEventListener: (\n    type: string,\n    listener: (event: MessageEvent) => void\n  ) => void;\n}\n\nexport type WebSocketChatTransportOptions<\n  ChatMessage extends UIMessage = UIMessage\n> = {\n  /** The framework-neutral AgentClient or useAgent connection. */\n  agent: AgentConnection;\n  /**\n   * Callback to add custom fields to the request body before sending.\n   */\n  prepareBody?: (options: {\n    messages: ChatMessage[];\n    trigger: \"submit-message\" | \"regenerate-message\";\n    messageId?: string;\n  }) => Promise<Record<string, unknown>> | Record<string, unknown>;\n  /**\n   * Optional set to track active request IDs.\n   * IDs are added when a request starts and removed when it completes.\n   * Used by the onAgentMessage handler to skip messages already handled by the transport.\n   */\n  activeRequestIds?: Set<string>;\n  /**\n   * Whether generic client-side abort/cancel lifecycle should cancel the\n   * server turn. Explicit cancellation via cancelActiveServerTurn() always\n   * sends CF_AGENT_CHAT_REQUEST_CANCEL.\n   * @default false\n   */\n  cancelOnClientAbort?: boolean;\n};\n\n/**\n * ChatTransport that sends messages over WebSocket and returns a\n * ReadableStream<UIMessageChunk> that the AI SDK's useChat consumes directly.\n *\n * This low-level transport handles new and regenerated request streams plus\n * cancellation. Higher-level protocol coordination such as automatic reconnect\n * resume, cross-tab transcript synchronization, and client-tool continuations\n * is provided by integrations such as `useAgentChat`.\n */\nexport class WebSocketChatTransport<\n  ChatMessage extends UIMessage = UIMessage\n> implements ChatTransport<ChatMessage> {\n  agent: AgentConnection;\n  private prepareBody?: WebSocketChatTransportOptions<ChatMessage>[\"prepareBody\"];\n  private activeRequestIds?: Set<string>;\n  private cancelOnClientAbort: boolean;\n\n  // Pending resume resolver — set by reconnectToStream, called by\n  // handleStreamResuming when onAgentMessage sees CF_AGENT_STREAM_RESUMING.\n  private _resumeResolver: ((data: { id: string }) => void) | null = null;\n  // Pending \"no stream\" resolver — called by handleStreamResumeNone\n  // when onAgentMessage sees CF_AGENT_STREAM_RESUME_NONE.\n  private _resumeNoneResolver:\n    | ((data: { probeId?: string }) => boolean)\n    | null = null;\n  // Keep-waiting hook (#1784) — set by whichever resume path is currently\n  // awaiting, called by handleStreamPending when onAgentMessage sees\n  // CF_AGENT_STREAM_PENDING. Extends the path's probe timeout so a slow\n  // pre-stream window (queue / MCP / model latency) does not resolve early.\n  private _onStreamPending: (() => void) | null = null;\n  // Retransmits the current handshake on a replacement socket without starting\n  // a second AI SDK resume request. Set only while a resolver is active.\n  private _retryResumeProbe: (() => void) | null = null;\n  // Set when a client-side tool result/approval is expected to trigger\n  // a new continuation stream. In this mode reconnectToStream() returns\n  // a deferred ReadableStream immediately so AI SDK status can transition\n  // to \"submitted\" before the server starts streaming.\n  private _expectToolContinuation = false;\n  private _abortToolContinuation: (() => boolean) | null = null;\n  private _activeServerTurnId: string | null = null;\n  private _cancelAttachedStream: (() => boolean) | null = null;\n  // Local-only detach for a resume stream owned by an obsolete Chat/agent\n  // generation. Unlike explicit cancellation, this never cancels server work.\n  private _detachResumeStream: (() => boolean) | null = null;\n\n  constructor(options: WebSocketChatTransportOptions<ChatMessage>) {\n    this.agent = options.agent;\n    this.prepareBody = options.prepareBody;\n    this.activeRequestIds = options.activeRequestIds;\n    this.cancelOnClientAbort = options.cancelOnClientAbort ?? false;\n  }\n\n  /**\n   * Point the singleton transport at a new Agent connection. A pending resolver\n   * belongs to the old Chat/socket generation and must settle before messages\n   * from the replacement connection can be consumed (#1914 review).\n   */\n  setAgent(agent: AgentConnection) {\n    if (this.agent === agent) return;\n    this.resetResumeState();\n    this.agent = agent;\n  }\n\n  setCancelOnClientAbort(cancelOnClientAbort: boolean) {\n    this.cancelOnClientAbort = cancelOnClientAbort;\n  }\n\n  /**\n   * Explicitly cancel the active server turn, if any.\n   * This is separate from generic client-side abort/cancel lifecycle so\n   * clients can detach locally without stopping server work.\n   */\n  cancelActiveServerTurn(): boolean {\n    const requestId = this._activeServerTurnId;\n    let cancelledRequest = false;\n\n    if (requestId) {\n      this.sendCancelFrame(requestId);\n      this._cancelAttachedStream?.();\n      this.clearActiveServerTurn(requestId);\n      cancelledRequest = true;\n    }\n\n    const cancelledToolContinuation = this.abortActiveToolContinuation();\n    return cancelledRequest || cancelledToolContinuation;\n  }\n\n  private sendCancelFrame(requestId: string) {\n    try {\n      this.agent.send(\n        JSON.stringify({\n          id: requestId,\n          type: MessageType.CF_AGENT_CHAT_REQUEST_CANCEL\n        })\n      );\n    } catch {\n      // Ignore failures (e.g. agent already disconnected)\n    }\n  }\n\n  private setActiveServerTurn(\n    requestId: string,\n    cancelAttachedStream: (() => boolean) | null\n  ) {\n    this._activeServerTurnId = requestId;\n    this._cancelAttachedStream = cancelAttachedStream;\n  }\n\n  private clearActiveServerTurn(requestId: string) {\n    if (this._activeServerTurnId === requestId) {\n      this._activeServerTurnId = null;\n      this._cancelAttachedStream = null;\n    }\n  }\n\n  /**\n   * Mark that the next reconnectToStream() call should attach to a\n   * server-initiated tool continuation rather than a page-load resume.\n   */\n  expectToolContinuation() {\n    this._expectToolContinuation = true;\n  }\n\n  /**\n   * Abort the active client-side tool continuation stream, if one is attached\n   * to a server request id.\n   */\n  abortActiveToolContinuation(): boolean {\n    return this._abortToolContinuation?.() ?? false;\n  }\n\n  /**\n   * True when the transport is waiting for a resume handshake.\n   */\n  isAwaitingResume(): boolean {\n    return this._resumeResolver !== null || this._resumeNoneResolver !== null;\n  }\n\n  /**\n   * Settle and detach the current handshake without interpreting it as a\n   * server-idle response. Used when the owning hook/agent generation changes.\n   */\n  cancelPendingResume(): boolean {\n    const resolveNone = this._resumeNoneResolver;\n    if (!resolveNone) return false;\n    return resolveNone({});\n  }\n\n  /**\n   * Invalidate all client-side resume state for an obsolete hook/agent\n   * generation without cancelling its durable server turn.\n   */\n  resetResumeState(): void {\n    this._expectToolContinuation = false;\n    this.cancelPendingResume();\n    this._detachResumeStream?.();\n  }\n\n  /**\n   * Re-send the active handshake request on the latest socket generation. This\n   * preserves one AI SDK resume operation while recovering a request/reply lost\n   * with the previous WebSocket.\n   */\n  retryPendingResume(): boolean {\n    const retry = this._retryResumeProbe;\n    if (!retry) return false;\n    retry();\n    return true;\n  }\n\n  /**\n   * Called by onAgentMessage when it receives CF_AGENT_STREAM_RESUMING.\n   * If reconnectToStream is waiting, this handles the resume handshake\n   * (ACK + stream creation) and returns true. Otherwise returns false\n   * so the caller can use its own fallback path.\n   */\n  handleStreamResuming(data: { id: string }): boolean {\n    if (!this._resumeResolver) return false;\n    this._resumeResolver(data);\n    return true;\n  }\n\n  /**\n   * Called by onAgentMessage when it receives CF_AGENT_STREAM_RESUME_NONE.\n   * If reconnectToStream is waiting, resolves the promise with null\n   * immediately (no 5-second timeout). Returns true if handled.\n   */\n  handleStreamResumeNone(data: { probeId?: string } = {}): boolean {\n    if (!this._resumeNoneResolver) return false;\n    return this._resumeNoneResolver(data);\n  }\n\n  /**\n   * Called by onAgentMessage when it receives CF_AGENT_STREAM_PENDING (#1784):\n   * the server accepted a turn but its stream has not started yet. If a resume\n   * path is awaiting, extend its probe timeout (so it keeps waiting for the\n   * eventual STREAM_RESUMING / STREAM_RESUME_NONE instead of resolving null\n   * after the short window). Returns true if a waiting path consumed it.\n   */\n  handleStreamPending(): boolean {\n    if (!this._onStreamPending) return false;\n    this._onStreamPending();\n    return true;\n  }\n\n  /**\n   * Called by the hook's shared message handler when a server turn finishes\n   * outside the currently attached transport stream, such as after local-only\n   * client cleanup.\n   */\n  handleServerTurnCompleted(requestId: string) {\n    this.clearActiveServerTurn(requestId);\n  }\n\n  /**\n   * Register a server turn that is being rendered outside a transport-owned\n   * stream, such as the hook's fallback cross-tab/resume observer path.\n   */\n  observeServerTurn(requestId: string) {\n    this.setActiveServerTurn(requestId, null);\n  }\n\n  async sendMessages(options: {\n    chatId: string;\n    messages: ChatMessage[];\n    abortSignal: AbortSignal | undefined;\n    trigger: \"submit-message\" | \"regenerate-message\";\n    messageId?: string;\n    body?: object;\n    headers?: Record<string, string> | Headers;\n    metadata?: unknown;\n  }): Promise<ReadableStream<UIMessageChunk>> {\n    const requestId = nanoid(8);\n    const abortController = new AbortController();\n    let completed = false;\n    let requestSent = false;\n\n    // Build the request body\n    let extraBody: Record<string, unknown> = {};\n    if (this.prepareBody) {\n      extraBody = await this.prepareBody({\n        messages: options.messages,\n        trigger: options.trigger,\n        messageId: options.messageId\n      });\n    }\n    if (options.body) {\n      extraBody = {\n        ...extraBody,\n        ...(options.body as Record<string, unknown>)\n      };\n    }\n\n    const bodyPayload = JSON.stringify({\n      messages: options.messages,\n      trigger: options.trigger,\n      ...extraBody\n    });\n\n    // Track this request so the onAgentMessage handler skips it\n    this.activeRequestIds?.add(requestId);\n\n    // Create a ReadableStream<UIMessageChunk> that emits parsed chunks\n    // as they arrive over the WebSocket\n    const agent = this.agent;\n    const activeIds = this.activeRequestIds;\n\n    // Single cleanup helper — every terminal path (done, error, abort)\n    // goes through here exactly once.\n    // keepId: when true, do NOT remove requestId from activeIds. Used by\n    // explicit cancellation so onAgentMessage skips in-flight chunks\n    // and the server's final done:true signal until cleanup happens there.\n    const finish = (\n      action: () => void,\n      keepId = false,\n      clearServerTurn = true\n    ) => {\n      if (completed) return;\n      completed = true;\n      if (clearServerTurn) {\n        this.clearActiveServerTurn(requestId);\n      }\n      try {\n        action();\n      } catch {\n        // Stream may already be closed\n      }\n      if (!keepId) {\n        activeIds?.delete(requestId);\n      }\n      abortController.abort();\n    };\n\n    const abortError = new Error(\"Aborted\");\n    abortError.name = \"AbortError\";\n\n    const cancelActiveRequest = () => {\n      if (completed) return false;\n      finish(() => streamController.error(abortError), true);\n      return true;\n    };\n    this.setActiveServerTurn(requestId, cancelActiveRequest);\n\n    // Abort handler: terminate the local stream. By default, generic AI SDK\n    // abort/cancel lifecycle is local-only so durable server turns can continue\n    // and be resumed. Use cancelActiveServerTurn() for explicit user/app\n    // cancellation, or cancelOnClientAbort for request-lifetime semantics.\n    const onAbort = () => {\n      if (completed) return;\n      if (this.cancelOnClientAbort) {\n        if (requestSent) {\n          this.sendCancelFrame(requestId);\n        }\n        finish(() => streamController.error(abortError), requestSent);\n      } else {\n        finish(() => streamController.error(abortError), false, !requestSent);\n      }\n    };\n\n    // streamController is assigned synchronously by start(), so it is\n    // always available by the time onAbort or onMessage can fire.\n    let streamController!: ReadableStreamDefaultController<UIMessageChunk>;\n\n    const stream = new ReadableStream<UIMessageChunk>({\n      start(controller) {\n        streamController = controller;\n\n        const onMessage = (event: MessageEvent) => {\n          try {\n            const data = JSON.parse(\n              event.data as string\n            ) as OutgoingMessage<ChatMessage>;\n\n            if (data.type !== MessageType.CF_AGENT_USE_CHAT_RESPONSE) return;\n            if (data.id !== requestId) return;\n\n            if (data.error) {\n              finish(() =>\n                controller.error(new Error(data.body || \"Stream error\"))\n              );\n              return;\n            }\n\n            // Parse the body as UIMessageChunk and enqueue\n            if (data.body?.trim()) {\n              try {\n                const chunk = JSON.parse(data.body) as UIMessageChunk;\n                controller.enqueue(chunk);\n              } catch {\n                // Skip malformed chunk bodies\n              }\n            }\n\n            if (data.done) {\n              finish(() => controller.close());\n            }\n          } catch {\n            // Ignore non-JSON messages\n          }\n        };\n\n        const onClose = () => {\n          finish(() => controller.close(), false, false);\n        };\n\n        agent.addEventListener(\"message\", onMessage, {\n          signal: abortController.signal\n        });\n        agent.addEventListener(\"close\", onClose, {\n          signal: abortController.signal\n        });\n      },\n      cancel() {\n        onAbort();\n      }\n    });\n\n    // Handle abort from the caller\n    if (options.abortSignal) {\n      options.abortSignal.addEventListener(\"abort\", onAbort, { once: true });\n      if (options.abortSignal.aborted) onAbort();\n    }\n\n    if (completed) {\n      return stream;\n    }\n\n    // Send the request over WebSocket\n    requestSent = true;\n    agent.send(\n      JSON.stringify({\n        id: requestId,\n        init: {\n          method: \"POST\",\n          body: bodyPayload\n        },\n        type: MessageType.CF_AGENT_USE_CHAT_REQUEST\n      })\n    );\n\n    return stream;\n  }\n\n  async reconnectToStream(_options: {\n    chatId: string;\n  }): Promise<ReadableStream<UIMessageChunk> | null> {\n    // A transport has one handshake slot. Returning null for an unexpected\n    // concurrent caller is safer than overwriting callbacks owned by the first\n    // AI SDK request (StrictMode/manual overlap); the hook serializes its normal\n    // mount, tool, public, and reconnect entry points.\n    if (this.isAwaitingResume()) return null;\n\n    if (this._expectToolContinuation) {\n      this._expectToolContinuation = false;\n      return this._createToolContinuationStream();\n    }\n\n    // Detect whether the server has an active stream for this chat.\n    // Instead of registering another message listener (which races with the\n    // hook), expose identity-owned callbacks consumed synchronously by the\n    // shared handler.\n    const activeIds = this.activeRequestIds;\n    const probeId = nanoid(8);\n\n    return new Promise<ReadableStream<UIMessageChunk> | null>((resolve) => {\n      let resolved = false;\n      let timeout: ReturnType<typeof setTimeout> | undefined;\n      let resumeResolver: ((data: { id: string }) => void) | null = null;\n      let resumeNoneResolver: ((data: { probeId?: string }) => boolean) | null =\n        null;\n      let onStreamPending: (() => void) | null = null;\n      let retryResumeProbe: (() => void) | null = null;\n\n      const clearOwnedCallbacks = () => {\n        if (resumeResolver && this._resumeResolver === resumeResolver) {\n          this._resumeResolver = null;\n        }\n        if (\n          resumeNoneResolver &&\n          this._resumeNoneResolver === resumeNoneResolver\n        ) {\n          this._resumeNoneResolver = null;\n        }\n        if (onStreamPending && this._onStreamPending === onStreamPending) {\n          this._onStreamPending = null;\n        }\n        if (retryResumeProbe && this._retryResumeProbe === retryResumeProbe) {\n          this._retryResumeProbe = null;\n        }\n      };\n\n      const done = (value: ReadableStream<UIMessageChunk> | null) => {\n        if (resolved) return;\n        resolved = true;\n        clearOwnedCallbacks();\n        if (timeout) clearTimeout(timeout);\n        resolve(value);\n      };\n\n      const armTimeout = (delay: number) => {\n        if (timeout) clearTimeout(timeout);\n        timeout = setTimeout(() => done(null), delay);\n      };\n\n      // Keep-waiting (#1784): the server says a turn is accepted but its stream\n      // has not started. Extend this operation's own timeout only.\n      onStreamPending = () => {\n        if (resolved) return;\n        armTimeout(RESUME_PENDING_TIMEOUT_MS);\n      };\n\n      resumeNoneResolver = (data) => {\n        if (data.probeId && data.probeId !== probeId) return false;\n        done(null);\n        return true;\n      };\n\n      resumeResolver = (data: { id: string }) => {\n        const requestId = data.id;\n        activeIds?.add(requestId);\n        const stream = this._createResumeStream(requestId);\n\n        this.agent.send(\n          JSON.stringify({\n            type: MessageType.CF_AGENT_STREAM_RESUME_ACK,\n            id: requestId\n          })\n        );\n\n        done(stream);\n      };\n\n      // Re-arm both the request and timeout on each replacement socket. This is\n      // a retransmission of this resolver's handshake, not a second Chat resume.\n      retryResumeProbe = () => {\n        if (resolved) return;\n        armTimeout(RESUME_PROBE_TIMEOUT_MS);\n        try {\n          this.agent.send(\n            JSON.stringify({\n              type: MessageType.CF_AGENT_STREAM_RESUME_REQUEST,\n              probeId\n            })\n          );\n        } catch {\n          // The next socket open retries again; the timeout remains a backstop.\n        }\n      };\n\n      this._onStreamPending = onStreamPending;\n      this._resumeNoneResolver = resumeNoneResolver;\n      this._resumeResolver = resumeResolver;\n      this._retryResumeProbe = retryResumeProbe;\n      retryResumeProbe();\n    });\n  }\n\n  /**\n   * Creates a deferred ReadableStream for client-side tool continuations.\n   * The stream is returned immediately so AI SDK status becomes \"submitted\"\n   * right after addToolOutput()/addToolApprovalResponse(), then it waits for\n   * the server to announce the continuation via STREAM_RESUMING.\n   */\n  private _createToolContinuationStream(): ReadableStream<UIMessageChunk> {\n    const agent = this.agent;\n    const activeIds = this.activeRequestIds;\n    const streamController = new AbortController();\n    const abortError = new Error(\"Aborted\");\n    abortError.name = \"AbortError\";\n    let completed = false;\n    let requestId: string | null = null;\n    let readerController: ReadableStreamDefaultController<UIMessageChunk> | null =\n      null;\n    const probeId = nanoid(8);\n    let onResumeRef: ((data: { id: string }) => void) | null = null;\n    let onResumeNoneRef: ((data: { probeId?: string }) => boolean) | null =\n      null;\n    let onStreamPendingRef: (() => void) | null = null;\n    let retryResumeProbeRef: (() => void) | null = null;\n    let abortToolContinuationRef: (() => boolean) | null = null;\n    let detachResumeStreamRef: (() => boolean) | null = null;\n\n    const clearOwnedHandshake = () => {\n      if (onResumeRef && this._resumeResolver === onResumeRef) {\n        this._resumeResolver = null;\n      }\n      if (onResumeNoneRef && this._resumeNoneResolver === onResumeNoneRef) {\n        this._resumeNoneResolver = null;\n      }\n      if (onStreamPendingRef && this._onStreamPending === onStreamPendingRef) {\n        this._onStreamPending = null;\n      }\n      if (\n        retryResumeProbeRef &&\n        this._retryResumeProbe === retryResumeProbeRef\n      ) {\n        this._retryResumeProbe = null;\n      }\n    };\n\n    const finish = (action: () => void, keepRequestId = false) => {\n      if (completed) return;\n      completed = true;\n      if (this._abortToolContinuation === abortToolContinuationRef) {\n        this._abortToolContinuation = null;\n      }\n      if (this._detachResumeStream === detachResumeStreamRef) {\n        this._detachResumeStream = null;\n      }\n      clearOwnedHandshake();\n      try {\n        action();\n      } catch {\n        // Stream may already be closed\n      }\n      if (requestId && !keepRequestId) {\n        activeIds?.delete(requestId);\n      }\n      streamController.abort();\n    };\n\n    const transport = this;\n\n    abortToolContinuationRef = () => {\n      if (completed) return false;\n\n      if (requestId === null) {\n        finish(() => readerController?.error(abortError));\n        return true;\n      }\n\n      try {\n        agent.send(\n          JSON.stringify({\n            type: MessageType.CF_AGENT_CHAT_REQUEST_CANCEL,\n            id: requestId\n          })\n        );\n      } catch {\n        // Ignore failures (e.g. agent already disconnected)\n      }\n\n      // Keep the ID so the shared message handler ignores in-flight chunks\n      // until the server's terminal frame performs its normal cleanup.\n      finish(() => readerController?.error(abortError), true);\n      return true;\n    };\n    this._abortToolContinuation = abortToolContinuationRef;\n    detachResumeStreamRef = () => {\n      if (completed) return false;\n      finish(() => readerController?.close());\n      return true;\n    };\n    this._detachResumeStream = detachResumeStreamRef;\n\n    return new ReadableStream<UIMessageChunk>({\n      start(controller) {\n        readerController = controller;\n        let timeout: ReturnType<typeof setTimeout> | undefined;\n\n        const armTimeout = (delay: number) => {\n          if (timeout) clearTimeout(timeout);\n          timeout = setTimeout(() => finish(() => controller.close()), delay);\n        };\n\n        const onResumeNone = (data: { probeId?: string }) => {\n          if (data.probeId && data.probeId !== probeId) return false;\n          finish(() => controller.close());\n          return true;\n        };\n\n        const onResume = (data: { id: string }) => {\n          if (requestId) return;\n\n          requestId = data.id;\n          activeIds?.add(requestId);\n          clearOwnedHandshake();\n          if (timeout) clearTimeout(timeout);\n\n          agent.send(\n            JSON.stringify({\n              type: MessageType.CF_AGENT_STREAM_RESUME_ACK,\n              id: requestId\n            })\n          );\n        };\n\n        const onStreamPending = () => {\n          if (completed) return;\n          armTimeout(RESUME_PENDING_TIMEOUT_MS);\n        };\n\n        const retryResumeProbe = () => {\n          if (completed || requestId !== null) return;\n          armTimeout(RESUME_PROBE_TIMEOUT_MS);\n          try {\n            transport.agent.send(\n              JSON.stringify({\n                type: MessageType.CF_AGENT_STREAM_RESUME_REQUEST,\n                probeId\n              })\n            );\n          } catch {\n            // A later socket open can retry this same deferred handshake.\n          }\n        };\n\n        onResumeRef = onResume;\n        onResumeNoneRef = onResumeNone;\n        onStreamPendingRef = onStreamPending;\n        retryResumeProbeRef = retryResumeProbe;\n        transport._resumeResolver = onResume;\n        transport._resumeNoneResolver = onResumeNone;\n        transport._onStreamPending = onStreamPending;\n        transport._retryResumeProbe = retryResumeProbe;\n\n        const onMessage = (event: MessageEvent) => {\n          try {\n            const data = JSON.parse(\n              event.data as string\n            ) as OutgoingMessage<UIMessage>;\n\n            if (\n              data.type !== MessageType.CF_AGENT_USE_CHAT_RESPONSE ||\n              requestId == null ||\n              data.id !== requestId\n            ) {\n              return;\n            }\n\n            if (data.error) {\n              finish(() =>\n                controller.error(new Error(data.body || \"Stream error\"))\n              );\n              return;\n            }\n\n            if (data.body?.trim()) {\n              try {\n                const chunk = JSON.parse(data.body) as UIMessageChunk;\n                controller.enqueue(chunk);\n              } catch {\n                // Skip malformed chunk bodies\n              }\n            }\n\n            if (data.done) {\n              finish(() => controller.close());\n            }\n          } catch {\n            // Ignore non-JSON messages\n          }\n        };\n\n        const onClose = () => finish(() => controller.close());\n\n        agent.addEventListener(\"message\", onMessage, {\n          signal: streamController.signal\n        });\n        agent.addEventListener(\"close\", onClose, {\n          signal: streamController.signal\n        });\n\n        retryResumeProbe();\n      },\n      cancel() {\n        if (requestId && transport.cancelOnClientAbort) {\n          transport.sendCancelFrame(requestId);\n          finish(() => {}, true);\n        } else {\n          finish(() => {});\n        }\n      }\n    });\n  }\n\n  /**\n   * Creates a ReadableStream that receives resumed stream chunks\n   * and forwards them to useChat as UIMessageChunk objects.\n   */\n  private _createResumeStream(\n    requestId: string\n  ): ReadableStream<UIMessageChunk> {\n    // Read agent at resolve time (not when reconnectToStream was called)\n    // so chunk listener attaches to the latest socket after _pk changes.\n    const agent = this.agent;\n    const activeIds = this.activeRequestIds;\n    const chunkController = new AbortController();\n    const abortError = new Error(\"Aborted\");\n    abortError.name = \"AbortError\";\n    let completed = false;\n    let detachResumeStream: (() => boolean) | null = null;\n\n    const finish = (\n      action: () => void,\n      keepId = false,\n      clearServerTurn = true\n    ) => {\n      if (completed) return;\n      completed = true;\n      if (clearServerTurn) {\n        this.clearActiveServerTurn(requestId);\n      }\n      if (this._detachResumeStream === detachResumeStream) {\n        this._detachResumeStream = null;\n      }\n      try {\n        action();\n      } catch {\n        // Stream may already be closed\n      }\n      if (!keepId) {\n        activeIds?.delete(requestId);\n      }\n      chunkController.abort();\n    };\n\n    let streamController: ReadableStreamDefaultController<UIMessageChunk> | null =\n      null;\n    const cancelActiveRequest = () => {\n      if (completed) return false;\n      finish(() => streamController?.error(abortError), true);\n      return true;\n    };\n    this.setActiveServerTurn(requestId, cancelActiveRequest);\n\n    const transport = this;\n    detachResumeStream = () => {\n      if (completed) return false;\n      finish(() => streamController?.close());\n      return true;\n    };\n    this._detachResumeStream = detachResumeStream;\n\n    return new ReadableStream<UIMessageChunk>({\n      start(controller) {\n        streamController = controller;\n\n        const onMessage = (event: MessageEvent) => {\n          try {\n            const data = JSON.parse(\n              event.data as string\n            ) as OutgoingMessage<UIMessage>;\n\n            if (data.type !== MessageType.CF_AGENT_USE_CHAT_RESPONSE) return;\n            if (data.id !== requestId) return;\n\n            if (data.error) {\n              finish(() =>\n                controller.error(new Error(data.body || \"Stream error\"))\n              );\n              return;\n            }\n\n            // Parse and enqueue the chunk\n            if (data.body?.trim()) {\n              try {\n                const chunk = JSON.parse(data.body) as UIMessageChunk;\n                controller.enqueue(chunk);\n              } catch {\n                // Skip malformed chunk bodies\n              }\n            }\n\n            if (data.done) {\n              finish(() => controller.close());\n            }\n          } catch {\n            // Ignore non-JSON messages\n          }\n        };\n\n        const onClose = () => {\n          finish(() => controller.close(), false, false);\n        };\n\n        agent.addEventListener(\"message\", onMessage, {\n          signal: chunkController.signal\n        });\n        agent.addEventListener(\"close\", onClose, {\n          signal: chunkController.signal\n        });\n      },\n      cancel() {\n        if (transport.cancelOnClientAbort) {\n          transport.sendCancelFrame(requestId);\n          finish(() => {}, true);\n        } else {\n          finish(() => {}, false, false);\n        }\n      }\n    });\n  }\n}\n"],"mappings":";;;;;;;;AAgBA,MAAM,0BAA0B;;;;;;;;;AAUhC,MAAM,4BAA4B;;;;;;;;;;AAyDlC,IAAa,yBAAb,MAEwC;CAkCtC,YAAY,SAAqD;EA1BjE,KAAQ,kBAA2D;EAGnE,KAAQ,sBAEG;EAKX,KAAQ,mBAAwC;EAGhD,KAAQ,oBAAyC;EAKjD,KAAQ,0BAA0B;EAClC,KAAQ,yBAAiD;EACzD,KAAQ,sBAAqC;EAC7C,KAAQ,wBAAgD;EAGxD,KAAQ,sBAA8C;EAGpD,KAAK,QAAQ,QAAQ;EACrB,KAAK,cAAc,QAAQ;EAC3B,KAAK,mBAAmB,QAAQ;EAChC,KAAK,sBAAsB,QAAQ,uBAAuB;CAC5D;;;;;;CAOA,SAAS,OAAwB;EAC/B,IAAI,KAAK,UAAU,OAAO;EAC1B,KAAK,iBAAiB;EACtB,KAAK,QAAQ;CACf;CAEA,uBAAuB,qBAA8B;EACnD,KAAK,sBAAsB;CAC7B;;;;;;CAOA,yBAAkC;EAChC,MAAM,YAAY,KAAK;EACvB,IAAI,mBAAmB;EAEvB,IAAI,WAAW;GACb,KAAK,gBAAgB,SAAS;GAC9B,KAAK,wBAAwB;GAC7B,KAAK,sBAAsB,SAAS;GACpC,mBAAmB;EACrB;EAEA,MAAM,4BAA4B,KAAK,4BAA4B;EACnE,OAAO,oBAAoB;CAC7B;CAEA,gBAAwB,WAAmB;EACzC,IAAI;GACF,KAAK,MAAM,KACT,KAAK,UAAU;IACb,IAAI;IACJ,MAAA;GACF,CAAC,CACH;EACF,QAAQ,CAER;CACF;CAEA,oBACE,WACA,sBACA;EACA,KAAK,sBAAsB;EAC3B,KAAK,wBAAwB;CAC/B;CAEA,sBAA8B,WAAmB;EAC/C,IAAI,KAAK,wBAAwB,WAAW;GAC1C,KAAK,sBAAsB;GAC3B,KAAK,wBAAwB;EAC/B;CACF;;;;;CAMA,yBAAyB;EACvB,KAAK,0BAA0B;CACjC;;;;;CAMA,8BAAuC;EACrC,OAAO,KAAK,yBAAyB,KAAK;CAC5C;;;;CAKA,mBAA4B;EAC1B,OAAO,KAAK,oBAAoB,QAAQ,KAAK,wBAAwB;CACvE;;;;;CAMA,sBAA+B;EAC7B,MAAM,cAAc,KAAK;EACzB,IAAI,CAAC,aAAa,OAAO;EACzB,OAAO,YAAY,CAAC,CAAC;CACvB;;;;;CAMA,mBAAyB;EACvB,KAAK,0BAA0B;EAC/B,KAAK,oBAAoB;EACzB,KAAK,sBAAsB;CAC7B;;;;;;CAOA,qBAA8B;EAC5B,MAAM,QAAQ,KAAK;EACnB,IAAI,CAAC,OAAO,OAAO;EACnB,MAAM;EACN,OAAO;CACT;;;;;;;CAQA,qBAAqB,MAA+B;EAClD,IAAI,CAAC,KAAK,iBAAiB,OAAO;EAClC,KAAK,gBAAgB,IAAI;EACzB,OAAO;CACT;;;;;;CAOA,uBAAuB,OAA6B,CAAC,GAAY;EAC/D,IAAI,CAAC,KAAK,qBAAqB,OAAO;EACtC,OAAO,KAAK,oBAAoB,IAAI;CACtC;;;;;;;;CASA,sBAA+B;EAC7B,IAAI,CAAC,KAAK,kBAAkB,OAAO;EACnC,KAAK,iBAAiB;EACtB,OAAO;CACT;;;;;;CAOA,0BAA0B,WAAmB;EAC3C,KAAK,sBAAsB,SAAS;CACtC;;;;;CAMA,kBAAkB,WAAmB;EACnC,KAAK,oBAAoB,WAAW,IAAI;CAC1C;CAEA,MAAM,aAAa,SASyB;EAC1C,MAAM,YAAY,OAAO,CAAC;EAC1B,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,IAAI,YAAY;EAChB,IAAI,cAAc;EAGlB,IAAI,YAAqC,CAAC;EAC1C,IAAI,KAAK,aACP,YAAY,MAAM,KAAK,YAAY;GACjC,UAAU,QAAQ;GAClB,SAAS,QAAQ;GACjB,WAAW,QAAQ;EACrB,CAAC;EAEH,IAAI,QAAQ,MACV,YAAY;GACV,GAAG;GACH,GAAI,QAAQ;EACd;EAGF,MAAM,cAAc,KAAK,UAAU;GACjC,UAAU,QAAQ;GAClB,SAAS,QAAQ;GACjB,GAAG;EACL,CAAC;EAGD,KAAK,kBAAkB,IAAI,SAAS;EAIpC,MAAM,QAAQ,KAAK;EACnB,MAAM,YAAY,KAAK;EAOvB,MAAM,UACJ,QACA,SAAS,OACT,kBAAkB,SACf;GACH,IAAI,WAAW;GACf,YAAY;GACZ,IAAI,iBACF,KAAK,sBAAsB,SAAS;GAEtC,IAAI;IACF,OAAO;GACT,QAAQ,CAER;GACA,IAAI,CAAC,QACH,WAAW,OAAO,SAAS;GAE7B,gBAAgB,MAAM;EACxB;EAEA,MAAM,6BAAa,IAAI,MAAM,SAAS;EACtC,WAAW,OAAO;EAElB,MAAM,4BAA4B;GAChC,IAAI,WAAW,OAAO;GACtB,aAAa,iBAAiB,MAAM,UAAU,GAAG,IAAI;GACrD,OAAO;EACT;EACA,KAAK,oBAAoB,WAAW,mBAAmB;EAMvD,MAAM,gBAAgB;GACpB,IAAI,WAAW;GACf,IAAI,KAAK,qBAAqB;IAC5B,IAAI,aACF,KAAK,gBAAgB,SAAS;IAEhC,aAAa,iBAAiB,MAAM,UAAU,GAAG,WAAW;GAC9D,OACE,aAAa,iBAAiB,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW;EAExE;EAIA,IAAI;EAEJ,MAAM,SAAS,IAAI,eAA+B;GAChD,MAAM,YAAY;IAChB,mBAAmB;IAEnB,MAAM,aAAa,UAAwB;KACzC,IAAI;MACF,MAAM,OAAO,KAAK,MAChB,MAAM,IACR;MAEA,IAAI,KAAK,SAAA,8BAAiD;MAC1D,IAAI,KAAK,OAAO,WAAW;MAE3B,IAAI,KAAK,OAAO;OACd,aACE,WAAW,MAAM,IAAI,MAAM,KAAK,QAAQ,cAAc,CAAC,CACzD;OACA;MACF;MAGA,IAAI,KAAK,MAAM,KAAK,GAClB,IAAI;OACF,MAAM,QAAQ,KAAK,MAAM,KAAK,IAAI;OAClC,WAAW,QAAQ,KAAK;MAC1B,QAAQ,CAER;MAGF,IAAI,KAAK,MACP,aAAa,WAAW,MAAM,CAAC;KAEnC,QAAQ,CAER;IACF;IAEA,MAAM,gBAAgB;KACpB,aAAa,WAAW,MAAM,GAAG,OAAO,KAAK;IAC/C;IAEA,MAAM,iBAAiB,WAAW,WAAW,EAC3C,QAAQ,gBAAgB,OAC1B,CAAC;IACD,MAAM,iBAAiB,SAAS,SAAS,EACvC,QAAQ,gBAAgB,OAC1B,CAAC;GACH;GACA,SAAS;IACP,QAAQ;GACV;EACF,CAAC;EAGD,IAAI,QAAQ,aAAa;GACvB,QAAQ,YAAY,iBAAiB,SAAS,SAAS,EAAE,MAAM,KAAK,CAAC;GACrE,IAAI,QAAQ,YAAY,SAAS,QAAQ;EAC3C;EAEA,IAAI,WACF,OAAO;EAIT,cAAc;EACd,MAAM,KACJ,KAAK,UAAU;GACb,IAAI;GACJ,MAAM;IACJ,QAAQ;IACR,MAAM;GACR;GACA,MAAA;EACF,CAAC,CACH;EAEA,OAAO;CACT;CAEA,MAAM,kBAAkB,UAE2B;EAKjD,IAAI,KAAK,iBAAiB,GAAG,OAAO;EAEpC,IAAI,KAAK,yBAAyB;GAChC,KAAK,0BAA0B;GAC/B,OAAO,KAAK,8BAA8B;EAC5C;EAMA,MAAM,YAAY,KAAK;EACvB,MAAM,UAAU,OAAO,CAAC;EAExB,OAAO,IAAI,SAAgD,YAAY;GACrE,IAAI,WAAW;GACf,IAAI;GACJ,IAAI,iBAA0D;GAC9D,IAAI,qBACF;GACF,IAAI,kBAAuC;GAC3C,IAAI,mBAAwC;GAE5C,MAAM,4BAA4B;IAChC,IAAI,kBAAkB,KAAK,oBAAoB,gBAC7C,KAAK,kBAAkB;IAEzB,IACE,sBACA,KAAK,wBAAwB,oBAE7B,KAAK,sBAAsB;IAE7B,IAAI,mBAAmB,KAAK,qBAAqB,iBAC/C,KAAK,mBAAmB;IAE1B,IAAI,oBAAoB,KAAK,sBAAsB,kBACjD,KAAK,oBAAoB;GAE7B;GAEA,MAAM,QAAQ,UAAiD;IAC7D,IAAI,UAAU;IACd,WAAW;IACX,oBAAoB;IACpB,IAAI,SAAS,aAAa,OAAO;IACjC,QAAQ,KAAK;GACf;GAEA,MAAM,cAAc,UAAkB;IACpC,IAAI,SAAS,aAAa,OAAO;IACjC,UAAU,iBAAiB,KAAK,IAAI,GAAG,KAAK;GAC9C;GAIA,wBAAwB;IACtB,IAAI,UAAU;IACd,WAAW,yBAAyB;GACtC;GAEA,sBAAsB,SAAS;IAC7B,IAAI,KAAK,WAAW,KAAK,YAAY,SAAS,OAAO;IACrD,KAAK,IAAI;IACT,OAAO;GACT;GAEA,kBAAkB,SAAyB;IACzC,MAAM,YAAY,KAAK;IACvB,WAAW,IAAI,SAAS;IACxB,MAAM,SAAS,KAAK,oBAAoB,SAAS;IAEjD,KAAK,MAAM,KACT,KAAK,UAAU;KACb,MAAA;KACA,IAAI;IACN,CAAC,CACH;IAEA,KAAK,MAAM;GACb;GAIA,yBAAyB;IACvB,IAAI,UAAU;IACd,WAAW,uBAAuB;IAClC,IAAI;KACF,KAAK,MAAM,KACT,KAAK,UAAU;MACb,MAAA;MACA;KACF,CAAC,CACH;IACF,QAAQ,CAER;GACF;GAEA,KAAK,mBAAmB;GACxB,KAAK,sBAAsB;GAC3B,KAAK,kBAAkB;GACvB,KAAK,oBAAoB;GACzB,iBAAiB;EACnB,CAAC;CACH;;;;;;;CAQA,gCAAwE;EACtE,MAAM,QAAQ,KAAK;EACnB,MAAM,YAAY,KAAK;EACvB,MAAM,mBAAmB,IAAI,gBAAgB;EAC7C,MAAM,6BAAa,IAAI,MAAM,SAAS;EACtC,WAAW,OAAO;EAClB,IAAI,YAAY;EAChB,IAAI,YAA2B;EAC/B,IAAI,mBACF;EACF,MAAM,UAAU,OAAO,CAAC;EACxB,IAAI,cAAuD;EAC3D,IAAI,kBACF;EACF,IAAI,qBAA0C;EAC9C,IAAI,sBAA2C;EAC/C,IAAI,2BAAmD;EACvD,IAAI,wBAAgD;EAEpD,MAAM,4BAA4B;GAChC,IAAI,eAAe,KAAK,oBAAoB,aAC1C,KAAK,kBAAkB;GAEzB,IAAI,mBAAmB,KAAK,wBAAwB,iBAClD,KAAK,sBAAsB;GAE7B,IAAI,sBAAsB,KAAK,qBAAqB,oBAClD,KAAK,mBAAmB;GAE1B,IACE,uBACA,KAAK,sBAAsB,qBAE3B,KAAK,oBAAoB;EAE7B;EAEA,MAAM,UAAU,QAAoB,gBAAgB,UAAU;GAC5D,IAAI,WAAW;GACf,YAAY;GACZ,IAAI,KAAK,2BAA2B,0BAClC,KAAK,yBAAyB;GAEhC,IAAI,KAAK,wBAAwB,uBAC/B,KAAK,sBAAsB;GAE7B,oBAAoB;GACpB,IAAI;IACF,OAAO;GACT,QAAQ,CAER;GACA,IAAI,aAAa,CAAC,eAChB,WAAW,OAAO,SAAS;GAE7B,iBAAiB,MAAM;EACzB;EAEA,MAAM,YAAY;EAElB,iCAAiC;GAC/B,IAAI,WAAW,OAAO;GAEtB,IAAI,cAAc,MAAM;IACtB,aAAa,kBAAkB,MAAM,UAAU,CAAC;IAChD,OAAO;GACT;GAEA,IAAI;IACF,MAAM,KACJ,KAAK,UAAU;KACb,MAAA;KACA,IAAI;IACN,CAAC,CACH;GACF,QAAQ,CAER;GAIA,aAAa,kBAAkB,MAAM,UAAU,GAAG,IAAI;GACtD,OAAO;EACT;EACA,KAAK,yBAAyB;EAC9B,8BAA8B;GAC5B,IAAI,WAAW,OAAO;GACtB,aAAa,kBAAkB,MAAM,CAAC;GACtC,OAAO;EACT;EACA,KAAK,sBAAsB;EAE3B,OAAO,IAAI,eAA+B;GACxC,MAAM,YAAY;IAChB,mBAAmB;IACnB,IAAI;IAEJ,MAAM,cAAc,UAAkB;KACpC,IAAI,SAAS,aAAa,OAAO;KACjC,UAAU,iBAAiB,aAAa,WAAW,MAAM,CAAC,GAAG,KAAK;IACpE;IAEA,MAAM,gBAAgB,SAA+B;KACnD,IAAI,KAAK,WAAW,KAAK,YAAY,SAAS,OAAO;KACrD,aAAa,WAAW,MAAM,CAAC;KAC/B,OAAO;IACT;IAEA,MAAM,YAAY,SAAyB;KACzC,IAAI,WAAW;KAEf,YAAY,KAAK;KACjB,WAAW,IAAI,SAAS;KACxB,oBAAoB;KACpB,IAAI,SAAS,aAAa,OAAO;KAEjC,MAAM,KACJ,KAAK,UAAU;MACb,MAAA;MACA,IAAI;KACN,CAAC,CACH;IACF;IAEA,MAAM,wBAAwB;KAC5B,IAAI,WAAW;KACf,WAAW,yBAAyB;IACtC;IAEA,MAAM,yBAAyB;KAC7B,IAAI,aAAa,cAAc,MAAM;KACrC,WAAW,uBAAuB;KAClC,IAAI;MACF,UAAU,MAAM,KACd,KAAK,UAAU;OACb,MAAA;OACA;MACF,CAAC,CACH;KACF,QAAQ,CAER;IACF;IAEA,cAAc;IACd,kBAAkB;IAClB,qBAAqB;IACrB,sBAAsB;IACtB,UAAU,kBAAkB;IAC5B,UAAU,sBAAsB;IAChC,UAAU,mBAAmB;IAC7B,UAAU,oBAAoB;IAE9B,MAAM,aAAa,UAAwB;KACzC,IAAI;MACF,MAAM,OAAO,KAAK,MAChB,MAAM,IACR;MAEA,IACE,KAAK,SAAA,gCACL,aAAa,QACb,KAAK,OAAO,WAEZ;MAGF,IAAI,KAAK,OAAO;OACd,aACE,WAAW,MAAM,IAAI,MAAM,KAAK,QAAQ,cAAc,CAAC,CACzD;OACA;MACF;MAEA,IAAI,KAAK,MAAM,KAAK,GAClB,IAAI;OACF,MAAM,QAAQ,KAAK,MAAM,KAAK,IAAI;OAClC,WAAW,QAAQ,KAAK;MAC1B,QAAQ,CAER;MAGF,IAAI,KAAK,MACP,aAAa,WAAW,MAAM,CAAC;KAEnC,QAAQ,CAER;IACF;IAEA,MAAM,gBAAgB,aAAa,WAAW,MAAM,CAAC;IAErD,MAAM,iBAAiB,WAAW,WAAW,EAC3C,QAAQ,iBAAiB,OAC3B,CAAC;IACD,MAAM,iBAAiB,SAAS,SAAS,EACvC,QAAQ,iBAAiB,OAC3B,CAAC;IAED,iBAAiB;GACnB;GACA,SAAS;IACP,IAAI,aAAa,UAAU,qBAAqB;KAC9C,UAAU,gBAAgB,SAAS;KACnC,aAAa,CAAC,GAAG,IAAI;IACvB,OACE,aAAa,CAAC,CAAC;GAEnB;EACF,CAAC;CACH;;;;;CAMA,oBACE,WACgC;EAGhC,MAAM,QAAQ,KAAK;EACnB,MAAM,YAAY,KAAK;EACvB,MAAM,kBAAkB,IAAI,gBAAgB;EAC5C,MAAM,6BAAa,IAAI,MAAM,SAAS;EACtC,WAAW,OAAO;EAClB,IAAI,YAAY;EAChB,IAAI,qBAA6C;EAEjD,MAAM,UACJ,QACA,SAAS,OACT,kBAAkB,SACf;GACH,IAAI,WAAW;GACf,YAAY;GACZ,IAAI,iBACF,KAAK,sBAAsB,SAAS;GAEtC,IAAI,KAAK,wBAAwB,oBAC/B,KAAK,sBAAsB;GAE7B,IAAI;IACF,OAAO;GACT,QAAQ,CAER;GACA,IAAI,CAAC,QACH,WAAW,OAAO,SAAS;GAE7B,gBAAgB,MAAM;EACxB;EAEA,IAAI,mBACF;EACF,MAAM,4BAA4B;GAChC,IAAI,WAAW,OAAO;GACtB,aAAa,kBAAkB,MAAM,UAAU,GAAG,IAAI;GACtD,OAAO;EACT;EACA,KAAK,oBAAoB,WAAW,mBAAmB;EAEvD,MAAM,YAAY;EAClB,2BAA2B;GACzB,IAAI,WAAW,OAAO;GACtB,aAAa,kBAAkB,MAAM,CAAC;GACtC,OAAO;EACT;EACA,KAAK,sBAAsB;EAE3B,OAAO,IAAI,eAA+B;GACxC,MAAM,YAAY;IAChB,mBAAmB;IAEnB,MAAM,aAAa,UAAwB;KACzC,IAAI;MACF,MAAM,OAAO,KAAK,MAChB,MAAM,IACR;MAEA,IAAI,KAAK,SAAA,8BAAiD;MAC1D,IAAI,KAAK,OAAO,WAAW;MAE3B,IAAI,KAAK,OAAO;OACd,aACE,WAAW,MAAM,IAAI,MAAM,KAAK,QAAQ,cAAc,CAAC,CACzD;OACA;MACF;MAGA,IAAI,KAAK,MAAM,KAAK,GAClB,IAAI;OACF,MAAM,QAAQ,KAAK,MAAM,KAAK,IAAI;OAClC,WAAW,QAAQ,KAAK;MAC1B,QAAQ,CAER;MAGF,IAAI,KAAK,MACP,aAAa,WAAW,MAAM,CAAC;KAEnC,QAAQ,CAER;IACF;IAEA,MAAM,gBAAgB;KACpB,aAAa,WAAW,MAAM,GAAG,OAAO,KAAK;IAC/C;IAEA,MAAM,iBAAiB,WAAW,WAAW,EAC3C,QAAQ,gBAAgB,OAC1B,CAAC;IACD,MAAM,iBAAiB,SAAS,SAAS,EACvC,QAAQ,gBAAgB,OAC1B,CAAC;GACH;GACA,SAAS;IACP,IAAI,UAAU,qBAAqB;KACjC,UAAU,gBAAgB,SAAS;KACnC,aAAa,CAAC,GAAG,IAAI;IACvB,OACE,aAAa,CAAC,GAAG,OAAO,KAAK;GAEjC;EACF,CAAC;CACH;AACF"}