@tanstack/ai 0.41.0 → 0.43.0

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

@@ -1,255 +1,296 @@
 import { parsePartialJSON } from "./json-parser.js";
+//#region src/activities/chat/stream/message-updaters.ts
+/**
+* Message Updaters (Internal)
+*
+* Internal helper functions for updating UIMessage parts.
+* These are used by StreamProcessor to manage the message array.
+*/
+/**
+* Update or add a text part to a message.
+*
+* If the last part is a text part, update it (continuing the same text segment).
+* Otherwise, create a new text part (starting a new text segment after tool calls).
+*/
 function updateTextPart(messages, messageId, content) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const lastPart = parts.length > 0 ? parts[parts.length - 1] : null;
-    if (lastPart && lastPart.type === "text") {
-      parts[parts.length - 1] = { type: "text", content };
-    } else {
-      parts.push({ type: "text", content });
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const lastPart = parts.length > 0 ? parts[parts.length - 1] : null;
+		if (lastPart && lastPart.type === "text") parts[parts.length - 1] = {
+			type: "text",
+			content
+		};
+		else parts.push({
+			type: "text",
+			content
+		});
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Update or add a tool call part to a message.
+*/
 function updateToolCallPart(messages, messageId, toolCall) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const existing = parts.find(
-      (p) => p.type === "tool-call" && p.id === toolCall.id
-    );
-    const metadata = toolCall.metadata ?? existing?.metadata;
-    const input = toolCall.input ?? existing?.input;
-    const toolCallPart = {
-      type: "tool-call",
-      id: toolCall.id,
-      name: toolCall.name,
-      arguments: toolCall.arguments,
-      state: toolCall.state,
-      // Carry forward approval, output and parsed input from the existing part
-      ...existing?.approval && { approval: { ...existing.approval } },
-      ...existing?.output !== void 0 && { output: existing.output },
-      ...input !== void 0 && { input },
-      ...metadata !== void 0 && { metadata }
-    };
-    if (existing) {
-      parts[parts.indexOf(existing)] = toolCallPart;
-    } else {
-      parts.push(toolCallPart);
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const existing = parts.find((p) => p.type === "tool-call" && p.id === toolCall.id);
+		const metadata = toolCall.metadata ?? existing?.metadata;
+		const input = toolCall.input ?? existing?.input;
+		const toolCallPart = {
+			type: "tool-call",
+			id: toolCall.id,
+			name: toolCall.name,
+			arguments: toolCall.arguments,
+			state: toolCall.state,
+			...existing?.approval && { approval: { ...existing.approval } },
+			...existing?.output !== void 0 && { output: existing.output },
+			...input !== void 0 && { input },
+			...metadata !== void 0 && { metadata }
+		};
+		if (existing) parts[parts.indexOf(existing)] = toolCallPart;
+		else parts.push(toolCallPart);
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Update or add a tool result part to a message.
+*/
 function updateToolResultPart(messages, messageId, toolCallId, content, state, error) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const resultPartIndex = parts.findIndex(
-      (p) => p.type === "tool-result" && p.toolCallId === toolCallId
-    );
-    const toolResultPart = {
-      type: "tool-result",
-      toolCallId,
-      content,
-      state,
-      ...error && { error }
-    };
-    if (resultPartIndex >= 0) {
-      parts[resultPartIndex] = toolResultPart;
-    } else {
-      parts.push(toolResultPart);
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const resultPartIndex = parts.findIndex((p) => p.type === "tool-result" && p.toolCallId === toolCallId);
+		const toolResultPart = {
+			type: "tool-result",
+			toolCallId,
+			content,
+			state,
+			...error && { error }
+		};
+		if (resultPartIndex >= 0) parts[resultPartIndex] = toolResultPart;
+		else parts.push(toolResultPart);
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Update a tool call part with approval request metadata.
+*/
 function updateToolCallApproval(messages, messageId, toolCallId, approvalId) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const toolCallPart = parts.find(
-      (p) => p.type === "tool-call" && p.id === toolCallId
-    );
-    if (toolCallPart) {
-      const index = parts.indexOf(toolCallPart);
-      parts[index] = {
-        ...toolCallPart,
-        state: "approval-requested",
-        approval: {
-          id: approvalId,
-          needsApproval: true
-        }
-      };
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const toolCallPart = parts.find((p) => p.type === "tool-call" && p.id === toolCallId);
+		if (toolCallPart) {
+			const index = parts.indexOf(toolCallPart);
+			parts[index] = {
+				...toolCallPart,
+				state: "approval-requested",
+				approval: {
+					id: approvalId,
+					needsApproval: true
+				}
+			};
+		}
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Update a tool call part with output.
+* Searches all messages to find the tool call by ID.
+*/
 function updateToolCallWithOutput(messages, toolCallId, output, state, errorText) {
-  return messages.map((msg) => {
-    const parts = [...msg.parts];
-    const toolCallPart = parts.find(
-      (p) => p.type === "tool-call" && p.id === toolCallId
-    );
-    if (toolCallPart) {
-      const index = parts.indexOf(toolCallPart);
-      parts[index] = {
-        ...toolCallPart,
-        output: errorText ? { error: errorText } : output,
-        state: state ?? (errorText ? "error" : "complete")
-      };
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		const parts = [...msg.parts];
+		const toolCallPart = parts.find((p) => p.type === "tool-call" && p.id === toolCallId);
+		if (toolCallPart) {
+			const index = parts.indexOf(toolCallPart);
+			parts[index] = {
+				...toolCallPart,
+				output: errorText ? { error: errorText } : output,
+				state: state ?? (errorText ? "error" : "complete")
+			};
+		}
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Update a tool call part with approval response.
+* Searches all messages to find the tool call by approval ID.
+*/
 function updateToolCallApprovalResponse(messages, approvalId, approved) {
-  return messages.map((msg) => {
-    const parts = [...msg.parts];
-    const toolCallPart = parts.find(
-      (p) => p.type === "tool-call" && p.approval?.id === approvalId
-    );
-    if (toolCallPart && toolCallPart.approval) {
-      const index = parts.indexOf(toolCallPart);
-      parts[index] = {
-        ...toolCallPart,
-        approval: { ...toolCallPart.approval, approved },
-        state: "approval-responded"
-      };
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		const parts = [...msg.parts];
+		const toolCallPart = parts.find((p) => p.type === "tool-call" && p.approval?.id === approvalId);
+		if (toolCallPart && toolCallPart.approval) {
+			const index = parts.indexOf(toolCallPart);
+			parts[index] = {
+				...toolCallPart,
+				approval: {
+					...toolCallPart.approval,
+					approved
+				},
+				state: "approval-responded"
+			};
+		}
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Append a delta to the structured-output part on `messageId`, or create one
+* if absent. Progressive parse of the accumulated buffer fills `partial`.
+*
+* Callers must only invoke this while the part is still in flight — the
+* helper unconditionally writes `status: 'streaming'`, so feeding it a delta
+* after a `complete`/`error` terminal would regress the part. In practice the
+* processor gates calls via `structuredMessageIds`, which is dropped on
+* terminal events.
+*
+* If the progressive parse returns null/undefined (the buffer is not yet a
+* parseable JSON prefix), the previously-good `partial` is preserved so the
+* UI doesn't flicker back to empty for a single render.
+*/
 function appendStructuredOutputDelta(messages, messageId, delta) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const existingIndex = parts.findIndex(
-      (p) => p.type === "structured-output"
-    );
-    const existing = existingIndex >= 0 ? parts[existingIndex] : null;
-    const nextRaw = (existing?.raw ?? "") + delta;
-    const progressive = parsePartialJSON(nextRaw);
-    const nextPartial = progressive !== void 0 && progressive !== null ? progressive : existing?.partial;
-    const nextPart = {
-      type: "structured-output",
-      status: "streaming",
-      raw: nextRaw,
-      ...nextPartial !== void 0 ? { partial: nextPartial } : {},
-      ...existing?.reasoning !== void 0 ? { reasoning: existing.reasoning } : {}
-    };
-    if (existingIndex >= 0) {
-      parts[existingIndex] = nextPart;
-    } else {
-      parts.push(nextPart);
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const existingIndex = parts.findIndex((p) => p.type === "structured-output");
+		const existing = existingIndex >= 0 ? parts[existingIndex] : null;
+		const nextRaw = (existing?.raw ?? "") + delta;
+		const progressive = parsePartialJSON(nextRaw);
+		const nextPartial = progressive !== void 0 && progressive !== null ? progressive : existing?.partial;
+		const nextPart = {
+			type: "structured-output",
+			status: "streaming",
+			raw: nextRaw,
+			...nextPartial !== void 0 ? { partial: nextPartial } : {},
+			...existing?.reasoning !== void 0 ? { reasoning: existing.reasoning } : {}
+		};
+		if (existingIndex >= 0) parts[existingIndex] = nextPart;
+		else parts.push(nextPart);
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Snap the structured-output part on `messageId` to `complete` with the
+* validated `data`. Picks the freshest available `raw` so the wire
+* round-trip stays internally consistent:
+*
+*   1. Caller-supplied `raw` (the original streamed bytes from the model).
+*   2. The existing part's `raw` (deltas accumulated before this terminal).
+*   3. `JSON.stringify(data)` as a defensive fallback for terminal-only
+*      completes that never shipped raw — keeps the part self-consistent
+*      so downstream consumers never see a complete part with empty raw.
+*/
 function completeStructuredOutputPart(messages, messageId, data, raw, reasoning) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const existingIndex = parts.findIndex(
-      (p) => p.type === "structured-output"
-    );
-    const existingRaw = existingIndex >= 0 ? parts[existingIndex].raw : "";
-    let resolvedRaw = raw || existingRaw;
-    if (resolvedRaw === "" && data !== void 0) {
-      try {
-        resolvedRaw = JSON.stringify(data);
-      } catch {
-      }
-    }
-    const nextPart = {
-      type: "structured-output",
-      status: "complete",
-      data,
-      partial: data,
-      raw: resolvedRaw,
-      ...reasoning !== void 0 ? { reasoning } : {}
-    };
-    if (existingIndex >= 0) {
-      parts[existingIndex] = nextPart;
-    } else {
-      parts.push(nextPart);
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const existingIndex = parts.findIndex((p) => p.type === "structured-output");
+		const existingRaw = existingIndex >= 0 ? parts[existingIndex].raw : "";
+		let resolvedRaw = raw || existingRaw;
+		if (resolvedRaw === "" && data !== void 0) try {
+			resolvedRaw = JSON.stringify(data);
+		} catch {}
+		const nextPart = {
+			type: "structured-output",
+			status: "complete",
+			data,
+			partial: data,
+			raw: resolvedRaw,
+			...reasoning !== void 0 ? { reasoning } : {}
+		};
+		if (existingIndex >= 0) parts[existingIndex] = nextPart;
+		else parts.push(nextPart);
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Mark the structured-output part on `messageId` as errored. If no part
+* exists yet — RUN_ERROR fired after `structured-output.start` but before
+* any delta — create an empty errored placeholder so consumers have
+* something renderable. Existing complete parts are left alone (an error
+* after a successful complete should not retroactively un-complete it).
+*/
 function errorStructuredOutputPart(messages, messageId, errorMessage) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const existingIndex = parts.findIndex(
-      (p) => p.type === "structured-output"
-    );
-    if (existingIndex < 0) {
-      parts.push({
-        type: "structured-output",
-        status: "error",
-        raw: "",
-        errorMessage
-      });
-      return { ...msg, parts };
-    }
-    const existing = parts[existingIndex];
-    if (existing.status === "complete") {
-      return msg;
-    }
-    parts[existingIndex] = {
-      ...existing,
-      status: "error",
-      errorMessage
-    };
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const existingIndex = parts.findIndex((p) => p.type === "structured-output");
+		if (existingIndex < 0) {
+			parts.push({
+				type: "structured-output",
+				status: "error",
+				raw: "",
+				errorMessage
+			});
+			return {
+				...msg,
+				parts
+			};
+		}
+		const existing = parts[existingIndex];
+		if (existing.status === "complete") return msg;
+		parts[existingIndex] = {
+			...existing,
+			status: "error",
+			errorMessage
+		};
+		return {
+			...msg,
+			parts
+		};
+	});
 }
+/**
+* Update or add a thinking part to a message, keyed by stepId.
+* Each distinct stepId produces its own ThinkingPart.
+*/
 function updateThinkingPart(messages, messageId, stepId, content, signature) {
-  return messages.map((msg) => {
-    if (msg.id !== messageId) {
-      return msg;
-    }
-    const parts = [...msg.parts];
-    const thinkingPartIndex = parts.findIndex(
-      (p) => p.type === "thinking" && p.stepId === stepId
-    );
-    const thinkingPart = {
-      type: "thinking",
-      content,
-      stepId,
-      ...signature && { signature }
-    };
-    if (thinkingPartIndex >= 0) {
-      parts[thinkingPartIndex] = thinkingPart;
-    } else {
-      parts.push(thinkingPart);
-    }
-    return { ...msg, parts };
-  });
+	return messages.map((msg) => {
+		if (msg.id !== messageId) return msg;
+		const parts = [...msg.parts];
+		const thinkingPartIndex = parts.findIndex((p) => p.type === "thinking" && p.stepId === stepId);
+		const thinkingPart = {
+			type: "thinking",
+			content,
+			stepId,
+			...signature && { signature }
+		};
+		if (thinkingPartIndex >= 0) parts[thinkingPartIndex] = thinkingPart;
+		else parts.push(thinkingPart);
+		return {
+			...msg,
+			parts
+		};
+	});
 }
-export {
-  appendStructuredOutputDelta,
-  completeStructuredOutputPart,
-  errorStructuredOutputPart,
-  updateTextPart,
-  updateThinkingPart,
-  updateToolCallApproval,
-  updateToolCallApprovalResponse,
-  updateToolCallPart,
-  updateToolCallWithOutput,
-  updateToolResultPart
-};
-//# sourceMappingURL=message-updaters.js.map
+//#endregion
+export { appendStructuredOutputDelta, completeStructuredOutputPart, errorStructuredOutputPart, updateTextPart, updateThinkingPart, updateToolCallApproval, updateToolCallApprovalResponse, updateToolCallPart, updateToolCallWithOutput, updateToolResultPart };
+
+//# sourceMappingURL=message-updaters.js.map

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    /** Parsed input — set when the arguments are complete. */\n    input?: unknown\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    // Same for the parsed input: it's supplied once at completion, so\n    // subsequent arg-less updates (approval, etc.) must not drop it.\n    const input = toolCall.input ?? existing?.input\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, output and parsed input from the existing part\n      ...(existing?.approval && { approval: { ...existing.approval } }),\n      ...(existing?.output !== undefined && { output: existing.output }),\n      ...(input !== undefined && { input }),\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,UASkB;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;AAGhD,UAAM,QAAQ,SAAS,SAAS,UAAU;AAE1C,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,UAAU,UAAa,EAAE,MAAA;AAAA,MAC7B,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;"}
+{"version":3,"file":"message-updaters.js","names":[],"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    /** Parsed input — set when the arguments are complete. */\n    input?: unknown\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    // Same for the parsed input: it's supplied once at completion, so\n    // subsequent arg-less updates (approval, etc.) must not drop it.\n    const input = toolCall.input ?? existing?.input\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, output and parsed input from the existing part\n      ...(existing?.approval && { approval: { ...existing.approval } }),\n      ...(existing?.output !== undefined && { output: existing.output }),\n      ...(input !== undefined && { input }),\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"],"mappings":";;;;;;;;;;;;;;AAwBA,SAAgB,eACd,UACA,WACA,SACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,WAAW,MAAM,SAAS,IAAI,MAAM,MAAM,SAAS,KAAK;EAE9D,IAAI,YAAY,SAAS,SAAS,QAEhC,MAAM,MAAM,SAAS,KAAK;GAAE,MAAM;GAAQ;EAAQ;OAGlD,MAAM,KAAK;GAAE,MAAM;GAAQ;EAAQ,CAAC;EAGtC,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;AAKA,SAAgB,mBACd,UACA,WACA,UASkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,WAAW,MAAM,MACpB,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO,SAAS,EACxE;EAMA,MAAM,WAAW,SAAS,YAAY,UAAU;EAGhD,MAAM,QAAQ,SAAS,SAAS,UAAU;EAE1C,MAAM,eAA6B;GACjC,MAAM;GACN,IAAI,SAAS;GACb,MAAM,SAAS;GACf,WAAW,SAAS;GACpB,OAAO,SAAS;GAEhB,GAAI,UAAU,YAAY,EAAE,UAAU,EAAE,GAAG,SAAS,SAAS,EAAE;GAC/D,GAAI,UAAU,WAAW,KAAA,KAAa,EAAE,QAAQ,SAAS,OAAO;GAChE,GAAI,UAAU,KAAA,KAAa,EAAE,MAAM;GACnC,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;EAC3C;EAEA,IAAI,UAEF,MAAM,MAAM,QAAQ,QAAQ,KAAK;OAGjC,MAAM,KAAK,YAAY;EAGzB,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;AAKA,SAAgB,qBACd,UACA,WACA,YACA,SACA,OACA,OACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,kBAAkB,MAAM,WAC3B,MACC,EAAE,SAAS,iBAAiB,EAAE,eAAe,UACjD;EAEA,MAAM,iBAAiC;GACrC,MAAM;GACN;GACA;GACA;GACA,GAAI,SAAS,EAAE,MAAM;EACvB;EAEA,IAAI,mBAAmB,GACrB,MAAM,mBAAmB;OAEzB,MAAM,KAAK,cAAc;EAG3B,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;AAKA,SAAgB,uBACd,UACA,WACA,YACA,YACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,eAAe,MAAM,MACxB,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO,UAC/D;EAEA,IAAI,cAAc;GAChB,MAAM,QAAQ,MAAM,QAAQ,YAAY;GACxC,MAAM,SAAS;IACb,GAAG;IACH,OAAO;IACP,UAAU;KACR,IAAI;KACJ,eAAe;IACjB;GACF;EACF;EAEA,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;;AAkCA,SAAgB,yBACd,UACA,YACA,QACA,OACA,WACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,eAAe,MAAM,MACxB,MAAyB,EAAE,SAAS,eAAe,EAAE,OAAO,UAC/D;EAEA,IAAI,cAAc;GAChB,MAAM,QAAQ,MAAM,QAAQ,YAAY;GACxC,MAAM,SAAS;IACb,GAAG;IACH,QAAQ,YAAY,EAAE,OAAO,UAAU,IAAI;IAC3C,OAAO,UAAU,YAAY,UAAU;GACzC;EACF;EAEA,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;;AAMA,SAAgB,+BACd,UACA,YACA,UACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,eAAe,MAAM,MACxB,MACC,EAAE,SAAS,eAAe,EAAE,UAAU,OAAO,UACjD;EAEA,IAAI,gBAAgB,aAAa,UAAU;GACzC,MAAM,QAAQ,MAAM,QAAQ,YAAY;GACxC,MAAM,SAAS;IACb,GAAG;IACH,UAAU;KAAE,GAAG,aAAa;KAAU;IAAS;IAC/C,OAAO;GACT;EACF;EAEA,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;;;;;;;;;;;;AAgBA,SAAgB,4BACd,UACA,WACA,OACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,gBAAgB,MAAM,WACzB,MAAiC,EAAE,SAAS,mBAC/C;EACA,MAAM,WACJ,iBAAiB,IAAK,MAAM,iBAA0C;EAExE,MAAM,WAAW,UAAU,OAAO,MAAM;EACxC,MAAM,cAAc,iBAAiB,OAAO;EAC5C,MAAM,cACJ,gBAAgB,KAAA,KAAa,gBAAgB,OACzC,cACA,UAAU;EAEhB,MAAM,WAAiC;GACrC,MAAM;GACN,QAAQ;GACR,KAAK;GACL,GAAI,gBAAgB,KAAA,IAAY,EAAE,SAAS,YAAY,IAAI,CAAC;GAC5D,GAAI,UAAU,cAAc,KAAA,IACxB,EAAE,WAAW,SAAS,UAAU,IAChC,CAAC;EACP;EAEA,IAAI,iBAAiB,GACnB,MAAM,iBAAiB;OAEvB,MAAM,KAAK,QAAQ;EAGrB,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;;;;;;;;;AAaA,SAAgB,6BACd,UACA,WACA,MACA,KACA,WACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,gBAAgB,MAAM,WACzB,MAAiC,EAAE,SAAS,mBAC/C;EAEA,MAAM,cACJ,iBAAiB,IACZ,MAAM,cAAc,CAA0B,MAC/C;EACN,IAAI,cAAc,OAAO;EACzB,IAAI,gBAAgB,MAAM,SAAS,KAAA,GACjC,IAAI;GACF,cAAc,KAAK,UAAU,IAAI;EACnC,QAAQ,CAQR;EAGF,MAAM,WAAiC;GACrC,MAAM;GACN,QAAQ;GACR;GACA,SAAS;GACT,KAAK;GACL,GAAI,cAAc,KAAA,IAAY,EAAE,UAAU,IAAI,CAAC;EACjD;EAEA,IAAI,iBAAiB,GACnB,MAAM,iBAAiB;OAEvB,MAAM,KAAK,QAAQ;EAGrB,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;;;;;AASA,SAAgB,0BACd,UACA,WACA,cACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,gBAAgB,MAAM,WACzB,MAAiC,EAAE,SAAS,mBAC/C;EAEA,IAAI,gBAAgB,GAAG;GACrB,MAAM,KAAK;IACT,MAAM;IACN,QAAQ;IACR,KAAK;IACL;GACF,CAAC;GACD,OAAO;IAAE,GAAG;IAAK;GAAM;EACzB;EAEA,MAAM,WAAW,MAAM;EACvB,IAAI,SAAS,WAAW,YACtB,OAAO;EAET,MAAM,iBAAiB;GACrB,GAAG;GACH,QAAQ;GACR;EACF;EACA,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH;;;;;AAMA,SAAgB,mBACd,UACA,WACA,QACA,SACA,WACkB;CAClB,OAAO,SAAS,KAAK,QAAQ;EAC3B,IAAI,IAAI,OAAO,WACb,OAAO;EAGT,MAAM,QAAQ,CAAC,GAAG,IAAI,KAAK;EAC3B,MAAM,oBAAoB,MAAM,WAC7B,MAAM,EAAE,SAAS,cAAc,EAAE,WAAW,MAC/C;EAEA,MAAM,eAA6B;GACjC,MAAM;GACN;GACA;GACA,GAAI,aAAa,EAAE,UAAU;EAC/B;EAEA,IAAI,qBAAqB,GAEvB,MAAM,qBAAqB;OAG3B,MAAM,KAAK,YAAY;EAGzB,OAAO;GAAE,GAAG;GAAK;EAAM;CACzB,CAAC;AACH"}