@tanstack/ai 0.47.3 → 0.48.0

package/dist/esm/utilities/normalize-stream-chunk.js

@@ -0,0 +1,100 @@
+import { EventType } from "../types.js";
+import { isTanstackUsage, toSpecTokenUsage } from "./ag-ui-usage.js";
+import { withTanstackMetadata } from "./merge-metadata.js";
+import { reasoningEncryptedValue } from "./reasoning-encrypted-value.js";
+import { specKeysFor } from "./spec-event-keys.js";
+//#region src/utilities/normalize-stream-chunk.ts
+function stringField(value) {
+	return typeof value === "string" && value !== "" ? value : void 0;
+}
+function encryptedValueExtras(chunk) {
+	const extras = [];
+	const timestamp = "timestamp" in chunk && typeof chunk.timestamp === "number" ? chunk.timestamp : void 0;
+	if (typeof chunk.signature === "string" && chunk.signature !== "") {
+		const source = chunk;
+		const toolCallId = stringField(source.toolCallId);
+		const entityId = stringField(chunk.stepId) ?? stringField(source.stepName) ?? stringField(source.messageId) ?? toolCallId;
+		if (entityId !== void 0) extras.push(reasoningEncryptedValue({
+			subtype: toolCallId && !chunk.stepId ? "tool-call" : "message",
+			entityId,
+			encryptedValue: chunk.signature,
+			timestamp
+		}));
+	}
+	if (chunk.type === EventType.TOOL_CALL_START) {
+		const thoughtSignature = stringField(chunk.metadata?.thoughtSignature);
+		if (thoughtSignature !== void 0 && chunk.toolCallId) extras.push(reasoningEncryptedValue({
+			subtype: "tool-call",
+			entityId: chunk.toolCallId,
+			encryptedValue: thoughtSignature,
+			timestamp
+		}));
+	}
+	return extras;
+}
+function normalizeStreamChunk(chunk) {
+	const specKeys = specKeysFor(chunk.type);
+	const source = chunk;
+	const specChunk = {};
+	for (const key of Object.keys(chunk)) if (specKeys.has(key)) specChunk[key] = source[key];
+	if (chunk.type === EventType.TOOL_CALL_START) {
+		if (specChunk.toolCallName === void 0 && chunk.toolName) specChunk.toolCallName = chunk.toolName;
+	}
+	if (chunk.type === EventType.RUN_ERROR && chunk.error != null) {
+		if (specChunk.message === void 0 && chunk.error.message) specChunk.message = chunk.error.message;
+		if (specChunk.code === void 0 && chunk.error.code !== void 0) specChunk.code = chunk.error.code;
+	}
+	const tanstack = {};
+	if (chunk.model !== void 0 && chunk.type !== EventType.TEXT_MESSAGE_CONTENT && chunk.type !== EventType.TOOL_CALL_ARGS) tanstack.model = chunk.model;
+	if (chunk.finishReason !== void 0) tanstack.finishReason = chunk.finishReason;
+	const interruptErrors = chunk["tanstack:interruptErrors"];
+	if (interruptErrors !== void 0) tanstack.interruptErrors = interruptErrors;
+	if (chunk.type === EventType.CUSTOM || chunk.type === EventType.RUN_ERROR) {
+		if (chunk.threadId !== void 0) tanstack.threadId = chunk.threadId;
+		if (chunk.runId !== void 0) tanstack.runId = chunk.runId;
+	}
+	const skipLeftover = /* @__PURE__ */ new Set([
+		"result",
+		"error",
+		"tanstack:interruptErrors"
+	]);
+	if (chunk.type === EventType.TEXT_MESSAGE_CONTENT || chunk.type === EventType.TOOL_CALL_ARGS) {
+		skipLeftover.add("model");
+		skipLeftover.add("content");
+		skipLeftover.add("args");
+	}
+	if (chunk.type === EventType.TOOL_CALL_START) skipLeftover.add("toolName");
+	for (const key of Object.keys(chunk)) {
+		if (specKeys.has(key) || skipLeftover.has(key)) continue;
+		if (tanstack[key] !== void 0) continue;
+		const value = source[key];
+		if (value !== void 0) tanstack[key] = value;
+	}
+	if ((chunk.type === EventType.RUN_FINISHED || chunk.type === EventType.RUN_ERROR) && isTanstackUsage(specChunk.usage)) {
+		const { usage, leftover } = toSpecTokenUsage(specChunk.usage, { model: typeof chunk.model === "string" ? chunk.model : void 0 });
+		specChunk.usage = usage;
+		if (leftover !== void 0) tanstack.usage = leftover;
+	}
+	const normalized = Object.keys(tanstack).length === 0 ? specChunk : withTanstackMetadata(specChunk, tanstack);
+	const extras = encryptedValueExtras(chunk);
+	const main = [normalized];
+	if (chunk.type === EventType.TOOL_CALL_END && chunk.result !== void 0) {
+		const parentMessageId = source.parentMessageId;
+		const resultChunk = {
+			type: EventType.TOOL_CALL_RESULT,
+			toolCallId: chunk.toolCallId,
+			content: Array.isArray(chunk.result) ? JSON.stringify(chunk.result) : chunk.result,
+			messageId: typeof parentMessageId === "string" && parentMessageId !== "" ? parentMessageId : chunk.toolCallId
+		};
+		if (chunk.state === "output-error") main.push({
+			...resultChunk,
+			metadata: { tanstack: { state: chunk.state } }
+		});
+		else main.push(resultChunk);
+	}
+	return [...main, ...extras];
+}
+//#endregion
+export { normalizeStreamChunk };
+
+//# sourceMappingURL=normalize-stream-chunk.js.map

package/dist/esm/utilities/normalize-stream-chunk.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"normalize-stream-chunk.js","names":[],"sources":["../../../src/utilities/normalize-stream-chunk.ts"],"sourcesContent":["import { EventType } from '../types'\nimport type { StreamChunk, ToolCallResultEvent } from '../types'\nimport type { AdapterYieldChunk } from './adapter-yield-chunk'\nimport { isTanstackUsage, toSpecTokenUsage } from './ag-ui-usage'\nimport type { MetadataRecord } from './merge-metadata'\nimport { withTanstackMetadata } from './merge-metadata'\nimport { reasoningEncryptedValue } from './reasoning-encrypted-value'\nimport { specKeysFor } from './spec-event-keys'\n\nfunction stringField(value: unknown): string | undefined {\n  return typeof value === 'string' && value !== '' ? value : undefined\n}\n\nfunction encryptedValueExtras(chunk: AdapterYieldChunk): Array<StreamChunk> {\n  const extras: Array<StreamChunk> = []\n  const timestamp =\n    'timestamp' in chunk && typeof chunk.timestamp === 'number'\n      ? chunk.timestamp\n      : undefined\n\n  if (typeof chunk.signature === 'string' && chunk.signature !== '') {\n    const source = chunk as Record<string, unknown>\n    const toolCallId = stringField(source.toolCallId)\n    const entityId =\n      stringField(chunk.stepId) ??\n      stringField(source.stepName) ??\n      stringField(source.messageId) ??\n      toolCallId\n    if (entityId !== undefined) {\n      extras.push(\n        reasoningEncryptedValue({\n          subtype: toolCallId && !chunk.stepId ? 'tool-call' : 'message',\n          entityId,\n          encryptedValue: chunk.signature,\n          timestamp,\n        }),\n      )\n    }\n  }\n\n  if (chunk.type === EventType.TOOL_CALL_START) {\n    const thoughtSignature = stringField(\n      (chunk.metadata as { thoughtSignature?: unknown } | undefined)\n        ?.thoughtSignature,\n    )\n    if (thoughtSignature !== undefined && chunk.toolCallId) {\n      extras.push(\n        reasoningEncryptedValue({\n          subtype: 'tool-call',\n          entityId: chunk.toolCallId,\n          encryptedValue: thoughtSignature,\n          timestamp,\n        }),\n      )\n    }\n  }\n\n  return extras\n}\n\nexport function normalizeStreamChunk(\n  chunk: AdapterYieldChunk,\n): Array<StreamChunk> {\n  const specKeys = specKeysFor(chunk.type)\n  const source = chunk as Record<string, unknown>\n  const specChunk: Record<string, unknown> & {\n    metadata?: MetadataRecord | null\n  } = {}\n\n  for (const key of Object.keys(chunk)) {\n    if (specKeys.has(key)) {\n      specChunk[key] = source[key]\n    }\n  }\n\n  if (chunk.type === EventType.TOOL_CALL_START) {\n    if (specChunk.toolCallName === undefined && chunk.toolName) {\n      specChunk.toolCallName = chunk.toolName\n    }\n  }\n\n  if (chunk.type === EventType.RUN_ERROR && chunk.error != null) {\n    if (specChunk.message === undefined && chunk.error.message) {\n      specChunk.message = chunk.error.message\n    }\n    if (specChunk.code === undefined && chunk.error.code !== undefined) {\n      specChunk.code = chunk.error.code\n    }\n  }\n\n  const tanstack: MetadataRecord = {}\n\n  if (\n    chunk.model !== undefined &&\n    chunk.type !== EventType.TEXT_MESSAGE_CONTENT &&\n    chunk.type !== EventType.TOOL_CALL_ARGS\n  ) {\n    tanstack.model = chunk.model\n  }\n\n  if (chunk.finishReason !== undefined) {\n    tanstack.finishReason = chunk.finishReason\n  }\n\n  const interruptErrors = chunk['tanstack:interruptErrors']\n  if (interruptErrors !== undefined) {\n    tanstack.interruptErrors = interruptErrors\n  }\n\n  if (chunk.type === EventType.CUSTOM || chunk.type === EventType.RUN_ERROR) {\n    if (chunk.threadId !== undefined) {\n      tanstack.threadId = chunk.threadId\n    }\n    if (chunk.runId !== undefined) {\n      tanstack.runId = chunk.runId\n    }\n  }\n\n  const skipLeftover = new Set(['result', 'error', 'tanstack:interruptErrors'])\n  if (\n    chunk.type === EventType.TEXT_MESSAGE_CONTENT ||\n    chunk.type === EventType.TOOL_CALL_ARGS\n  ) {\n    skipLeftover.add('model')\n    skipLeftover.add('content')\n    skipLeftover.add('args')\n  }\n  if (chunk.type === EventType.TOOL_CALL_START) {\n    skipLeftover.add('toolName')\n  }\n  for (const key of Object.keys(chunk)) {\n    if (specKeys.has(key) || skipLeftover.has(key)) continue\n    if (tanstack[key] !== undefined) continue\n    const value = source[key]\n    if (value !== undefined) {\n      tanstack[key] = value\n    }\n  }\n\n  if (\n    (chunk.type === EventType.RUN_FINISHED ||\n      chunk.type === EventType.RUN_ERROR) &&\n    isTanstackUsage(specChunk.usage)\n  ) {\n    const { usage, leftover } = toSpecTokenUsage(specChunk.usage, {\n      model: typeof chunk.model === 'string' ? chunk.model : undefined,\n    })\n    specChunk.usage = usage\n    if (leftover !== undefined) {\n      tanstack.usage = leftover\n    }\n  }\n\n  const normalized =\n    Object.keys(tanstack).length === 0\n      ? specChunk\n      : withTanstackMetadata(specChunk, tanstack)\n\n  const extras = encryptedValueExtras(chunk)\n  const main: Array<StreamChunk> = [normalized as StreamChunk]\n\n  if (chunk.type === EventType.TOOL_CALL_END && chunk.result !== undefined) {\n    const parentMessageId = source.parentMessageId\n    const resultChunk: ToolCallResultEvent = {\n      type: EventType.TOOL_CALL_RESULT,\n      toolCallId: chunk.toolCallId,\n      content: Array.isArray(chunk.result)\n        ? JSON.stringify(chunk.result)\n        : chunk.result,\n      messageId:\n        typeof parentMessageId === 'string' && parentMessageId !== ''\n          ? parentMessageId\n          : chunk.toolCallId,\n    }\n    if (chunk.state === 'output-error') {\n      main.push({\n        ...resultChunk,\n        metadata: { tanstack: { state: chunk.state } },\n      })\n    } else {\n      main.push(resultChunk)\n    }\n  }\n\n  return [...main, ...extras]\n}\n"],"mappings":";;;;;;AASA,SAAS,YAAY,OAAoC;CACvD,OAAO,OAAO,UAAU,YAAY,UAAU,KAAK,QAAQ,KAAA;AAC7D;AAEA,SAAS,qBAAqB,OAA8C;CAC1E,MAAM,SAA6B,CAAC;CACpC,MAAM,YACJ,eAAe,SAAS,OAAO,MAAM,cAAc,WAC/C,MAAM,YACN,KAAA;CAEN,IAAI,OAAO,MAAM,cAAc,YAAY,MAAM,cAAc,IAAI;EACjE,MAAM,SAAS;EACf,MAAM,aAAa,YAAY,OAAO,UAAU;EAChD,MAAM,WACJ,YAAY,MAAM,MAAM,KACxB,YAAY,OAAO,QAAQ,KAC3B,YAAY,OAAO,SAAS,KAC5B;EACF,IAAI,aAAa,KAAA,GACf,OAAO,KACL,wBAAwB;GACtB,SAAS,cAAc,CAAC,MAAM,SAAS,cAAc;GACrD;GACA,gBAAgB,MAAM;GACtB;EACF,CAAC,CACH;CAEJ;CAEA,IAAI,MAAM,SAAS,UAAU,iBAAiB;EAC5C,MAAM,mBAAmB,YACtB,MAAM,UACH,gBACN;EACA,IAAI,qBAAqB,KAAA,KAAa,MAAM,YAC1C,OAAO,KACL,wBAAwB;GACtB,SAAS;GACT,UAAU,MAAM;GAChB,gBAAgB;GAChB;EACF,CAAC,CACH;CAEJ;CAEA,OAAO;AACT;AAEA,SAAgB,qBACd,OACoB;CACpB,MAAM,WAAW,YAAY,MAAM,IAAI;CACvC,MAAM,SAAS;CACf,MAAM,YAEF,CAAC;CAEL,KAAK,MAAM,OAAO,OAAO,KAAK,KAAK,GACjC,IAAI,SAAS,IAAI,GAAG,GAClB,UAAU,OAAO,OAAO;CAI5B,IAAI,MAAM,SAAS,UAAU,iBACvB;MAAA,UAAU,iBAAiB,KAAA,KAAa,MAAM,UAChD,UAAU,eAAe,MAAM;CAAA;CAInC,IAAI,MAAM,SAAS,UAAU,aAAa,MAAM,SAAS,MAAM;EAC7D,IAAI,UAAU,YAAY,KAAA,KAAa,MAAM,MAAM,SACjD,UAAU,UAAU,MAAM,MAAM;EAElC,IAAI,UAAU,SAAS,KAAA,KAAa,MAAM,MAAM,SAAS,KAAA,GACvD,UAAU,OAAO,MAAM,MAAM;CAEjC;CAEA,MAAM,WAA2B,CAAC;CAElC,IACE,MAAM,UAAU,KAAA,KAChB,MAAM,SAAS,UAAU,wBACzB,MAAM,SAAS,UAAU,gBAEzB,SAAS,QAAQ,MAAM;CAGzB,IAAI,MAAM,iBAAiB,KAAA,GACzB,SAAS,eAAe,MAAM;CAGhC,MAAM,kBAAkB,MAAM;CAC9B,IAAI,oBAAoB,KAAA,GACtB,SAAS,kBAAkB;CAG7B,IAAI,MAAM,SAAS,UAAU,UAAU,MAAM,SAAS,UAAU,WAAW;EACzE,IAAI,MAAM,aAAa,KAAA,GACrB,SAAS,WAAW,MAAM;EAE5B,IAAI,MAAM,UAAU,KAAA,GAClB,SAAS,QAAQ,MAAM;CAE3B;CAEA,MAAM,+BAAe,IAAI,IAAI;EAAC;EAAU;EAAS;CAA0B,CAAC;CAC5E,IACE,MAAM,SAAS,UAAU,wBACzB,MAAM,SAAS,UAAU,gBACzB;EACA,aAAa,IAAI,OAAO;EACxB,aAAa,IAAI,SAAS;EAC1B,aAAa,IAAI,MAAM;CACzB;CACA,IAAI,MAAM,SAAS,UAAU,iBAC3B,aAAa,IAAI,UAAU;CAE7B,KAAK,MAAM,OAAO,OAAO,KAAK,KAAK,GAAG;EACpC,IAAI,SAAS,IAAI,GAAG,KAAK,aAAa,IAAI,GAAG,GAAG;EAChD,IAAI,SAAS,SAAS,KAAA,GAAW;EACjC,MAAM,QAAQ,OAAO;EACrB,IAAI,UAAU,KAAA,GACZ,SAAS,OAAO;CAEpB;CAEA,KACG,MAAM,SAAS,UAAU,gBACxB,MAAM,SAAS,UAAU,cAC3B,gBAAgB,UAAU,KAAK,GAC/B;EACA,MAAM,EAAE,OAAO,aAAa,iBAAiB,UAAU,OAAO,EAC5D,OAAO,OAAO,MAAM,UAAU,WAAW,MAAM,QAAQ,KAAA,EACzD,CAAC;EACD,UAAU,QAAQ;EAClB,IAAI,aAAa,KAAA,GACf,SAAS,QAAQ;CAErB;CAEA,MAAM,aACJ,OAAO,KAAK,QAAQ,CAAC,CAAC,WAAW,IAC7B,YACA,qBAAqB,WAAW,QAAQ;CAE9C,MAAM,SAAS,qBAAqB,KAAK;CACzC,MAAM,OAA2B,CAAC,UAAyB;CAE3D,IAAI,MAAM,SAAS,UAAU,iBAAiB,MAAM,WAAW,KAAA,GAAW;EACxE,MAAM,kBAAkB,OAAO;EAC/B,MAAM,cAAmC;GACvC,MAAM,UAAU;GAChB,YAAY,MAAM;GAClB,SAAS,MAAM,QAAQ,MAAM,MAAM,IAC/B,KAAK,UAAU,MAAM,MAAM,IAC3B,MAAM;GACV,WACE,OAAO,oBAAoB,YAAY,oBAAoB,KACvD,kBACA,MAAM;EACd;EACA,IAAI,MAAM,UAAU,gBAClB,KAAK,KAAK;GACR,GAAG;GACH,UAAU,EAAE,UAAU,EAAE,OAAO,MAAM,MAAM,EAAE;EAC/C,CAAC;OAED,KAAK,KAAK,WAAW;CAEzB;CAEA,OAAO,CAAC,GAAG,MAAM,GAAG,MAAM;AAC5B"}

package/dist/esm/utilities/reasoning-encrypted-value.d.ts

@@ -0,0 +1,8 @@
+import { ReasoningEncryptedValueEvent } from '../types.js';
+/** Spec event that carries a provider thinking / tool-call signature blob. */
+export declare function reasoningEncryptedValue(opts: {
+    subtype: 'message' | 'tool-call';
+    entityId: string;
+    encryptedValue: string;
+    timestamp?: number;
+}): ReasoningEncryptedValueEvent;

package/dist/esm/utilities/reasoning-encrypted-value.js

@@ -0,0 +1,16 @@
+import { EventType } from "../types.js";
+//#region src/utilities/reasoning-encrypted-value.ts
+/** Spec event that carries a provider thinking / tool-call signature blob. */
+function reasoningEncryptedValue(opts) {
+	return {
+		type: EventType.REASONING_ENCRYPTED_VALUE,
+		subtype: opts.subtype,
+		entityId: opts.entityId,
+		encryptedValue: opts.encryptedValue,
+		...opts.timestamp !== void 0 ? { timestamp: opts.timestamp } : {}
+	};
+}
+//#endregion
+export { reasoningEncryptedValue };
+
+//# sourceMappingURL=reasoning-encrypted-value.js.map

package/dist/esm/utilities/reasoning-encrypted-value.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reasoning-encrypted-value.js","names":[],"sources":["../../../src/utilities/reasoning-encrypted-value.ts"],"sourcesContent":["import { EventType } from '../types'\nimport type { ReasoningEncryptedValueEvent } from '../types'\n\n/** Spec event that carries a provider thinking / tool-call signature blob. */\nexport function reasoningEncryptedValue(opts: {\n  subtype: 'message' | 'tool-call'\n  entityId: string\n  encryptedValue: string\n  timestamp?: number\n}): ReasoningEncryptedValueEvent {\n  return {\n    type: EventType.REASONING_ENCRYPTED_VALUE,\n    subtype: opts.subtype,\n    entityId: opts.entityId,\n    encryptedValue: opts.encryptedValue,\n    ...(opts.timestamp !== undefined ? { timestamp: opts.timestamp } : {}),\n  }\n}\n"],"mappings":";;;AAIA,SAAgB,wBAAwB,MAKP;CAC/B,OAAO;EACL,MAAM,UAAU;EAChB,SAAS,KAAK;EACd,UAAU,KAAK;EACf,gBAAgB,KAAK;EACrB,GAAI,KAAK,cAAc,KAAA,IAAY,EAAE,WAAW,KAAK,UAAU,IAAI,CAAC;CACtE;AACF"}

package/dist/esm/utilities/restore-inbound-chunk.d.ts

@@ -0,0 +1,15 @@
+import { StreamChunk } from '../types.js';
+import { AdapterYieldChunk } from './adapter-yield-chunk.js';
+/**
+ * Rebuild TanStack `TokenUsage` (`promptTokens`) from spec `usage[]` plus
+ * leftover fields in `metadata.tanstack.usage`. Restore in-process aliases
+ * that the wire does not keep (`toolName`, `TOOL_CALL_END.input`). Mutates
+ * in place so WeakMap run-id stamps stay attached.
+ */
+export declare function restorePublicUsage(chunk: StreamChunk): StreamChunk;
+/**
+ * Rebuild the pre-wire chunk shape after SSE/HTTP/WS ingest.
+ * Copies `metadata.tanstack` extras back to top-level fields and rebuilds
+ * TanStack `TokenUsage` from spec `usage[]` when present.
+ */
+export declare function restoreInboundChunk(chunk: StreamChunk): AdapterYieldChunk;

package/dist/esm/utilities/restore-inbound-chunk.js

@@ -0,0 +1,43 @@
+import { EventType } from "../types.js";
+import { isTanstackUsage, rebuildTokenUsage } from "./ag-ui-usage.js";
+import { tanstackMetadata } from "./merge-metadata.js";
+//#region src/utilities/restore-inbound-chunk.ts
+/**
+* Rebuild TanStack `TokenUsage` (`promptTokens`) from spec `usage[]` plus
+* leftover fields in `metadata.tanstack.usage`. Restore in-process aliases
+* that the wire does not keep (`toolName`, `TOOL_CALL_END.input`). Mutates
+* in place so WeakMap run-id stamps stay attached.
+*/
+function restorePublicUsage(chunk) {
+	if ((chunk.type === EventType.RUN_FINISHED || chunk.type === EventType.RUN_ERROR) && (Array.isArray(chunk.usage) || isTanstackUsage(chunk.usage))) {
+		const rebuilt = rebuildTokenUsage(chunk.usage, tanstackMetadata(chunk)?.usage);
+		if (rebuilt !== void 0) chunk.usage = rebuilt;
+	}
+	if (chunk.type === EventType.TOOL_CALL_START && chunk.toolName === void 0 && chunk.toolCallName) chunk.toolName = chunk.toolCallName;
+	if (chunk.type === EventType.TOOL_CALL_END && chunk.input === void 0) {
+		const input = tanstackMetadata(chunk)?.input;
+		if (input !== void 0) chunk.input = input;
+	}
+	return chunk;
+}
+/**
+* Rebuild the pre-wire chunk shape after SSE/HTTP/WS ingest.
+* Copies `metadata.tanstack` extras back to top-level fields and rebuilds
+* TanStack `TokenUsage` from spec `usage[]` when present.
+*/
+function restoreInboundChunk(chunk) {
+	restorePublicUsage(chunk);
+	const tanstack = tanstackMetadata(chunk);
+	const next = chunk;
+	if (tanstack == null) return next;
+	for (const [key, value] of Object.entries(tanstack)) {
+		if (key === "usage" || key === "interruptErrors") continue;
+		if (next[key] === void 0 && value !== void 0) next[key] = value;
+	}
+	if (tanstack.interruptErrors !== void 0) next["tanstack:interruptErrors"] = tanstack.interruptErrors;
+	return next;
+}
+//#endregion
+export { restoreInboundChunk, restorePublicUsage };
+
+//# sourceMappingURL=restore-inbound-chunk.js.map

package/dist/esm/utilities/restore-inbound-chunk.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"restore-inbound-chunk.js","names":[],"sources":["../../../src/utilities/restore-inbound-chunk.ts"],"sourcesContent":["import { EventType } from '../types'\nimport type { StreamChunk } from '../types'\nimport type { AdapterYieldChunk } from './adapter-yield-chunk'\nimport { isTanstackUsage, rebuildTokenUsage } from './ag-ui-usage'\nimport { tanstackMetadata } from './merge-metadata'\n\n/**\n * Rebuild TanStack `TokenUsage` (`promptTokens`) from spec `usage[]` plus\n * leftover fields in `metadata.tanstack.usage`. Restore in-process aliases\n * that the wire does not keep (`toolName`, `TOOL_CALL_END.input`). Mutates\n * in place so WeakMap run-id stamps stay attached.\n */\nexport function restorePublicUsage(chunk: StreamChunk): StreamChunk {\n  if (\n    (chunk.type === EventType.RUN_FINISHED ||\n      chunk.type === EventType.RUN_ERROR) &&\n    (Array.isArray(chunk.usage) || isTanstackUsage(chunk.usage))\n  ) {\n    const rebuilt = rebuildTokenUsage(\n      chunk.usage,\n      tanstackMetadata(chunk)?.usage,\n    )\n    if (rebuilt !== undefined) {\n      chunk.usage = rebuilt\n    }\n  }\n\n  if (\n    chunk.type === EventType.TOOL_CALL_START &&\n    chunk.toolName === undefined &&\n    chunk.toolCallName\n  ) {\n    chunk.toolName = chunk.toolCallName\n  }\n\n  if (chunk.type === EventType.TOOL_CALL_END && chunk.input === undefined) {\n    const input = tanstackMetadata(chunk)?.input\n    if (input !== undefined) {\n      chunk.input = input\n    }\n  }\n\n  return chunk\n}\n\n/**\n * Rebuild the pre-wire chunk shape after SSE/HTTP/WS ingest.\n * Copies `metadata.tanstack` extras back to top-level fields and rebuilds\n * TanStack `TokenUsage` from spec `usage[]` when present.\n */\nexport function restoreInboundChunk(chunk: StreamChunk): AdapterYieldChunk {\n  restorePublicUsage(chunk)\n  const tanstack = tanstackMetadata(chunk)\n  const next = chunk as AdapterYieldChunk & Record<string, unknown>\n\n  if (tanstack == null) {\n    return next\n  }\n\n  for (const [key, value] of Object.entries(tanstack)) {\n    if (key === 'usage' || key === 'interruptErrors') continue\n    if (next[key] === undefined && value !== undefined) {\n      next[key] = value\n    }\n  }\n\n  if (tanstack.interruptErrors !== undefined) {\n    next['tanstack:interruptErrors'] = tanstack.interruptErrors\n  }\n\n  return next\n}\n"],"mappings":";;;;;;;;;;AAYA,SAAgB,mBAAmB,OAAiC;CAClE,KACG,MAAM,SAAS,UAAU,gBACxB,MAAM,SAAS,UAAU,eAC1B,MAAM,QAAQ,MAAM,KAAK,KAAK,gBAAgB,MAAM,KAAK,IAC1D;EACA,MAAM,UAAU,kBACd,MAAM,OACN,iBAAiB,KAAK,CAAC,EAAE,KAC3B;EACA,IAAI,YAAY,KAAA,GACd,MAAM,QAAQ;CAElB;CAEA,IACE,MAAM,SAAS,UAAU,mBACzB,MAAM,aAAa,KAAA,KACnB,MAAM,cAEN,MAAM,WAAW,MAAM;CAGzB,IAAI,MAAM,SAAS,UAAU,iBAAiB,MAAM,UAAU,KAAA,GAAW;EACvE,MAAM,QAAQ,iBAAiB,KAAK,CAAC,EAAE;EACvC,IAAI,UAAU,KAAA,GACZ,MAAM,QAAQ;CAElB;CAEA,OAAO;AACT;;;;;;AAOA,SAAgB,oBAAoB,OAAuC;CACzE,mBAAmB,KAAK;CACxB,MAAM,WAAW,iBAAiB,KAAK;CACvC,MAAM,OAAO;CAEb,IAAI,YAAY,MACd,OAAO;CAGT,KAAK,MAAM,CAAC,KAAK,UAAU,OAAO,QAAQ,QAAQ,GAAG;EACnD,IAAI,QAAQ,WAAW,QAAQ,mBAAmB;EAClD,IAAI,KAAK,SAAS,KAAA,KAAa,UAAU,KAAA,GACvC,KAAK,OAAO;CAEhB;CAEA,IAAI,SAAS,oBAAoB,KAAA,GAC/B,KAAK,8BAA8B,SAAS;CAG9C,OAAO;AACT"}

package/dist/esm/utilities/spec-event-keys.d.ts

@@ -0,0 +1,2 @@
+export declare function specKeysFor(type: string): ReadonlySet<string>;
+export declare function isSpecTopLevelKey(type: string, key: string): boolean;

package/dist/esm/utilities/spec-event-keys.js

@@ -0,0 +1,56 @@
+import { EventType } from "../types.js";
+//#region src/utilities/spec-event-keys.ts
+var SHARED = /* @__PURE__ */ new Set([
+	"type",
+	"timestamp",
+	"rawEvent",
+	"metadata"
+]);
+function keys(...fields) {
+	return /* @__PURE__ */ new Set([...SHARED, ...fields]);
+}
+var SPEC_KEYS = /* @__PURE__ */ new Map([
+	[EventType.TEXT_MESSAGE_START, keys("messageId", "role", "name")],
+	[EventType.TEXT_MESSAGE_CONTENT, keys("messageId", "delta")],
+	[EventType.TEXT_MESSAGE_END, keys("messageId")],
+	[EventType.TEXT_MESSAGE_CHUNK, keys("messageId", "role", "delta", "name")],
+	[EventType.TOOL_CALL_START, keys("toolCallId", "toolCallName", "parentMessageId")],
+	[EventType.TOOL_CALL_ARGS, keys("toolCallId", "delta")],
+	[EventType.TOOL_CALL_END, keys("toolCallId")],
+	[EventType.TOOL_CALL_CHUNK, keys("toolCallId", "toolCallName", "parentMessageId", "delta")],
+	[EventType.TOOL_CALL_RESULT, keys("messageId", "toolCallId", "content", "role")],
+	[EventType.THINKING_START, keys("title")],
+	[EventType.THINKING_END, SHARED],
+	[EventType.THINKING_TEXT_MESSAGE_START, SHARED],
+	[EventType.THINKING_TEXT_MESSAGE_CONTENT, keys("delta")],
+	[EventType.THINKING_TEXT_MESSAGE_END, SHARED],
+	[EventType.STATE_SNAPSHOT, keys("snapshot")],
+	[EventType.STATE_DELTA, keys("delta")],
+	[EventType.MESSAGES_SNAPSHOT, keys("messages")],
+	[EventType.ACTIVITY_SNAPSHOT, keys("messageId", "activityType", "content", "replace")],
+	[EventType.ACTIVITY_DELTA, keys("messageId", "activityType", "patch")],
+	[EventType.RAW, keys("event", "source")],
+	[EventType.CUSTOM, keys("name", "value")],
+	[EventType.RUN_STARTED, keys("threadId", "runId", "parentRunId", "input")],
+	[EventType.RUN_FINISHED, keys("threadId", "runId", "result", "outcome", "usage")],
+	[EventType.RUN_ERROR, keys("message", "code", "usage")],
+	[EventType.STEP_STARTED, keys("stepName")],
+	[EventType.STEP_FINISHED, keys("stepName")],
+	[EventType.REASONING_START, keys("messageId")],
+	[EventType.REASONING_MESSAGE_START, keys("messageId", "role")],
+	[EventType.REASONING_MESSAGE_CONTENT, keys("messageId", "delta")],
+	[EventType.REASONING_MESSAGE_END, keys("messageId")],
+	[EventType.REASONING_MESSAGE_CHUNK, keys("messageId", "delta")],
+	[EventType.REASONING_END, keys("messageId")],
+	[EventType.REASONING_ENCRYPTED_VALUE, keys("subtype", "entityId", "encryptedValue")]
+]);
+function specKeysFor(type) {
+	return SPEC_KEYS.get(type) ?? SHARED;
+}
+function isSpecTopLevelKey(type, key) {
+	return specKeysFor(type).has(key);
+}
+//#endregion
+export { isSpecTopLevelKey, specKeysFor };
+
+//# sourceMappingURL=spec-event-keys.js.map

package/dist/esm/utilities/spec-event-keys.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"spec-event-keys.js","names":[],"sources":["../../../src/utilities/spec-event-keys.ts"],"sourcesContent":["import { EventType } from '../types'\n\nconst SHARED: ReadonlySet<string> = new Set([\n  'type',\n  'timestamp',\n  'rawEvent',\n  'metadata',\n])\n\nfunction keys(...fields: Array<string>): ReadonlySet<string> {\n  return new Set([...SHARED, ...fields])\n}\n\nconst SPEC_KEYS: ReadonlyMap<string, ReadonlySet<string>> = new Map([\n  [EventType.TEXT_MESSAGE_START, keys('messageId', 'role', 'name')],\n  [EventType.TEXT_MESSAGE_CONTENT, keys('messageId', 'delta')],\n  [EventType.TEXT_MESSAGE_END, keys('messageId')],\n  [EventType.TEXT_MESSAGE_CHUNK, keys('messageId', 'role', 'delta', 'name')],\n  [\n    EventType.TOOL_CALL_START,\n    keys('toolCallId', 'toolCallName', 'parentMessageId'),\n  ],\n  [EventType.TOOL_CALL_ARGS, keys('toolCallId', 'delta')],\n  [EventType.TOOL_CALL_END, keys('toolCallId')],\n  [\n    EventType.TOOL_CALL_CHUNK,\n    keys('toolCallId', 'toolCallName', 'parentMessageId', 'delta'),\n  ],\n  [\n    EventType.TOOL_CALL_RESULT,\n    keys('messageId', 'toolCallId', 'content', 'role'),\n  ],\n  [EventType.THINKING_START, keys('title')],\n  [EventType.THINKING_END, SHARED],\n  [EventType.THINKING_TEXT_MESSAGE_START, SHARED],\n  [EventType.THINKING_TEXT_MESSAGE_CONTENT, keys('delta')],\n  [EventType.THINKING_TEXT_MESSAGE_END, SHARED],\n  [EventType.STATE_SNAPSHOT, keys('snapshot')],\n  [EventType.STATE_DELTA, keys('delta')],\n  [EventType.MESSAGES_SNAPSHOT, keys('messages')],\n  [\n    EventType.ACTIVITY_SNAPSHOT,\n    keys('messageId', 'activityType', 'content', 'replace'),\n  ],\n  [EventType.ACTIVITY_DELTA, keys('messageId', 'activityType', 'patch')],\n  [EventType.RAW, keys('event', 'source')],\n  [EventType.CUSTOM, keys('name', 'value')],\n  [EventType.RUN_STARTED, keys('threadId', 'runId', 'parentRunId', 'input')],\n  [\n    EventType.RUN_FINISHED,\n    keys('threadId', 'runId', 'result', 'outcome', 'usage'),\n  ],\n  [EventType.RUN_ERROR, keys('message', 'code', 'usage')],\n  [EventType.STEP_STARTED, keys('stepName')],\n  [EventType.STEP_FINISHED, keys('stepName')],\n  [EventType.REASONING_START, keys('messageId')],\n  [EventType.REASONING_MESSAGE_START, keys('messageId', 'role')],\n  [EventType.REASONING_MESSAGE_CONTENT, keys('messageId', 'delta')],\n  [EventType.REASONING_MESSAGE_END, keys('messageId')],\n  [EventType.REASONING_MESSAGE_CHUNK, keys('messageId', 'delta')],\n  [EventType.REASONING_END, keys('messageId')],\n  [\n    EventType.REASONING_ENCRYPTED_VALUE,\n    keys('subtype', 'entityId', 'encryptedValue'),\n  ],\n])\n\nexport function specKeysFor(type: string): ReadonlySet<string> {\n  return SPEC_KEYS.get(type) ?? SHARED\n}\n\nexport function isSpecTopLevelKey(type: string, key: string): boolean {\n  return specKeysFor(type).has(key)\n}\n"],"mappings":";;AAEA,IAAM,yBAA8B,IAAI,IAAI;CAC1C;CACA;CACA;CACA;AACF,CAAC;AAED,SAAS,KAAK,GAAG,QAA4C;CAC3D,uBAAO,IAAI,IAAI,CAAC,GAAG,QAAQ,GAAG,MAAM,CAAC;AACvC;AAEA,IAAM,4BAAsD,IAAI,IAAI;CAClE,CAAC,UAAU,oBAAoB,KAAK,aAAa,QAAQ,MAAM,CAAC;CAChE,CAAC,UAAU,sBAAsB,KAAK,aAAa,OAAO,CAAC;CAC3D,CAAC,UAAU,kBAAkB,KAAK,WAAW,CAAC;CAC9C,CAAC,UAAU,oBAAoB,KAAK,aAAa,QAAQ,SAAS,MAAM,CAAC;CACzE,CACE,UAAU,iBACV,KAAK,cAAc,gBAAgB,iBAAiB,CACtD;CACA,CAAC,UAAU,gBAAgB,KAAK,cAAc,OAAO,CAAC;CACtD,CAAC,UAAU,eAAe,KAAK,YAAY,CAAC;CAC5C,CACE,UAAU,iBACV,KAAK,cAAc,gBAAgB,mBAAmB,OAAO,CAC/D;CACA,CACE,UAAU,kBACV,KAAK,aAAa,cAAc,WAAW,MAAM,CACnD;CACA,CAAC,UAAU,gBAAgB,KAAK,OAAO,CAAC;CACxC,CAAC,UAAU,cAAc,MAAM;CAC/B,CAAC,UAAU,6BAA6B,MAAM;CAC9C,CAAC,UAAU,+BAA+B,KAAK,OAAO,CAAC;CACvD,CAAC,UAAU,2BAA2B,MAAM;CAC5C,CAAC,UAAU,gBAAgB,KAAK,UAAU,CAAC;CAC3C,CAAC,UAAU,aAAa,KAAK,OAAO,CAAC;CACrC,CAAC,UAAU,mBAAmB,KAAK,UAAU,CAAC;CAC9C,CACE,UAAU,mBACV,KAAK,aAAa,gBAAgB,WAAW,SAAS,CACxD;CACA,CAAC,UAAU,gBAAgB,KAAK,aAAa,gBAAgB,OAAO,CAAC;CACrE,CAAC,UAAU,KAAK,KAAK,SAAS,QAAQ,CAAC;CACvC,CAAC,UAAU,QAAQ,KAAK,QAAQ,OAAO,CAAC;CACxC,CAAC,UAAU,aAAa,KAAK,YAAY,SAAS,eAAe,OAAO,CAAC;CACzE,CACE,UAAU,cACV,KAAK,YAAY,SAAS,UAAU,WAAW,OAAO,CACxD;CACA,CAAC,UAAU,WAAW,KAAK,WAAW,QAAQ,OAAO,CAAC;CACtD,CAAC,UAAU,cAAc,KAAK,UAAU,CAAC;CACzC,CAAC,UAAU,eAAe,KAAK,UAAU,CAAC;CAC1C,CAAC,UAAU,iBAAiB,KAAK,WAAW,CAAC;CAC7C,CAAC,UAAU,yBAAyB,KAAK,aAAa,MAAM,CAAC;CAC7D,CAAC,UAAU,2BAA2B,KAAK,aAAa,OAAO,CAAC;CAChE,CAAC,UAAU,uBAAuB,KAAK,WAAW,CAAC;CACnD,CAAC,UAAU,yBAAyB,KAAK,aAAa,OAAO,CAAC;CAC9D,CAAC,UAAU,eAAe,KAAK,WAAW,CAAC;CAC3C,CACE,UAAU,2BACV,KAAK,WAAW,YAAY,gBAAgB,CAC9C;AACF,CAAC;AAED,SAAgB,YAAY,MAAmC;CAC7D,OAAO,UAAU,IAAI,IAAI,KAAK;AAChC;AAEA,SAAgB,kBAAkB,MAAc,KAAsB;CACpE,OAAO,YAAY,IAAI,CAAC,CAAC,IAAI,GAAG;AAClC"}

package/dist/esm/utilities/structured-output-events.d.ts

@@ -1,11 +1,11 @@
-import { StreamChunk } from '../types.js';
+import { AdapterYieldChunk } from './adapter-yield-chunk.js';
 export declare function structuredOutputStartChunk(args: {
     messageId: string;
     model: string;
     threadId: string;
     runId: string;
     timestamp?: number;
-}): StreamChunk;
+}): AdapterYieldChunk;
 export declare function structuredOutputCompleteChunk(args: {
     messageId: string;
     model: string;
@@ -14,4 +14,4 @@ export declare function structuredOutputCompleteChunk(args: {
     object: unknown;
     raw: string;
     timestamp?: number;
-}): StreamChunk;
+}): AdapterYieldChunk;

package/dist/esm/utilities/structured-output-events.js.map

@@ -1 +1 @@
-{"version":3,"file":"structured-output-events.js","names":[],"sources":["../../../src/utilities/structured-output-events.ts"],"sourcesContent":["import { EventType } from '../types'\nimport type { StreamChunk } from '../types'\n\nexport function structuredOutputStartChunk(args: {\n  messageId: string\n  model: string\n  threadId: string\n  runId: string\n  timestamp?: number\n}): StreamChunk {\n  return {\n    type: EventType.CUSTOM,\n    name: 'structured-output.start',\n    value: { messageId: args.messageId },\n    model: args.model,\n    timestamp: args.timestamp ?? Date.now(),\n    threadId: args.threadId,\n    runId: args.runId,\n  }\n}\n\nexport function structuredOutputCompleteChunk(args: {\n  messageId: string\n  model: string\n  threadId: string\n  runId: string\n  object: unknown\n  raw: string\n  timestamp?: number\n}): StreamChunk {\n  return {\n    type: EventType.CUSTOM,\n    name: 'structured-output.complete',\n    value: {\n      object: args.object,\n      raw: args.raw,\n      messageId: args.messageId,\n    },\n    model: args.model,\n    timestamp: args.timestamp ?? Date.now(),\n    threadId: args.threadId,\n    runId: args.runId,\n  }\n}\n"],"mappings":";;AAGA,SAAgB,2BAA2B,MAM3B;CACd,OAAO;EACL,MAAM,UAAU;EAChB,MAAM;EACN,OAAO,EAAE,WAAW,KAAK,UAAU;EACnC,OAAO,KAAK;EACZ,WAAW,KAAK,aAAa,KAAK,IAAI;EACtC,UAAU,KAAK;EACf,OAAO,KAAK;CACd;AACF;AAEA,SAAgB,8BAA8B,MAQ9B;CACd,OAAO;EACL,MAAM,UAAU;EAChB,MAAM;EACN,OAAO;GACL,QAAQ,KAAK;GACb,KAAK,KAAK;GACV,WAAW,KAAK;EAClB;EACA,OAAO,KAAK;EACZ,WAAW,KAAK,aAAa,KAAK,IAAI;EACtC,UAAU,KAAK;EACf,OAAO,KAAK;CACd;AACF"}
+{"version":3,"file":"structured-output-events.js","names":[],"sources":["../../../src/utilities/structured-output-events.ts"],"sourcesContent":["import { EventType } from '../types'\nimport type { AdapterYieldChunk } from './adapter-yield-chunk'\n\nexport function structuredOutputStartChunk(args: {\n  messageId: string\n  model: string\n  threadId: string\n  runId: string\n  timestamp?: number\n}): AdapterYieldChunk {\n  return {\n    type: EventType.CUSTOM,\n    name: 'structured-output.start',\n    value: { messageId: args.messageId },\n    model: args.model,\n    timestamp: args.timestamp ?? Date.now(),\n    threadId: args.threadId,\n    runId: args.runId,\n  }\n}\n\nexport function structuredOutputCompleteChunk(args: {\n  messageId: string\n  model: string\n  threadId: string\n  runId: string\n  object: unknown\n  raw: string\n  timestamp?: number\n}): AdapterYieldChunk {\n  return {\n    type: EventType.CUSTOM,\n    name: 'structured-output.complete',\n    value: {\n      object: args.object,\n      raw: args.raw,\n      messageId: args.messageId,\n    },\n    model: args.model,\n    timestamp: args.timestamp ?? Date.now(),\n    threadId: args.threadId,\n    runId: args.runId,\n  }\n}\n"],"mappings":";;AAGA,SAAgB,2BAA2B,MAMrB;CACpB,OAAO;EACL,MAAM,UAAU;EAChB,MAAM;EACN,OAAO,EAAE,WAAW,KAAK,UAAU;EACnC,OAAO,KAAK;EACZ,WAAW,KAAK,aAAa,KAAK,IAAI;EACtC,UAAU,KAAK;EACf,OAAO,KAAK;CACd;AACF;AAEA,SAAgB,8BAA8B,MAQxB;CACpB,OAAO;EACL,MAAM,UAAU;EAChB,MAAM;EACN,OAAO;GACL,QAAQ,KAAK;GACb,KAAK,KAAK;GACV,WAAW,KAAK;EAClB;EACA,OAAO,KAAK;EACZ,WAAW,KAAK,aAAa,KAAK,IAAI;EACtC,UAAU,KAAK;EACf,OAAO,KAAK;CACd;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai",
-  "version": "0.47.3",
+  "version": "0.48.0",
   "description": "Type-safe TypeScript AI SDK for streaming chat, tool calling, agents, structured outputs, and multimodal generation.",
   "author": "Tanner Linsley",
   "license": "MIT",
@@ -80,7 +80,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.9.0",
+    "@tanstack/ai-event-client": "^0.10.0",
     "@tanstack/ai-utils": "^0.4.0"
   },
   "peerDependencies": {

package/src/activities/chat/adapter.ts

@@ -2,10 +2,10 @@ import type {
   DefaultMessageMetadataByModality,
   JSONSchema,
   Modality,
-  StreamChunk,
   TextOptions,
   TokenUsage,
 } from '../../types'
+import type { AdapterYieldChunk } from '../../utilities/adapter-yield-chunk'
 import type { CapabilityHandle } from './middleware/capabilities'
 
 /**
@@ -106,7 +106,7 @@ export interface TextAdapter<
    */
   chatStream: (
     options: TextOptions<TProviderOptions>,
-  ) => AsyncIterable<StreamChunk>
+  ) => AsyncIterable<AdapterYieldChunk>
 
   /**
    * Generate structured output using the provider's native structured output API.
@@ -136,7 +136,7 @@ export interface TextAdapter<
    */
   structuredOutputStream?: (
     options: StructuredOutputOptions<TProviderOptions>,
-  ) => AsyncIterable<StreamChunk>
+  ) => AsyncIterable<AdapterYieldChunk>
 
   /**
    * Declares whether the adapter supports combining `tools` and a
@@ -229,7 +229,7 @@ export abstract class BaseTextAdapter<
 
   abstract chatStream(
     options: TextOptions<TProviderOptions>,
-  ): AsyncIterable<StreamChunk>
+  ): AsyncIterable<AdapterYieldChunk>
 
   /**
    * Generate structured output using the provider's native structured output API.