@cloudflare/ai-chat 0.0.4 → 0.0.5

package/dist/index.js

@@ -80,6 +80,7 @@ var AIChatAgent = class extends Agent {
 		this._chunkBuffer = [];
 		this._isFlushingChunks = false;
 		this._lastCleanupTime = 0;
+		this._pendingResumeConnections = /* @__PURE__ */ new Set();
 		this.sql`create table if not exists cf_ai_chat_agent_messages (
       id text primary key,
       message text not null,
@@ -105,9 +106,14 @@ var AIChatAgent = class extends Agent {
 		this._chatMessageAbortControllers = /* @__PURE__ */ new Map();
 		this._restoreActiveStream();
 		const _onConnect = this.onConnect.bind(this);
-		this.onConnect = async (connection, ctx$1) => {
+		this.onConnect = async (connection, ctx) => {
 			if (this._activeStreamId) this._notifyStreamResuming(connection);
-			return _onConnect(connection, ctx$1);
+			return _onConnect(connection, ctx);
+		};
+		const _onClose = this.onClose.bind(this);
+		this.onClose = async (connection, code, reason, wasClean) => {
+			this._pendingResumeConnections.delete(connection.id);
+			return _onClose(connection, code, reason, wasClean);
 		};
 		const _onMessage = this.onMessage.bind(this);
 		this.onMessage = async (connection, message) => {
@@ -177,6 +183,7 @@ var AIChatAgent = class extends Agent {
 					this._activeStreamId = null;
 					this._activeRequestId = null;
 					this._streamChunkIndex = 0;
+					this._pendingResumeConnections.clear();
 					this.messages = [];
 					this._broadcastChatMessage({ type: MessageType.CF_AGENT_CHAT_CLEAR }, [connection.id]);
 					return;
@@ -191,6 +198,7 @@ var AIChatAgent = class extends Agent {
 					return;
 				}
 				if (data.type === MessageType.CF_AGENT_STREAM_RESUME_ACK) {
+					this._pendingResumeConnections.delete(connection.id);
 					if (this._activeStreamId && this._activeRequestId && this._activeRequestId === data.id) this._sendStreamChunks(connection, this._activeStreamId, this._activeRequestId);
 					return;
 				}
@@ -274,6 +282,7 @@ var AIChatAgent = class extends Agent {
 	*/
 	_notifyStreamResuming(connection) {
 		if (!this._activeStreamId || !this._activeRequestId) return;
+		this._pendingResumeConnections.add(connection.id);
 		connection.send(JSON.stringify({
 			type: MessageType.CF_AGENT_STREAM_RESUMING,
 			id: this._activeRequestId
@@ -376,6 +385,7 @@ var AIChatAgent = class extends Agent {
 		this._activeStreamId = null;
 		this._activeRequestId = null;
 		this._streamChunkIndex = 0;
+		this._pendingResumeConnections.clear();
 		this._maybeCleanupOldStreams();
 	}
 	/**
@@ -400,7 +410,8 @@ var AIChatAgent = class extends Agent {
     `;
 	}
 	_broadcastChatMessage(message, exclude) {
-		this.broadcast(JSON.stringify(message), exclude);
+		const allExclusions = [...exclude || [], ...this._pendingResumeConnections];
+		this.broadcast(JSON.stringify(message), allExclusions);
 	}
 	/**
 	* Broadcasts a text event for non-SSE responses.
@@ -689,6 +700,374 @@ var AIChatAgent = class extends Agent {
 		}
 		return true;
 	}
+	async _streamSSEReply(id, streamId, reader, message, streamCompleted, continuation = false) {
+		let activeTextParts = {};
+		let activeReasoningParts = {};
+		const partialToolCalls = {};
+		const { getToolName, isToolUIPart, parsePartialJson } = await import("ai");
+		streamCompleted.value = false;
+		while (true) {
+			const { done, value } = await reader.read();
+			if (done) {
+				this._completeStream(streamId);
+				streamCompleted.value = true;
+				this._broadcastChatMessage({
+					body: "",
+					done: true,
+					id,
+					type: MessageType.CF_AGENT_USE_CHAT_RESPONSE,
+					...continuation && { continuation: true }
+				});
+				break;
+			}
+			const lines = decoder.decode(value).split("\n");
+			for (const line of lines) if (line.startsWith("data: ") && line !== "data: [DONE]") try {
+				const data = JSON.parse(line.slice(6));
+				switch (data.type) {
+					case "text-start": {
+						const textPart = {
+							type: "text",
+							text: "",
+							providerMetadata: data.providerMetadata,
+							state: "streaming"
+						};
+						activeTextParts[data.id] = textPart;
+						message.parts.push(textPart);
+						break;
+					}
+					case "text-delta": {
+						const textPart = activeTextParts[data.id];
+						textPart.text += data.delta;
+						textPart.providerMetadata = data.providerMetadata ?? textPart.providerMetadata;
+						break;
+					}
+					case "text-end": {
+						const textPart = activeTextParts[data.id];
+						textPart.state = "done";
+						textPart.providerMetadata = data.providerMetadata ?? textPart.providerMetadata;
+						delete activeTextParts[data.id];
+						break;
+					}
+					case "reasoning-start": {
+						const reasoningPart = {
+							type: "reasoning",
+							text: "",
+							providerMetadata: data.providerMetadata,
+							state: "streaming"
+						};
+						activeReasoningParts[data.id] = reasoningPart;
+						message.parts.push(reasoningPart);
+						break;
+					}
+					case "reasoning-delta": {
+						const reasoningPart = activeReasoningParts[data.id];
+						reasoningPart.text += data.delta;
+						reasoningPart.providerMetadata = data.providerMetadata ?? reasoningPart.providerMetadata;
+						break;
+					}
+					case "reasoning-end": {
+						const reasoningPart = activeReasoningParts[data.id];
+						reasoningPart.providerMetadata = data.providerMetadata ?? reasoningPart.providerMetadata;
+						reasoningPart.state = "done";
+						delete activeReasoningParts[data.id];
+						break;
+					}
+					case "file":
+						message.parts.push({
+							type: "file",
+							mediaType: data.mediaType,
+							url: data.url
+						});
+						break;
+					case "source-url":
+						message.parts.push({
+							type: "source-url",
+							sourceId: data.sourceId,
+							url: data.url,
+							title: data.title,
+							providerMetadata: data.providerMetadata
+						});
+						break;
+					case "source-document":
+						message.parts.push({
+							type: "source-document",
+							sourceId: data.sourceId,
+							mediaType: data.mediaType,
+							title: data.title,
+							filename: data.filename,
+							providerMetadata: data.providerMetadata
+						});
+						break;
+					case "tool-input-start": {
+						const toolInvocations = message.parts.filter(isToolUIPart);
+						partialToolCalls[data.toolCallId] = {
+							text: "",
+							toolName: data.toolName,
+							index: toolInvocations.length,
+							dynamic: data.dynamic
+						};
+						if (data.dynamic) this.updateDynamicToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: data.toolName,
+							state: "input-streaming",
+							input: void 0
+						});
+						else await this.updateToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: data.toolName,
+							state: "input-streaming",
+							input: void 0
+						});
+						break;
+					}
+					case "tool-input-delta": {
+						const partialToolCall = partialToolCalls[data.toolCallId];
+						partialToolCall.text += data.inputTextDelta;
+						const partialArgs = (await parsePartialJson(partialToolCall.text)).value;
+						if (partialToolCall.dynamic) this.updateDynamicToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: partialToolCall.toolName,
+							state: "input-streaming",
+							input: partialArgs
+						});
+						else await this.updateToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: partialToolCall.toolName,
+							state: "input-streaming",
+							input: partialArgs
+						});
+						break;
+					}
+					case "tool-input-available":
+						if (data.dynamic) this.updateDynamicToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: data.toolName,
+							state: "input-available",
+							input: data.input,
+							providerMetadata: data.providerMetadata
+						});
+						else await this.updateToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: data.toolName,
+							state: "input-available",
+							input: data.input,
+							providerExecuted: data.providerExecuted,
+							providerMetadata: data.providerMetadata
+						});
+						break;
+					case "tool-input-error":
+						if (data.dynamic) this.updateDynamicToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: data.toolName,
+							state: "output-error",
+							input: data.input,
+							errorText: data.errorText,
+							providerMetadata: data.providerMetadata
+						});
+						else await this.updateToolPart(message, {
+							toolCallId: data.toolCallId,
+							toolName: data.toolName,
+							state: "output-error",
+							input: void 0,
+							rawInput: data.input,
+							errorText: data.errorText,
+							providerExecuted: data.providerExecuted,
+							providerMetadata: data.providerMetadata
+						});
+						break;
+					case "tool-output-available":
+						if (data.dynamic) {
+							const toolInvocation = message.parts.filter((part) => part.type === "dynamic-tool").find((invocation) => invocation.toolCallId === data.toolCallId);
+							if (!toolInvocation) throw new Error("Tool invocation not found");
+							this.updateDynamicToolPart(message, {
+								toolCallId: data.toolCallId,
+								toolName: toolInvocation.toolName,
+								state: "output-available",
+								input: toolInvocation.input,
+								output: data.output,
+								preliminary: data.preliminary
+							});
+						} else {
+							const toolInvocation = message.parts.filter(isToolUIPart).find((invocation) => invocation.toolCallId === data.toolCallId);
+							if (!toolInvocation) throw new Error("Tool invocation not found");
+							await this.updateToolPart(message, {
+								toolCallId: data.toolCallId,
+								toolName: getToolName(toolInvocation),
+								state: "output-available",
+								input: toolInvocation.input,
+								output: data.output,
+								providerExecuted: data.providerExecuted,
+								preliminary: data.preliminary
+							});
+						}
+						break;
+					case "tool-output-error":
+						if (data.dynamic) {
+							const toolInvocation = message.parts.filter((part) => part.type === "dynamic-tool").find((invocation) => invocation.toolCallId === data.toolCallId);
+							if (!toolInvocation) throw new Error("Tool invocation not found");
+							this.updateDynamicToolPart(message, {
+								toolCallId: data.toolCallId,
+								toolName: toolInvocation.toolName,
+								state: "output-error",
+								input: toolInvocation.input,
+								errorText: data.errorText
+							});
+						} else {
+							const toolInvocation = message.parts.filter(isToolUIPart).find((invocation) => invocation.toolCallId === data.toolCallId);
+							if (!toolInvocation) throw new Error("Tool invocation not found");
+							await this.updateToolPart(message, {
+								toolCallId: data.toolCallId,
+								toolName: getToolName(toolInvocation),
+								state: "output-error",
+								input: toolInvocation.input,
+								rawInput: "rawInput" in toolInvocation ? toolInvocation.rawInput : void 0,
+								errorText: data.errorText
+							});
+						}
+						break;
+					case "start-step":
+						message.parts.push({ type: "step-start" });
+						break;
+					case "finish-step":
+						activeTextParts = {};
+						activeReasoningParts = {};
+						break;
+					case "start":
+						if (data.messageId != null) message.id = data.messageId;
+						await this.updateMessageMetadata(message, data.messageMetadata);
+						break;
+					case "finish":
+						await this.updateMessageMetadata(message, data.messageMetadata);
+						break;
+					case "message-metadata":
+						await this.updateMessageMetadata(message, data.messageMetadata);
+						break;
+					case "error":
+						this._broadcastChatMessage({
+							error: true,
+							body: data.errorText ?? JSON.stringify(data),
+							done: false,
+							id,
+							type: MessageType.CF_AGENT_USE_CHAT_RESPONSE
+						});
+						break;
+				}
+				let eventToSend = data;
+				if (data.type === "finish" && "finishReason" in data) {
+					const { finishReason, ...rest } = data;
+					eventToSend = {
+						...rest,
+						type: "finish",
+						messageMetadata: { finishReason }
+					};
+				}
+				const chunkBody = JSON.stringify(eventToSend);
+				this._storeStreamChunk(streamId, chunkBody);
+				this._broadcastChatMessage({
+					body: chunkBody,
+					done: false,
+					id,
+					type: MessageType.CF_AGENT_USE_CHAT_RESPONSE,
+					...continuation && { continuation: true }
+				});
+			} catch (_error) {}
+		}
+	}
+	async _sendPlaintextReply(id, streamId, reader, message, streamCompleted, continuation = false) {
+		this._broadcastTextEvent(streamId, {
+			type: "text-start",
+			id
+		}, continuation);
+		while (true) {
+			const { done, value } = await reader.read();
+			if (done) {
+				this._broadcastTextEvent(streamId, {
+					type: "text-end",
+					id
+				}, continuation);
+				this._completeStream(streamId);
+				streamCompleted.value = true;
+				this._broadcastChatMessage({
+					body: "",
+					done: true,
+					id,
+					type: MessageType.CF_AGENT_USE_CHAT_RESPONSE,
+					...continuation && { continuation: true }
+				});
+				break;
+			}
+			const chunk = decoder.decode(value);
+			if (chunk.length > 0) {
+				message.parts.push({
+					type: "text",
+					text: chunk
+				});
+				this._broadcastTextEvent(streamId, {
+					type: "text-delta",
+					id,
+					delta: chunk
+				}, continuation);
+			}
+		}
+	}
+	updateDynamicToolPart(message, options) {
+		const part = message.parts.find((part) => part.type === "dynamic-tool" && part.toolCallId === options.toolCallId);
+		const anyOptions = options;
+		const anyPart = part;
+		if (part != null) {
+			part.state = options.state;
+			anyPart.toolName = options.toolName;
+			anyPart.input = anyOptions.input;
+			anyPart.output = anyOptions.output;
+			anyPart.errorText = anyOptions.errorText;
+			anyPart.rawInput = anyOptions.rawInput ?? anyPart.rawInput;
+			anyPart.preliminary = anyOptions.preliminary;
+			if (anyOptions.providerMetadata != null && part.state === "input-available") part.callProviderMetadata = anyOptions.providerMetadata;
+		} else message.parts.push({
+			type: "dynamic-tool",
+			toolName: options.toolName,
+			toolCallId: options.toolCallId,
+			state: options.state,
+			input: anyOptions.input,
+			output: anyOptions.output,
+			errorText: anyOptions.errorText,
+			preliminary: anyOptions.preliminary,
+			...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
+		});
+	}
+	async updateToolPart(message, options) {
+		const { isToolUIPart } = await import("ai");
+		const part = message.parts.find((part) => isToolUIPart(part) && part.toolCallId === options.toolCallId);
+		const anyOptions = options;
+		const anyPart = part;
+		if (part != null) {
+			part.state = options.state;
+			anyPart.input = anyOptions.input;
+			anyPart.output = anyOptions.output;
+			anyPart.errorText = anyOptions.errorText;
+			anyPart.rawInput = anyOptions.rawInput;
+			anyPart.preliminary = anyOptions.preliminary;
+			anyPart.providerExecuted = anyOptions.providerExecuted ?? part.providerExecuted;
+			if (anyOptions.providerMetadata != null && part.state === "input-available") part.callProviderMetadata = anyOptions.providerMetadata;
+		} else message.parts.push({
+			type: `tool-${options.toolName}`,
+			toolCallId: options.toolCallId,
+			state: options.state,
+			input: anyOptions.input,
+			output: anyOptions.output,
+			rawInput: anyOptions.rawInput,
+			errorText: anyOptions.errorText,
+			providerExecuted: anyOptions.providerExecuted,
+			preliminary: anyOptions.preliminary,
+			...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
+		});
+	}
+	async updateMessageMetadata(message, metadata) {
+		if (metadata != null) message.metadata = message.metadata != null ? {
+			...message.metadata,
+			...metadata
+		} : metadata;
+	}
 	async _reply(id, response, excludeBroadcastIds = [], options = {}) {
 		const { continuation = false } = options;
 		return this._tryCatchChat(async () => {
@@ -703,7 +1082,6 @@ var AIChatAgent = class extends Agent {
 				return;
 			}
 			const streamId = this._startStream(id);
-			const { getToolName: getToolName$1, isToolUIPart: isToolUIPart$1, parsePartialJson } = await import("ai");
 			const reader = response.body.getReader();
 			const message = {
 				id: `assistant_${Date.now()}_${Math.random().toString(36).slice(2, 11)}`,
@@ -714,358 +1092,13 @@ var AIChatAgent = class extends Agent {
 			this._streamCompletionPromise = new Promise((resolve) => {
 				this._streamCompletionResolve = resolve;
 			});
-			let activeTextParts = {};
-			let activeReasoningParts = {};
-			const partialToolCalls = {};
-			function updateDynamicToolPart(options$1) {
-				const part = message.parts.find((part$1) => part$1.type === "dynamic-tool" && part$1.toolCallId === options$1.toolCallId);
-				const anyOptions = options$1;
-				const anyPart = part;
-				if (part != null) {
-					part.state = options$1.state;
-					anyPart.toolName = options$1.toolName;
-					anyPart.input = anyOptions.input;
-					anyPart.output = anyOptions.output;
-					anyPart.errorText = anyOptions.errorText;
-					anyPart.rawInput = anyOptions.rawInput ?? anyPart.rawInput;
-					anyPart.preliminary = anyOptions.preliminary;
-					if (anyOptions.providerMetadata != null && part.state === "input-available") part.callProviderMetadata = anyOptions.providerMetadata;
-				} else message.parts.push({
-					type: "dynamic-tool",
-					toolName: options$1.toolName,
-					toolCallId: options$1.toolCallId,
-					state: options$1.state,
-					input: anyOptions.input,
-					output: anyOptions.output,
-					errorText: anyOptions.errorText,
-					preliminary: anyOptions.preliminary,
-					...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
-				});
-			}
-			function updateToolPart(options$1) {
-				const part = message.parts.find((part$1) => isToolUIPart$1(part$1) && part$1.toolCallId === options$1.toolCallId);
-				const anyOptions = options$1;
-				const anyPart = part;
-				if (part != null) {
-					part.state = options$1.state;
-					anyPart.input = anyOptions.input;
-					anyPart.output = anyOptions.output;
-					anyPart.errorText = anyOptions.errorText;
-					anyPart.rawInput = anyOptions.rawInput;
-					anyPart.preliminary = anyOptions.preliminary;
-					anyPart.providerExecuted = anyOptions.providerExecuted ?? part.providerExecuted;
-					if (anyOptions.providerMetadata != null && part.state === "input-available") part.callProviderMetadata = anyOptions.providerMetadata;
-				} else message.parts.push({
-					type: `tool-${options$1.toolName}`,
-					toolCallId: options$1.toolCallId,
-					state: options$1.state,
-					input: anyOptions.input,
-					output: anyOptions.output,
-					rawInput: anyOptions.rawInput,
-					errorText: anyOptions.errorText,
-					providerExecuted: anyOptions.providerExecuted,
-					preliminary: anyOptions.preliminary,
-					...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
-				});
-			}
-			async function updateMessageMetadata(metadata) {
-				if (metadata != null) message.metadata = message.metadata != null ? {
-					...message.metadata,
-					...metadata
-				} : metadata;
-			}
 			const isSSE = (response.headers.get("content-type") || "").includes("text/event-stream");
-			if (!isSSE) this._broadcastTextEvent(streamId, {
-				type: "text-start",
-				id
-			}, continuation);
-			let streamCompleted = false;
+			const streamCompleted = { value: false };
 			try {
-				while (true) {
-					const { done, value } = await reader.read();
-					if (done) {
-						if (!isSSE) this._broadcastTextEvent(streamId, {
-							type: "text-end",
-							id
-						}, continuation);
-						this._completeStream(streamId);
-						streamCompleted = true;
-						this._broadcastChatMessage({
-							body: "",
-							done: true,
-							id,
-							type: MessageType.CF_AGENT_USE_CHAT_RESPONSE,
-							...continuation && { continuation: true }
-						});
-						break;
-					}
-					const chunk = decoder.decode(value);
-					if (isSSE) {
-						const lines = chunk.split("\n");
-						for (const line of lines) if (line.startsWith("data: ") && line !== "data: [DONE]") try {
-							const data = JSON.parse(line.slice(6));
-							switch (data.type) {
-								case "text-start": {
-									const textPart = {
-										type: "text",
-										text: "",
-										providerMetadata: data.providerMetadata,
-										state: "streaming"
-									};
-									activeTextParts[data.id] = textPart;
-									message.parts.push(textPart);
-									break;
-								}
-								case "text-delta": {
-									const textPart = activeTextParts[data.id];
-									textPart.text += data.delta;
-									textPart.providerMetadata = data.providerMetadata ?? textPart.providerMetadata;
-									break;
-								}
-								case "text-end": {
-									const textPart = activeTextParts[data.id];
-									textPart.state = "done";
-									textPart.providerMetadata = data.providerMetadata ?? textPart.providerMetadata;
-									delete activeTextParts[data.id];
-									break;
-								}
-								case "reasoning-start": {
-									const reasoningPart = {
-										type: "reasoning",
-										text: "",
-										providerMetadata: data.providerMetadata,
-										state: "streaming"
-									};
-									activeReasoningParts[data.id] = reasoningPart;
-									message.parts.push(reasoningPart);
-									break;
-								}
-								case "reasoning-delta": {
-									const reasoningPart = activeReasoningParts[data.id];
-									reasoningPart.text += data.delta;
-									reasoningPart.providerMetadata = data.providerMetadata ?? reasoningPart.providerMetadata;
-									break;
-								}
-								case "reasoning-end": {
-									const reasoningPart = activeReasoningParts[data.id];
-									reasoningPart.providerMetadata = data.providerMetadata ?? reasoningPart.providerMetadata;
-									reasoningPart.state = "done";
-									delete activeReasoningParts[data.id];
-									break;
-								}
-								case "file":
-									message.parts.push({
-										type: "file",
-										mediaType: data.mediaType,
-										url: data.url
-									});
-									break;
-								case "source-url":
-									message.parts.push({
-										type: "source-url",
-										sourceId: data.sourceId,
-										url: data.url,
-										title: data.title,
-										providerMetadata: data.providerMetadata
-									});
-									break;
-								case "source-document":
-									message.parts.push({
-										type: "source-document",
-										sourceId: data.sourceId,
-										mediaType: data.mediaType,
-										title: data.title,
-										filename: data.filename,
-										providerMetadata: data.providerMetadata
-									});
-									break;
-								case "tool-input-start": {
-									const toolInvocations = message.parts.filter(isToolUIPart$1);
-									partialToolCalls[data.toolCallId] = {
-										text: "",
-										toolName: data.toolName,
-										index: toolInvocations.length,
-										dynamic: data.dynamic
-									};
-									if (data.dynamic) updateDynamicToolPart({
-										toolCallId: data.toolCallId,
-										toolName: data.toolName,
-										state: "input-streaming",
-										input: void 0
-									});
-									else updateToolPart({
-										toolCallId: data.toolCallId,
-										toolName: data.toolName,
-										state: "input-streaming",
-										input: void 0
-									});
-									break;
-								}
-								case "tool-input-delta": {
-									const partialToolCall = partialToolCalls[data.toolCallId];
-									partialToolCall.text += data.inputTextDelta;
-									const partialArgs = (await parsePartialJson(partialToolCall.text)).value;
-									if (partialToolCall.dynamic) updateDynamicToolPart({
-										toolCallId: data.toolCallId,
-										toolName: partialToolCall.toolName,
-										state: "input-streaming",
-										input: partialArgs
-									});
-									else updateToolPart({
-										toolCallId: data.toolCallId,
-										toolName: partialToolCall.toolName,
-										state: "input-streaming",
-										input: partialArgs
-									});
-									break;
-								}
-								case "tool-input-available":
-									if (data.dynamic) updateDynamicToolPart({
-										toolCallId: data.toolCallId,
-										toolName: data.toolName,
-										state: "input-available",
-										input: data.input,
-										providerMetadata: data.providerMetadata
-									});
-									else updateToolPart({
-										toolCallId: data.toolCallId,
-										toolName: data.toolName,
-										state: "input-available",
-										input: data.input,
-										providerExecuted: data.providerExecuted,
-										providerMetadata: data.providerMetadata
-									});
-									break;
-								case "tool-input-error":
-									if (data.dynamic) updateDynamicToolPart({
-										toolCallId: data.toolCallId,
-										toolName: data.toolName,
-										state: "output-error",
-										input: data.input,
-										errorText: data.errorText,
-										providerMetadata: data.providerMetadata
-									});
-									else updateToolPart({
-										toolCallId: data.toolCallId,
-										toolName: data.toolName,
-										state: "output-error",
-										input: void 0,
-										rawInput: data.input,
-										errorText: data.errorText,
-										providerExecuted: data.providerExecuted,
-										providerMetadata: data.providerMetadata
-									});
-									break;
-								case "tool-output-available":
-									if (data.dynamic) {
-										const toolInvocation = message.parts.filter((part) => part.type === "dynamic-tool").find((invocation) => invocation.toolCallId === data.toolCallId);
-										if (!toolInvocation) throw new Error("Tool invocation not found");
-										updateDynamicToolPart({
-											toolCallId: data.toolCallId,
-											toolName: toolInvocation.toolName,
-											state: "output-available",
-											input: toolInvocation.input,
-											output: data.output,
-											preliminary: data.preliminary
-										});
-									} else {
-										const toolInvocation = message.parts.filter(isToolUIPart$1).find((invocation) => invocation.toolCallId === data.toolCallId);
-										if (!toolInvocation) throw new Error("Tool invocation not found");
-										updateToolPart({
-											toolCallId: data.toolCallId,
-											toolName: getToolName$1(toolInvocation),
-											state: "output-available",
-											input: toolInvocation.input,
-											output: data.output,
-											providerExecuted: data.providerExecuted,
-											preliminary: data.preliminary
-										});
-									}
-									break;
-								case "tool-output-error":
-									if (data.dynamic) {
-										const toolInvocation = message.parts.filter((part) => part.type === "dynamic-tool").find((invocation) => invocation.toolCallId === data.toolCallId);
-										if (!toolInvocation) throw new Error("Tool invocation not found");
-										updateDynamicToolPart({
-											toolCallId: data.toolCallId,
-											toolName: toolInvocation.toolName,
-											state: "output-error",
-											input: toolInvocation.input,
-											errorText: data.errorText
-										});
-									} else {
-										const toolInvocation = message.parts.filter(isToolUIPart$1).find((invocation) => invocation.toolCallId === data.toolCallId);
-										if (!toolInvocation) throw new Error("Tool invocation not found");
-										updateToolPart({
-											toolCallId: data.toolCallId,
-											toolName: getToolName$1(toolInvocation),
-											state: "output-error",
-											input: toolInvocation.input,
-											rawInput: "rawInput" in toolInvocation ? toolInvocation.rawInput : void 0,
-											errorText: data.errorText
-										});
-									}
-									break;
-								case "start-step":
-									message.parts.push({ type: "step-start" });
-									break;
-								case "finish-step":
-									activeTextParts = {};
-									activeReasoningParts = {};
-									break;
-								case "start":
-									if (data.messageId != null) message.id = data.messageId;
-									await updateMessageMetadata(data.messageMetadata);
-									break;
-								case "finish":
-									await updateMessageMetadata(data.messageMetadata);
-									break;
-								case "message-metadata":
-									await updateMessageMetadata(data.messageMetadata);
-									break;
-								case "error":
-									this._broadcastChatMessage({
-										error: true,
-										body: data.errorText ?? JSON.stringify(data),
-										done: false,
-										id,
-										type: MessageType.CF_AGENT_USE_CHAT_RESPONSE
-									});
-									break;
-							}
-							let eventToSend = data;
-							if (data.type === "finish" && "finishReason" in data) {
-								const { finishReason, ...rest } = data;
-								eventToSend = {
-									...rest,
-									type: "finish",
-									messageMetadata: { finishReason }
-								};
-							}
-							const chunkBody = JSON.stringify(eventToSend);
-							this._storeStreamChunk(streamId, chunkBody);
-							this._broadcastChatMessage({
-								body: chunkBody,
-								done: false,
-								id,
-								type: MessageType.CF_AGENT_USE_CHAT_RESPONSE,
-								...continuation && { continuation: true }
-							});
-						} catch (_error) {}
-					} else if (chunk.length > 0) {
-						message.parts.push({
-							type: "text",
-							text: chunk
-						});
-						this._broadcastTextEvent(streamId, {
-							type: "text-delta",
-							id,
-							delta: chunk
-						}, continuation);
-					}
-				}
+				if (isSSE) await this._streamSSEReply(id, streamId, reader, message, streamCompleted, continuation);
+				else await this._sendPlaintextReply(id, streamId, reader, message, streamCompleted, continuation);
 			} catch (error) {
-				if (!streamCompleted) {
+				if (!streamCompleted.value) {
 					this._markStreamError(streamId);
 					this._broadcastChatMessage({
 						body: error instanceof Error ? error.message : "Stream error",