@tanstack/ai 0.28.0 → 0.31.0

package/dist/esm/activities/chat/middleware/index.d.ts

@@ -1,2 +1,10 @@
 export type { ChatMiddleware, ChatMiddlewareContext, ChatMiddlewarePhase, ChatMiddlewareConfig, StructuredOutputMiddlewareConfig, ToolCallHookContext, BeforeToolCallDecision, AfterToolCallInfo, IterationInfo, ToolPhaseCompleteInfo, UsageInfo, FinishInfo, AbortInfo, ErrorInfo, } from './types.js';
 export { MiddlewareRunner } from './compose.js';
+export { createCapability, CapabilityRegistry } from './capabilities.js';
+export type { Capability, CapabilityHandle, CapabilityContext, CapabilityGetter, CapabilityProvider, CapabilityGetOptions, } from './capabilities.js';
+export { defineChatMiddleware } from './define.js';
+export type { DefinedChatMiddleware } from './define.js';
+export { createChatMiddleware } from './builder.js';
+export type { ChatMiddlewareBuilder, MissingCapabilities, NamesOf, } from './builder.js';
+export { validateCapabilities } from './validate.js';
+export type { AnyChatMiddleware } from './types.js';

package/dist/esm/activities/chat/middleware/types.d.ts

@@ -1,5 +1,6 @@
 import { JSONSchema, ModelMessage, StreamChunk, TokenUsage, Tool, ToolCall } from '../../../types.js';
 import { SystemPrompt } from '../../../system-prompts.js';
+import { Capability, CapabilityHandle, CapabilityRegistry } from './capabilities.js';
 /**
  * Phase of the chat middleware lifecycle.
  * - 'init': Initial config transform before the chat engine starts
@@ -81,6 +82,28 @@ export interface ChatMiddlewareContext<TContext = unknown> {
     messages: ReadonlyArray<ModelMessage>;
     /** Generate a unique ID with the given prefix */
     createId: (prefix: string) => string;
+    /**
+     * Capability bookkeeping for this request. Populated by middleware `setup`
+     * hooks (via `provide` accessors) and read by later middleware (via `get`
+     * accessors). Prefer the accessors returned by `createCapability` over using
+     * this directly. Orthogonal to `context` (the user runtime context).
+     */
+    capabilities: CapabilityRegistry;
+    /**
+     * Read a provided capability by its handle. Equivalent to the handle's own
+     * `get` accessor (`getX(ctx)`); throws if the capability was never provided.
+     */
+    get: <TValue>(capability: Capability<TValue>) => TValue;
+    /**
+     * Read a capability by its handle, returning `undefined` if it was never
+     * provided (never throws).
+     */
+    getOptional: <TValue>(capability: Capability<TValue>) => TValue | undefined;
+    /**
+     * Provide a capability value. Equivalent to the handle's own `provide`
+     * accessor (`provideX(ctx, value)`). Typically called from `setup`.
+     */
+    provide: <TValue>(capability: Capability<TValue>, value: TValue) => void;
 }
 /**
  * Chat configuration that middleware can observe or transform.
@@ -270,6 +293,32 @@ export interface ErrorInfo {
 export interface ChatMiddleware<TContext = unknown> {
     /** Optional name for debugging and identification */
     name?: string;
+    /**
+     * Capabilities this middleware requires. `chat()` validates that some
+     * middleware (or the adapter) provides each one; unsatisfied requirements are
+     * a compile-time error (array coverage / builder) and a runtime error before
+     * the adapter runs.
+     */
+    requires?: ReadonlyArray<CapabilityHandle>;
+    /**
+     * Capabilities this middleware provides. Each declared capability MUST be
+     * provided (via its `provide` accessor) inside `setup`, or `chat()` throws
+     * after the setup phase.
+     */
+    provides?: ReadonlyArray<CapabilityHandle>;
+    /**
+     * Capabilities this middleware uses if present but does not require.
+     * Non-gating: never causes a validation error. Read with
+     * `getX(ctx, { optional: true })`.
+     */
+    optionalRequires?: ReadonlyArray<CapabilityHandle>;
+    /**
+     * Provisioning hook. Runs FIRST — before `onConfig` (init) — across all
+     * middleware in array order. Use it to call `provide` accessors so later
+     * middleware (`onConfig` onward) can consume the capabilities. Receives the
+     * stable context; does NOT receive the mutable config.
+     */
+    setup?: (ctx: ChatMiddlewareContext<TContext>) => void | Promise<void>;
     /**
      * Called to observe or transform the chat configuration.
      * Called at init and at the beginning of each agent iteration.
@@ -345,3 +394,5 @@ export interface ChatMiddleware<TContext = unknown> {
      */
     onError?: (ctx: ChatMiddlewareContext<TContext>, info: ErrorInfo) => void | Promise<void>;
 }
+/** A `ChatMiddleware` with a permissive context — for use as a constraint. */
+export type AnyChatMiddleware = ChatMiddleware<any>;

package/dist/esm/activities/chat/middleware/validate.d.ts

@@ -0,0 +1,19 @@
+import { CapabilityHandle } from './capabilities.js';
+import { AnyChatMiddleware } from './types.js';
+/** Minimal adapter shape needed for capability validation. */
+interface CapabilityRequiringAdapter {
+    name: string;
+    requires?: ReadonlyArray<CapabilityHandle>;
+}
+/**
+ * Runtime validation: every required capability (from middleware `requires` and
+ * the adapter's `requires`) must be provided by some middleware's `provides`.
+ * `optionalRequires` is never gating. Throws a clear error otherwise.
+ *
+ * Presence only — ORDER is not validated here (a provider may appear after its
+ * consumer in the array and still pass). Use the `createChatMiddleware()`
+ * builder for compile-time order enforcement; at runtime, a consumer that reads
+ * a not-yet-provided capability during `setup` fails loud via its getter.
+ */
+export declare function validateCapabilities(middlewares: ReadonlyArray<AnyChatMiddleware>, adapter: CapabilityRequiringAdapter): void;
+export {};

package/dist/esm/activities/chat/middleware/validate.js

@@ -0,0 +1,30 @@
+function validateCapabilities(middlewares, adapter) {
+  const provided = /* @__PURE__ */ new Set();
+  for (const mw of middlewares) {
+    for (const handle of mw.provides ?? []) provided.add(handle);
+  }
+  const providedNames = () => {
+    const names = [...provided].map((h) => h.capabilityName);
+    return names.length ? names.join(", ") : "none";
+  };
+  for (const handle of adapter.requires ?? []) {
+    if (!provided.has(handle)) {
+      throw new Error(
+        `Adapter "${adapter.name}" requires capability "${handle.capabilityName}". Provided capabilities: ${providedNames()}. Add a middleware that provides "${handle.capabilityName}".`
+      );
+    }
+  }
+  for (const mw of middlewares) {
+    for (const handle of mw.requires ?? []) {
+      if (!provided.has(handle)) {
+        throw new Error(
+          `Middleware "${mw.name ?? "unnamed"}" requires capability "${handle.capabilityName}". Provided capabilities: ${providedNames()}. Add a middleware that provides "${handle.capabilityName}".`
+        );
+      }
+    }
+  }
+}
+export {
+  validateCapabilities
+};
+//# sourceMappingURL=validate.js.map

package/dist/esm/activities/chat/middleware/validate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"validate.js","sources":["../../../../../src/activities/chat/middleware/validate.ts"],"sourcesContent":["import type { CapabilityHandle } from './capabilities'\nimport type { AnyChatMiddleware } from './types'\n\n/** Minimal adapter shape needed for capability validation. */\ninterface CapabilityRequiringAdapter {\n  name: string\n  requires?: ReadonlyArray<CapabilityHandle>\n}\n\n/**\n * Runtime validation: every required capability (from middleware `requires` and\n * the adapter's `requires`) must be provided by some middleware's `provides`.\n * `optionalRequires` is never gating. Throws a clear error otherwise.\n *\n * Presence only — ORDER is not validated here (a provider may appear after its\n * consumer in the array and still pass). Use the `createChatMiddleware()`\n * builder for compile-time order enforcement; at runtime, a consumer that reads\n * a not-yet-provided capability during `setup` fails loud via its getter.\n */\nexport function validateCapabilities(\n  middlewares: ReadonlyArray<AnyChatMiddleware>,\n  adapter: CapabilityRequiringAdapter,\n): void {\n  const provided = new Set<CapabilityHandle>()\n  for (const mw of middlewares) {\n    for (const handle of mw.provides ?? []) provided.add(handle)\n  }\n\n  const providedNames = (): string => {\n    const names = [...provided].map((h) => h.capabilityName)\n    return names.length ? names.join(', ') : 'none'\n  }\n\n  for (const handle of adapter.requires ?? []) {\n    if (!provided.has(handle)) {\n      throw new Error(\n        `Adapter \"${adapter.name}\" requires capability \"${handle.capabilityName}\". ` +\n          `Provided capabilities: ${providedNames()}. ` +\n          `Add a middleware that provides \"${handle.capabilityName}\".`,\n      )\n    }\n  }\n\n  for (const mw of middlewares) {\n    for (const handle of mw.requires ?? []) {\n      if (!provided.has(handle)) {\n        throw new Error(\n          `Middleware \"${mw.name ?? 'unnamed'}\" requires capability ` +\n            `\"${handle.capabilityName}\". Provided capabilities: ${providedNames()}. ` +\n            `Add a middleware that provides \"${handle.capabilityName}\".`,\n        )\n      }\n    }\n  }\n}\n"],"names":[],"mappings":"AAmBO,SAAS,qBACd,aACA,SACM;AACN,QAAM,+BAAe,IAAA;AACrB,aAAW,MAAM,aAAa;AAC5B,eAAW,UAAU,GAAG,YAAY,CAAA,EAAI,UAAS,IAAI,MAAM;AAAA,EAC7D;AAEA,QAAM,gBAAgB,MAAc;AAClC,UAAM,QAAQ,CAAC,GAAG,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE,cAAc;AACvD,WAAO,MAAM,SAAS,MAAM,KAAK,IAAI,IAAI;AAAA,EAC3C;AAEA,aAAW,UAAU,QAAQ,YAAY,CAAA,GAAI;AAC3C,QAAI,CAAC,SAAS,IAAI,MAAM,GAAG;AACzB,YAAM,IAAI;AAAA,QACR,YAAY,QAAQ,IAAI,0BAA0B,OAAO,cAAc,6BAC3C,cAAA,CAAe,qCACN,OAAO,cAAc;AAAA,MAAA;AAAA,IAE9D;AAAA,EACF;AAEA,aAAW,MAAM,aAAa;AAC5B,eAAW,UAAU,GAAG,YAAY,CAAA,GAAI;AACtC,UAAI,CAAC,SAAS,IAAI,MAAM,GAAG;AACzB,cAAM,IAAI;AAAA,UACR,eAAe,GAAG,QAAQ,SAAS,0BAC7B,OAAO,cAAc,6BAA6B,cAAA,CAAe,qCAClC,OAAO,cAAc;AAAA,QAAA;AAAA,MAE9D;AAAA,IACF;AAAA,EACF;AACF;"}

package/dist/esm/activities/chat/stream/message-updaters.js

@@ -101,7 +101,7 @@ function updateToolCallWithOutput(messages, toolCallId, output, state, errorText
       parts[index] = {
         ...toolCallPart,
         output: errorText ? { error: errorText } : output,
-        state: state ?? (errorText ? "input-complete" : "complete")
+        state: state ?? (errorText ? "error" : "complete")
       };
     }
     return { ...msg, parts };

package/dist/esm/activities/chat/stream/message-updaters.js.map

@@ -1 +1 @@
-{"version":3,"file":"message-updaters.js","sources":["../../../../../src/activities/chat/stream/message-updaters.ts"],"sourcesContent":["/**\n * Message Updaters (Internal)\n *\n * Internal helper functions for updating UIMessage parts.\n * These are used by StreamProcessor to manage the message array.\n */\n\nimport { parsePartialJSON } from './json-parser'\nimport type {\n  ContentPart,\n  StructuredOutputPart,\n  ThinkingPart,\n  ToolCallPart,\n  ToolResultPart,\n  UIMessage,\n} from '../../../types'\nimport type { ToolCallState, ToolResultState } from './types'\n\n/**\n * Update or add a text part to a message.\n *\n * If the last part is a text part, update it (continuing the same text segment).\n * Otherwise, create a new text part (starting a new text segment after tool calls).\n */\nexport function updateTextPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  content: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const lastPart = parts.length > 0 ? parts[parts.length - 1] : null\n\n    if (lastPart && lastPart.type === 'text') {\n      // Update the last text part (continuing same text segment)\n      parts[parts.length - 1] = { type: 'text', content }\n    } else {\n      // Create new text part (starting new text segment after tool calls/results)\n      parts.push({ type: 'text', content })\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a tool call part to a message.\n */\nexport function updateToolCallPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCall: {\n    id: string\n    name: string\n    arguments: string\n    state: ToolCallState\n    metadata?: Record<string, unknown>\n  },\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existing = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCall.id,\n    )\n\n    // Carry forward metadata from either the new toolCall or the existing\n    // part. Once the adapter has emitted metadata for a tool call (e.g.\n    // Gemini's thoughtSignature on TOOL_CALL_START) we must not lose it on\n    // subsequent updates that don't re-supply it.\n    const metadata = toolCall.metadata ?? existing?.metadata\n\n    const toolCallPart: ToolCallPart = {\n      type: 'tool-call',\n      id: toolCall.id,\n      name: toolCall.name,\n      arguments: toolCall.arguments,\n      state: toolCall.state,\n      // Carry forward approval and output from the existing part\n      ...(existing?.approval && { approval: { ...existing.approval } }),\n      ...(existing?.output !== undefined && { output: existing.output }),\n      ...(metadata !== undefined && { metadata }),\n    }\n\n    if (existing) {\n      // Update existing tool call\n      parts[parts.indexOf(existing)] = toolCallPart\n    } else {\n      // Add new tool call at the end (preserve natural streaming order)\n      parts.push(toolCallPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a tool result part to a message.\n */\nexport function updateToolResultPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  content: string | Array<ContentPart>,\n  state: ToolResultState,\n  error?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const resultPartIndex = parts.findIndex(\n      (p): p is ToolResultPart =>\n        p.type === 'tool-result' && p.toolCallId === toolCallId,\n    )\n\n    const toolResultPart: ToolResultPart = {\n      type: 'tool-result',\n      toolCallId,\n      content,\n      state,\n      ...(error && { error }),\n    }\n\n    if (resultPartIndex >= 0) {\n      parts[resultPartIndex] = toolResultPart\n    } else {\n      parts.push(toolResultPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with approval request metadata.\n */\nexport function updateToolCallApproval(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  approvalId: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = {\n        ...toolCallPart,\n        state: 'approval-requested',\n        approval: {\n          id: approvalId,\n          needsApproval: true,\n        },\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part's state (e.g., to \"input-complete\").\n */\nexport function updateToolCallState(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  state: ToolCallState,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = { ...toolCallPart, state }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with output.\n * Searches all messages to find the tool call by ID.\n */\nexport function updateToolCallWithOutput(\n  messages: Array<UIMessage>,\n  toolCallId: string,\n  output: any,\n  state?: ToolCallState,\n  errorText?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = {\n        ...toolCallPart,\n        output: errorText ? { error: errorText } : output,\n        state: state ?? (errorText ? 'input-complete' : 'complete'),\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with approval response.\n * Searches all messages to find the tool call by approval ID.\n */\nexport function updateToolCallApprovalResponse(\n  messages: Array<UIMessage>,\n  approvalId: string,\n  approved: boolean,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart =>\n        p.type === 'tool-call' && p.approval?.id === approvalId,\n    )\n\n    if (toolCallPart && toolCallPart.approval) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = {\n        ...toolCallPart,\n        approval: { ...toolCallPart.approval, approved },\n        state: 'approval-responded',\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Append a delta to the structured-output part on `messageId`, or create one\n * if absent. Progressive parse of the accumulated buffer fills `partial`.\n *\n * Callers must only invoke this while the part is still in flight — the\n * helper unconditionally writes `status: 'streaming'`, so feeding it a delta\n * after a `complete`/`error` terminal would regress the part. In practice the\n * processor gates calls via `structuredMessageIds`, which is dropped on\n * terminal events.\n *\n * If the progressive parse returns null/undefined (the buffer is not yet a\n * parseable JSON prefix), the previously-good `partial` is preserved so the\n * UI doesn't flicker back to empty for a single render.\n */\nexport function appendStructuredOutputDelta(\n  messages: Array<UIMessage>,\n  messageId: string,\n  delta: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existingIndex = parts.findIndex(\n      (p): p is StructuredOutputPart => p.type === 'structured-output',\n    )\n    const existing =\n      existingIndex >= 0 ? (parts[existingIndex] as StructuredOutputPart) : null\n\n    const nextRaw = (existing?.raw ?? '') + delta\n    const progressive = parsePartialJSON(nextRaw)\n    const nextPartial =\n      progressive !== undefined && progressive !== null\n        ? progressive\n        : existing?.partial\n\n    const nextPart: StructuredOutputPart = {\n      type: 'structured-output',\n      status: 'streaming',\n      raw: nextRaw,\n      ...(nextPartial !== undefined ? { partial: nextPartial } : {}),\n      ...(existing?.reasoning !== undefined\n        ? { reasoning: existing.reasoning }\n        : {}),\n    }\n\n    if (existingIndex >= 0) {\n      parts[existingIndex] = nextPart\n    } else {\n      parts.push(nextPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Snap the structured-output part on `messageId` to `complete` with the\n * validated `data`. Picks the freshest available `raw` so the wire\n * round-trip stays internally consistent:\n *\n *   1. Caller-supplied `raw` (the original streamed bytes from the model).\n *   2. The existing part's `raw` (deltas accumulated before this terminal).\n *   3. `JSON.stringify(data)` as a defensive fallback for terminal-only\n *      completes that never shipped raw — keeps the part self-consistent\n *      so downstream consumers never see a complete part with empty raw.\n */\nexport function completeStructuredOutputPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  data: unknown,\n  raw: string,\n  reasoning?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existingIndex = parts.findIndex(\n      (p): p is StructuredOutputPart => p.type === 'structured-output',\n    )\n\n    const existingRaw =\n      existingIndex >= 0\n        ? (parts[existingIndex] as StructuredOutputPart).raw\n        : ''\n    let resolvedRaw = raw || existingRaw\n    if (resolvedRaw === '' && data !== undefined) {\n      try {\n        resolvedRaw = JSON.stringify(data)\n      } catch {\n        // Unserializable (circular, BigInt, throwing toJSON). Leave raw\n        // empty. Both downstream paths handle this: `ag-ui-wire.ts`\n        // `collectText` skips complete parts with empty raw entirely, and\n        // `uiMessageToModelMessages` falls back to a defensive\n        // `safeJsonStringify(data)` which itself returns `''` for the same\n        // unserializable inputs — so the turn is silently dropped from the\n        // next request rather than shipping garbage or crashing the stream.\n      }\n    }\n\n    const nextPart: StructuredOutputPart = {\n      type: 'structured-output',\n      status: 'complete',\n      data,\n      partial: data,\n      raw: resolvedRaw,\n      ...(reasoning !== undefined ? { reasoning } : {}),\n    }\n\n    if (existingIndex >= 0) {\n      parts[existingIndex] = nextPart\n    } else {\n      parts.push(nextPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Mark the structured-output part on `messageId` as errored. If no part\n * exists yet — RUN_ERROR fired after `structured-output.start` but before\n * any delta — create an empty errored placeholder so consumers have\n * something renderable. Existing complete parts are left alone (an error\n * after a successful complete should not retroactively un-complete it).\n */\nexport function errorStructuredOutputPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  errorMessage: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existingIndex = parts.findIndex(\n      (p): p is StructuredOutputPart => p.type === 'structured-output',\n    )\n\n    if (existingIndex < 0) {\n      parts.push({\n        type: 'structured-output',\n        status: 'error',\n        raw: '',\n        errorMessage,\n      })\n      return { ...msg, parts }\n    }\n\n    const existing = parts[existingIndex] as StructuredOutputPart\n    if (existing.status === 'complete') {\n      return msg\n    }\n    parts[existingIndex] = {\n      ...existing,\n      status: 'error',\n      errorMessage,\n    }\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a thinking part to a message, keyed by stepId.\n * Each distinct stepId produces its own ThinkingPart.\n */\nexport function updateThinkingPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  stepId: string,\n  content: string,\n  signature?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const thinkingPartIndex = parts.findIndex(\n      (p) => p.type === 'thinking' && p.stepId === stepId,\n    )\n\n    const thinkingPart: ThinkingPart = {\n      type: 'thinking',\n      content,\n      stepId,\n      ...(signature && { signature }),\n    }\n\n    if (thinkingPartIndex >= 0) {\n      // Update existing thinking part for this step\n      parts[thinkingPartIndex] = thinkingPart\n    } else {\n      // Add new thinking part at the end (preserve natural streaming order)\n      parts.push(thinkingPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n"],"names":[],"mappings":";AAwBO,SAAS,eACd,UACA,WACA,SACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,WAAW,MAAM,SAAS,IAAI,MAAM,MAAM,SAAS,CAAC,IAAI;AAE9D,QAAI,YAAY,SAAS,SAAS,QAAQ;AAExC,YAAM,MAAM,SAAS,CAAC,IAAI,EAAE,MAAM,QAAQ,QAAA;AAAA,IAC5C,OAAO;AAEL,YAAM,KAAK,EAAE,MAAM,QAAQ,SAAS;AAAA,IACtC;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,mBACd,UACA,WACA,UAOkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,WAAW,MAAM;AAAA,MACrB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO,SAAS;AAAA,IAAA;AAOxE,UAAM,WAAW,SAAS,YAAY,UAAU;AAEhD,UAAM,eAA6B;AAAA,MACjC,MAAM;AAAA,MACN,IAAI,SAAS;AAAA,MACb,MAAM,SAAS;AAAA,MACf,WAAW,SAAS;AAAA,MACpB,OAAO,SAAS;AAAA;AAAA,MAEhB,GAAI,UAAU,YAAY,EAAE,UAAU,EAAE,GAAG,SAAS,WAAS;AAAA,MAC7D,GAAI,UAAU,WAAW,UAAa,EAAE,QAAQ,SAAS,OAAA;AAAA,MACzD,GAAI,aAAa,UAAa,EAAE,SAAA;AAAA,IAAS;AAG3C,QAAI,UAAU;AAEZ,YAAM,MAAM,QAAQ,QAAQ,CAAC,IAAI;AAAA,IACnC,OAAO;AAEL,YAAM,KAAK,YAAY;AAAA,IACzB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,qBACd,UACA,WACA,YACA,SACA,OACA,OACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,kBAAkB,MAAM;AAAA,MAC5B,CAAC,MACC,EAAE,SAAS,iBAAiB,EAAE,eAAe;AAAA,IAAA;AAGjD,UAAM,iBAAiC;AAAA,MACrC,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA;AAAA,MACA,GAAI,SAAS,EAAE,MAAA;AAAA,IAAM;AAGvB,QAAI,mBAAmB,GAAG;AACxB,YAAM,eAAe,IAAI;AAAA,IAC3B,OAAO;AACL,YAAM,KAAK,cAAc;AAAA,IAC3B;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,uBACd,UACA,WACA,YACA,YACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,IAAA;AAG/D,QAAI,cAAc;AAChB,YAAM,QAAQ,MAAM,QAAQ,YAAY;AACxC,YAAM,KAAK,IAAI;AAAA,QACb,GAAG;AAAA,QACH,OAAO;AAAA,QACP,UAAU;AAAA,UACR,IAAI;AAAA,UACJ,eAAe;AAAA,QAAA;AAAA,MACjB;AAAA,IAEJ;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAkCO,SAAS,yBACd,UACA,YACA,QACA,OACA,WACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,IAAA;AAG/D,QAAI,cAAc;AAChB,YAAM,QAAQ,MAAM,QAAQ,YAAY;AACxC,YAAM,KAAK,IAAI;AAAA,QACb,GAAG;AAAA,QACH,QAAQ,YAAY,EAAE,OAAO,cAAc;AAAA,QAC3C,OAAO,UAAU,YAAY,mBAAmB;AAAA,MAAA;AAAA,IAEpD;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAMO,SAAS,+BACd,UACA,YACA,UACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MACC,EAAE,SAAS,eAAe,EAAE,UAAU,OAAO;AAAA,IAAA;AAGjD,QAAI,gBAAgB,aAAa,UAAU;AACzC,YAAM,QAAQ,MAAM,QAAQ,YAAY;AACxC,YAAM,KAAK,IAAI;AAAA,QACb,GAAG;AAAA,QACH,UAAU,EAAE,GAAG,aAAa,UAAU,SAAA;AAAA,QACtC,OAAO;AAAA,MAAA;AAAA,IAEX;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAgBO,SAAS,4BACd,UACA,WACA,OACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,gBAAgB,MAAM;AAAA,MAC1B,CAAC,MAAiC,EAAE,SAAS;AAAA,IAAA;AAE/C,UAAM,WACJ,iBAAiB,IAAK,MAAM,aAAa,IAA6B;AAExE,UAAM,WAAW,UAAU,OAAO,MAAM;AACxC,UAAM,cAAc,iBAAiB,OAAO;AAC5C,UAAM,cACJ,gBAAgB,UAAa,gBAAgB,OACzC,cACA,UAAU;AAEhB,UAAM,WAAiC;AAAA,MACrC,MAAM;AAAA,MACN,QAAQ;AAAA,MACR,KAAK;AAAA,MACL,GAAI,gBAAgB,SAAY,EAAE,SAAS,YAAA,IAAgB,CAAA;AAAA,MAC3D,GAAI,UAAU,cAAc,SACxB,EAAE,WAAW,SAAS,cACtB,CAAA;AAAA,IAAC;AAGP,QAAI,iBAAiB,GAAG;AACtB,YAAM,aAAa,IAAI;AAAA,IACzB,OAAO;AACL,YAAM,KAAK,QAAQ;AAAA,IACrB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAaO,SAAS,6BACd,UACA,WACA,MACA,KACA,WACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,gBAAgB,MAAM;AAAA,MAC1B,CAAC,MAAiC,EAAE,SAAS;AAAA,IAAA;AAG/C,UAAM,cACJ,iBAAiB,IACZ,MAAM,aAAa,EAA2B,MAC/C;AACN,QAAI,cAAc,OAAO;AACzB,QAAI,gBAAgB,MAAM,SAAS,QAAW;AAC5C,UAAI;AACF,sBAAc,KAAK,UAAU,IAAI;AAAA,MACnC,QAAQ;AAAA,MAQR;AAAA,IACF;AAEA,UAAM,WAAiC;AAAA,MACrC,MAAM;AAAA,MACN,QAAQ;AAAA,MACR;AAAA,MACA,SAAS;AAAA,MACT,KAAK;AAAA,MACL,GAAI,cAAc,SAAY,EAAE,cAAc,CAAA;AAAA,IAAC;AAGjD,QAAI,iBAAiB,GAAG;AACtB,YAAM,aAAa,IAAI;AAAA,IACzB,OAAO;AACL,YAAM,KAAK,QAAQ;AAAA,IACrB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AASO,SAAS,0BACd,UACA,WACA,cACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,gBAAgB,MAAM;AAAA,MAC1B,CAAC,MAAiC,EAAE,SAAS;AAAA,IAAA;AAG/C,QAAI,gBAAgB,GAAG;AACrB,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,QAAQ;AAAA,QACR,KAAK;AAAA,QACL;AAAA,MAAA,CACD;AACD,aAAO,EAAE,GAAG,KAAK,MAAA;AAAA,IACnB;AAEA,UAAM,WAAW,MAAM,aAAa;AACpC,QAAI,SAAS,WAAW,YAAY;AAClC,aAAO;AAAA,IACT;AACA,UAAM,aAAa,IAAI;AAAA,MACrB,GAAG;AAAA,MACH,QAAQ;AAAA,MACR;AAAA,IAAA;AAEF,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAMO,SAAS,mBACd,UACA,WACA,QACA,SACA,WACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,oBAAoB,MAAM;AAAA,MAC9B,CAAC,MAAM,EAAE,SAAS,cAAc,EAAE,WAAW;AAAA,IAAA;AAG/C,UAAM,eAA6B;AAAA,MACjC,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA,GAAI,aAAa,EAAE,UAAA;AAAA,IAAU;AAG/B,QAAI,qBAAqB,GAAG;AAE1B,YAAM,iBAAiB,IAAI;AAAA,IAC7B,OAAO;AAEL,YAAM,KAAK,YAAY;AAAA,IACzB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;"}
+{"version":3,"file":"message-updaters.js","sources":["../../../../../src/activities/chat/stream/message-updaters.ts"],"sourcesContent":["/**\n * Message Updaters (Internal)\n *\n * Internal helper functions for updating UIMessage parts.\n * These are used by StreamProcessor to manage the message array.\n */\n\nimport { parsePartialJSON } from './json-parser'\nimport type {\n  ContentPart,\n  StructuredOutputPart,\n  ThinkingPart,\n  ToolCallPart,\n  ToolResultPart,\n  UIMessage,\n} from '../../../types'\nimport type { ToolCallState, ToolResultState } from './types'\n\n/**\n * Update or add a text part to a message.\n *\n * If the last part is a text part, update it (continuing the same text segment).\n * Otherwise, create a new text part (starting a new text segment after tool calls).\n */\nexport function updateTextPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  content: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const lastPart = parts.length > 0 ? parts[parts.length - 1] : null\n\n    if (lastPart && lastPart.type === 'text') {\n      // Update the last text part (continuing same text segment)\n      parts[parts.length - 1] = { type: 'text', content }\n    } else {\n      // Create new text part (starting new text segment after tool calls/results)\n      parts.push({ type: 'text', content })\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a tool call part to a message.\n */\nexport function updateToolCallPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCall: {\n    id: string\n    name: string\n    arguments: string\n    state: ToolCallState\n    metadata?: Record<string, unknown>\n  },\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existing = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCall.id,\n    )\n\n    // Carry forward metadata from either the new toolCall or the existing\n    // part. Once the adapter has emitted metadata for a tool call (e.g.\n    // Gemini's thoughtSignature on TOOL_CALL_START) we must not lose it on\n    // subsequent updates that don't re-supply it.\n    const metadata = toolCall.metadata ?? existing?.metadata\n\n    const toolCallPart: ToolCallPart = {\n      type: 'tool-call',\n      id: toolCall.id,\n      name: toolCall.name,\n      arguments: toolCall.arguments,\n      state: toolCall.state,\n      // Carry forward approval and output from the existing part\n      ...(existing?.approval && { approval: { ...existing.approval } }),\n      ...(existing?.output !== undefined && { output: existing.output }),\n      ...(metadata !== undefined && { metadata }),\n    }\n\n    if (existing) {\n      // Update existing tool call\n      parts[parts.indexOf(existing)] = toolCallPart\n    } else {\n      // Add new tool call at the end (preserve natural streaming order)\n      parts.push(toolCallPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a tool result part to a message.\n */\nexport function updateToolResultPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  content: string | Array<ContentPart>,\n  state: ToolResultState,\n  error?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const resultPartIndex = parts.findIndex(\n      (p): p is ToolResultPart =>\n        p.type === 'tool-result' && p.toolCallId === toolCallId,\n    )\n\n    const toolResultPart: ToolResultPart = {\n      type: 'tool-result',\n      toolCallId,\n      content,\n      state,\n      ...(error && { error }),\n    }\n\n    if (resultPartIndex >= 0) {\n      parts[resultPartIndex] = toolResultPart\n    } else {\n      parts.push(toolResultPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with approval request metadata.\n */\nexport function updateToolCallApproval(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  approvalId: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = {\n        ...toolCallPart,\n        state: 'approval-requested',\n        approval: {\n          id: approvalId,\n          needsApproval: true,\n        },\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part's state (e.g., to \"input-complete\").\n */\nexport function updateToolCallState(\n  messages: Array<UIMessage>,\n  messageId: string,\n  toolCallId: string,\n  state: ToolCallState,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = { ...toolCallPart, state }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with output.\n * Searches all messages to find the tool call by ID.\n */\nexport function updateToolCallWithOutput(\n  messages: Array<UIMessage>,\n  toolCallId: string,\n  output: any,\n  state?: ToolCallState,\n  errorText?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart => p.type === 'tool-call' && p.id === toolCallId,\n    )\n\n    if (toolCallPart) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = {\n        ...toolCallPart,\n        output: errorText ? { error: errorText } : output,\n        state: state ?? (errorText ? 'error' : 'complete'),\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update a tool call part with approval response.\n * Searches all messages to find the tool call by approval ID.\n */\nexport function updateToolCallApprovalResponse(\n  messages: Array<UIMessage>,\n  approvalId: string,\n  approved: boolean,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    const parts = [...msg.parts]\n    const toolCallPart = parts.find(\n      (p): p is ToolCallPart =>\n        p.type === 'tool-call' && p.approval?.id === approvalId,\n    )\n\n    if (toolCallPart && toolCallPart.approval) {\n      const index = parts.indexOf(toolCallPart)\n      parts[index] = {\n        ...toolCallPart,\n        approval: { ...toolCallPart.approval, approved },\n        state: 'approval-responded',\n      }\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Append a delta to the structured-output part on `messageId`, or create one\n * if absent. Progressive parse of the accumulated buffer fills `partial`.\n *\n * Callers must only invoke this while the part is still in flight — the\n * helper unconditionally writes `status: 'streaming'`, so feeding it a delta\n * after a `complete`/`error` terminal would regress the part. In practice the\n * processor gates calls via `structuredMessageIds`, which is dropped on\n * terminal events.\n *\n * If the progressive parse returns null/undefined (the buffer is not yet a\n * parseable JSON prefix), the previously-good `partial` is preserved so the\n * UI doesn't flicker back to empty for a single render.\n */\nexport function appendStructuredOutputDelta(\n  messages: Array<UIMessage>,\n  messageId: string,\n  delta: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existingIndex = parts.findIndex(\n      (p): p is StructuredOutputPart => p.type === 'structured-output',\n    )\n    const existing =\n      existingIndex >= 0 ? (parts[existingIndex] as StructuredOutputPart) : null\n\n    const nextRaw = (existing?.raw ?? '') + delta\n    const progressive = parsePartialJSON(nextRaw)\n    const nextPartial =\n      progressive !== undefined && progressive !== null\n        ? progressive\n        : existing?.partial\n\n    const nextPart: StructuredOutputPart = {\n      type: 'structured-output',\n      status: 'streaming',\n      raw: nextRaw,\n      ...(nextPartial !== undefined ? { partial: nextPartial } : {}),\n      ...(existing?.reasoning !== undefined\n        ? { reasoning: existing.reasoning }\n        : {}),\n    }\n\n    if (existingIndex >= 0) {\n      parts[existingIndex] = nextPart\n    } else {\n      parts.push(nextPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Snap the structured-output part on `messageId` to `complete` with the\n * validated `data`. Picks the freshest available `raw` so the wire\n * round-trip stays internally consistent:\n *\n *   1. Caller-supplied `raw` (the original streamed bytes from the model).\n *   2. The existing part's `raw` (deltas accumulated before this terminal).\n *   3. `JSON.stringify(data)` as a defensive fallback for terminal-only\n *      completes that never shipped raw — keeps the part self-consistent\n *      so downstream consumers never see a complete part with empty raw.\n */\nexport function completeStructuredOutputPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  data: unknown,\n  raw: string,\n  reasoning?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existingIndex = parts.findIndex(\n      (p): p is StructuredOutputPart => p.type === 'structured-output',\n    )\n\n    const existingRaw =\n      existingIndex >= 0\n        ? (parts[existingIndex] as StructuredOutputPart).raw\n        : ''\n    let resolvedRaw = raw || existingRaw\n    if (resolvedRaw === '' && data !== undefined) {\n      try {\n        resolvedRaw = JSON.stringify(data)\n      } catch {\n        // Unserializable (circular, BigInt, throwing toJSON). Leave raw\n        // empty. Both downstream paths handle this: `ag-ui-wire.ts`\n        // `collectText` skips complete parts with empty raw entirely, and\n        // `uiMessageToModelMessages` falls back to a defensive\n        // `safeJsonStringify(data)` which itself returns `''` for the same\n        // unserializable inputs — so the turn is silently dropped from the\n        // next request rather than shipping garbage or crashing the stream.\n      }\n    }\n\n    const nextPart: StructuredOutputPart = {\n      type: 'structured-output',\n      status: 'complete',\n      data,\n      partial: data,\n      raw: resolvedRaw,\n      ...(reasoning !== undefined ? { reasoning } : {}),\n    }\n\n    if (existingIndex >= 0) {\n      parts[existingIndex] = nextPart\n    } else {\n      parts.push(nextPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Mark the structured-output part on `messageId` as errored. If no part\n * exists yet — RUN_ERROR fired after `structured-output.start` but before\n * any delta — create an empty errored placeholder so consumers have\n * something renderable. Existing complete parts are left alone (an error\n * after a successful complete should not retroactively un-complete it).\n */\nexport function errorStructuredOutputPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  errorMessage: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const existingIndex = parts.findIndex(\n      (p): p is StructuredOutputPart => p.type === 'structured-output',\n    )\n\n    if (existingIndex < 0) {\n      parts.push({\n        type: 'structured-output',\n        status: 'error',\n        raw: '',\n        errorMessage,\n      })\n      return { ...msg, parts }\n    }\n\n    const existing = parts[existingIndex] as StructuredOutputPart\n    if (existing.status === 'complete') {\n      return msg\n    }\n    parts[existingIndex] = {\n      ...existing,\n      status: 'error',\n      errorMessage,\n    }\n    return { ...msg, parts }\n  })\n}\n\n/**\n * Update or add a thinking part to a message, keyed by stepId.\n * Each distinct stepId produces its own ThinkingPart.\n */\nexport function updateThinkingPart(\n  messages: Array<UIMessage>,\n  messageId: string,\n  stepId: string,\n  content: string,\n  signature?: string,\n): Array<UIMessage> {\n  return messages.map((msg) => {\n    if (msg.id !== messageId) {\n      return msg\n    }\n\n    const parts = [...msg.parts]\n    const thinkingPartIndex = parts.findIndex(\n      (p) => p.type === 'thinking' && p.stepId === stepId,\n    )\n\n    const thinkingPart: ThinkingPart = {\n      type: 'thinking',\n      content,\n      stepId,\n      ...(signature && { signature }),\n    }\n\n    if (thinkingPartIndex >= 0) {\n      // Update existing thinking part for this step\n      parts[thinkingPartIndex] = thinkingPart\n    } else {\n      // Add new thinking part at the end (preserve natural streaming order)\n      parts.push(thinkingPart)\n    }\n\n    return { ...msg, parts }\n  })\n}\n"],"names":[],"mappings":";AAwBO,SAAS,eACd,UACA,WACA,SACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,WAAW,MAAM,SAAS,IAAI,MAAM,MAAM,SAAS,CAAC,IAAI;AAE9D,QAAI,YAAY,SAAS,SAAS,QAAQ;AAExC,YAAM,MAAM,SAAS,CAAC,IAAI,EAAE,MAAM,QAAQ,QAAA;AAAA,IAC5C,OAAO;AAEL,YAAM,KAAK,EAAE,MAAM,QAAQ,SAAS;AAAA,IACtC;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,mBACd,UACA,WACA,UAOkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,WAAW,MAAM;AAAA,MACrB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO,SAAS;AAAA,IAAA;AAOxE,UAAM,WAAW,SAAS,YAAY,UAAU;AAEhD,UAAM,eAA6B;AAAA,MACjC,MAAM;AAAA,MACN,IAAI,SAAS;AAAA,MACb,MAAM,SAAS;AAAA,MACf,WAAW,SAAS;AAAA,MACpB,OAAO,SAAS;AAAA;AAAA,MAEhB,GAAI,UAAU,YAAY,EAAE,UAAU,EAAE,GAAG,SAAS,WAAS;AAAA,MAC7D,GAAI,UAAU,WAAW,UAAa,EAAE,QAAQ,SAAS,OAAA;AAAA,MACzD,GAAI,aAAa,UAAa,EAAE,SAAA;AAAA,IAAS;AAG3C,QAAI,UAAU;AAEZ,YAAM,MAAM,QAAQ,QAAQ,CAAC,IAAI;AAAA,IACnC,OAAO;AAEL,YAAM,KAAK,YAAY;AAAA,IACzB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,qBACd,UACA,WACA,YACA,SACA,OACA,OACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,kBAAkB,MAAM;AAAA,MAC5B,CAAC,MACC,EAAE,SAAS,iBAAiB,EAAE,eAAe;AAAA,IAAA;AAGjD,UAAM,iBAAiC;AAAA,MACrC,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA;AAAA,MACA,GAAI,SAAS,EAAE,MAAA;AAAA,IAAM;AAGvB,QAAI,mBAAmB,GAAG;AACxB,YAAM,eAAe,IAAI;AAAA,IAC3B,OAAO;AACL,YAAM,KAAK,cAAc;AAAA,IAC3B;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAKO,SAAS,uBACd,UACA,WACA,YACA,YACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,IAAA;AAG/D,QAAI,cAAc;AAChB,YAAM,QAAQ,MAAM,QAAQ,YAAY;AACxC,YAAM,KAAK,IAAI;AAAA,QACb,GAAG;AAAA,QACH,OAAO;AAAA,QACP,UAAU;AAAA,UACR,IAAI;AAAA,UACJ,eAAe;AAAA,QAAA;AAAA,MACjB;AAAA,IAEJ;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAkCO,SAAS,yBACd,UACA,YACA,QACA,OACA,WACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO;AAAA,IAAA;AAG/D,QAAI,cAAc;AAChB,YAAM,QAAQ,MAAM,QAAQ,YAAY;AACxC,YAAM,KAAK,IAAI;AAAA,QACb,GAAG;AAAA,QACH,QAAQ,YAAY,EAAE,OAAO,cAAc;AAAA,QAC3C,OAAO,UAAU,YAAY,UAAU;AAAA,MAAA;AAAA,IAE3C;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAMO,SAAS,+BACd,UACA,YACA,UACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,eAAe,MAAM;AAAA,MACzB,CAAC,MACC,EAAE,SAAS,eAAe,EAAE,UAAU,OAAO;AAAA,IAAA;AAGjD,QAAI,gBAAgB,aAAa,UAAU;AACzC,YAAM,QAAQ,MAAM,QAAQ,YAAY;AACxC,YAAM,KAAK,IAAI;AAAA,QACb,GAAG;AAAA,QACH,UAAU,EAAE,GAAG,aAAa,UAAU,SAAA;AAAA,QACtC,OAAO;AAAA,MAAA;AAAA,IAEX;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAgBO,SAAS,4BACd,UACA,WACA,OACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,gBAAgB,MAAM;AAAA,MAC1B,CAAC,MAAiC,EAAE,SAAS;AAAA,IAAA;AAE/C,UAAM,WACJ,iBAAiB,IAAK,MAAM,aAAa,IAA6B;AAExE,UAAM,WAAW,UAAU,OAAO,MAAM;AACxC,UAAM,cAAc,iBAAiB,OAAO;AAC5C,UAAM,cACJ,gBAAgB,UAAa,gBAAgB,OACzC,cACA,UAAU;AAEhB,UAAM,WAAiC;AAAA,MACrC,MAAM;AAAA,MACN,QAAQ;AAAA,MACR,KAAK;AAAA,MACL,GAAI,gBAAgB,SAAY,EAAE,SAAS,YAAA,IAAgB,CAAA;AAAA,MAC3D,GAAI,UAAU,cAAc,SACxB,EAAE,WAAW,SAAS,cACtB,CAAA;AAAA,IAAC;AAGP,QAAI,iBAAiB,GAAG;AACtB,YAAM,aAAa,IAAI;AAAA,IACzB,OAAO;AACL,YAAM,KAAK,QAAQ;AAAA,IACrB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAaO,SAAS,6BACd,UACA,WACA,MACA,KACA,WACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,gBAAgB,MAAM;AAAA,MAC1B,CAAC,MAAiC,EAAE,SAAS;AAAA,IAAA;AAG/C,UAAM,cACJ,iBAAiB,IACZ,MAAM,aAAa,EAA2B,MAC/C;AACN,QAAI,cAAc,OAAO;AACzB,QAAI,gBAAgB,MAAM,SAAS,QAAW;AAC5C,UAAI;AACF,sBAAc,KAAK,UAAU,IAAI;AAAA,MACnC,QAAQ;AAAA,MAQR;AAAA,IACF;AAEA,UAAM,WAAiC;AAAA,MACrC,MAAM;AAAA,MACN,QAAQ;AAAA,MACR;AAAA,MACA,SAAS;AAAA,MACT,KAAK;AAAA,MACL,GAAI,cAAc,SAAY,EAAE,cAAc,CAAA;AAAA,IAAC;AAGjD,QAAI,iBAAiB,GAAG;AACtB,YAAM,aAAa,IAAI;AAAA,IACzB,OAAO;AACL,YAAM,KAAK,QAAQ;AAAA,IACrB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AASO,SAAS,0BACd,UACA,WACA,cACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,gBAAgB,MAAM;AAAA,MAC1B,CAAC,MAAiC,EAAE,SAAS;AAAA,IAAA;AAG/C,QAAI,gBAAgB,GAAG;AACrB,YAAM,KAAK;AAAA,QACT,MAAM;AAAA,QACN,QAAQ;AAAA,QACR,KAAK;AAAA,QACL;AAAA,MAAA,CACD;AACD,aAAO,EAAE,GAAG,KAAK,MAAA;AAAA,IACnB;AAEA,UAAM,WAAW,MAAM,aAAa;AACpC,QAAI,SAAS,WAAW,YAAY;AAClC,aAAO;AAAA,IACT;AACA,UAAM,aAAa,IAAI;AAAA,MACrB,GAAG;AAAA,MACH,QAAQ;AAAA,MACR;AAAA,IAAA;AAEF,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;AAMO,SAAS,mBACd,UACA,WACA,QACA,SACA,WACkB;AAClB,SAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,QAAI,IAAI,OAAO,WAAW;AACxB,aAAO;AAAA,IACT;AAEA,UAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;AAC3B,UAAM,oBAAoB,MAAM;AAAA,MAC9B,CAAC,MAAM,EAAE,SAAS,cAAc,EAAE,WAAW;AAAA,IAAA;AAG/C,UAAM,eAA6B;AAAA,MACjC,MAAM;AAAA,MACN;AAAA,MACA;AAAA,MACA,GAAI,aAAa,EAAE,UAAA;AAAA,IAAU;AAG/B,QAAI,qBAAqB,GAAG;AAE1B,YAAM,iBAAiB,IAAI;AAAA,IAC7B,OAAO;AAEL,YAAM,KAAK,YAAY;AAAA,IACzB;AAEA,WAAO,EAAE,GAAG,KAAK,MAAA;AAAA,EACnB,CAAC;AACH;"}

package/dist/esm/activities/chat/stream/processor.d.ts

@@ -362,6 +362,12 @@ export declare class StreamProcessor {
      * Mark a tool call as complete and emit event
      */
     private completeToolCall;
+    /**
+     * Whether the rendered tool-call part for the given id has reached the
+     * terminal 'error' state. Used to prevent the completion safety net from
+     * downgrading a failed call back to 'input-complete'.
+     */
+    private isToolCallPartErrored;
     /**
      * Emit pending text update for a specific message.
      *

package/dist/esm/activities/chat/stream/processor.js

@@ -137,7 +137,7 @@ class StreamProcessor {
       this.messages,
       toolCallId,
       output,
-      error ? "input-complete" : void 0,
+      error ? "error" : void 0,
       error
     );
     const content = normalizeToolResult(output);
@@ -767,7 +767,7 @@ class StreamProcessor {
         this.messages,
         chunk.toolCallId,
         output,
-        chunk.state === "output-error" ? "input-complete" : void 0
+        chunk.state === "output-error" ? "error" : void 0
       );
       const resultState = chunk.state === "output-error" ? "error" : "complete";
       this.messages = updateToolResultPart(
@@ -807,7 +807,7 @@ class StreamProcessor {
       this.messages,
       chunk.toolCallId,
       output,
-      chunk.state === "output-error" ? "input-complete" : void 0
+      chunk.state === "output-error" ? "error" : void 0
     );
     const resultState = chunk.state === "output-error" ? "error" : "complete";
     this.messages = updateToolResultPart(
@@ -1137,6 +1137,9 @@ class StreamProcessor {
   completeToolCall(messageId, _index, toolCall) {
     toolCall.state = "input-complete";
     toolCall.parsedArguments = this.jsonParser.parse(toolCall.arguments);
+    if (this.isToolCallPartErrored(toolCall.id)) {
+      return;
+    }
     this.messages = updateToolCallPart(this.messages, messageId, {
       id: toolCall.id,
       name: toolCall.name,
@@ -1152,6 +1155,18 @@ class StreamProcessor {
       toolCall.arguments
     );
   }
+  /**
+   * Whether the rendered tool-call part for the given id has reached the
+   * terminal 'error' state. Used to prevent the completion safety net from
+   * downgrading a failed call back to 'input-complete'.
+   */
+  isToolCallPartErrored(toolCallId) {
+    return this.messages.some(
+      (msg) => msg.parts.some(
+        (part) => part.type === "tool-call" && part.id === toolCallId && part.state === "error"
+      )
+    );
+  }
   /**
    * Emit pending text update for a specific message.
    *