agents 0.11.9 → 0.12.1

package/dist/browser/ai.d.ts

@@ -1,4 +1,4 @@
-import { t as BrowserToolsOptions } from "../shared-BUHZFGTk.js";
+import { t as BrowserToolsOptions } from "../shared-Ch9slKdI.js";
 import { ToolSet } from "ai";
 
 //#region src/browser/ai.d.ts

package/dist/browser/ai.js

@@ -1,6 +1,6 @@
-import { n as SEARCH_DESCRIPTION, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-mfBbxjS1.js";
-import { z } from "zod";
+import { n as SEARCH_DESCRIPTION, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-C6l4ZKRN.js";
 import { tool } from "ai";
+import { z } from "zod";
 //#region src/browser/ai.ts
 /**
 * Create AI SDK tools for browser automation via CDP code mode.

package/dist/browser/index.d.ts

@@ -1,4 +1,4 @@
-import { a as createBrowserToolHandlers, i as ToolResult, n as EXECUTE_DESCRIPTION, r as SEARCH_DESCRIPTION, t as BrowserToolsOptions } from "../shared-BUHZFGTk.js";
+import { a as createBrowserToolHandlers, i as ToolResult, n as EXECUTE_DESCRIPTION, r as SEARCH_DESCRIPTION, t as BrowserToolsOptions } from "../shared-Ch9slKdI.js";
 
 //#region src/browser/cdp-session.d.ts
 interface DebugEntry {

package/dist/browser/index.js

@@ -1,2 +1,2 @@
-import { a as connectBrowser, i as CdpSession, n as SEARCH_DESCRIPTION, o as connectUrl, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-mfBbxjS1.js";
+import { a as connectBrowser, i as CdpSession, n as SEARCH_DESCRIPTION, o as connectUrl, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-C6l4ZKRN.js";
 export { CdpSession, EXECUTE_DESCRIPTION, SEARCH_DESCRIPTION, connectBrowser, connectUrl, createBrowserToolHandlers };

package/dist/browser/tanstack-ai.d.ts

@@ -1,4 +1,4 @@
-import { t as BrowserToolsOptions } from "../shared-BUHZFGTk.js";
+import { t as BrowserToolsOptions } from "../shared-Ch9slKdI.js";
 import { ServerTool } from "@tanstack/ai";
 
 //#region src/browser/tanstack-ai.d.ts

package/dist/browser/tanstack-ai.js

@@ -1,4 +1,4 @@
-import { n as SEARCH_DESCRIPTION, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-mfBbxjS1.js";
+import { n as SEARCH_DESCRIPTION, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-C6l4ZKRN.js";
 import { z } from "zod";
 import { toolDefinition } from "@tanstack/ai";
 //#region src/browser/tanstack-ai.ts

package/dist/chat/index.d.ts

@@ -1,3 +1,5 @@
+import { a as AgentToolEventState, i as AgentToolEventMessage, l as AgentToolRunState, r as AgentToolEvent } from "../agent-tool-types-CB7nISDE.js";
+import { n as createAgentToolEventState, t as applyAgentToolEvent } from "../agent-tools-Bb1O8blK.js";
 import { JSONSchema7, Tool, ToolSet, UIMessage } from "ai";
 import { Connection } from "agents";
 
@@ -501,6 +503,7 @@ declare class ResumableStream {
    *          When non-null the caller should reconstruct the message from chunks.
    */
   replayChunks(connection: Connection, requestId: string): string | null;
+  replayCompletedChunksByRequestId(connection: Connection, requestId: string): boolean;
   /**
    * Restore active stream state if the agent was restarted during streaming.
    * All streams are restored regardless of age — stale cleanup happens
@@ -836,5 +839,5 @@ declare function resolveToolMergeId(message: UIMessage, serverMessages: readonly
  */
 declare function assistantContentKey(message: UIMessage, sanitize?: (message: UIMessage) => UIMessage): string | undefined;
 //#endregion
-export { AbortRegistry, type BroadcastStreamEvent, type BroadcastStreamState, type TransitionResult as BroadcastTransitionResult, CHAT_MESSAGE_TYPES, type ChatProtocolEvent, type ChatRecoveryContext, type ChatRecoveryOptions, type ChatResponseResult, type ChunkAction, type ChunkResult, type ClientToolSchema, type ContinuationConnection, type ContinuationDeferred, type ContinuationPending, ContinuationState, type EnqueueOptions, type MessageConcurrency, type MessagePart, type MessageParts, type NormalizedMessageConcurrency, ROW_MAX_BYTES, ResumableStream, type SaveMessagesOptions, type SaveMessagesResult, type SqlTaggedTemplate, StreamAccumulator, type StreamAccumulatorOptions, type StreamChunkData, SubmitConcurrencyController, type SubmitConcurrencyDecision, type ToolPartUpdate, TurnQueue, type TurnResult, applyChunkToParts, applyToolUpdate, assistantContentKey, transition as broadcastTransition, byteLength, createToolsFromClientSchemas, enforceRowSizeLimit, isReplayChunk, parseProtocolMessage, reconcileMessages, resolveToolMergeId, sanitizeMessage, toolApprovalUpdate, toolResultUpdate };
+export { AbortRegistry, type AgentToolEvent, type AgentToolEventMessage, type AgentToolEventState, type AgentToolRunState, type BroadcastStreamEvent, type BroadcastStreamState, type TransitionResult as BroadcastTransitionResult, CHAT_MESSAGE_TYPES, type ChatProtocolEvent, type ChatRecoveryContext, type ChatRecoveryOptions, type ChatResponseResult, type ChunkAction, type ChunkResult, type ClientToolSchema, type ContinuationConnection, type ContinuationDeferred, type ContinuationPending, ContinuationState, type EnqueueOptions, type MessageConcurrency, type MessagePart, type MessageParts, type NormalizedMessageConcurrency, ROW_MAX_BYTES, ResumableStream, type SaveMessagesOptions, type SaveMessagesResult, type SqlTaggedTemplate, StreamAccumulator, type StreamAccumulatorOptions, type StreamChunkData, SubmitConcurrencyController, type SubmitConcurrencyDecision, type ToolPartUpdate, TurnQueue, type TurnResult, applyAgentToolEvent, applyChunkToParts, applyToolUpdate, assistantContentKey, transition as broadcastTransition, byteLength, createAgentToolEventState, createToolsFromClientSchemas, enforceRowSizeLimit, isReplayChunk, parseProtocolMessage, reconcileMessages, resolveToolMergeId, sanitizeMessage, toolApprovalUpdate, toolResultUpdate };
 //# sourceMappingURL=index.d.ts.map

package/dist/chat/index.js

@@ -1,303 +1,6 @@
-import { nanoid } from "nanoid";
+import { i as isReplayChunk, n as createAgentToolEventState, r as applyChunkToParts, t as applyAgentToolEvent } from "../agent-tools-BAdX1vdI.js";
 import { jsonSchema, tool } from "ai";
-//#region src/chat/message-builder.ts
-/**
-* Applies a stream chunk to a mutable parts array, building up the message
-* incrementally. Returns true if the chunk was handled, false if it was
-* an unrecognized type (caller may handle it with additional logic).
-*
-* Handles all common chunk types that both server and client need:
-* - text-start / text-delta / text-end
-* - reasoning-start / reasoning-delta / reasoning-end
-* - file
-* - source-url / source-document
-* - tool-input-start / tool-input-delta / tool-input-available / tool-input-error
-* - tool-output-available / tool-output-error
-* - step-start (aliased from start-step)
-* - data-* (developer-defined typed JSON blobs)
-*
-* @param parts - The mutable parts array to update
-* @param chunk - The parsed stream chunk data
-* @returns true if handled, false if the chunk type is not recognized
-*/
-function applyChunkToParts(parts, chunk) {
-	switch (chunk.type) {
-		case "text-start":
-			parts.push({
-				type: "text",
-				text: "",
-				state: "streaming"
-			});
-			return true;
-		case "text-delta": {
-			const lastTextPart = findLastPartByType(parts, "text");
-			if (lastTextPart && lastTextPart.type === "text") lastTextPart.text += chunk.delta ?? "";
-			else parts.push({
-				type: "text",
-				text: chunk.delta ?? "",
-				state: "streaming"
-			});
-			return true;
-		}
-		case "text-end": {
-			const lastTextPart = findLastPartByType(parts, "text");
-			if (lastTextPart && "state" in lastTextPart) lastTextPart.state = "done";
-			return true;
-		}
-		case "reasoning-start":
-			parts.push({
-				type: "reasoning",
-				text: "",
-				state: "streaming"
-			});
-			return true;
-		case "reasoning-delta": {
-			const lastReasoningPart = findLastPartByType(parts, "reasoning");
-			if (lastReasoningPart && lastReasoningPart.type === "reasoning") {
-				lastReasoningPart.text += chunk.delta ?? "";
-				mergeProviderMetadata(lastReasoningPart, chunk.providerMetadata);
-			} else parts.push({
-				type: "reasoning",
-				text: chunk.delta ?? "",
-				state: "streaming",
-				...chunk.providerMetadata != null ? { providerMetadata: chunk.providerMetadata } : {}
-			});
-			return true;
-		}
-		case "reasoning-end": {
-			const lastReasoningPart = findLastPartByType(parts, "reasoning");
-			if (lastReasoningPart && "state" in lastReasoningPart) {
-				lastReasoningPart.state = "done";
-				mergeProviderMetadata(lastReasoningPart, chunk.providerMetadata);
-			}
-			return true;
-		}
-		case "file":
-			parts.push({
-				type: "file",
-				mediaType: chunk.mediaType,
-				url: chunk.url
-			});
-			return true;
-		case "source-url":
-			parts.push({
-				type: "source-url",
-				sourceId: chunk.sourceId,
-				url: chunk.url,
-				title: chunk.title,
-				providerMetadata: chunk.providerMetadata
-			});
-			return true;
-		case "source-document":
-			parts.push({
-				type: "source-document",
-				sourceId: chunk.sourceId,
-				mediaType: chunk.mediaType,
-				title: chunk.title,
-				filename: chunk.filename,
-				providerMetadata: chunk.providerMetadata
-			});
-			return true;
-		case "tool-input-start":
-			if (findToolPartByCallId(parts, chunk.toolCallId)) return true;
-			parts.push({
-				type: `tool-${chunk.toolName}`,
-				toolCallId: chunk.toolCallId,
-				toolName: chunk.toolName,
-				state: "input-streaming",
-				input: void 0,
-				...chunk.providerExecuted != null ? { providerExecuted: chunk.providerExecuted } : {},
-				...chunk.providerMetadata != null ? { callProviderMetadata: chunk.providerMetadata } : {},
-				...chunk.title != null ? { title: chunk.title } : {}
-			});
-			return true;
-		case "tool-input-delta": {
-			const toolPart = findToolPartByCallId(parts, chunk.toolCallId);
-			if (toolPart && toolPart.state === "input-streaming") toolPart.input = chunk.input;
-			return true;
-		}
-		case "tool-input-available": {
-			const existing = findToolPartByCallId(parts, chunk.toolCallId);
-			if (existing) {
-				const p = existing;
-				if (p.state === "input-streaming") {
-					p.state = "input-available";
-					p.input = chunk.input;
-					if (chunk.providerExecuted != null) p.providerExecuted = chunk.providerExecuted;
-					if (chunk.providerMetadata != null) p.callProviderMetadata = chunk.providerMetadata;
-					if (chunk.title != null) p.title = chunk.title;
-				}
-				return true;
-			}
-			parts.push({
-				type: `tool-${chunk.toolName}`,
-				toolCallId: chunk.toolCallId,
-				toolName: chunk.toolName,
-				state: "input-available",
-				input: chunk.input,
-				...chunk.providerExecuted != null ? { providerExecuted: chunk.providerExecuted } : {},
-				...chunk.providerMetadata != null ? { callProviderMetadata: chunk.providerMetadata } : {},
-				...chunk.title != null ? { title: chunk.title } : {}
-			});
-			return true;
-		}
-		case "tool-input-error": {
-			const existing = findToolPartByCallId(parts, chunk.toolCallId);
-			if (existing) {
-				const p = existing;
-				if (p.state === "output-available" || p.state === "output-error" || p.state === "output-denied") return true;
-				p.state = "output-error";
-				p.errorText = chunk.errorText;
-				p.input = chunk.input;
-				if (chunk.providerExecuted != null) p.providerExecuted = chunk.providerExecuted;
-				if (chunk.providerMetadata != null) p.callProviderMetadata = chunk.providerMetadata;
-			} else parts.push({
-				type: `tool-${chunk.toolName}`,
-				toolCallId: chunk.toolCallId,
-				toolName: chunk.toolName,
-				state: "output-error",
-				input: chunk.input,
-				errorText: chunk.errorText,
-				...chunk.providerExecuted != null ? { providerExecuted: chunk.providerExecuted } : {},
-				...chunk.providerMetadata != null ? { callProviderMetadata: chunk.providerMetadata } : {}
-			});
-			return true;
-		}
-		case "tool-approval-request": {
-			const toolPart = findToolPartByCallId(parts, chunk.toolCallId);
-			if (toolPart) {
-				const p = toolPart;
-				p.state = "approval-requested";
-				p.approval = { id: chunk.approvalId };
-			}
-			return true;
-		}
-		case "tool-output-denied": {
-			const toolPart = findToolPartByCallId(parts, chunk.toolCallId);
-			if (toolPart) {
-				const p = toolPart;
-				p.state = "output-denied";
-			}
-			return true;
-		}
-		case "tool-output-available": {
-			const toolPart = findToolPartByCallId(parts, chunk.toolCallId);
-			if (toolPart) {
-				const p = toolPart;
-				p.state = "output-available";
-				p.output = chunk.output;
-				if (chunk.preliminary !== void 0) p.preliminary = chunk.preliminary;
-			}
-			return true;
-		}
-		case "tool-output-error": {
-			const toolPart = findToolPartByCallId(parts, chunk.toolCallId);
-			if (toolPart) {
-				const p = toolPart;
-				p.state = "output-error";
-				p.errorText = chunk.errorText;
-			}
-			return true;
-		}
-		case "step-start":
-		case "start-step":
-			parts.push({ type: "step-start" });
-			return true;
-		default:
-			if (chunk.type.startsWith("data-")) {
-				if (chunk.transient) return true;
-				if (chunk.id != null) {
-					const existing = findDataPartByTypeAndId(parts, chunk.type, chunk.id);
-					if (existing) {
-						existing.data = chunk.data;
-						return true;
-					}
-				}
-				parts.push({
-					type: chunk.type,
-					...chunk.id != null && { id: chunk.id },
-					data: chunk.data
-				});
-				return true;
-			}
-			return false;
-	}
-}
-/**
-* Returns true if `chunk` would be a no-op replay against the already-known
-* `parts` — i.e. some upstream is re-emitting events for a tool call that
-* the message has already advanced past.
-*
-* Used by stream broadcasters to suppress re-broadcasting these chunks to
-* connected clients. AI SDK v6's `updateToolPart` mutates an existing tool
-* part in place when a chunk arrives with a matching `toolCallId`, so a
-* replayed `tool-input-start` would clobber an `output-available` part back
-* to `input-streaming` on the client (issue #1404).
-*
-* Only returns true when re-broadcasting would *visibly regress* state on
-* a v6 client. Safe-by-construction chunk types (e.g. `tool-output-available`
-* carrying the same output the part already has) return false.
-*
-* Conditions:
-* - `tool-input-start` for a `toolCallId` that already exists in `parts`.
-* - `tool-input-delta` for a `toolCallId` whose existing part is no longer
-*   `input-streaming`.
-* - `tool-input-available` for a `toolCallId` whose existing part is no
-*   longer `input-streaming` (i.e. has already advanced to `input-available`
-*   or any terminal state).
-*/
-function isReplayChunk(parts, chunk) {
-	if (chunk.type !== "tool-input-start" && chunk.type !== "tool-input-delta" && chunk.type !== "tool-input-available") return false;
-	if (!chunk.toolCallId) return false;
-	const existing = findToolPartByCallId(parts, chunk.toolCallId);
-	if (!existing) return false;
-	if (chunk.type === "tool-input-start") return true;
-	return existing.state !== "input-streaming";
-}
-/**
-* Finds the last part in the array matching the given type.
-* Searches from the end for efficiency (the part we want is usually recent).
-*/
-function findLastPartByType(parts, type) {
-	for (let i = parts.length - 1; i >= 0; i--) if (parts[i].type === type) return parts[i];
-}
-/**
-* Finds a tool part by its toolCallId.
-* Searches from the end since the tool part is usually recent.
-*/
-function findToolPartByCallId(parts, toolCallId) {
-	if (!toolCallId) return void 0;
-	for (let i = parts.length - 1; i >= 0; i--) {
-		const p = parts[i];
-		if ("toolCallId" in p && p.toolCallId === toolCallId) return p;
-	}
-}
-/**
-* Shallow-merges providerMetadata from a chunk onto an existing part.
-* Preserves any metadata already on the part (e.g. from earlier deltas)
-* while adding new keys from the chunk. This is critical for providers
-* like Anthropic that emit the thinking block signature on reasoning-end.
-*/
-function mergeProviderMetadata(part, metadata) {
-	if (metadata == null) return;
-	const p = part;
-	p.providerMetadata = {
-		...p.providerMetadata,
-		...metadata
-	};
-}
-/**
-* Finds a data part by its type and id for reconciliation.
-* Data parts use type+id as a composite key so when the same combination
-* is seen again, the existing part's data is updated in-place.
-*/
-function findDataPartByTypeAndId(parts, type, id) {
-	for (let i = parts.length - 1; i >= 0; i--) {
-		const p = parts[i];
-		if (p.type === type && "id" in p && p.id === id) return p;
-	}
-}
-//#endregion
+import { nanoid } from "nanoid";
 //#region src/chat/sanitize.ts
 const textEncoder$1 = new TextEncoder();
 /** Maximum serialized message size before compaction (bytes). 1.8MB with headroom below SQLite's 2MB limit. */
@@ -1043,6 +746,37 @@ var ResumableStream = class ResumableStream {
 		}));
 		return null;
 	}
+	replayCompletedChunksByRequestId(connection, requestId) {
+		this.flushBuffer();
+		const stream = this.sql`
+      select * from cf_ai_chat_stream_metadata
+      where request_id = ${requestId}
+      and status = 'completed'
+      order by created_at desc
+      limit 1
+    `[0];
+		if (!stream) return false;
+		const chunks = this.sql`
+      select * from cf_ai_chat_stream_chunks
+      where stream_id = ${stream.id}
+      order by chunk_index asc
+    `;
+		for (const chunk of chunks || []) connection.send(JSON.stringify({
+			body: chunk.body,
+			done: false,
+			id: requestId,
+			type: CHAT_MESSAGE_TYPES.USE_CHAT_RESPONSE,
+			replay: true
+		}));
+		connection.send(JSON.stringify({
+			body: "",
+			done: true,
+			id: requestId,
+			type: CHAT_MESSAGE_TYPES.USE_CHAT_RESPONSE,
+			replay: true
+		}));
+		return true;
+	}
 	/**
 	* Restore active stream state if the agent was restarted during streaming.
 	* All streams are restored regardless of age — stale cleanup happens
@@ -1615,6 +1349,6 @@ function findMessageByToolCallId(messages, toolCallId) {
 	}
 }
 //#endregion
-export { AbortRegistry, CHAT_MESSAGE_TYPES, ContinuationState, ROW_MAX_BYTES, ResumableStream, StreamAccumulator, SubmitConcurrencyController, TurnQueue, applyChunkToParts, applyToolUpdate, assistantContentKey, transition as broadcastTransition, byteLength, createToolsFromClientSchemas, enforceRowSizeLimit, isReplayChunk, parseProtocolMessage, reconcileMessages, resolveToolMergeId, sanitizeMessage, toolApprovalUpdate, toolResultUpdate };
+export { AbortRegistry, CHAT_MESSAGE_TYPES, ContinuationState, ROW_MAX_BYTES, ResumableStream, StreamAccumulator, SubmitConcurrencyController, TurnQueue, applyAgentToolEvent, applyChunkToParts, applyToolUpdate, assistantContentKey, transition as broadcastTransition, byteLength, createAgentToolEventState, createToolsFromClientSchemas, enforceRowSizeLimit, isReplayChunk, parseProtocolMessage, reconcileMessages, resolveToolMergeId, sanitizeMessage, toolApprovalUpdate, toolResultUpdate };
 
 //# sourceMappingURL=index.js.map