@tanstack/ai 0.49.1 → 0.50.0

package/dist/esm/index.js

@@ -15,15 +15,15 @@ import { hashSchemaInput, normalizeApprovalSchema } from "./activities/chat/tool
 import { INTERRUPT_BINDING_METADATA_KEY, InterruptResumeValidationError, interruptItemError, readInterruptBinding, readUnopenedInterruptBinding, validateInterruptResumeBatch, withInterruptBinding, withoutInterruptBinding } from "./interrupt-resume.js";
 import { INTERRUPT_PAYLOAD_METADATA_KEY, createInterruptBinding, defineInterrupt } from "./interrupt-definition.js";
 import { INTERRUPT_CONTINUATION_METADATA_KEY, INTERRUPT_CONTINUATION_VERSION, genericInterruptContinuationFromDescriptor, readGenericInterruptContinuation, wrapGenericInterruptContinuation } from "./generic-interrupt-continuation.js";
-import { uiMessagesToWire } from "./utilities/ag-ui-wire.js";
 import { isContentPart, isContentPartArray, normalizeToolResult } from "./utilities/tool-result.js";
 import { getProviderExecutedMetadata, isProviderExecutedToolCall } from "./utilities/provider-executed.js";
+import { convertMessagesToModelMessages, generateMessageId, modelMessageToUIMessage, modelMessagesToUIMessages, normalizeToUIMessage, uiMessageToModelMessages } from "./activities/chat/messages.js";
+import { uiMessagesToWire } from "./utilities/ag-ui-wire.js";
 import { firstSentence, renderLazyCatalogEntry } from "./activities/chat/tools/lazy-tools.js";
 import { DISCOVERY_TOOL_NAME } from "./activities/chat/tools/lazy-tool-manager.js";
 import { DuplicateToolNameError } from "./activities/chat/tools/unique-tool-names.js";
 import { ToolCallManager } from "./activities/chat/tools/tool-calls.js";
 import { combineStrategies, maxIterations, untilFinishReason } from "./activities/chat/agent-loop-strategies.js";
-import { convertMessagesToModelMessages, generateMessageId, modelMessageToUIMessage, modelMessagesToUIMessages, normalizeToUIMessage, uiMessageToModelMessages } from "./activities/chat/messages.js";
 import { MCPDuplicateToolNameError } from "./activities/chat/mcp/manager.js";
 import { chat, createChatOptions } from "./activities/chat/index.js";
 import { createSummarizeOptions, summarize } from "./activities/summarize/index.js";

package/dist/esm/types.d.ts

@@ -250,6 +250,10 @@ export interface ModelMessage<TContent extends string | null | Array<ContentPart
         content: string;
         signature?: string;
     }>;
+    /** Error reported by an AG-UI tool message. */
+    error?: string;
+    /** Optional AG-UI message metadata. TanStack-owned fields live under `tanstack`. */
+    metadata?: Record<string, any>;
     /**
      * Completed structured output represented by this assistant message.
      * `content` remains the provider-facing JSON text; this field preserves the
@@ -310,10 +314,14 @@ export interface ToolCallPart<TMetadata = unknown> {
 }
 export interface ToolResultPart {
     type: 'tool-result';
+    id?: string;
+    name?: string;
     toolCallId: string;
     content: string | Array<ContentPart>;
     state: ToolResultState;
     error?: string;
+    metadata?: Record<string, unknown>;
+    createdAt?: Date;
 }
 export interface ThinkingPart {
     type: 'thinking';
@@ -385,9 +393,17 @@ export interface TanStackMessageMetadata {
     signature?: string;
     /** Per-tool-call provider metadata keyed by tool call id (e.g. Gemini thoughtSignature). */
     toolCallMetadata?: Record<string, unknown>;
+    toolResult?: {
+        id?: string;
+        createdAt?: string;
+        content?: Array<ContentPart>;
+    };
     structuredOutput?: {
         status?: 'streaming' | 'complete' | 'error';
+        partial?: unknown;
+        data?: unknown;
         raw?: string;
+        reasoning?: string;
         errorMessage?: string;
     };
     uiResources?: Array<UIResourcePart>;
@@ -421,6 +437,8 @@ export interface UIMessage<TData = unknown> {
     role: 'system' | 'user' | 'assistant';
     parts: Array<MessagePart<TData>>;
     createdAt?: Date;
+    /** Optional AG-UI sender name. Converters preserve it across wire and persist. */
+    name?: string;
     /**
      * Optional AG-UI metadata bag. TanStack writes the `tanstack` key.
      * User keys stay at the top.

package/dist/esm/utilities/ag-ui-wire.d.ts

@@ -1,51 +1,27 @@
-import { ContentPart, UIMessage } from '../types.js';
+import { AssistantMessage, ReasoningMessage, SystemMessage, ToolMessage, UserMessage } from '@ag-ui/core';
+import { ModelMessage, UIMessage } from '../types.js';
 import { MetadataRecord } from './merge-metadata.js';
-type AGUITextInputContent = {
-    type: 'text';
-    text: string;
-};
-type AGUIInputContent = AGUITextInputContent | (ContentPart & {
-    type: 'image' | 'audio' | 'video' | 'document';
-});
-type AGUIToolCallMirror = {
-    id: string;
-    type: 'function';
-    function: {
-        name: string;
-        arguments: string;
-    };
-    encryptedValue?: string;
-};
-type AGUIToolMessage = {
-    role: 'tool';
-    id: string;
-    toolCallId: string;
-    content: string;
-    error?: string;
-};
-type AGUIReasoningMessage = {
-    role: 'reasoning';
-    id: string;
-    content: string;
-    encryptedValue?: string;
+type WithMetadata<T> = T & {
     metadata?: MetadataRecord;
 };
-/** Spec AG-UI message. No `parts`, no `createdAt` Date. */
-type WireAnchorMessage = {
-    id: string;
-    role: UIMessage['role'];
+type WireSystemMessage = WithMetadata<SystemMessage>;
+type WireUserMessage = WithMetadata<UserMessage>;
+type WireAssistantMessage = WithMetadata<AssistantMessage>;
+type WireToolMessage = WithMetadata<ToolMessage & {
     name?: string;
-    content?: string | Array<AGUIInputContent>;
-    toolCalls?: Array<AGUIToolCallMirror>;
-    metadata?: MetadataRecord;
-};
-export type WireMessage = WireAnchorMessage | AGUIToolMessage | AGUIReasoningMessage;
+}>;
+type WireReasoningMessage = WithMetadata<ReasoningMessage>;
+export type WireMessage = WireSystemMessage | WireUserMessage | WireAssistantMessage | WireToolMessage | WireReasoningMessage;
 /**
- * Serialize TanStack `UIMessage`s into the AG-UI `RunAgentInput.messages`
- * wire shape. Anchors are spec-only (`id`, `role`, `name`, `content`,
- * `toolCalls`, `metadata`). Tool results and thinking parts on assistant
- * messages are additionally emitted as fan-out `{role:'tool',...}` and
- * `{role:'reasoning',...}` entries for strict AG-UI server consumers.
+ * Serialize TanStack `UIMessage`s and `ModelMessage`s into the AG-UI
+ * `RunAgentInput.messages` wire shape. Anchors are spec-only (`id`, `role`,
+ * `name`, `content`, `toolCalls`, `metadata`). Tool results and thinking parts
+ * on assistant messages are additionally emitted as fan-out
+ * `{role:'tool',...}` and `{role:'reasoning',...}` entries for strict AG-UI
+ * server consumers. Set `includeSnapshotStructuredOutput` to retain complete
+ * structured-output metadata for UI snapshots.
  */
-export declare function uiMessagesToWire(messages: Array<UIMessage>): Array<WireMessage>;
+export declare function uiMessagesToWire(messages: Array<UIMessage | ModelMessage>, options?: {
+    includeSnapshotStructuredOutput: boolean;
+}): Array<WireMessage>;
 export {};

package/dist/esm/utilities/ag-ui-wire.js

@@ -1,28 +1,74 @@
 import { tanstackMetadata } from "./merge-metadata.js";
+import { normalizeToolResult } from "./tool-result.js";
+import { coerceCreatedAt, modelMessageToUIMessage } from "../activities/chat/messages.js";
 //#region src/utilities/ag-ui-wire.ts
+function isRecord(value) {
+	return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function rebuiltToolMetadata(metadata, createdAt, id, content, anchorOwnsUiResources = false) {
+	const source = isRecord(metadata) ? metadata : {};
+	const tanstack = isRecord(source.tanstack) ? { ...source.tanstack } : {};
+	if (anchorOwnsUiResources) delete tanstack.uiResources;
+	const date = coerceCreatedAt(createdAt);
+	const toolResult = {
+		...id !== void 0 ? { id } : {},
+		...date && { createdAt: date.toISOString() },
+		...Array.isArray(content) && { content }
+	};
+	const result = {
+		...source,
+		tanstack: {
+			...tanstack,
+			toolResult
+		}
+	};
+	return Object.keys(result).length ? result : void 0;
+}
 /**
-* Serialize TanStack `UIMessage`s into the AG-UI `RunAgentInput.messages`
-* wire shape. Anchors are spec-only (`id`, `role`, `name`, `content`,
-* `toolCalls`, `metadata`). Tool results and thinking parts on assistant
-* messages are additionally emitted as fan-out `{role:'tool',...}` and
-* `{role:'reasoning',...}` entries for strict AG-UI server consumers.
+* Serialize TanStack `UIMessage`s and `ModelMessage`s into the AG-UI
+* `RunAgentInput.messages` wire shape. Anchors are spec-only (`id`, `role`,
+* `name`, `content`, `toolCalls`, `metadata`). Tool results and thinking parts
+* on assistant messages are additionally emitted as fan-out
+* `{role:'tool',...}` and `{role:'reasoning',...}` entries for strict AG-UI
+* server consumers. Set `includeSnapshotStructuredOutput` to retain complete
+* structured-output metadata for UI snapshots.
 */
-function uiMessagesToWire(messages) {
+function uiMessagesToWire(messages, options) {
 	const wire = [];
+	const usedWireIds = new Set(messages.flatMap((message) => "id" in message && message.id && message.role !== "tool" ? [message.id] : []));
+	const includeSnapshotStructuredOutput = options?.includeSnapshotStructuredOutput ?? false;
+	const assistantIds = /* @__PURE__ */ new Set();
+	for (const msg of messages) if (msg.role === "assistant" && msg.id !== void 0) assistantIds.add(msg.id);
 	for (const msg of messages) {
-		const parts = msg.parts ?? [];
+		if (!("parts" in msg) && msg.role === "tool" && msg.toolCallId) {
+			const id = uniqueToolWireId(toolWireId(msg.id, msg.toolCallId, assistantIds), usedWireIds);
+			const metadata = rebuiltToolMetadata(msg.metadata, msg.createdAt, msg.id, msg.content);
+			wire.push({
+				role: "tool",
+				id,
+				...msg.name !== void 0 && { name: msg.name },
+				toolCallId: msg.toolCallId,
+				content: typeof msg.content === "string" ? msg.content : JSON.stringify(msg.content),
+				...msg.error !== void 0 && { error: msg.error },
+				...metadata !== void 0 && { metadata }
+			});
+			continue;
+		}
+		const uiMessage = "parts" in msg ? msg : modelMessageToUIMessage(msg, msg.id);
+		const parts = uiMessage.parts;
+		usedWireIds.add(uiMessage.id);
 		if (msg.role === "system") {
-			wire.push(toAnchor(msg, { content: parts.length > 0 ? collectText(parts) : msg.content ?? "" }, parts));
+			wire.push(toAnchor(uiMessage, "system", { content: parts.length > 0 ? collectText(parts) : msg.content ?? "" }, parts, includeSnapshotStructuredOutput));
 			continue;
 		}
 		if (msg.role === "user") {
-			wire.push(toAnchor(msg, { content: parts.length > 0 ? collectUserContent(parts) : msg.content ?? "" }, parts));
+			wire.push(toAnchor(uiMessage, "user", { content: parts.length > 0 ? collectUserContent(parts) : msg.content ?? "" }, parts, includeSnapshotStructuredOutput));
 			continue;
 		}
 		for (const part of parts) if (part.type === "thinking") {
 			const reasoning = {
 				role: "reasoning",
-				id: deriveReasoningId(msg.id, part),
+				id: uniqueWireId(deriveReasoningId(uiMessage.id, part), usedWireIds),
 				content: part.content
 			};
 			if (part.signature) reasoning.encryptedValue = part.signature;
@@ -30,52 +76,96 @@ function uiMessagesToWire(messages) {
 		}
 		const text = collectText(parts);
 		const toolCalls = collectToolCalls(parts);
-		wire.push(toAnchor(msg, {
+		wire.push(toAnchor(uiMessage, "assistant", {
 			...text !== "" && { content: text },
 			...toolCalls && { toolCalls }
-		}, parts));
-		for (const part of parts) if (part.type === "tool-result") wire.push({
-			role: "tool",
-			id: deriveToolMessageId(part.toolCallId),
-			toolCallId: part.toolCallId,
-			content: typeof part.content === "string" ? part.content : JSON.stringify(part.content),
-			...part.error !== void 0 && { error: part.error }
-		});
+		}, parts, includeSnapshotStructuredOutput));
+		const explicitToolResults = new Set(parts.flatMap((part) => part.type === "tool-result" ? [part.toolCallId] : []));
+		for (const part of parts) if (part.type === "tool-result") {
+			const id = uniqueToolWireId(part.id ?? deriveToolMessageId(part.toolCallId), usedWireIds);
+			const metadata = rebuiltToolMetadata(part.metadata, part.createdAt, part.id, part.content, true);
+			wire.push({
+				role: "tool",
+				id,
+				toolCallId: part.toolCallId,
+				...part.name !== void 0 && { name: part.name },
+				content: typeof part.content === "string" ? part.content : JSON.stringify(part.content),
+				...part.error !== void 0 && { error: part.error },
+				...metadata !== void 0 && { metadata }
+			});
+		} else if (part.type === "tool-call") {
+			const approved = part.approval?.approved;
+			if (explicitToolResults.has(part.id) || part.output === void 0 && (part.state !== "approval-responded" || approved === void 0)) continue;
+			const result = part.output !== void 0 ? normalizeToolResult(part.output) : JSON.stringify({
+				approved,
+				...approved && { pendingExecution: true },
+				message: approved ? "User approved this action" : "User denied this action"
+			});
+			const content = typeof result === "string" ? result : JSON.stringify(result);
+			wire.push({
+				role: "tool",
+				id: uniqueToolWireId(deriveToolMessageId(part.id), usedWireIds),
+				toolCallId: part.id,
+				content,
+				metadata: rebuiltToolMetadata(void 0, void 0, void 0, result)
+			});
+		}
 	}
 	return wire;
 }
-function toAnchor(msg, extras, parts) {
-	const metadata = messageMetadata(msg, parts);
-	const name = msg.name;
-	return {
+function toAnchor(msg, role, extras, parts, includeSnapshotStructuredOutput) {
+	const metadata = messageMetadata(msg, parts, includeSnapshotStructuredOutput);
+	const base = {
 		id: msg.id,
-		role: msg.role,
-		...name !== void 0 && { name },
-		...extras,
+		...msg.name !== void 0 && { name: msg.name },
 		...metadata !== void 0 && { metadata }
 	};
+	if (role === "system") return {
+		...base,
+		role,
+		content: String(extras.content ?? "")
+	};
+	if (role === "user") return {
+		...base,
+		role,
+		content: extras.content ?? ""
+	};
+	return {
+		...base,
+		role,
+		...typeof extras.content === "string" && { content: extras.content },
+		...extras.toolCalls !== void 0 && { toolCalls: extras.toolCalls }
+	};
 }
-function messageMetadata(msg, parts) {
+function messageMetadata(msg, parts, includeSnapshotStructuredOutput) {
 	const base = { ...msg.metadata ?? {} };
-	const tanstack = { ...tanstackMetadata(msg) ?? {} };
-	if (msg.createdAt) tanstack.createdAt = msg.createdAt.toISOString();
-	const leftover = unfinishedStructuredOutput(parts);
-	if (leftover) tanstack.structuredOutput = leftover;
+	const previousTanstack = tanstackMetadata(msg);
+	const tanstack = {};
+	if (previousTanstack?.model !== void 0) tanstack.model = previousTanstack.model;
+	if (previousTanstack?.signature !== void 0) tanstack.signature = previousTanstack.signature;
+	const createdAt = coerceCreatedAt(msg.createdAt);
+	if (createdAt !== void 0) tanstack.createdAt = createdAt.toISOString();
+	const structuredOutput = serializedStructuredOutput(parts, includeSnapshotStructuredOutput);
+	if (structuredOutput) tanstack.structuredOutput = structuredOutput;
 	const toolCallMetadata = {};
 	for (const part of parts) if (part.type === "tool-call" && part.metadata !== void 0) toolCallMetadata[part.id] = part.metadata;
 	if (Object.keys(toolCallMetadata).length > 0) tanstack.toolCallMetadata = toolCallMetadata;
 	const uiResources = parts.filter((p) => p.type === "ui-resource");
 	if (uiResources.length > 0) tanstack.uiResources = uiResources;
 	if (Object.keys(tanstack).length > 0) base.tanstack = tanstack;
+	else delete base.tanstack;
 	return Object.keys(base).length > 0 ? base : void 0;
 }
-function unfinishedStructuredOutput(parts) {
-	for (const p of parts) if (p.type === "structured-output" && p.status !== "complete") return structuredOutputLeftover(p);
+function serializedStructuredOutput(parts, includeSnapshotStructuredOutput) {
+	for (const p of parts) if (p.type === "structured-output" && (includeSnapshotStructuredOutput || p.status !== "complete")) return structuredOutputMetadata(p, includeSnapshotStructuredOutput);
 }
-function structuredOutputLeftover(part) {
+function structuredOutputMetadata(part, includeSnapshotStructuredOutput) {
 	return {
 		status: part.status,
 		raw: part.raw,
+		...includeSnapshotStructuredOutput && part.partial !== void 0 ? { partial: part.partial } : {},
+		...includeSnapshotStructuredOutput && part.data !== void 0 ? { data: part.data } : {},
+		...includeSnapshotStructuredOutput && part.reasoning ? { reasoning: part.reasoning } : {},
 		...part.errorMessage !== void 0 && { errorMessage: part.errorMessage }
 	};
 }
@@ -123,6 +213,25 @@ function deriveReasoningId(messageId, part) {
 function deriveToolMessageId(toolCallId) {
 	return `tool-${toolCallId}`;
 }
+function uniqueToolWireId(id, used) {
+	return uniqueWireId(id, used);
+}
+function uniqueWireId(id, used) {
+	if (!used.has(id)) {
+		used.add(id);
+		return id;
+	}
+	let suffix = 2;
+	while (used.has(`${id}-${suffix}`)) suffix++;
+	const unique = `${id}-${suffix}`;
+	used.add(unique);
+	return unique;
+}
+function toolWireId(id, toolCallId, assistantIds) {
+	const derived = deriveToolMessageId(toolCallId);
+	if (id === void 0 || assistantIds.has(id)) return derived;
+	return id;
+}
 function hashContent(s) {
 	let h = 0;
 	for (let i = 0; i < s.length; i++) h = h * 31 + s.charCodeAt(i) | 0;

package/dist/esm/utilities/ag-ui-wire.js.map

@@ -1 +1 @@
-{"version":3,"file":"ag-ui-wire.js","names":[],"sources":["../../../src/utilities/ag-ui-wire.ts"],"sourcesContent":["import type {\n  ContentPart,\n  MessagePart,\n  StructuredOutputPart,\n  TanStackMessageMetadata,\n  UIMessage,\n  UIResourcePart,\n} from '../types'\nimport type { MetadataRecord } from './merge-metadata'\nimport { tanstackMetadata } from './merge-metadata'\n\ntype AGUITextInputContent = { type: 'text'; text: string }\ntype AGUIInputContent =\n  | AGUITextInputContent\n  | (ContentPart & { type: 'image' | 'audio' | 'video' | 'document' })\n\ntype AGUIToolCallMirror = {\n  id: string\n  type: 'function'\n  function: { name: string; arguments: string }\n  encryptedValue?: string\n}\n\ntype AGUIToolMessage = {\n  role: 'tool'\n  id: string\n  toolCallId: string\n  content: string\n  error?: string\n}\n\ntype AGUIReasoningMessage = {\n  role: 'reasoning'\n  id: string\n  content: string\n  encryptedValue?: string\n  metadata?: MetadataRecord\n}\n\n/** Spec AG-UI message. No `parts`, no `createdAt` Date. */\ntype WireAnchorMessage = {\n  id: string\n  role: UIMessage['role']\n  name?: string\n  content?: string | Array<AGUIInputContent>\n  toolCalls?: Array<AGUIToolCallMirror>\n  metadata?: MetadataRecord\n}\n\nexport type WireMessage =\n  | WireAnchorMessage\n  | AGUIToolMessage\n  | AGUIReasoningMessage\n\n/**\n * Serialize TanStack `UIMessage`s into the AG-UI `RunAgentInput.messages`\n * wire shape. Anchors are spec-only (`id`, `role`, `name`, `content`,\n * `toolCalls`, `metadata`). Tool results and thinking parts on assistant\n * messages are additionally emitted as fan-out `{role:'tool',...}` and\n * `{role:'reasoning',...}` entries for strict AG-UI server consumers.\n */\nexport function uiMessagesToWire(\n  messages: Array<UIMessage>,\n): Array<WireMessage> {\n  const wire: Array<WireMessage> = []\n\n  for (const msg of messages) {\n    // Defensive: ModelMessage-shaped input has no `parts`; fall back to `content`.\n    const parts: ReadonlyArray<MessagePart> =\n      // eslint-disable-next-line @typescript-eslint/no-unnecessary-type-assertion -- runtime input may be ModelMessage-shaped (no `parts`); cast forces the optional-chain fallback below to remain in scope\n      (msg.parts as ReadonlyArray<MessagePart> | undefined) ?? []\n\n    if (msg.role === 'system') {\n      wire.push(\n        toAnchor(\n          msg,\n          {\n            content:\n              parts.length > 0\n                ? collectText(parts)\n                : ((msg as { content?: string }).content ?? ''),\n          },\n          parts,\n        ),\n      )\n      continue\n    }\n\n    if (msg.role === 'user') {\n      wire.push(\n        toAnchor(\n          msg,\n          {\n            content:\n              parts.length > 0\n                ? collectUserContent(parts)\n                : ((msg as { content?: string }).content ?? ''),\n          },\n          parts,\n        ),\n      )\n      continue\n    }\n\n    // assistant: emit reasoning fan-outs first, then anchor, then tool fan-outs\n    for (const part of parts) {\n      if (part.type === 'thinking') {\n        const reasoning: AGUIReasoningMessage = {\n          role: 'reasoning',\n          id: deriveReasoningId(msg.id, part),\n          content: part.content,\n        }\n        if (part.signature) {\n          reasoning.encryptedValue = part.signature\n        }\n        wire.push(reasoning)\n      }\n    }\n\n    const text = collectText(parts)\n    const toolCalls = collectToolCalls(parts)\n    wire.push(\n      toAnchor(\n        msg,\n        {\n          ...(text !== '' && { content: text }),\n          ...(toolCalls && { toolCalls }),\n        },\n        parts,\n      ),\n    )\n\n    for (const part of parts) {\n      if (part.type === 'tool-result') {\n        wire.push({\n          role: 'tool',\n          id: deriveToolMessageId(part.toolCallId),\n          toolCallId: part.toolCallId,\n          content:\n            typeof part.content === 'string'\n              ? part.content\n              : JSON.stringify(part.content),\n          ...(part.error !== undefined && { error: part.error }),\n        })\n      }\n    }\n  }\n\n  return wire\n}\n\nfunction toAnchor(\n  msg: UIMessage,\n  extras: {\n    content?: string | Array<AGUIInputContent>\n    toolCalls?: Array<AGUIToolCallMirror>\n  },\n  parts: ReadonlyArray<MessagePart>,\n): WireAnchorMessage {\n  const metadata = messageMetadata(msg, parts)\n  const name = (msg as { name?: string }).name\n  return {\n    id: msg.id,\n    role: msg.role,\n    ...(name !== undefined && { name }),\n    ...extras,\n    ...(metadata !== undefined && { metadata }),\n  }\n}\n\nfunction messageMetadata(\n  msg: UIMessage,\n  parts: ReadonlyArray<MessagePart>,\n): MetadataRecord | undefined {\n  const base: MetadataRecord = { ...(msg.metadata ?? {}) }\n  const tanstack: MetadataRecord = { ...(tanstackMetadata(msg) ?? {}) }\n  if (msg.createdAt) tanstack.createdAt = msg.createdAt.toISOString()\n\n  const leftover = unfinishedStructuredOutput(parts)\n  if (leftover) tanstack.structuredOutput = leftover\n\n  const toolCallMetadata: Record<string, unknown> = {}\n  for (const part of parts) {\n    if (part.type === 'tool-call' && part.metadata !== undefined) {\n      toolCallMetadata[part.id] = part.metadata\n    }\n  }\n  if (Object.keys(toolCallMetadata).length > 0) {\n    tanstack.toolCallMetadata = toolCallMetadata\n  }\n\n  const uiResources = parts.filter(\n    (p): p is UIResourcePart => p.type === 'ui-resource',\n  )\n  if (uiResources.length > 0) tanstack.uiResources = uiResources\n\n  if (Object.keys(tanstack).length > 0) base.tanstack = tanstack\n  return Object.keys(base).length > 0 ? base : undefined\n}\n\nfunction unfinishedStructuredOutput(\n  parts: ReadonlyArray<MessagePart>,\n): TanStackMessageMetadata['structuredOutput'] | undefined {\n  for (const p of parts) {\n    if (p.type === 'structured-output' && p.status !== 'complete') {\n      return structuredOutputLeftover(p)\n    }\n  }\n  return undefined\n}\n\nfunction structuredOutputLeftover(\n  part: StructuredOutputPart,\n): NonNullable<TanStackMessageMetadata['structuredOutput']> {\n  return {\n    status: part.status,\n    raw: part.raw,\n    ...(part.errorMessage !== undefined && { errorMessage: part.errorMessage }),\n  }\n}\n\nfunction collectText(parts: ReadonlyArray<MessagePart>): string {\n  // The streamed JSON of a completed structured-output part is the source of\n  // truth for multi-turn coherence — emitting it back as assistant content\n  // lets the LLM see its own prior structured response. Streaming/errored\n  // parts are skipped: they'd ship malformed JSON fragments and confuse the\n  // model. `completeStructuredOutputPart` tries hard to populate `raw`\n  // (caller → existing buffer → `JSON.stringify(data)`), but the stringify\n  // fallback can leave it empty when `data` is unserializable (BigInt,\n  // circular). The `p.raw !== ''` guard below is what enforces \"no malformed\n  // round-trip\" in that case — without it we'd ship `''` and the model would\n  // see an empty assistant turn.\n  const out: Array<string> = []\n  for (const p of parts) {\n    if (p.type === 'text') {\n      out.push(p.content)\n    } else if (\n      p.type === 'structured-output' &&\n      p.status === 'complete' &&\n      p.raw !== ''\n    ) {\n      out.push(p.raw)\n    }\n  }\n  return out.join('')\n}\n\nfunction collectUserContent(\n  parts: ReadonlyArray<MessagePart>,\n): string | Array<AGUIInputContent> {\n  const hasMultimodal = parts.some(\n    (p) =>\n      p.type === 'image' ||\n      p.type === 'audio' ||\n      p.type === 'video' ||\n      p.type === 'document',\n  )\n  if (!hasMultimodal) {\n    return collectText(parts)\n  }\n  const out: Array<AGUIInputContent> = []\n  for (const p of parts) {\n    if (p.type === 'text') {\n      out.push({ type: 'text', text: p.content })\n    } else if (\n      p.type === 'image' ||\n      p.type === 'audio' ||\n      p.type === 'video' ||\n      p.type === 'document'\n    ) {\n      out.push(p)\n    }\n  }\n  return out\n}\n\nfunction thoughtSignatureFromMetadata(metadata: unknown): string | undefined {\n  if (\n    metadata == null ||\n    typeof metadata !== 'object' ||\n    Array.isArray(metadata)\n  ) {\n    return undefined\n  }\n  if (!('thoughtSignature' in metadata)) return undefined\n  const value = metadata.thoughtSignature\n  return typeof value === 'string' && value !== '' ? value : undefined\n}\n\nfunction collectToolCalls(\n  parts: ReadonlyArray<MessagePart>,\n): Array<AGUIToolCallMirror> | undefined {\n  const calls: Array<AGUIToolCallMirror> = []\n  for (const p of parts) {\n    if (p.type === 'tool-call') {\n      const encryptedValue = thoughtSignatureFromMetadata(p.metadata)\n      calls.push({\n        id: p.id,\n        type: 'function',\n        function: { name: p.name, arguments: p.arguments },\n        ...(encryptedValue !== undefined ? { encryptedValue } : {}),\n      })\n    }\n  }\n  return calls.length > 0 ? calls : undefined\n}\n\nfunction deriveReasoningId(messageId: string, part: MessagePart): string {\n  return `${messageId}-reasoning-${(part as { id?: string }).id ?? hashContent((part as { content: string }).content)}`\n}\n\nfunction deriveToolMessageId(toolCallId: string): string {\n  return `tool-${toolCallId}`\n}\n\nfunction hashContent(s: string): string {\n  // Cheap deterministic id suffix; collisions are tolerable since\n  // reasoning ids only matter for AG-UI server consumers, not for our\n  // own server's dedup logic (which keys on toolCallId, not reasoning id).\n  let h = 0\n  for (let i = 0; i < s.length; i++) h = (h * 31 + s.charCodeAt(i)) | 0\n  return Math.abs(h).toString(36)\n}\n"],"mappings":";;;;;;;;;AA6DA,SAAgB,iBACd,UACoB;CACpB,MAAM,OAA2B,CAAC;CAElC,KAAK,MAAM,OAAO,UAAU;EAE1B,MAAM,QAEH,IAAI,SAAoD,CAAC;EAE5D,IAAI,IAAI,SAAS,UAAU;GACzB,KAAK,KACH,SACE,KACA,EACE,SACE,MAAM,SAAS,IACX,YAAY,KAAK,IACf,IAA6B,WAAW,GAClD,GACA,KACF,CACF;GACA;EACF;EAEA,IAAI,IAAI,SAAS,QAAQ;GACvB,KAAK,KACH,SACE,KACA,EACE,SACE,MAAM,SAAS,IACX,mBAAmB,KAAK,IACtB,IAA6B,WAAW,GAClD,GACA,KACF,CACF;GACA;EACF;EAGA,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,YAAY;GAC5B,MAAM,YAAkC;IACtC,MAAM;IACN,IAAI,kBAAkB,IAAI,IAAI,IAAI;IAClC,SAAS,KAAK;GAChB;GACA,IAAI,KAAK,WACP,UAAU,iBAAiB,KAAK;GAElC,KAAK,KAAK,SAAS;EACrB;EAGF,MAAM,OAAO,YAAY,KAAK;EAC9B,MAAM,YAAY,iBAAiB,KAAK;EACxC,KAAK,KACH,SACE,KACA;GACE,GAAI,SAAS,MAAM,EAAE,SAAS,KAAK;GACnC,GAAI,aAAa,EAAE,UAAU;EAC/B,GACA,KACF,CACF;EAEA,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,eAChB,KAAK,KAAK;GACR,MAAM;GACN,IAAI,oBAAoB,KAAK,UAAU;GACvC,YAAY,KAAK;GACjB,SACE,OAAO,KAAK,YAAY,WACpB,KAAK,UACL,KAAK,UAAU,KAAK,OAAO;GACjC,GAAI,KAAK,UAAU,KAAA,KAAa,EAAE,OAAO,KAAK,MAAM;EACtD,CAAC;CAGP;CAEA,OAAO;AACT;AAEA,SAAS,SACP,KACA,QAIA,OACmB;CACnB,MAAM,WAAW,gBAAgB,KAAK,KAAK;CAC3C,MAAM,OAAQ,IAA0B;CACxC,OAAO;EACL,IAAI,IAAI;EACR,MAAM,IAAI;EACV,GAAI,SAAS,KAAA,KAAa,EAAE,KAAK;EACjC,GAAG;EACH,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;CAC3C;AACF;AAEA,SAAS,gBACP,KACA,OAC4B;CAC5B,MAAM,OAAuB,EAAE,GAAI,IAAI,YAAY,CAAC,EAAG;CACvD,MAAM,WAA2B,EAAE,GAAI,iBAAiB,GAAG,KAAK,CAAC,EAAG;CACpE,IAAI,IAAI,WAAW,SAAS,YAAY,IAAI,UAAU,YAAY;CAElE,MAAM,WAAW,2BAA2B,KAAK;CACjD,IAAI,UAAU,SAAS,mBAAmB;CAE1C,MAAM,mBAA4C,CAAC;CACnD,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,eAAe,KAAK,aAAa,KAAA,GACjD,iBAAiB,KAAK,MAAM,KAAK;CAGrC,IAAI,OAAO,KAAK,gBAAgB,CAAC,CAAC,SAAS,GACzC,SAAS,mBAAmB;CAG9B,MAAM,cAAc,MAAM,QACvB,MAA2B,EAAE,SAAS,aACzC;CACA,IAAI,YAAY,SAAS,GAAG,SAAS,cAAc;CAEnD,IAAI,OAAO,KAAK,QAAQ,CAAC,CAAC,SAAS,GAAG,KAAK,WAAW;CACtD,OAAO,OAAO,KAAK,IAAI,CAAC,CAAC,SAAS,IAAI,OAAO,KAAA;AAC/C;AAEA,SAAS,2BACP,OACyD;CACzD,KAAK,MAAM,KAAK,OACd,IAAI,EAAE,SAAS,uBAAuB,EAAE,WAAW,YACjD,OAAO,yBAAyB,CAAC;AAIvC;AAEA,SAAS,yBACP,MAC0D;CAC1D,OAAO;EACL,QAAQ,KAAK;EACb,KAAK,KAAK;EACV,GAAI,KAAK,iBAAiB,KAAA,KAAa,EAAE,cAAc,KAAK,aAAa;CAC3E;AACF;AAEA,SAAS,YAAY,OAA2C;CAW9D,MAAM,MAAqB,CAAC;CAC5B,KAAK,MAAM,KAAK,OACd,IAAI,EAAE,SAAS,QACb,IAAI,KAAK,EAAE,OAAO;MACb,IACL,EAAE,SAAS,uBACX,EAAE,WAAW,cACb,EAAE,QAAQ,IAEV,IAAI,KAAK,EAAE,GAAG;CAGlB,OAAO,IAAI,KAAK,EAAE;AACpB;AAEA,SAAS,mBACP,OACkC;CAQlC,IAAI,CAPkB,MAAM,MACzB,MACC,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,UAEV,GACH,OAAO,YAAY,KAAK;CAE1B,MAAM,MAA+B,CAAC;CACtC,KAAK,MAAM,KAAK,OACd,IAAI,EAAE,SAAS,QACb,IAAI,KAAK;EAAE,MAAM;EAAQ,MAAM,EAAE;CAAQ,CAAC;MACrC,IACL,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,YAEX,IAAI,KAAK,CAAC;CAGd,OAAO;AACT;AAEA,SAAS,6BAA6B,UAAuC;CAC3E,IACE,YAAY,QACZ,OAAO,aAAa,YACpB,MAAM,QAAQ,QAAQ,GAEtB;CAEF,IAAI,EAAE,sBAAsB,WAAW,OAAO,KAAA;CAC9C,MAAM,QAAQ,SAAS;CACvB,OAAO,OAAO,UAAU,YAAY,UAAU,KAAK,QAAQ,KAAA;AAC7D;AAEA,SAAS,iBACP,OACuC;CACvC,MAAM,QAAmC,CAAC;CAC1C,KAAK,MAAM,KAAK,OACd,IAAI,EAAE,SAAS,aAAa;EAC1B,MAAM,iBAAiB,6BAA6B,EAAE,QAAQ;EAC9D,MAAM,KAAK;GACT,IAAI,EAAE;GACN,MAAM;GACN,UAAU;IAAE,MAAM,EAAE;IAAM,WAAW,EAAE;GAAU;GACjD,GAAI,mBAAmB,KAAA,IAAY,EAAE,eAAe,IAAI,CAAC;EAC3D,CAAC;CACH;CAEF,OAAO,MAAM,SAAS,IAAI,QAAQ,KAAA;AACpC;AAEA,SAAS,kBAAkB,WAAmB,MAA2B;CACvE,OAAO,GAAG,UAAU,aAAc,KAAyB,MAAM,YAAa,KAA6B,OAAO;AACpH;AAEA,SAAS,oBAAoB,YAA4B;CACvD,OAAO,QAAQ;AACjB;AAEA,SAAS,YAAY,GAAmB;CAItC,IAAI,IAAI;CACR,KAAK,IAAI,IAAI,GAAG,IAAI,EAAE,QAAQ,KAAK,IAAK,IAAI,KAAK,EAAE,WAAW,CAAC,IAAK;CACpE,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,EAAE;AAChC"}
+{"version":3,"file":"ag-ui-wire.js","names":[],"sources":["../../../src/utilities/ag-ui-wire.ts"],"sourcesContent":["import type {\n  AssistantMessage,\n  InputContent,\n  ReasoningMessage,\n  SystemMessage,\n  ToolCall,\n  ToolMessage,\n  UserMessage,\n} from '@ag-ui/core'\nimport type {\n  ContentPart,\n  MessagePart,\n  ModelMessage,\n  StructuredOutputPart,\n  TanStackMessageMetadata,\n  UIMessage,\n  UIResourcePart,\n} from '../types'\nimport type { MetadataRecord } from './merge-metadata'\nimport { tanstackMetadata } from './merge-metadata'\nimport { normalizeToolResult } from './tool-result'\nimport {\n  coerceCreatedAt,\n  modelMessageToUIMessage,\n} from '../activities/chat/messages'\n\ntype WithMetadata<T> = T & { metadata?: MetadataRecord }\ntype WireSystemMessage = WithMetadata<SystemMessage>\ntype WireUserMessage = WithMetadata<UserMessage>\ntype WireAssistantMessage = WithMetadata<AssistantMessage>\ntype WireToolMessage = WithMetadata<\n  ToolMessage & {\n    name?: string\n  }\n>\ntype WireReasoningMessage = WithMetadata<ReasoningMessage>\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return typeof value === 'object' && value !== null && !Array.isArray(value)\n}\n\nfunction rebuiltToolMetadata(\n  metadata: unknown,\n  createdAt: unknown,\n  id: string | undefined,\n  content: string | null | Array<ContentPart>,\n  anchorOwnsUiResources = false,\n): MetadataRecord | undefined {\n  const source: MetadataRecord = isRecord(metadata) ? metadata : {}\n  const tanstack = isRecord(source.tanstack) ? { ...source.tanstack } : {}\n  if (anchorOwnsUiResources) delete tanstack.uiResources\n  const date = coerceCreatedAt(createdAt)\n  const toolResult: NonNullable<TanStackMessageMetadata['toolResult']> = {\n    ...(id !== undefined ? { id } : {}),\n    ...(date && { createdAt: date.toISOString() }),\n    ...(Array.isArray(content) && { content }),\n  }\n  const result = {\n    ...source,\n    tanstack: { ...tanstack, toolResult },\n  }\n  return Object.keys(result).length ? result : undefined\n}\n\nexport type WireMessage =\n  | WireSystemMessage\n  | WireUserMessage\n  | WireAssistantMessage\n  | WireToolMessage\n  | WireReasoningMessage\n\n/**\n * Serialize TanStack `UIMessage`s and `ModelMessage`s into the AG-UI\n * `RunAgentInput.messages` wire shape. Anchors are spec-only (`id`, `role`,\n * `name`, `content`, `toolCalls`, `metadata`). Tool results and thinking parts\n * on assistant messages are additionally emitted as fan-out\n * `{role:'tool',...}` and `{role:'reasoning',...}` entries for strict AG-UI\n * server consumers. Set `includeSnapshotStructuredOutput` to retain complete\n * structured-output metadata for UI snapshots.\n */\nexport function uiMessagesToWire(\n  messages: Array<UIMessage | ModelMessage>,\n  options?: { includeSnapshotStructuredOutput: boolean },\n): Array<WireMessage> {\n  const wire: Array<WireMessage> = []\n  const usedWireIds = new Set<string>(\n    messages.flatMap((message) =>\n      'id' in message && message.id && message.role !== 'tool'\n        ? [message.id]\n        : [],\n    ),\n  )\n  const includeSnapshotStructuredOutput =\n    options?.includeSnapshotStructuredOutput ?? false\n\n  const assistantIds = new Set<string>()\n  for (const msg of messages) {\n    if (msg.role === 'assistant' && msg.id !== undefined) {\n      assistantIds.add(msg.id)\n    }\n  }\n\n  for (const msg of messages) {\n    if (!('parts' in msg) && msg.role === 'tool' && msg.toolCallId) {\n      const id = uniqueToolWireId(\n        toolWireId(msg.id, msg.toolCallId, assistantIds),\n        usedWireIds,\n      )\n      const metadata = rebuiltToolMetadata(\n        msg.metadata,\n        msg.createdAt,\n        msg.id,\n        msg.content,\n      )\n      wire.push({\n        role: 'tool',\n        id,\n        ...(msg.name !== undefined && { name: msg.name }),\n        toolCallId: msg.toolCallId,\n        content:\n          typeof msg.content === 'string'\n            ? msg.content\n            : JSON.stringify(msg.content),\n        ...(msg.error !== undefined && { error: msg.error }),\n        ...(metadata !== undefined && { metadata }),\n      })\n      continue\n    }\n\n    const uiMessage: UIMessage =\n      'parts' in msg ? msg : modelMessageToUIMessage(msg, msg.id)\n    const parts: ReadonlyArray<MessagePart> = uiMessage.parts\n    usedWireIds.add(uiMessage.id)\n\n    if (msg.role === 'system') {\n      wire.push(\n        toAnchor(\n          uiMessage,\n          'system',\n          {\n            content:\n              parts.length > 0\n                ? collectText(parts)\n                : ((msg as { content?: string }).content ?? ''),\n          },\n          parts,\n          includeSnapshotStructuredOutput,\n        ),\n      )\n      continue\n    }\n\n    if (msg.role === 'user') {\n      wire.push(\n        toAnchor(\n          uiMessage,\n          'user',\n          {\n            content:\n              parts.length > 0\n                ? collectUserContent(parts)\n                : ((msg as { content?: string }).content ?? ''),\n          },\n          parts,\n          includeSnapshotStructuredOutput,\n        ),\n      )\n      continue\n    }\n\n    // assistant: emit reasoning fan-outs first, then anchor, then tool fan-outs\n    for (const part of parts) {\n      if (part.type === 'thinking') {\n        const reasoning: WireReasoningMessage = {\n          role: 'reasoning',\n          id: uniqueWireId(deriveReasoningId(uiMessage.id, part), usedWireIds),\n          content: part.content,\n        }\n        if (part.signature) {\n          reasoning.encryptedValue = part.signature\n        }\n        wire.push(reasoning)\n      }\n    }\n\n    const text = collectText(parts)\n    const toolCalls = collectToolCalls(parts)\n    wire.push(\n      toAnchor(\n        uiMessage,\n        'assistant',\n        {\n          ...(text !== '' && { content: text }),\n          ...(toolCalls && { toolCalls }),\n        },\n        parts,\n        includeSnapshotStructuredOutput,\n      ),\n    )\n\n    const explicitToolResults = new Set(\n      parts.flatMap((part) =>\n        part.type === 'tool-result' ? [part.toolCallId] : [],\n      ),\n    )\n    for (const part of parts) {\n      if (part.type === 'tool-result') {\n        const id = uniqueToolWireId(\n          part.id ?? deriveToolMessageId(part.toolCallId),\n          usedWireIds,\n        )\n        const metadata = rebuiltToolMetadata(\n          part.metadata,\n          part.createdAt,\n          part.id,\n          part.content,\n          true,\n        )\n        wire.push({\n          role: 'tool',\n          id,\n          toolCallId: part.toolCallId,\n          ...(part.name !== undefined && { name: part.name }),\n          content:\n            typeof part.content === 'string'\n              ? part.content\n              : JSON.stringify(part.content),\n          ...(part.error !== undefined && { error: part.error }),\n          ...(metadata !== undefined && { metadata }),\n        })\n      } else if (part.type === 'tool-call') {\n        const approved = part.approval?.approved\n        if (\n          explicitToolResults.has(part.id) ||\n          (part.output === undefined &&\n            (part.state !== 'approval-responded' || approved === undefined))\n        ) {\n          continue\n        }\n        const result =\n          part.output !== undefined\n            ? normalizeToolResult(part.output)\n            : JSON.stringify({\n                approved,\n                ...(approved && { pendingExecution: true }),\n                message: approved\n                  ? 'User approved this action'\n                  : 'User denied this action',\n              })\n        const content =\n          typeof result === 'string' ? result : JSON.stringify(result)\n        wire.push({\n          role: 'tool',\n          id: uniqueToolWireId(deriveToolMessageId(part.id), usedWireIds),\n          toolCallId: part.id,\n          content,\n          metadata: rebuiltToolMetadata(\n            undefined,\n            undefined,\n            undefined,\n            result,\n          ),\n        })\n      }\n    }\n  }\n\n  return wire\n}\n\nfunction toAnchor(\n  msg: UIMessage,\n  role: 'system',\n  extras: { content: string },\n  parts: ReadonlyArray<MessagePart>,\n  includeSnapshotStructuredOutput: boolean,\n): WireSystemMessage\nfunction toAnchor(\n  msg: UIMessage,\n  role: 'user',\n  extras: { content: string | Array<InputContent> },\n  parts: ReadonlyArray<MessagePart>,\n  includeSnapshotStructuredOutput: boolean,\n): WireUserMessage\nfunction toAnchor(\n  msg: UIMessage,\n  role: 'assistant',\n  extras: {\n    content?: string\n    toolCalls?: Array<ToolCall>\n  },\n  parts: ReadonlyArray<MessagePart>,\n  includeSnapshotStructuredOutput: boolean,\n): WireAssistantMessage\nfunction toAnchor(\n  msg: UIMessage,\n  role: UIMessage['role'],\n  extras: {\n    content?: string | Array<InputContent>\n    toolCalls?: Array<ToolCall>\n  },\n  parts: ReadonlyArray<MessagePart>,\n  includeSnapshotStructuredOutput: boolean,\n): WireSystemMessage | WireUserMessage | WireAssistantMessage {\n  const metadata = messageMetadata(msg, parts, includeSnapshotStructuredOutput)\n  const base = {\n    id: msg.id,\n    ...(msg.name !== undefined && { name: msg.name }),\n    ...(metadata !== undefined && { metadata }),\n  }\n  if (role === 'system') {\n    return { ...base, role, content: String(extras.content ?? '') }\n  }\n  if (role === 'user') {\n    return { ...base, role, content: extras.content ?? '' }\n  }\n  return {\n    ...base,\n    role,\n    ...(typeof extras.content === 'string' && { content: extras.content }),\n    ...(extras.toolCalls !== undefined && { toolCalls: extras.toolCalls }),\n  }\n}\n\nfunction messageMetadata(\n  msg: UIMessage,\n  parts: ReadonlyArray<MessagePart>,\n  includeSnapshotStructuredOutput: boolean,\n): MetadataRecord | undefined {\n  const base: MetadataRecord = { ...(msg.metadata ?? {}) }\n  const previousTanstack = tanstackMetadata(msg)\n  const tanstack: TanStackMessageMetadata = {}\n  if (previousTanstack?.model !== undefined)\n    tanstack.model = previousTanstack.model\n  if (previousTanstack?.signature !== undefined)\n    tanstack.signature = previousTanstack.signature\n  const createdAt = coerceCreatedAt(msg.createdAt)\n  if (createdAt !== undefined) tanstack.createdAt = createdAt.toISOString()\n\n  const structuredOutput = serializedStructuredOutput(\n    parts,\n    includeSnapshotStructuredOutput,\n  )\n  if (structuredOutput) tanstack.structuredOutput = structuredOutput\n\n  const toolCallMetadata: Record<string, unknown> = {}\n  for (const part of parts) {\n    if (part.type === 'tool-call' && part.metadata !== undefined) {\n      toolCallMetadata[part.id] = part.metadata\n    }\n  }\n  if (Object.keys(toolCallMetadata).length > 0) {\n    tanstack.toolCallMetadata = toolCallMetadata\n  }\n\n  const uiResources = parts.filter(\n    (p): p is UIResourcePart => p.type === 'ui-resource',\n  )\n  if (uiResources.length > 0) tanstack.uiResources = uiResources\n\n  if (Object.keys(tanstack).length > 0) base.tanstack = tanstack\n  else delete base.tanstack\n  return Object.keys(base).length > 0 ? base : undefined\n}\n\nfunction serializedStructuredOutput(\n  parts: ReadonlyArray<MessagePart>,\n  includeSnapshotStructuredOutput: boolean,\n): TanStackMessageMetadata['structuredOutput'] | undefined {\n  for (const p of parts) {\n    if (\n      p.type === 'structured-output' &&\n      (includeSnapshotStructuredOutput || p.status !== 'complete')\n    ) {\n      return structuredOutputMetadata(p, includeSnapshotStructuredOutput)\n    }\n  }\n  return undefined\n}\n\nfunction structuredOutputMetadata(\n  part: StructuredOutputPart,\n  includeSnapshotStructuredOutput: boolean,\n): NonNullable<TanStackMessageMetadata['structuredOutput']> {\n  return {\n    status: part.status,\n    raw: part.raw,\n    ...(includeSnapshotStructuredOutput && part.partial !== undefined\n      ? { partial: part.partial }\n      : {}),\n    ...(includeSnapshotStructuredOutput && part.data !== undefined\n      ? { data: part.data }\n      : {}),\n    ...(includeSnapshotStructuredOutput && part.reasoning\n      ? { reasoning: part.reasoning }\n      : {}),\n    ...(part.errorMessage !== undefined && { errorMessage: part.errorMessage }),\n  }\n}\n\nfunction collectText(parts: ReadonlyArray<MessagePart>): string {\n  // The streamed JSON of a completed structured-output part is the source of\n  // truth for multi-turn coherence — emitting it back as assistant content\n  // lets the LLM see its own prior structured response. Streaming/errored\n  // parts are skipped: they'd ship malformed JSON fragments and confuse the\n  // model. `completeStructuredOutputPart` tries hard to populate `raw`\n  // (caller → existing buffer → `JSON.stringify(data)`), but the stringify\n  // fallback can leave it empty when `data` is unserializable (BigInt,\n  // circular). The `p.raw !== ''` guard below is what enforces \"no malformed\n  // round-trip\" in that case — without it we'd ship `''` and the model would\n  // see an empty assistant turn.\n  const out: Array<string> = []\n  for (const p of parts) {\n    if (p.type === 'text') {\n      out.push(p.content)\n    } else if (\n      p.type === 'structured-output' &&\n      p.status === 'complete' &&\n      p.raw !== ''\n    ) {\n      out.push(p.raw)\n    }\n  }\n  return out.join('')\n}\n\nfunction collectUserContent(\n  parts: ReadonlyArray<MessagePart>,\n): string | Array<InputContent> {\n  const hasMultimodal = parts.some(\n    (p) =>\n      p.type === 'image' ||\n      p.type === 'audio' ||\n      p.type === 'video' ||\n      p.type === 'document',\n  )\n  if (!hasMultimodal) {\n    return collectText(parts)\n  }\n  const out: Array<InputContent> = []\n  for (const p of parts) {\n    if (p.type === 'text') {\n      out.push({ type: 'text', text: p.content })\n    } else if (\n      p.type === 'image' ||\n      p.type === 'audio' ||\n      p.type === 'video' ||\n      p.type === 'document'\n    ) {\n      out.push(p)\n    }\n  }\n  return out\n}\n\nfunction thoughtSignatureFromMetadata(metadata: unknown): string | undefined {\n  if (\n    metadata == null ||\n    typeof metadata !== 'object' ||\n    Array.isArray(metadata)\n  ) {\n    return undefined\n  }\n  if (!('thoughtSignature' in metadata)) return undefined\n  const value = metadata.thoughtSignature\n  return typeof value === 'string' && value !== '' ? value : undefined\n}\n\nfunction collectToolCalls(\n  parts: ReadonlyArray<MessagePart>,\n): Array<ToolCall> | undefined {\n  const calls: Array<ToolCall> = []\n  for (const p of parts) {\n    if (p.type === 'tool-call') {\n      const encryptedValue = thoughtSignatureFromMetadata(p.metadata)\n      calls.push({\n        id: p.id,\n        type: 'function',\n        function: { name: p.name, arguments: p.arguments },\n        ...(encryptedValue !== undefined ? { encryptedValue } : {}),\n      })\n    }\n  }\n  return calls.length > 0 ? calls : undefined\n}\n\nfunction deriveReasoningId(messageId: string, part: MessagePart): string {\n  return `${messageId}-reasoning-${(part as { id?: string }).id ?? hashContent((part as { content: string }).content)}`\n}\n\nfunction deriveToolMessageId(toolCallId: string): string {\n  return `tool-${toolCallId}`\n}\n\nfunction uniqueToolWireId(id: string, used: Set<string>): string {\n  return uniqueWireId(id, used)\n}\n\nfunction uniqueWireId(id: string, used: Set<string>): string {\n  if (!used.has(id)) {\n    used.add(id)\n    return id\n  }\n  let suffix = 2\n  while (used.has(`${id}-${suffix}`)) suffix++\n  const unique = `${id}-${suffix}`\n  used.add(unique)\n  return unique\n}\n\nfunction toolWireId(\n  id: string | undefined,\n  toolCallId: string,\n  assistantIds: ReadonlySet<string>,\n): string {\n  const derived = deriveToolMessageId(toolCallId)\n  if (id === undefined || assistantIds.has(id)) return derived\n  return id\n}\n\nfunction hashContent(s: string): string {\n  // Cheap deterministic id suffix; collisions are tolerable since\n  // reasoning ids only matter for AG-UI server consumers, not for our\n  // own server's dedup logic (which keys on toolCallId, not reasoning id).\n  let h = 0\n  for (let i = 0; i < s.length; i++) h = (h * 31 + s.charCodeAt(i)) | 0\n  return Math.abs(h).toString(36)\n}\n"],"mappings":";;;;AAqCA,SAAS,SAAS,OAAkD;CAClE,OAAO,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,KAAK;AAC5E;AAEA,SAAS,oBACP,UACA,WACA,IACA,SACA,wBAAwB,OACI;CAC5B,MAAM,SAAyB,SAAS,QAAQ,IAAI,WAAW,CAAC;CAChE,MAAM,WAAW,SAAS,OAAO,QAAQ,IAAI,EAAE,GAAG,OAAO,SAAS,IAAI,CAAC;CACvE,IAAI,uBAAuB,OAAO,SAAS;CAC3C,MAAM,OAAO,gBAAgB,SAAS;CACtC,MAAM,aAAiE;EACrE,GAAI,OAAO,KAAA,IAAY,EAAE,GAAG,IAAI,CAAC;EACjC,GAAI,QAAQ,EAAE,WAAW,KAAK,YAAY,EAAE;EAC5C,GAAI,MAAM,QAAQ,OAAO,KAAK,EAAE,QAAQ;CAC1C;CACA,MAAM,SAAS;EACb,GAAG;EACH,UAAU;GAAE,GAAG;GAAU;EAAW;CACtC;CACA,OAAO,OAAO,KAAK,MAAM,CAAC,CAAC,SAAS,SAAS,KAAA;AAC/C;;;;;;;;;;AAkBA,SAAgB,iBACd,UACA,SACoB;CACpB,MAAM,OAA2B,CAAC;CAClC,MAAM,cAAc,IAAI,IACtB,SAAS,SAAS,YAChB,QAAQ,WAAW,QAAQ,MAAM,QAAQ,SAAS,SAC9C,CAAC,QAAQ,EAAE,IACX,CAAC,CACP,CACF;CACA,MAAM,kCACJ,SAAS,mCAAmC;CAE9C,MAAM,+BAAe,IAAI,IAAY;CACrC,KAAK,MAAM,OAAO,UAChB,IAAI,IAAI,SAAS,eAAe,IAAI,OAAO,KAAA,GACzC,aAAa,IAAI,IAAI,EAAE;CAI3B,KAAK,MAAM,OAAO,UAAU;EAC1B,IAAI,EAAE,WAAW,QAAQ,IAAI,SAAS,UAAU,IAAI,YAAY;GAC9D,MAAM,KAAK,iBACT,WAAW,IAAI,IAAI,IAAI,YAAY,YAAY,GAC/C,WACF;GACA,MAAM,WAAW,oBACf,IAAI,UACJ,IAAI,WACJ,IAAI,IACJ,IAAI,OACN;GACA,KAAK,KAAK;IACR,MAAM;IACN;IACA,GAAI,IAAI,SAAS,KAAA,KAAa,EAAE,MAAM,IAAI,KAAK;IAC/C,YAAY,IAAI;IAChB,SACE,OAAO,IAAI,YAAY,WACnB,IAAI,UACJ,KAAK,UAAU,IAAI,OAAO;IAChC,GAAI,IAAI,UAAU,KAAA,KAAa,EAAE,OAAO,IAAI,MAAM;IAClD,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;GAC3C,CAAC;GACD;EACF;EAEA,MAAM,YACJ,WAAW,MAAM,MAAM,wBAAwB,KAAK,IAAI,EAAE;EAC5D,MAAM,QAAoC,UAAU;EACpD,YAAY,IAAI,UAAU,EAAE;EAE5B,IAAI,IAAI,SAAS,UAAU;GACzB,KAAK,KACH,SACE,WACA,UACA,EACE,SACE,MAAM,SAAS,IACX,YAAY,KAAK,IACf,IAA6B,WAAW,GAClD,GACA,OACA,+BACF,CACF;GACA;EACF;EAEA,IAAI,IAAI,SAAS,QAAQ;GACvB,KAAK,KACH,SACE,WACA,QACA,EACE,SACE,MAAM,SAAS,IACX,mBAAmB,KAAK,IACtB,IAA6B,WAAW,GAClD,GACA,OACA,+BACF,CACF;GACA;EACF;EAGA,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,YAAY;GAC5B,MAAM,YAAkC;IACtC,MAAM;IACN,IAAI,aAAa,kBAAkB,UAAU,IAAI,IAAI,GAAG,WAAW;IACnE,SAAS,KAAK;GAChB;GACA,IAAI,KAAK,WACP,UAAU,iBAAiB,KAAK;GAElC,KAAK,KAAK,SAAS;EACrB;EAGF,MAAM,OAAO,YAAY,KAAK;EAC9B,MAAM,YAAY,iBAAiB,KAAK;EACxC,KAAK,KACH,SACE,WACA,aACA;GACE,GAAI,SAAS,MAAM,EAAE,SAAS,KAAK;GACnC,GAAI,aAAa,EAAE,UAAU;EAC/B,GACA,OACA,+BACF,CACF;EAEA,MAAM,sBAAsB,IAAI,IAC9B,MAAM,SAAS,SACb,KAAK,SAAS,gBAAgB,CAAC,KAAK,UAAU,IAAI,CAAC,CACrD,CACF;EACA,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,eAAe;GAC/B,MAAM,KAAK,iBACT,KAAK,MAAM,oBAAoB,KAAK,UAAU,GAC9C,WACF;GACA,MAAM,WAAW,oBACf,KAAK,UACL,KAAK,WACL,KAAK,IACL,KAAK,SACL,IACF;GACA,KAAK,KAAK;IACR,MAAM;IACN;IACA,YAAY,KAAK;IACjB,GAAI,KAAK,SAAS,KAAA,KAAa,EAAE,MAAM,KAAK,KAAK;IACjD,SACE,OAAO,KAAK,YAAY,WACpB,KAAK,UACL,KAAK,UAAU,KAAK,OAAO;IACjC,GAAI,KAAK,UAAU,KAAA,KAAa,EAAE,OAAO,KAAK,MAAM;IACpD,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;GAC3C,CAAC;EACH,OAAO,IAAI,KAAK,SAAS,aAAa;GACpC,MAAM,WAAW,KAAK,UAAU;GAChC,IACE,oBAAoB,IAAI,KAAK,EAAE,KAC9B,KAAK,WAAW,KAAA,MACd,KAAK,UAAU,wBAAwB,aAAa,KAAA,IAEvD;GAEF,MAAM,SACJ,KAAK,WAAW,KAAA,IACZ,oBAAoB,KAAK,MAAM,IAC/B,KAAK,UAAU;IACb;IACA,GAAI,YAAY,EAAE,kBAAkB,KAAK;IACzC,SAAS,WACL,8BACA;GACN,CAAC;GACP,MAAM,UACJ,OAAO,WAAW,WAAW,SAAS,KAAK,UAAU,MAAM;GAC7D,KAAK,KAAK;IACR,MAAM;IACN,IAAI,iBAAiB,oBAAoB,KAAK,EAAE,GAAG,WAAW;IAC9D,YAAY,KAAK;IACjB;IACA,UAAU,oBACR,KAAA,GACA,KAAA,GACA,KAAA,GACA,MACF;GACF,CAAC;EACH;CAEJ;CAEA,OAAO;AACT;AA0BA,SAAS,SACP,KACA,MACA,QAIA,OACA,iCAC4D;CAC5D,MAAM,WAAW,gBAAgB,KAAK,OAAO,+BAA+B;CAC5E,MAAM,OAAO;EACX,IAAI,IAAI;EACR,GAAI,IAAI,SAAS,KAAA,KAAa,EAAE,MAAM,IAAI,KAAK;EAC/C,GAAI,aAAa,KAAA,KAAa,EAAE,SAAS;CAC3C;CACA,IAAI,SAAS,UACX,OAAO;EAAE,GAAG;EAAM;EAAM,SAAS,OAAO,OAAO,WAAW,EAAE;CAAE;CAEhE,IAAI,SAAS,QACX,OAAO;EAAE,GAAG;EAAM;EAAM,SAAS,OAAO,WAAW;CAAG;CAExD,OAAO;EACL,GAAG;EACH;EACA,GAAI,OAAO,OAAO,YAAY,YAAY,EAAE,SAAS,OAAO,QAAQ;EACpE,GAAI,OAAO,cAAc,KAAA,KAAa,EAAE,WAAW,OAAO,UAAU;CACtE;AACF;AAEA,SAAS,gBACP,KACA,OACA,iCAC4B;CAC5B,MAAM,OAAuB,EAAE,GAAI,IAAI,YAAY,CAAC,EAAG;CACvD,MAAM,mBAAmB,iBAAiB,GAAG;CAC7C,MAAM,WAAoC,CAAC;CAC3C,IAAI,kBAAkB,UAAU,KAAA,GAC9B,SAAS,QAAQ,iBAAiB;CACpC,IAAI,kBAAkB,cAAc,KAAA,GAClC,SAAS,YAAY,iBAAiB;CACxC,MAAM,YAAY,gBAAgB,IAAI,SAAS;CAC/C,IAAI,cAAc,KAAA,GAAW,SAAS,YAAY,UAAU,YAAY;CAExE,MAAM,mBAAmB,2BACvB,OACA,+BACF;CACA,IAAI,kBAAkB,SAAS,mBAAmB;CAElD,MAAM,mBAA4C,CAAC;CACnD,KAAK,MAAM,QAAQ,OACjB,IAAI,KAAK,SAAS,eAAe,KAAK,aAAa,KAAA,GACjD,iBAAiB,KAAK,MAAM,KAAK;CAGrC,IAAI,OAAO,KAAK,gBAAgB,CAAC,CAAC,SAAS,GACzC,SAAS,mBAAmB;CAG9B,MAAM,cAAc,MAAM,QACvB,MAA2B,EAAE,SAAS,aACzC;CACA,IAAI,YAAY,SAAS,GAAG,SAAS,cAAc;CAEnD,IAAI,OAAO,KAAK,QAAQ,CAAC,CAAC,SAAS,GAAG,KAAK,WAAW;MACjD,OAAO,KAAK;CACjB,OAAO,OAAO,KAAK,IAAI,CAAC,CAAC,SAAS,IAAI,OAAO,KAAA;AAC/C;AAEA,SAAS,2BACP,OACA,iCACyD;CACzD,KAAK,MAAM,KAAK,OACd,IACE,EAAE,SAAS,wBACV,mCAAmC,EAAE,WAAW,aAEjD,OAAO,yBAAyB,GAAG,+BAA+B;AAIxE;AAEA,SAAS,yBACP,MACA,iCAC0D;CAC1D,OAAO;EACL,QAAQ,KAAK;EACb,KAAK,KAAK;EACV,GAAI,mCAAmC,KAAK,YAAY,KAAA,IACpD,EAAE,SAAS,KAAK,QAAQ,IACxB,CAAC;EACL,GAAI,mCAAmC,KAAK,SAAS,KAAA,IACjD,EAAE,MAAM,KAAK,KAAK,IAClB,CAAC;EACL,GAAI,mCAAmC,KAAK,YACxC,EAAE,WAAW,KAAK,UAAU,IAC5B,CAAC;EACL,GAAI,KAAK,iBAAiB,KAAA,KAAa,EAAE,cAAc,KAAK,aAAa;CAC3E;AACF;AAEA,SAAS,YAAY,OAA2C;CAW9D,MAAM,MAAqB,CAAC;CAC5B,KAAK,MAAM,KAAK,OACd,IAAI,EAAE,SAAS,QACb,IAAI,KAAK,EAAE,OAAO;MACb,IACL,EAAE,SAAS,uBACX,EAAE,WAAW,cACb,EAAE,QAAQ,IAEV,IAAI,KAAK,EAAE,GAAG;CAGlB,OAAO,IAAI,KAAK,EAAE;AACpB;AAEA,SAAS,mBACP,OAC8B;CAQ9B,IAAI,CAPkB,MAAM,MACzB,MACC,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,UAEV,GACH,OAAO,YAAY,KAAK;CAE1B,MAAM,MAA2B,CAAC;CAClC,KAAK,MAAM,KAAK,OACd,IAAI,EAAE,SAAS,QACb,IAAI,KAAK;EAAE,MAAM;EAAQ,MAAM,EAAE;CAAQ,CAAC;MACrC,IACL,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,WACX,EAAE,SAAS,YAEX,IAAI,KAAK,CAAC;CAGd,OAAO;AACT;AAEA,SAAS,6BAA6B,UAAuC;CAC3E,IACE,YAAY,QACZ,OAAO,aAAa,YACpB,MAAM,QAAQ,QAAQ,GAEtB;CAEF,IAAI,EAAE,sBAAsB,WAAW,OAAO,KAAA;CAC9C,MAAM,QAAQ,SAAS;CACvB,OAAO,OAAO,UAAU,YAAY,UAAU,KAAK,QAAQ,KAAA;AAC7D;AAEA,SAAS,iBACP,OAC6B;CAC7B,MAAM,QAAyB,CAAC;CAChC,KAAK,MAAM,KAAK,OACd,IAAI,EAAE,SAAS,aAAa;EAC1B,MAAM,iBAAiB,6BAA6B,EAAE,QAAQ;EAC9D,MAAM,KAAK;GACT,IAAI,EAAE;GACN,MAAM;GACN,UAAU;IAAE,MAAM,EAAE;IAAM,WAAW,EAAE;GAAU;GACjD,GAAI,mBAAmB,KAAA,IAAY,EAAE,eAAe,IAAI,CAAC;EAC3D,CAAC;CACH;CAEF,OAAO,MAAM,SAAS,IAAI,QAAQ,KAAA;AACpC;AAEA,SAAS,kBAAkB,WAAmB,MAA2B;CACvE,OAAO,GAAG,UAAU,aAAc,KAAyB,MAAM,YAAa,KAA6B,OAAO;AACpH;AAEA,SAAS,oBAAoB,YAA4B;CACvD,OAAO,QAAQ;AACjB;AAEA,SAAS,iBAAiB,IAAY,MAA2B;CAC/D,OAAO,aAAa,IAAI,IAAI;AAC9B;AAEA,SAAS,aAAa,IAAY,MAA2B;CAC3D,IAAI,CAAC,KAAK,IAAI,EAAE,GAAG;EACjB,KAAK,IAAI,EAAE;EACX,OAAO;CACT;CACA,IAAI,SAAS;CACb,OAAO,KAAK,IAAI,GAAG,GAAG,GAAG,QAAQ,GAAG;CACpC,MAAM,SAAS,GAAG,GAAG,GAAG;CACxB,KAAK,IAAI,MAAM;CACf,OAAO;AACT;AAEA,SAAS,WACP,IACA,YACA,cACQ;CACR,MAAM,UAAU,oBAAoB,UAAU;CAC9C,IAAI,OAAO,KAAA,KAAa,aAAa,IAAI,EAAE,GAAG,OAAO;CACrD,OAAO;AACT;AAEA,SAAS,YAAY,GAAmB;CAItC,IAAI,IAAI;CACR,KAAK,IAAI,IAAI,GAAG,IAAI,EAAE,QAAQ,KAAK,IAAK,IAAI,KAAK,EAAE,WAAW,CAAC,IAAK;CACpE,OAAO,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,EAAE;AAChC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai",
-  "version": "0.49.1",
+  "version": "0.50.0",
   "description": "Type-safe TypeScript AI SDK for streaming chat, tool calling, agents, structured outputs, and multimodal generation.",
   "author": "Tanner Linsley",
   "license": "MIT",
@@ -88,7 +88,7 @@
     "@ag-ui/core": "0.1.1-canary.beta.0",
     "@standard-schema/spec": "^1.1.0",
     "partial-json": "^0.1.7",
-    "@tanstack/ai-event-client": "^0.10.0",
+    "@tanstack/ai-event-client": "^0.11.0",
     "@tanstack/ai-utils": "^0.4.0"
   },
   "peerDependencies": {

package/src/activities/chat/index.ts

@@ -103,7 +103,6 @@ import type {
   Interrupt,
   JSONSchema,
   LazyToolsConfig,
-  MessagesSnapshotEvent,
   ModelMessage,
   ProviderTool,
   RunFinishedEvent,
@@ -2756,9 +2755,9 @@ class TextEngine<
     return {
       type: EventType.MESSAGES_SNAPSHOT,
       timestamp: Date.now(),
-      messages: uiMessagesToWire(
-        modelMessagesToUIMessages(withIds),
-      ) as MessagesSnapshotEvent['messages'],
+      messages: uiMessagesToWire(modelMessagesToUIMessages(withIds), {
+        includeSnapshotStructuredOutput: true,
+      }),
     }
   }