agents 0.13.3 → 0.14.1

package/dist/agent-tool-types.d.ts

@@ -1,26 +1,28 @@
 import {
   a as AgentToolEventState,
-  c as AgentToolRunInspection,
-  d as AgentToolStoredChunk,
-  f as AgentToolTerminalStatus,
-  h as RunAgentToolResult,
+  c as AgentToolRunInfo,
+  d as AgentToolRunStatus,
+  f as AgentToolStoredChunk,
+  g as RunAgentToolResult,
+  h as RunAgentToolOptions,
   i as AgentToolEventMessage,
-  l as AgentToolRunState,
-  m as RunAgentToolOptions,
+  l as AgentToolRunInspection,
+  m as ChatCapableAgentClass,
   n as AgentToolDisplayMetadata,
-  o as AgentToolLifecycleResult,
-  p as ChatCapableAgentClass,
+  o as AgentToolFailure,
+  p as AgentToolTerminalStatus,
   r as AgentToolEvent,
-  s as AgentToolRunInfo,
+  s as AgentToolLifecycleResult,
   t as AgentToolChildAdapter,
-  u as AgentToolRunStatus
-} from "./agent-tool-types-l98LCbBl.js";
+  u as AgentToolRunState
+} from "./agent-tool-types-BAJWu8s4.js";
 export {
   AgentToolChildAdapter,
   AgentToolDisplayMetadata,
   AgentToolEvent,
   AgentToolEventMessage,
   AgentToolEventState,
+  AgentToolFailure,
   AgentToolLifecycleResult,
   AgentToolRunInfo,
   AgentToolRunInspection,

package/dist/{agent-tools-Bg5ilERh.d.ts → agent-tools-0R6KEert.d.ts}

@@ -1,7 +1,7 @@
 import {
   a as AgentToolEventState,
   i as AgentToolEventMessage
-} from "./agent-tool-types-l98LCbBl.js";
+} from "./agent-tool-types-BAJWu8s4.js";
 
 //#region src/chat/agent-tools.d.ts
 declare function createAgentToolEventState(): AgentToolEventState;
@@ -11,4 +11,4 @@ declare function applyAgentToolEvent(
 ): AgentToolEventState;
 //#endregion
 export { createAgentToolEventState as n, applyAgentToolEvent as t };
-//# sourceMappingURL=agent-tools-Bg5ilERh.d.ts.map
+//# sourceMappingURL=agent-tools-0R6KEert.d.ts.map

package/dist/{agent-tools-BAdX1vdI.js → agent-tools-DYrkT-Kx.js}

@@ -1,4 +1,44 @@
 //#region src/chat/message-builder.ts
+/** Whether a value is a plain (non-array, non-null) object. */
+function isPlainObject(value) {
+	return value !== null && typeof value === "object" && !Array.isArray(value);
+}
+/**
+* Coerce a tool part's `input` into a provider-acceptable object.
+*
+* The Anthropic Messages API requires `tool_use.input` to be a JSON **object** —
+* `null`, `undefined`, `""`, a raw string, **or an array** are all rejected with
+* `tool_use.input: Input should be an object` (verified empirically against the
+* live API: `{}` → 200, but `""`, `[]`, and `[{...}]` all → 400). A streamed
+* tool call that finishes with no `input_json_delta` events (the model called
+* the tool with no args), or whose input surfaces as a stringified JSON blob,
+* can persist one of these shapes — and because it lives in durable storage, the
+* session is then wedged across reconnects, redeploys, and DO evictions.
+* Enforcing the invariant at the write boundary (and as a read-side repair
+* backstop) keeps the transcript valid.
+*
+* - A plain (non-array) object is returned untouched (`changed: false`).
+* - A string that parses to a plain object is parsed.
+* - Everything else (`null`, `undefined`, `""`, arrays, primitives, non-object
+*   or unparseable JSON) collapses to `{}`.
+*/
+function normalizeToolInput(raw) {
+	if (isPlainObject(raw)) return {
+		input: raw,
+		changed: false
+	};
+	if (typeof raw === "string" && raw.trim().startsWith("{")) try {
+		const parsed = JSON.parse(raw);
+		if (isPlainObject(parsed)) return {
+			input: parsed,
+			changed: true
+		};
+	} catch {}
+	return {
+		input: {},
+		changed: true
+	};
+}
 /**
 * 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
@@ -120,7 +160,7 @@ function applyChunkToParts(parts, chunk) {
 				const p = existing;
 				if (p.state === "input-streaming") {
 					p.state = "input-available";
-					p.input = chunk.input;
+					p.input = normalizeToolInput(chunk.input).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;
@@ -132,7 +172,7 @@ function applyChunkToParts(parts, chunk) {
 				toolCallId: chunk.toolCallId,
 				toolName: chunk.toolName,
 				state: "input-available",
-				input: chunk.input,
+				input: normalizeToolInput(chunk.input).input,
 				...chunk.providerExecuted != null ? { providerExecuted: chunk.providerExecuted } : {},
 				...chunk.providerMetadata != null ? { callProviderMetadata: chunk.providerMetadata } : {},
 				...chunk.title != null ? { title: chunk.title } : {}
@@ -146,7 +186,7 @@ function applyChunkToParts(parts, chunk) {
 				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;
+				p.input = normalizeToolInput(chunk.input).input;
 				if (chunk.providerExecuted != null) p.providerExecuted = chunk.providerExecuted;
 				if (chunk.providerMetadata != null) p.callProviderMetadata = chunk.providerMetadata;
 			} else parts.push({
@@ -154,7 +194,7 @@ function applyChunkToParts(parts, chunk) {
 				toolCallId: chunk.toolCallId,
 				toolName: chunk.toolName,
 				state: "output-error",
-				input: chunk.input,
+				input: normalizeToolInput(chunk.input).input,
 				errorText: chunk.errorText,
 				...chunk.providerExecuted != null ? { providerExecuted: chunk.providerExecuted } : {},
 				...chunk.providerMetadata != null ? { callProviderMetadata: chunk.providerMetadata } : {}
@@ -420,6 +460,6 @@ function applyAgentToolEvent(state, message) {
 	};
 }
 //#endregion
-export { isReplayChunk as i, createAgentToolEventState as n, applyChunkToParts as r, applyAgentToolEvent as t };
+export { normalizeToolInput as a, isReplayChunk as i, createAgentToolEventState as n, applyChunkToParts as r, applyAgentToolEvent as t };
 
-//# sourceMappingURL=agent-tools-BAdX1vdI.js.map
+//# sourceMappingURL=agent-tools-DYrkT-Kx.js.map

package/dist/agent-tools-DYrkT-Kx.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-tools-DYrkT-Kx.js","names":[],"sources":["../src/chat/message-builder.ts","../src/chat/agent-tools.ts"],"sourcesContent":["/**\n * Shared message builder for reconstructing UIMessage parts from stream chunks.\n *\n * Used by both @cloudflare/ai-chat (server + client) and @cloudflare/think\n * to avoid duplicating the chunk-type switch/case logic.\n *\n * Operates on a mutable parts array for performance (avoids allocating new arrays\n * on every chunk during streaming).\n */\n\nimport type { UIMessage } from \"ai\";\n\n/** The parts array type from UIMessage */\nexport type MessageParts = UIMessage[\"parts\"];\n\n/** A single part from the UIMessage parts array */\nexport type MessagePart = MessageParts[number];\n\n/**\n * Parsed chunk data from an AI SDK stream event.\n * This is the JSON-parsed body of a CF_AGENT_USE_CHAT_RESPONSE message,\n * or the `data:` payload of an SSE line.\n */\nexport type StreamChunkData = {\n  type: string;\n  id?: string;\n  delta?: string;\n  text?: string;\n  mediaType?: string;\n  url?: string;\n  sourceId?: string;\n  title?: string;\n  filename?: string;\n  toolCallId?: string;\n  toolName?: string;\n  input?: unknown;\n  inputTextDelta?: string;\n  output?: unknown;\n  state?: string;\n  errorText?: string;\n  /** When true, the output is preliminary (may be updated by a later chunk) */\n  preliminary?: boolean;\n  /** Approval ID for tools with needsApproval */\n  approvalId?: string;\n  providerMetadata?: Record<string, unknown>;\n  /** Whether the tool was executed by the provider (e.g. Gemini code execution) */\n  providerExecuted?: boolean;\n  /** Payload for data-* parts (developer-defined typed JSON) */\n  data?: unknown;\n  /** When true, data parts are ephemeral and not persisted to message.parts */\n  transient?: boolean;\n  /** Message ID assigned by the server at stream start */\n  messageId?: string;\n  /** Per-message metadata attached by start/finish/message-metadata chunks */\n  messageMetadata?: unknown;\n  [key: string]: unknown;\n};\n\n/** Whether a value is a plain (non-array, non-null) object. */\nfunction isPlainObject(value: unknown): value is Record<string, unknown> {\n  return value !== null && typeof value === \"object\" && !Array.isArray(value);\n}\n\n/**\n * Coerce a tool part's `input` into a provider-acceptable object.\n *\n * The Anthropic Messages API requires `tool_use.input` to be a JSON **object** —\n * `null`, `undefined`, `\"\"`, a raw string, **or an array** are all rejected with\n * `tool_use.input: Input should be an object` (verified empirically against the\n * live API: `{}` → 200, but `\"\"`, `[]`, and `[{...}]` all → 400). A streamed\n * tool call that finishes with no `input_json_delta` events (the model called\n * the tool with no args), or whose input surfaces as a stringified JSON blob,\n * can persist one of these shapes — and because it lives in durable storage, the\n * session is then wedged across reconnects, redeploys, and DO evictions.\n * Enforcing the invariant at the write boundary (and as a read-side repair\n * backstop) keeps the transcript valid.\n *\n * - A plain (non-array) object is returned untouched (`changed: false`).\n * - A string that parses to a plain object is parsed.\n * - Everything else (`null`, `undefined`, `\"\"`, arrays, primitives, non-object\n *   or unparseable JSON) collapses to `{}`.\n */\nexport function normalizeToolInput(raw: unknown): {\n  input: unknown;\n  changed: boolean;\n} {\n  if (isPlainObject(raw)) return { input: raw, changed: false };\n  if (typeof raw === \"string\" && raw.trim().startsWith(\"{\")) {\n    try {\n      const parsed: unknown = JSON.parse(raw);\n      if (isPlainObject(parsed)) return { input: parsed, changed: true };\n    } catch {\n      // Unparseable / partial JSON — fall through to the empty-object default.\n    }\n  }\n  return { input: {}, changed: true };\n}\n\n/**\n * Applies a stream chunk to a mutable parts array, building up the message\n * incrementally. Returns true if the chunk was handled, false if it was\n * an unrecognized type (caller may handle it with additional logic).\n *\n * Handles all common chunk types that both server and client need:\n * - text-start / text-delta / text-end\n * - reasoning-start / reasoning-delta / reasoning-end\n * - file\n * - source-url / source-document\n * - tool-input-start / tool-input-delta / tool-input-available / tool-input-error\n * - tool-output-available / tool-output-error\n * - step-start (aliased from start-step)\n * - data-* (developer-defined typed JSON blobs)\n *\n * @param parts - The mutable parts array to update\n * @param chunk - The parsed stream chunk data\n * @returns true if handled, false if the chunk type is not recognized\n */\nexport function applyChunkToParts(\n  parts: MessagePart[],\n  chunk: StreamChunkData\n): boolean {\n  switch (chunk.type) {\n    case \"text-start\": {\n      parts.push({\n        type: \"text\",\n        text: \"\",\n        state: \"streaming\"\n      } as MessagePart);\n      return true;\n    }\n\n    case \"text-delta\": {\n      const lastTextPart = findLastPartByType(parts, \"text\");\n      if (lastTextPart && lastTextPart.type === \"text\") {\n        (lastTextPart as { text: string }).text += chunk.delta ?? \"\";\n      } else {\n        // No text-start received — create a new text part (stream resumption fallback)\n        parts.push({\n          type: \"text\",\n          text: chunk.delta ?? \"\",\n          state: \"streaming\"\n        } as MessagePart);\n      }\n      return true;\n    }\n\n    case \"text-end\": {\n      const lastTextPart = findLastPartByType(parts, \"text\");\n      if (lastTextPart && \"state\" in lastTextPart) {\n        (lastTextPart as { state: string }).state = \"done\";\n      }\n      return true;\n    }\n\n    case \"reasoning-start\": {\n      parts.push({\n        type: \"reasoning\",\n        text: \"\",\n        state: \"streaming\"\n      } as MessagePart);\n      return true;\n    }\n\n    case \"reasoning-delta\": {\n      const lastReasoningPart = findLastPartByType(parts, \"reasoning\");\n      if (lastReasoningPart && lastReasoningPart.type === \"reasoning\") {\n        (lastReasoningPart as { text: string }).text += chunk.delta ?? \"\";\n        mergeProviderMetadata(lastReasoningPart, chunk.providerMetadata);\n      } else {\n        // No reasoning-start received — create a new reasoning part (stream resumption fallback)\n        parts.push({\n          type: \"reasoning\",\n          text: chunk.delta ?? \"\",\n          state: \"streaming\",\n          ...(chunk.providerMetadata != null\n            ? { providerMetadata: chunk.providerMetadata }\n            : {})\n        } as MessagePart);\n      }\n      return true;\n    }\n\n    case \"reasoning-end\": {\n      const lastReasoningPart = findLastPartByType(parts, \"reasoning\");\n      if (lastReasoningPart && \"state\" in lastReasoningPart) {\n        (lastReasoningPart as { state: string }).state = \"done\";\n        mergeProviderMetadata(lastReasoningPart, chunk.providerMetadata);\n      }\n      return true;\n    }\n\n    case \"file\": {\n      parts.push({\n        type: \"file\",\n        mediaType: chunk.mediaType,\n        url: chunk.url\n      } as MessagePart);\n      return true;\n    }\n\n    case \"source-url\": {\n      parts.push({\n        type: \"source-url\",\n        sourceId: chunk.sourceId,\n        url: chunk.url,\n        title: chunk.title,\n        providerMetadata: chunk.providerMetadata\n      } as MessagePart);\n      return true;\n    }\n\n    case \"source-document\": {\n      parts.push({\n        type: \"source-document\",\n        sourceId: chunk.sourceId,\n        mediaType: chunk.mediaType,\n        title: chunk.title,\n        filename: chunk.filename,\n        providerMetadata: chunk.providerMetadata\n      } as MessagePart);\n      return true;\n    }\n\n    case \"tool-input-start\": {\n      // Idempotent against an existing tool part with the same toolCallId.\n      // Some providers (notably the OpenAI Responses API) replay prior\n      // tool calls in continuation streams as a fresh `tool-input-start`\n      // → `tool-input-delta` → `tool-input-available` →\n      // `tool-output-available` sequence carrying the original toolCallId\n      // and original output. Without this guard a replay would push a\n      // duplicate part into the streaming message *and* clobber the\n      // original part's state when the AI SDK's mutate-in-place\n      // `updateToolPart` processes the replay on the client (issue #1404).\n      // A model that genuinely wants a fresh tool call always emits a\n      // new toolCallId, so an existing match is never a legitimate\n      // \"start over\".\n      const existing = findToolPartByCallId(parts, chunk.toolCallId);\n      if (existing) {\n        return true;\n      }\n      parts.push({\n        type: `tool-${chunk.toolName}`,\n        toolCallId: chunk.toolCallId,\n        toolName: chunk.toolName,\n        state: \"input-streaming\",\n        input: undefined,\n        ...(chunk.providerExecuted != null\n          ? { providerExecuted: chunk.providerExecuted }\n          : {}),\n        ...(chunk.providerMetadata != null\n          ? { callProviderMetadata: chunk.providerMetadata }\n          : {}),\n        ...(chunk.title != null ? { title: chunk.title } : {})\n      } as MessagePart);\n      return true;\n    }\n\n    case \"tool-input-delta\": {\n      // Only mutate input while the tool is still actively input-streaming.\n      // Deltas arriving after the tool has already advanced (input-available\n      // or any terminal state) are provider replay and must not regress\n      // a fully-formed input back to a partial one.\n      const toolPart = findToolPartByCallId(parts, chunk.toolCallId);\n      if (\n        toolPart &&\n        (toolPart as Record<string, unknown>).state === \"input-streaming\"\n      ) {\n        (toolPart as Record<string, unknown>).input = chunk.input;\n      }\n      return true;\n    }\n\n    case \"tool-input-available\": {\n      const existing = findToolPartByCallId(parts, chunk.toolCallId);\n      if (existing) {\n        const p = existing as Record<string, unknown>;\n        // Only advance from the streaming-input phase. Once the tool is\n        // already at input-available or any terminal state\n        // (output-available, output-error, output-denied,\n        // approval-requested, approval-responded), this chunk is a\n        // provider replay and must not regress state or overwrite a\n        // resolved input/output. See the comment on tool-input-start.\n        if (p.state === \"input-streaming\") {\n          p.state = \"input-available\";\n          p.input = normalizeToolInput(chunk.input).input;\n          if (chunk.providerExecuted != null) {\n            p.providerExecuted = chunk.providerExecuted;\n          }\n          if (chunk.providerMetadata != null) {\n            p.callProviderMetadata = chunk.providerMetadata;\n          }\n          if (chunk.title != null) {\n            p.title = chunk.title;\n          }\n        }\n        return true;\n      }\n      parts.push({\n        type: `tool-${chunk.toolName}`,\n        toolCallId: chunk.toolCallId,\n        toolName: chunk.toolName,\n        state: \"input-available\",\n        input: normalizeToolInput(chunk.input).input,\n        ...(chunk.providerExecuted != null\n          ? { providerExecuted: chunk.providerExecuted }\n          : {}),\n        ...(chunk.providerMetadata != null\n          ? { callProviderMetadata: chunk.providerMetadata }\n          : {}),\n        ...(chunk.title != null ? { title: chunk.title } : {})\n      } as MessagePart);\n      return true;\n    }\n\n    case \"tool-input-error\": {\n      const existing = findToolPartByCallId(parts, chunk.toolCallId);\n      if (existing) {\n        const p = existing as Record<string, unknown>;\n        // First-write-wins: a tool that's already terminal must not be\n        // regressed (or re-decided as an error) by a later chunk. A\n        // tool-input-error here is either provider replay or a confused\n        // upstream — preserve the existing terminal state.\n        if (\n          p.state === \"output-available\" ||\n          p.state === \"output-error\" ||\n          p.state === \"output-denied\"\n        ) {\n          return true;\n        }\n        p.state = \"output-error\";\n        p.errorText = chunk.errorText;\n        p.input = normalizeToolInput(chunk.input).input;\n        if (chunk.providerExecuted != null) {\n          p.providerExecuted = chunk.providerExecuted;\n        }\n        if (chunk.providerMetadata != null) {\n          p.callProviderMetadata = chunk.providerMetadata;\n        }\n      } else {\n        parts.push({\n          type: `tool-${chunk.toolName}`,\n          toolCallId: chunk.toolCallId,\n          toolName: chunk.toolName,\n          state: \"output-error\",\n          input: normalizeToolInput(chunk.input).input,\n          errorText: chunk.errorText,\n          ...(chunk.providerExecuted != null\n            ? { providerExecuted: chunk.providerExecuted }\n            : {}),\n          ...(chunk.providerMetadata != null\n            ? { callProviderMetadata: chunk.providerMetadata }\n            : {})\n        } as MessagePart);\n      }\n      return true;\n    }\n\n    case \"tool-approval-request\": {\n      const toolPart = findToolPartByCallId(parts, chunk.toolCallId);\n      if (toolPart) {\n        const p = toolPart as Record<string, unknown>;\n        p.state = \"approval-requested\";\n        p.approval = { id: chunk.approvalId };\n      }\n      return true;\n    }\n\n    case \"tool-output-denied\": {\n      const toolPart = findToolPartByCallId(parts, chunk.toolCallId);\n      if (toolPart) {\n        const p = toolPart as Record<string, unknown>;\n        p.state = \"output-denied\";\n      }\n      return true;\n    }\n\n    case \"tool-output-available\": {\n      const toolPart = findToolPartByCallId(parts, chunk.toolCallId);\n      if (toolPart) {\n        const p = toolPart as Record<string, unknown>;\n        p.state = \"output-available\";\n        p.output = chunk.output;\n        if (chunk.preliminary !== undefined) {\n          p.preliminary = chunk.preliminary;\n        }\n      }\n      return true;\n    }\n\n    case \"tool-output-error\": {\n      const toolPart = findToolPartByCallId(parts, chunk.toolCallId);\n      if (toolPart) {\n        const p = toolPart as Record<string, unknown>;\n        p.state = \"output-error\";\n        p.errorText = chunk.errorText;\n      }\n      return true;\n    }\n\n    // Both \"step-start\" (client convention) and \"start-step\" (server convention)\n    case \"step-start\":\n    case \"start-step\": {\n      parts.push({ type: \"step-start\" } as MessagePart);\n      return true;\n    }\n\n    default: {\n      // https://ai-sdk.dev/docs/ai-sdk-ui/streaming-data\n      if (chunk.type.startsWith(\"data-\")) {\n        // Transient parts are ephemeral — the AI SDK client fires an onData\n        // callback instead of adding them to message.parts. On the server we\n        // still broadcast them (so connected clients see them in real time)\n        // but skip persisting them into the stored message parts.\n        if (chunk.transient) {\n          return true;\n        }\n\n        // Reconciliation: if a part with the same type AND id already exists,\n        // update its data in-place instead of appending a duplicate.\n        if (chunk.id != null) {\n          const existing = findDataPartByTypeAndId(parts, chunk.type, chunk.id);\n          if (existing) {\n            (existing as Record<string, unknown>).data = chunk.data;\n            return true;\n          }\n        }\n\n        // Append new data parts to the array directly.\n        // Note: `chunk.data` should always be provided — if omitted, the\n        // persisted part will have `data: undefined` which JSON.stringify\n        // drops, so the part will have no `data` field on reload.\n        // The cast is needed because UIMessage[\"parts\"] doesn't include\n        // data-* types in its union because they're an open extension point.\n        parts.push({\n          type: chunk.type,\n          ...(chunk.id != null && { id: chunk.id }),\n          data: chunk.data\n        } as MessagePart);\n        return true;\n      }\n\n      return false;\n    }\n  }\n}\n\n/**\n * Returns true if `chunk` would be a no-op replay against the already-known\n * `parts` — i.e. some upstream is re-emitting events for a tool call that\n * the message has already advanced past.\n *\n * Used by stream broadcasters to suppress re-broadcasting these chunks to\n * connected clients. AI SDK v6's `updateToolPart` mutates an existing tool\n * part in place when a chunk arrives with a matching `toolCallId`, so a\n * replayed `tool-input-start` would clobber an `output-available` part back\n * to `input-streaming` on the client (issue #1404).\n *\n * Only returns true when re-broadcasting would *visibly regress* state on\n * a v6 client. Safe-by-construction chunk types (e.g. `tool-output-available`\n * carrying the same output the part already has) return false.\n *\n * Conditions:\n * - `tool-input-start` for a `toolCallId` that already exists in `parts`.\n * - `tool-input-delta` for a `toolCallId` whose existing part is no longer\n *   `input-streaming`.\n * - `tool-input-available` for a `toolCallId` whose existing part is no\n *   longer `input-streaming` (i.e. has already advanced to `input-available`\n *   or any terminal state).\n */\nexport function isReplayChunk(\n  parts: MessagePart[],\n  chunk: StreamChunkData\n): boolean {\n  if (\n    chunk.type !== \"tool-input-start\" &&\n    chunk.type !== \"tool-input-delta\" &&\n    chunk.type !== \"tool-input-available\"\n  ) {\n    return false;\n  }\n  if (!chunk.toolCallId) return false;\n  const existing = findToolPartByCallId(parts, chunk.toolCallId);\n  if (!existing) return false;\n  if (chunk.type === \"tool-input-start\") return true;\n  const state = (existing as Record<string, unknown>).state;\n  return state !== \"input-streaming\";\n}\n\n/**\n * Finds the last part in the array matching the given type.\n * Searches from the end for efficiency (the part we want is usually recent).\n */\nfunction findLastPartByType(\n  parts: MessagePart[],\n  type: string\n): MessagePart | undefined {\n  for (let i = parts.length - 1; i >= 0; i--) {\n    if (parts[i].type === type) {\n      return parts[i];\n    }\n  }\n  return undefined;\n}\n\n/**\n * Finds a tool part by its toolCallId.\n * Searches from the end since the tool part is usually recent.\n */\nfunction findToolPartByCallId(\n  parts: MessagePart[],\n  toolCallId: string | undefined\n): MessagePart | undefined {\n  if (!toolCallId) return undefined;\n  for (let i = parts.length - 1; i >= 0; i--) {\n    const p = parts[i];\n    if (\"toolCallId\" in p && p.toolCallId === toolCallId) {\n      return p;\n    }\n  }\n  return undefined;\n}\n\n/**\n * Shallow-merges providerMetadata from a chunk onto an existing part.\n * Preserves any metadata already on the part (e.g. from earlier deltas)\n * while adding new keys from the chunk. This is critical for providers\n * like Anthropic that emit the thinking block signature on reasoning-end.\n */\nfunction mergeProviderMetadata(\n  part: MessagePart,\n  metadata: Record<string, unknown> | undefined\n): void {\n  if (metadata == null) return;\n  const p = part as Record<string, unknown>;\n  p.providerMetadata = {\n    ...(p.providerMetadata as Record<string, unknown> | undefined),\n    ...metadata\n  };\n}\n\n/**\n * Finds a data part by its type and id for reconciliation.\n * Data parts use type+id as a composite key so when the same combination\n * is seen again, the existing part's data is updated in-place.\n */\nfunction findDataPartByTypeAndId(\n  parts: MessagePart[],\n  type: string,\n  id: string\n): MessagePart | undefined {\n  for (let i = parts.length - 1; i >= 0; i--) {\n    const p = parts[i];\n    if (p.type === type && \"id\" in p && (p as { id: string }).id === id) {\n      return p;\n    }\n  }\n  return undefined;\n}\n","import { applyChunkToParts } from \"./message-builder\";\nimport type {\n  AgentToolEventMessage,\n  AgentToolEventState,\n  AgentToolRunState\n} from \"../agent-tool-types\";\n\nfunction sortRuns(runs: AgentToolRunState[]): AgentToolRunState[] {\n  return [...runs].sort((a, b) => {\n    if (a.order !== b.order) return a.order - b.order;\n    return a.runId.localeCompare(b.runId);\n  });\n}\n\nfunction rebuildIndexes(\n  runsById: Record<string, AgentToolRunState>\n): Pick<AgentToolEventState, \"runsByToolCallId\" | \"unboundRuns\"> {\n  const grouped: Record<string, AgentToolRunState[]> = {};\n  const unboundRuns: AgentToolRunState[] = [];\n  for (const run of Object.values(runsById)) {\n    if (run.parentToolCallId) {\n      grouped[run.parentToolCallId] = grouped[run.parentToolCallId] ?? [];\n      grouped[run.parentToolCallId].push(run);\n    } else {\n      unboundRuns.push(run);\n    }\n  }\n  for (const [toolCallId, runs] of Object.entries(grouped)) {\n    grouped[toolCallId] = sortRuns(runs);\n  }\n  return { runsByToolCallId: grouped, unboundRuns: sortRuns(unboundRuns) };\n}\n\nfunction emptyRun(\n  message: AgentToolEventMessage\n): AgentToolRunState | undefined {\n  const { event } = message;\n  if (event.kind === \"started\") {\n    return {\n      runId: event.runId,\n      agentType: event.agentType,\n      parentToolCallId: message.parentToolCallId,\n      inputPreview: event.inputPreview,\n      order: event.order,\n      display: event.display,\n      status: \"running\",\n      parts: [],\n      subAgent: { agent: event.agentType, name: event.runId }\n    };\n  }\n  return undefined;\n}\n\nfunction applyToRun(\n  prev: AgentToolRunState | undefined,\n  message: AgentToolEventMessage\n): AgentToolRunState | undefined {\n  const seeded = prev ?? emptyRun(message);\n  const { event } = message;\n\n  switch (event.kind) {\n    case \"started\":\n      if (\n        seeded?.status === \"completed\" ||\n        seeded?.status === \"error\" ||\n        seeded?.status === \"aborted\" ||\n        seeded?.status === \"interrupted\"\n      ) {\n        return seeded;\n      }\n      return {\n        ...seeded,\n        runId: event.runId,\n        agentType: event.agentType,\n        parentToolCallId: message.parentToolCallId,\n        inputPreview: event.inputPreview,\n        order: event.order,\n        display: event.display,\n        status: \"running\",\n        parts: seeded?.parts ?? [],\n        subAgent: { agent: event.agentType, name: event.runId }\n      };\n    case \"chunk\": {\n      if (!seeded) return undefined;\n      const parts = [...seeded.parts];\n      try {\n        applyChunkToParts(parts, JSON.parse(event.body));\n      } catch {\n        return seeded;\n      }\n      return { ...seeded, parts };\n    }\n    case \"finished\":\n      if (!seeded) return undefined;\n      return {\n        ...seeded,\n        status: \"completed\",\n        summary: event.summary,\n        error: undefined\n      };\n    case \"error\":\n      if (!seeded) return undefined;\n      return { ...seeded, status: \"error\", error: event.error };\n    case \"aborted\":\n      if (!seeded) return undefined;\n      return { ...seeded, status: \"aborted\", error: event.reason };\n    case \"interrupted\":\n      if (!seeded) return undefined;\n      return { ...seeded, status: \"interrupted\", error: event.error };\n  }\n}\n\nexport function createAgentToolEventState(): AgentToolEventState {\n  return {\n    runsById: {},\n    runsByToolCallId: {},\n    unboundRuns: []\n  };\n}\n\nexport function applyAgentToolEvent(\n  state: AgentToolEventState,\n  message: AgentToolEventMessage\n): AgentToolEventState {\n  if (message.type !== \"agent-tool-event\") return state;\n  const runId = message.event.runId;\n  const nextRun = applyToRun(state.runsById[runId], message);\n  if (!nextRun) return state;\n\n  const runsById = { ...state.runsById, [runId]: nextRun };\n  return { runsById, ...rebuildIndexes(runsById) };\n}\n\nexport type {\n  AgentToolEvent,\n  AgentToolEventMessage,\n  AgentToolEventState,\n  AgentToolRunState\n} from \"../agent-tool-types\";\n"],"mappings":";;AA2DA,SAAS,cAAc,OAAkD;CACvE,OAAO,UAAU,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,KAAK;AAC5E;;;;;;;;;;;;;;;;;;;;AAqBA,SAAgB,mBAAmB,KAGjC;CACA,IAAI,cAAc,GAAG,GAAG,OAAO;EAAE,OAAO;EAAK,SAAS;CAAM;CAC5D,IAAI,OAAO,QAAQ,YAAY,IAAI,KAAK,EAAE,WAAW,GAAG,GACtD,IAAI;EACF,MAAM,SAAkB,KAAK,MAAM,GAAG;EACtC,IAAI,cAAc,MAAM,GAAG,OAAO;GAAE,OAAO;GAAQ,SAAS;EAAK;CACnE,QAAQ,CAER;CAEF,OAAO;EAAE,OAAO,CAAC;EAAG,SAAS;CAAK;AACpC;;;;;;;;;;;;;;;;;;;;AAqBA,SAAgB,kBACd,OACA,OACS;CACT,QAAQ,MAAM,MAAd;EACE,KAAK;GACH,MAAM,KAAK;IACT,MAAM;IACN,MAAM;IACN,OAAO;GACT,CAAgB;GAChB,OAAO;EAGT,KAAK,cAAc;GACjB,MAAM,eAAe,mBAAmB,OAAO,MAAM;GACrD,IAAI,gBAAgB,aAAa,SAAS,QACxC,aAAmC,QAAQ,MAAM,SAAS;QAG1D,MAAM,KAAK;IACT,MAAM;IACN,MAAM,MAAM,SAAS;IACrB,OAAO;GACT,CAAgB;GAElB,OAAO;EACT;EAEA,KAAK,YAAY;GACf,MAAM,eAAe,mBAAmB,OAAO,MAAM;GACrD,IAAI,gBAAgB,WAAW,cAC7B,aAAoC,QAAQ;GAE9C,OAAO;EACT;EAEA,KAAK;GACH,MAAM,KAAK;IACT,MAAM;IACN,MAAM;IACN,OAAO;GACT,CAAgB;GAChB,OAAO;EAGT,KAAK,mBAAmB;GACtB,MAAM,oBAAoB,mBAAmB,OAAO,WAAW;GAC/D,IAAI,qBAAqB,kBAAkB,SAAS,aAAa;IAC/D,kBAAwC,QAAQ,MAAM,SAAS;IAC/D,sBAAsB,mBAAmB,MAAM,gBAAgB;GACjE,OAEE,MAAM,KAAK;IACT,MAAM;IACN,MAAM,MAAM,SAAS;IACrB,OAAO;IACP,GAAI,MAAM,oBAAoB,OAC1B,EAAE,kBAAkB,MAAM,iBAAiB,IAC3C,CAAC;GACP,CAAgB;GAElB,OAAO;EACT;EAEA,KAAK,iBAAiB;GACpB,MAAM,oBAAoB,mBAAmB,OAAO,WAAW;GAC/D,IAAI,qBAAqB,WAAW,mBAAmB;IACrD,kBAAyC,QAAQ;IACjD,sBAAsB,mBAAmB,MAAM,gBAAgB;GACjE;GACA,OAAO;EACT;EAEA,KAAK;GACH,MAAM,KAAK;IACT,MAAM;IACN,WAAW,MAAM;IACjB,KAAK,MAAM;GACb,CAAgB;GAChB,OAAO;EAGT,KAAK;GACH,MAAM,KAAK;IACT,MAAM;IACN,UAAU,MAAM;IAChB,KAAK,MAAM;IACX,OAAO,MAAM;IACb,kBAAkB,MAAM;GAC1B,CAAgB;GAChB,OAAO;EAGT,KAAK;GACH,MAAM,KAAK;IACT,MAAM;IACN,UAAU,MAAM;IAChB,WAAW,MAAM;IACjB,OAAO,MAAM;IACb,UAAU,MAAM;IAChB,kBAAkB,MAAM;GAC1B,CAAgB;GAChB,OAAO;EAGT,KAAK;GAcH,IADiB,qBAAqB,OAAO,MAAM,UACxC,GACT,OAAO;GAET,MAAM,KAAK;IACT,MAAM,QAAQ,MAAM;IACpB,YAAY,MAAM;IAClB,UAAU,MAAM;IAChB,OAAO;IACP,OAAO,KAAA;IACP,GAAI,MAAM,oBAAoB,OAC1B,EAAE,kBAAkB,MAAM,iBAAiB,IAC3C,CAAC;IACL,GAAI,MAAM,oBAAoB,OAC1B,EAAE,sBAAsB,MAAM,iBAAiB,IAC/C,CAAC;IACL,GAAI,MAAM,SAAS,OAAO,EAAE,OAAO,MAAM,MAAM,IAAI,CAAC;GACtD,CAAgB;GAChB,OAAO;EAGT,KAAK,oBAAoB;GAKvB,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;GAC7D,IACE,YACC,SAAqC,UAAU,mBAEhD,SAAsC,QAAQ,MAAM;GAEtD,OAAO;EACT;EAEA,KAAK,wBAAwB;GAC3B,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;GAC7D,IAAI,UAAU;IACZ,MAAM,IAAI;IAOV,IAAI,EAAE,UAAU,mBAAmB;KACjC,EAAE,QAAQ;KACV,EAAE,QAAQ,mBAAmB,MAAM,KAAK,EAAE;KAC1C,IAAI,MAAM,oBAAoB,MAC5B,EAAE,mBAAmB,MAAM;KAE7B,IAAI,MAAM,oBAAoB,MAC5B,EAAE,uBAAuB,MAAM;KAEjC,IAAI,MAAM,SAAS,MACjB,EAAE,QAAQ,MAAM;IAEpB;IACA,OAAO;GACT;GACA,MAAM,KAAK;IACT,MAAM,QAAQ,MAAM;IACpB,YAAY,MAAM;IAClB,UAAU,MAAM;IAChB,OAAO;IACP,OAAO,mBAAmB,MAAM,KAAK,EAAE;IACvC,GAAI,MAAM,oBAAoB,OAC1B,EAAE,kBAAkB,MAAM,iBAAiB,IAC3C,CAAC;IACL,GAAI,MAAM,oBAAoB,OAC1B,EAAE,sBAAsB,MAAM,iBAAiB,IAC/C,CAAC;IACL,GAAI,MAAM,SAAS,OAAO,EAAE,OAAO,MAAM,MAAM,IAAI,CAAC;GACtD,CAAgB;GAChB,OAAO;EACT;EAEA,KAAK,oBAAoB;GACvB,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;GAC7D,IAAI,UAAU;IACZ,MAAM,IAAI;IAKV,IACE,EAAE,UAAU,sBACZ,EAAE,UAAU,kBACZ,EAAE,UAAU,iBAEZ,OAAO;IAET,EAAE,QAAQ;IACV,EAAE,YAAY,MAAM;IACpB,EAAE,QAAQ,mBAAmB,MAAM,KAAK,EAAE;IAC1C,IAAI,MAAM,oBAAoB,MAC5B,EAAE,mBAAmB,MAAM;IAE7B,IAAI,MAAM,oBAAoB,MAC5B,EAAE,uBAAuB,MAAM;GAEnC,OACE,MAAM,KAAK;IACT,MAAM,QAAQ,MAAM;IACpB,YAAY,MAAM;IAClB,UAAU,MAAM;IAChB,OAAO;IACP,OAAO,mBAAmB,MAAM,KAAK,EAAE;IACvC,WAAW,MAAM;IACjB,GAAI,MAAM,oBAAoB,OAC1B,EAAE,kBAAkB,MAAM,iBAAiB,IAC3C,CAAC;IACL,GAAI,MAAM,oBAAoB,OAC1B,EAAE,sBAAsB,MAAM,iBAAiB,IAC/C,CAAC;GACP,CAAgB;GAElB,OAAO;EACT;EAEA,KAAK,yBAAyB;GAC5B,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;GAC7D,IAAI,UAAU;IACZ,MAAM,IAAI;IACV,EAAE,QAAQ;IACV,EAAE,WAAW,EAAE,IAAI,MAAM,WAAW;GACtC;GACA,OAAO;EACT;EAEA,KAAK,sBAAsB;GACzB,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;GAC7D,IAAI,UAAU;IACZ,MAAM,IAAI;IACV,EAAE,QAAQ;GACZ;GACA,OAAO;EACT;EAEA,KAAK,yBAAyB;GAC5B,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;GAC7D,IAAI,UAAU;IACZ,MAAM,IAAI;IACV,EAAE,QAAQ;IACV,EAAE,SAAS,MAAM;IACjB,IAAI,MAAM,gBAAgB,KAAA,GACxB,EAAE,cAAc,MAAM;GAE1B;GACA,OAAO;EACT;EAEA,KAAK,qBAAqB;GACxB,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;GAC7D,IAAI,UAAU;IACZ,MAAM,IAAI;IACV,EAAE,QAAQ;IACV,EAAE,YAAY,MAAM;GACtB;GACA,OAAO;EACT;EAGA,KAAK;EACL,KAAK;GACH,MAAM,KAAK,EAAE,MAAM,aAAa,CAAgB;GAChD,OAAO;EAGT;GAEE,IAAI,MAAM,KAAK,WAAW,OAAO,GAAG;IAKlC,IAAI,MAAM,WACR,OAAO;IAKT,IAAI,MAAM,MAAM,MAAM;KACpB,MAAM,WAAW,wBAAwB,OAAO,MAAM,MAAM,MAAM,EAAE;KACpE,IAAI,UAAU;MACZ,SAAsC,OAAO,MAAM;MACnD,OAAO;KACT;IACF;IAQA,MAAM,KAAK;KACT,MAAM,MAAM;KACZ,GAAI,MAAM,MAAM,QAAQ,EAAE,IAAI,MAAM,GAAG;KACvC,MAAM,MAAM;IACd,CAAgB;IAChB,OAAO;GACT;GAEA,OAAO;CAEX;AACF;;;;;;;;;;;;;;;;;;;;;;;;AAyBA,SAAgB,cACd,OACA,OACS;CACT,IACE,MAAM,SAAS,sBACf,MAAM,SAAS,sBACf,MAAM,SAAS,wBAEf,OAAO;CAET,IAAI,CAAC,MAAM,YAAY,OAAO;CAC9B,MAAM,WAAW,qBAAqB,OAAO,MAAM,UAAU;CAC7D,IAAI,CAAC,UAAU,OAAO;CACtB,IAAI,MAAM,SAAS,oBAAoB,OAAO;CAE9C,OADe,SAAqC,UACnC;AACnB;;;;;AAMA,SAAS,mBACP,OACA,MACyB;CACzB,KAAK,IAAI,IAAI,MAAM,SAAS,GAAG,KAAK,GAAG,KACrC,IAAI,MAAM,GAAG,SAAS,MACpB,OAAO,MAAM;AAInB;;;;;AAMA,SAAS,qBACP,OACA,YACyB;CACzB,IAAI,CAAC,YAAY,OAAO,KAAA;CACxB,KAAK,IAAI,IAAI,MAAM,SAAS,GAAG,KAAK,GAAG,KAAK;EAC1C,MAAM,IAAI,MAAM;EAChB,IAAI,gBAAgB,KAAK,EAAE,eAAe,YACxC,OAAO;CAEX;AAEF;;;;;;;AAQA,SAAS,sBACP,MACA,UACM;CACN,IAAI,YAAY,MAAM;CACtB,MAAM,IAAI;CACV,EAAE,mBAAmB;EACnB,GAAI,EAAE;EACN,GAAG;CACL;AACF;;;;;;AAOA,SAAS,wBACP,OACA,MACA,IACyB;CACzB,KAAK,IAAI,IAAI,MAAM,SAAS,GAAG,KAAK,GAAG,KAAK;EAC1C,MAAM,IAAI,MAAM;EAChB,IAAI,EAAE,SAAS,QAAQ,QAAQ,KAAM,EAAqB,OAAO,IAC/D,OAAO;CAEX;AAEF;;;ACtiBA,SAAS,SAAS,MAAgD;CAChE,OAAO,CAAC,GAAG,IAAI,EAAE,MAAM,GAAG,MAAM;EAC9B,IAAI,EAAE,UAAU,EAAE,OAAO,OAAO,EAAE,QAAQ,EAAE;EAC5C,OAAO,EAAE,MAAM,cAAc,EAAE,KAAK;CACtC,CAAC;AACH;AAEA,SAAS,eACP,UAC+D;CAC/D,MAAM,UAA+C,CAAC;CACtD,MAAM,cAAmC,CAAC;CAC1C,KAAK,MAAM,OAAO,OAAO,OAAO,QAAQ,GACtC,IAAI,IAAI,kBAAkB;EACxB,QAAQ,IAAI,oBAAoB,QAAQ,IAAI,qBAAqB,CAAC;EAClE,QAAQ,IAAI,kBAAkB,KAAK,GAAG;CACxC,OACE,YAAY,KAAK,GAAG;CAGxB,KAAK,MAAM,CAAC,YAAY,SAAS,OAAO,QAAQ,OAAO,GACrD,QAAQ,cAAc,SAAS,IAAI;CAErC,OAAO;EAAE,kBAAkB;EAAS,aAAa,SAAS,WAAW;CAAE;AACzE;AAEA,SAAS,SACP,SAC+B;CAC/B,MAAM,EAAE,UAAU;CAClB,IAAI,MAAM,SAAS,WACjB,OAAO;EACL,OAAO,MAAM;EACb,WAAW,MAAM;EACjB,kBAAkB,QAAQ;EAC1B,cAAc,MAAM;EACpB,OAAO,MAAM;EACb,SAAS,MAAM;EACf,QAAQ;EACR,OAAO,CAAC;EACR,UAAU;GAAE,OAAO,MAAM;GAAW,MAAM,MAAM;EAAM;CACxD;AAGJ;AAEA,SAAS,WACP,MACA,SAC+B;CAC/B,MAAM,SAAS,QAAQ,SAAS,OAAO;CACvC,MAAM,EAAE,UAAU;CAElB,QAAQ,MAAM,MAAd;EACE,KAAK;GACH,IACE,QAAQ,WAAW,eACnB,QAAQ,WAAW,WACnB,QAAQ,WAAW,aACnB,QAAQ,WAAW,eAEnB,OAAO;GAET,OAAO;IACL,GAAG;IACH,OAAO,MAAM;IACb,WAAW,MAAM;IACjB,kBAAkB,QAAQ;IAC1B,cAAc,MAAM;IACpB,OAAO,MAAM;IACb,SAAS,MAAM;IACf,QAAQ;IACR,OAAO,QAAQ,SAAS,CAAC;IACzB,UAAU;KAAE,OAAO,MAAM;KAAW,MAAM,MAAM;IAAM;GACxD;EACF,KAAK,SAAS;GACZ,IAAI,CAAC,QAAQ,OAAO,KAAA;GACpB,MAAM,QAAQ,CAAC,GAAG,OAAO,KAAK;GAC9B,IAAI;IACF,kBAAkB,OAAO,KAAK,MAAM,MAAM,IAAI,CAAC;GACjD,QAAQ;IACN,OAAO;GACT;GACA,OAAO;IAAE,GAAG;IAAQ;GAAM;EAC5B;EACA,KAAK;GACH,IAAI,CAAC,QAAQ,OAAO,KAAA;GACpB,OAAO;IACL,GAAG;IACH,QAAQ;IACR,SAAS,MAAM;IACf,OAAO,KAAA;GACT;EACF,KAAK;GACH,IAAI,CAAC,QAAQ,OAAO,KAAA;GACpB,OAAO;IAAE,GAAG;IAAQ,QAAQ;IAAS,OAAO,MAAM;GAAM;EAC1D,KAAK;GACH,IAAI,CAAC,QAAQ,OAAO,KAAA;GACpB,OAAO;IAAE,GAAG;IAAQ,QAAQ;IAAW,OAAO,MAAM;GAAO;EAC7D,KAAK;GACH,IAAI,CAAC,QAAQ,OAAO,KAAA;GACpB,OAAO;IAAE,GAAG;IAAQ,QAAQ;IAAe,OAAO,MAAM;GAAM;CAClE;AACF;AAEA,SAAgB,4BAAiD;CAC/D,OAAO;EACL,UAAU,CAAC;EACX,kBAAkB,CAAC;EACnB,aAAa,CAAC;CAChB;AACF;AAEA,SAAgB,oBACd,OACA,SACqB;CACrB,IAAI,QAAQ,SAAS,oBAAoB,OAAO;CAChD,MAAM,QAAQ,QAAQ,MAAM;CAC5B,MAAM,UAAU,WAAW,MAAM,SAAS,QAAQ,OAAO;CACzD,IAAI,CAAC,SAAS,OAAO;CAErB,MAAM,WAAW;EAAE,GAAG,MAAM;GAAW,QAAQ;CAAQ;CACvD,OAAO;EAAE;EAAU,GAAG,eAAe,QAAQ;CAAE;AACjD"}

package/dist/agent-tools.d.ts

@@ -1,20 +1,21 @@
 import {
   a as AgentToolEventState,
-  c as AgentToolRunInspection,
-  d as AgentToolStoredChunk,
-  f as AgentToolTerminalStatus,
-  h as RunAgentToolResult,
+  c as AgentToolRunInfo,
+  d as AgentToolRunStatus,
+  f as AgentToolStoredChunk,
+  g as RunAgentToolResult,
+  h as RunAgentToolOptions,
   i as AgentToolEventMessage,
-  l as AgentToolRunState,
-  m as RunAgentToolOptions,
+  l as AgentToolRunInspection,
+  m as ChatCapableAgentClass,
   n as AgentToolDisplayMetadata,
-  o as AgentToolLifecycleResult,
-  p as ChatCapableAgentClass,
+  o as AgentToolFailure,
+  p as AgentToolTerminalStatus,
   r as AgentToolEvent,
-  s as AgentToolRunInfo,
+  s as AgentToolLifecycleResult,
   t as AgentToolChildAdapter,
-  u as AgentToolRunStatus
-} from "./agent-tool-types-l98LCbBl.js";
+  u as AgentToolRunState
+} from "./agent-tool-types-BAJWu8s4.js";
 import { Tool } from "ai";
 
 //#region src/agent-tools.d.ts
@@ -36,15 +37,7 @@ type AgentToolFactoryOptions<Output = unknown> = {
 declare function agentTool<Input = unknown, Output = unknown>(
   cls: ChatCapableAgentClass,
   options: AgentToolFactoryOptions<Output>
-): Tool<
-  Input,
-  | string
-  | Output
-  | {
-      ok: false;
-      error: string;
-    }
->;
+): Tool<Input, string | Output | AgentToolFailure>;
 //#endregion
 export {
   type AgentToolChildAdapter,
@@ -53,6 +46,7 @@ export {
   type AgentToolEventMessage,
   type AgentToolEventState,
   type AgentToolFactoryOptions,
+  type AgentToolFailure,
   type AgentToolLifecycleResult,
   type AgentToolRunInfo,
   type AgentToolRunInspection,

package/dist/agent-tools.js

@@ -6,10 +6,12 @@ function currentAgentToolRunner() {
 	if (agent === null || typeof agent !== "object" || typeof agent.runAgentTool !== "function") throw new Error("agentTool() can only run inside an Agent turn. Use it from getTools() on an Agent subclass.");
 	return agent;
 }
-function failure(error) {
+function failure(status, error, retryable) {
 	return {
 		ok: false,
-		error
+		status,
+		error,
+		retryable
 	};
 }
 /**
@@ -26,22 +28,24 @@ function agentTool(cls, options) {
 				...options.displayName ? { name: options.displayName } : {},
 				...options.icon ? { icon: options.icon } : {}
 			} : void 0;
+			const runId = executeOptions?.toolCallId ? `agent-tool:${executeOptions.toolCallId}` : void 0;
 			const result = await currentAgentToolRunner().runAgentTool(cls, {
 				input,
+				runId,
 				parentToolCallId: executeOptions?.toolCallId,
 				signal: executeOptions?.abortSignal,
 				display
 			});
 			if (result.status === "completed") {
 				if (options.outputSchema) {
-					if (result.output === void 0) return failure("agent tool completed without structured output required by outputSchema");
+					if (result.output === void 0) return failure("error", "agent tool completed without structured output required by outputSchema", false);
 					return options.outputSchema.parse(result.output);
 				}
 				return result.summary ?? "";
 			}
-			if (result.status === "aborted") return failure("agent tool run was cancelled");
-			if (result.status === "interrupted") return failure("agent tool run was interrupted; no recoverable output");
-			return failure(result.error ?? "agent tool run failed");
+			if (result.status === "aborted") return failure("aborted", result.error ?? "agent tool run was cancelled", false);
+			if (result.status === "interrupted") return failure("interrupted", result.error ?? "agent tool run was interrupted before it finished; it can be retried", true);
+			return failure("error", result.error ?? "agent tool run failed", false);
 		}
 	});
 }

package/dist/agent-tools.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-tools.js","names":["agentContext","createTool"],"sources":["../src/agent-tools.ts"],"sourcesContent":["import { tool, type Tool } from \"ai\";\nimport { __DO_NOT_USE_WILL_BREAK__agentContext as agentContext } from \"./internal_context\";\nimport type {\n  ChatCapableAgentClass,\n  RunAgentToolOptions,\n  RunAgentToolResult,\n  AgentToolDisplayMetadata\n} from \"./agent-tool-types\";\n\ntype SchemaLike<T = unknown> = {\n  parse(value: unknown): T;\n};\n\ntype AgentToolFactoryOptions<Output = unknown> = {\n  description: string;\n  inputSchema: unknown;\n  outputSchema?: SchemaLike<Output>;\n  displayName?: string;\n  icon?: string;\n  display?: AgentToolDisplayMetadata;\n};\n\ntype ToolExecutionOptions = {\n  toolCallId?: string;\n  abortSignal?: AbortSignal;\n};\n\ntype AgentToolRunner = {\n  runAgentTool<Input, Output>(\n    cls: ChatCapableAgentClass,\n    options: RunAgentToolOptions<Input>\n  ): Promise<RunAgentToolResult<Output>>;\n};\n\nfunction currentAgentToolRunner(): AgentToolRunner {\n  const agent = agentContext.getStore()?.agent;\n  if (\n    agent === null ||\n    typeof agent !== \"object\" ||\n    typeof (agent as { runAgentTool?: unknown }).runAgentTool !== \"function\"\n  ) {\n    throw new Error(\n      \"agentTool() can only run inside an Agent turn. Use it from getTools() on an Agent subclass.\"\n    );\n  }\n  return agent as AgentToolRunner;\n}\n\nfunction failure(error: string): { ok: false; error: string } {\n  return { ok: false, error };\n}\n\n/**\n * Create an AI SDK tool that dispatches a chat-capable sub-agent through\n * `Agent.runAgentTool`.\n */\nexport function agentTool<Input = unknown, Output = unknown>(\n  cls: ChatCapableAgentClass,\n  options: AgentToolFactoryOptions<Output>\n): Tool<Input, string | Output | { ok: false; error: string }> {\n  const createTool = tool as unknown as <I, O>(config: {\n    description: string;\n    inputSchema: unknown;\n    execute: (input: I, options?: ToolExecutionOptions) => Promise<O>;\n  }) => Tool<I, O>;\n\n  return createTool<Input, string | Output | { ok: false; error: string }>({\n    description: options.description,\n    inputSchema: options.inputSchema,\n    execute: async (input: Input, executeOptions?: ToolExecutionOptions) => {\n      const display: AgentToolDisplayMetadata | undefined =\n        options.displayName || options.icon || options.display\n          ? {\n              ...options.display,\n              ...(options.displayName ? { name: options.displayName } : {}),\n              ...(options.icon ? { icon: options.icon } : {})\n            }\n          : undefined;\n\n      const result = await currentAgentToolRunner().runAgentTool<Input, Output>(\n        cls,\n        {\n          input,\n          parentToolCallId: executeOptions?.toolCallId,\n          signal: executeOptions?.abortSignal,\n          display\n        }\n      );\n\n      if (result.status === \"completed\") {\n        if (options.outputSchema) {\n          if (result.output === undefined) {\n            return failure(\n              \"agent tool completed without structured output required by outputSchema\"\n            );\n          }\n          return options.outputSchema.parse(result.output);\n        }\n        return result.summary ?? \"\";\n      }\n\n      if (result.status === \"aborted\") {\n        return failure(\"agent tool run was cancelled\");\n      }\n      if (result.status === \"interrupted\") {\n        return failure(\"agent tool run was interrupted; no recoverable output\");\n      }\n      return failure(result.error ?? \"agent tool run failed\");\n    }\n  });\n}\n\nexport type { AgentToolFactoryOptions };\nexport type {\n  AgentToolChildAdapter,\n  AgentToolDisplayMetadata,\n  AgentToolEvent,\n  AgentToolEventMessage,\n  AgentToolEventState,\n  AgentToolLifecycleResult,\n  AgentToolRunInfo,\n  AgentToolRunInspection,\n  AgentToolRunState,\n  AgentToolRunStatus,\n  AgentToolStoredChunk,\n  AgentToolTerminalStatus,\n  ChatCapableAgentClass,\n  RunAgentToolOptions,\n  RunAgentToolResult\n} from \"./agent-tool-types\";\n"],"mappings":";;;AAkCA,SAAS,yBAA0C;CACjD,MAAM,QAAQA,sCAAa,SAAS,GAAG;CACvC,IACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAQ,MAAqC,iBAAiB,YAE9D,MAAM,IAAI,MACR,6FACF;CAEF,OAAO;AACT;AAEA,SAAS,QAAQ,OAA6C;CAC5D,OAAO;EAAE,IAAI;EAAO;CAAM;AAC5B;;;;;AAMA,SAAgB,UACd,KACA,SAC6D;CAO7D,OAAOC,KAAkE;EACvE,aAAa,QAAQ;EACrB,aAAa,QAAQ;EACrB,SAAS,OAAO,OAAc,mBAA0C;GACtE,MAAM,UACJ,QAAQ,eAAe,QAAQ,QAAQ,QAAQ,UAC3C;IACE,GAAG,QAAQ;IACX,GAAI,QAAQ,cAAc,EAAE,MAAM,QAAQ,YAAY,IAAI,CAAC;IAC3D,GAAI,QAAQ,OAAO,EAAE,MAAM,QAAQ,KAAK,IAAI,CAAC;GAC/C,IACA,KAAA;GAEN,MAAM,SAAS,MAAM,uBAAuB,EAAE,aAC5C,KACA;IACE;IACA,kBAAkB,gBAAgB;IAClC,QAAQ,gBAAgB;IACxB;GACF,CACF;GAEA,IAAI,OAAO,WAAW,aAAa;IACjC,IAAI,QAAQ,cAAc;KACxB,IAAI,OAAO,WAAW,KAAA,GACpB,OAAO,QACL,yEACF;KAEF,OAAO,QAAQ,aAAa,MAAM,OAAO,MAAM;IACjD;IACA,OAAO,OAAO,WAAW;GAC3B;GAEA,IAAI,OAAO,WAAW,WACpB,OAAO,QAAQ,8BAA8B;GAE/C,IAAI,OAAO,WAAW,eACpB,OAAO,QAAQ,uDAAuD;GAExE,OAAO,QAAQ,OAAO,SAAS,uBAAuB;EACxD;CACF,CAAC;AACH"}
+{"version":3,"file":"agent-tools.js","names":["agentContext","createTool"],"sources":["../src/agent-tools.ts"],"sourcesContent":["import { tool, type Tool } from \"ai\";\nimport { __DO_NOT_USE_WILL_BREAK__agentContext as agentContext } from \"./internal_context\";\nimport type {\n  ChatCapableAgentClass,\n  RunAgentToolOptions,\n  RunAgentToolResult,\n  AgentToolDisplayMetadata,\n  AgentToolFailure\n} from \"./agent-tool-types\";\n\ntype SchemaLike<T = unknown> = {\n  parse(value: unknown): T;\n};\n\ntype AgentToolFactoryOptions<Output = unknown> = {\n  description: string;\n  inputSchema: unknown;\n  outputSchema?: SchemaLike<Output>;\n  displayName?: string;\n  icon?: string;\n  display?: AgentToolDisplayMetadata;\n};\n\ntype ToolExecutionOptions = {\n  toolCallId?: string;\n  abortSignal?: AbortSignal;\n};\n\ntype AgentToolRunner = {\n  runAgentTool<Input, Output>(\n    cls: ChatCapableAgentClass,\n    options: RunAgentToolOptions<Input>\n  ): Promise<RunAgentToolResult<Output>>;\n};\n\nfunction currentAgentToolRunner(): AgentToolRunner {\n  const agent = agentContext.getStore()?.agent;\n  if (\n    agent === null ||\n    typeof agent !== \"object\" ||\n    typeof (agent as { runAgentTool?: unknown }).runAgentTool !== \"function\"\n  ) {\n    throw new Error(\n      \"agentTool() can only run inside an Agent turn. Use it from getTools() on an Agent subclass.\"\n    );\n  }\n  return agent as AgentToolRunner;\n}\n\nfunction failure(\n  status: AgentToolFailure[\"status\"],\n  error: string,\n  retryable: boolean\n): AgentToolFailure {\n  return { ok: false, status, error, retryable };\n}\n\n/**\n * Create an AI SDK tool that dispatches a chat-capable sub-agent through\n * `Agent.runAgentTool`.\n */\nexport function agentTool<Input = unknown, Output = unknown>(\n  cls: ChatCapableAgentClass,\n  options: AgentToolFactoryOptions<Output>\n): Tool<Input, string | Output | AgentToolFailure> {\n  const createTool = tool as unknown as <I, O>(config: {\n    description: string;\n    inputSchema: unknown;\n    execute: (input: I, options?: ToolExecutionOptions) => Promise<O>;\n  }) => Tool<I, O>;\n\n  return createTool<Input, string | Output | AgentToolFailure>({\n    description: options.description,\n    inputSchema: options.inputSchema,\n    execute: async (input: Input, executeOptions?: ToolExecutionOptions) => {\n      const display: AgentToolDisplayMetadata | undefined =\n        options.displayName || options.icon || options.display\n          ? {\n              ...options.display,\n              ...(options.displayName ? { name: options.displayName } : {}),\n              ...(options.icon ? { icon: options.icon } : {})\n            }\n          : undefined;\n\n      // Derive a STABLE runId from the tool call id (#1630). A tool call's id is\n      // preserved in the transcript, so when a parent turn is re-run by chat\n      // recovery after a deploy / eviction, the same `agentTool()` call resolves\n      // to the same runId — turning the re-issue into a duplicate that\n      // `runAgentTool` re-attaches to the still-running child, instead of a\n      // fresh `nanoid` that spawns a brand-new child and re-runs already-\n      // completed work (\"the agent went all the way back\"). Falls back to a\n      // fresh id only when there is no tool call id (rare; preserves prior\n      // behavior).\n      const runId = executeOptions?.toolCallId\n        ? `agent-tool:${executeOptions.toolCallId}`\n        : undefined;\n\n      const result = await currentAgentToolRunner().runAgentTool<Input, Output>(\n        cls,\n        {\n          input,\n          runId,\n          parentToolCallId: executeOptions?.toolCallId,\n          signal: executeOptions?.abortSignal,\n          display\n        }\n      );\n\n      if (result.status === \"completed\") {\n        if (options.outputSchema) {\n          if (result.output === undefined) {\n            return failure(\n              \"error\",\n              \"agent tool completed without structured output required by outputSchema\",\n              false\n            );\n          }\n          return options.outputSchema.parse(result.output);\n        }\n        return result.summary ?? \"\";\n      }\n\n      if (result.status === \"aborted\") {\n        // Intentional cancellation (parent/user stopped the run) — not retryable.\n        return failure(\n          \"aborted\",\n          result.error ?? \"agent tool run was cancelled\",\n          false\n        );\n      }\n      if (result.status === \"interrupted\") {\n        // The child was reset/superseded by a deploy or parent recovery before\n        // it reached a logical outcome. Re-dispatching the run can succeed, so\n        // surface it as retryable rather than a terminal failure the parent\n        // would report to the user as final.\n        return failure(\n          \"interrupted\",\n          result.error ??\n            \"agent tool run was interrupted before it finished; it can be retried\",\n          true\n        );\n      }\n      return failure(\"error\", result.error ?? \"agent tool run failed\", false);\n    }\n  });\n}\n\nexport type { AgentToolFactoryOptions };\nexport type {\n  AgentToolChildAdapter,\n  AgentToolDisplayMetadata,\n  AgentToolEvent,\n  AgentToolEventMessage,\n  AgentToolEventState,\n  AgentToolFailure,\n  AgentToolLifecycleResult,\n  AgentToolRunInfo,\n  AgentToolRunInspection,\n  AgentToolRunState,\n  AgentToolRunStatus,\n  AgentToolStoredChunk,\n  AgentToolTerminalStatus,\n  ChatCapableAgentClass,\n  RunAgentToolOptions,\n  RunAgentToolResult\n} from \"./agent-tool-types\";\n"],"mappings":";;;AAmCA,SAAS,yBAA0C;CACjD,MAAM,QAAQA,sCAAa,SAAS,GAAG;CACvC,IACE,UAAU,QACV,OAAO,UAAU,YACjB,OAAQ,MAAqC,iBAAiB,YAE9D,MAAM,IAAI,MACR,6FACF;CAEF,OAAO;AACT;AAEA,SAAS,QACP,QACA,OACA,WACkB;CAClB,OAAO;EAAE,IAAI;EAAO;EAAQ;EAAO;CAAU;AAC/C;;;;;AAMA,SAAgB,UACd,KACA,SACiD;CAOjD,OAAOC,KAAsD;EAC3D,aAAa,QAAQ;EACrB,aAAa,QAAQ;EACrB,SAAS,OAAO,OAAc,mBAA0C;GACtE,MAAM,UACJ,QAAQ,eAAe,QAAQ,QAAQ,QAAQ,UAC3C;IACE,GAAG,QAAQ;IACX,GAAI,QAAQ,cAAc,EAAE,MAAM,QAAQ,YAAY,IAAI,CAAC;IAC3D,GAAI,QAAQ,OAAO,EAAE,MAAM,QAAQ,KAAK,IAAI,CAAC;GAC/C,IACA,KAAA;GAWN,MAAM,QAAQ,gBAAgB,aAC1B,cAAc,eAAe,eAC7B,KAAA;GAEJ,MAAM,SAAS,MAAM,uBAAuB,EAAE,aAC5C,KACA;IACE;IACA;IACA,kBAAkB,gBAAgB;IAClC,QAAQ,gBAAgB;IACxB;GACF,CACF;GAEA,IAAI,OAAO,WAAW,aAAa;IACjC,IAAI,QAAQ,cAAc;KACxB,IAAI,OAAO,WAAW,KAAA,GACpB,OAAO,QACL,SACA,2EACA,KACF;KAEF,OAAO,QAAQ,aAAa,MAAM,OAAO,MAAM;IACjD;IACA,OAAO,OAAO,WAAW;GAC3B;GAEA,IAAI,OAAO,WAAW,WAEpB,OAAO,QACL,WACA,OAAO,SAAS,gCAChB,KACF;GAEF,IAAI,OAAO,WAAW,eAKpB,OAAO,QACL,eACA,OAAO,SACL,wEACF,IACF;GAEF,OAAO,QAAQ,SAAS,OAAO,SAAS,yBAAyB,KAAK;EACxE;CACF,CAAC;AACH"}

package/dist/browser/ai.d.ts

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

package/dist/browser/ai.js

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

package/dist/browser/index.d.ts

@@ -4,7 +4,7 @@ import {
   n as EXECUTE_DESCRIPTION,
   r as SEARCH_DESCRIPTION,
   t as BrowserToolsOptions
-} from "../shared-Cvj92byG.js";
+} from "../shared-4CAYLCTO.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-CiKaIK4h.js";
+import { a as connectBrowser, i as CdpSession, n as SEARCH_DESCRIPTION, o as connectUrl, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-BIpUk4G5.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-Cvj92byG.js";
+import { t as BrowserToolsOptions } from "../shared-4CAYLCTO.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-CiKaIK4h.js";
+import { n as SEARCH_DESCRIPTION, r as createBrowserToolHandlers, t as EXECUTE_DESCRIPTION } from "../shared-BIpUk4G5.js";
 import { z } from "zod";
 import { toolDefinition } from "@tanstack/ai";
 //#region src/browser/tanstack-ai.ts