@tanstack/ai 0.53.0 → 0.55.0

package/README.md

@@ -202,19 +202,20 @@ Learn more in the
 
 Official adapters include:
 
-| Package                                                                              | Use it for                                                                     |
-| ------------------------------------------------------------------------------------ | ------------------------------------------------------------------------------ |
-| [`@tanstack/ai-openrouter`](https://tanstack.com/ai/latest/docs/adapters/openrouter) | 300+ models through one OpenRouter API, with per-request cost tracking         |
-| [`@tanstack/ai-openai`](https://tanstack.com/ai/latest/docs/adapters/openai)         | OpenAI chat, image, video, speech, transcription, realtime, and provider tools |
-| [`@tanstack/ai-anthropic`](https://tanstack.com/ai/latest/docs/adapters/anthropic)   | Anthropic Claude chat, thinking, tools, structured outputs, and Vertex Claude  |
-| [`@tanstack/ai-gemini`](https://tanstack.com/ai/latest/docs/adapters/gemini)         | Google Gemini chat, image, speech, and audio generation                        |
-| [`@tanstack/ai-vertex`](https://tanstack.com/ai/latest/docs/adapters/vertex)         | Gemini on Vertex AI with regional endpoints and Google Cloud credentials       |
-| [`@tanstack/ai-ollama`](https://tanstack.com/ai/latest/docs/adapters/ollama)         | Local Ollama models                                                            |
-| [`@tanstack/ai-grok`](https://tanstack.com/ai/latest/docs/adapters/grok)             | xAI Grok chat, images, and realtime                                            |
-| [`@tanstack/ai-groq`](https://tanstack.com/ai/latest/docs/adapters/groq)             | Groq low-latency inference                                                     |
-| [`@tanstack/ai-elevenlabs`](https://tanstack.com/ai/latest/docs/adapters/elevenlabs) | ElevenLabs realtime voice, speech, transcription, music, and sound effects     |
-| [`@tanstack/ai-byteplus`](https://tanstack.com/ai/latest/docs/adapters/byteplus)     | BytePlus Seed chat, Seedance video, Seedream image, and Seed Speech TTS/ASR    |
-| [`@tanstack/ai-fal`](https://tanstack.com/ai/latest/docs/adapters/fal)               | fal.ai image, video, audio, speech, and transcription models                   |
+| Package                                                                              | Use it for                                                                           |
+| ------------------------------------------------------------------------------------ | ------------------------------------------------------------------------------------ |
+| [`@tanstack/ai-openrouter`](https://tanstack.com/ai/latest/docs/adapters/openrouter) | 300+ models through one OpenRouter API, with per-request cost tracking               |
+| [`@tanstack/ai-openai`](https://tanstack.com/ai/latest/docs/adapters/openai)         | OpenAI chat, image, video, speech, transcription, realtime, and provider tools       |
+| [`@tanstack/ai-anthropic`](https://tanstack.com/ai/latest/docs/adapters/anthropic)   | Anthropic Claude chat, thinking, tools, structured outputs, and Vertex Claude        |
+| [`@tanstack/ai-gemini`](https://tanstack.com/ai/latest/docs/adapters/gemini)         | Google Gemini chat, image, speech, and audio generation                              |
+| [`@tanstack/ai-vertex`](https://tanstack.com/ai/latest/docs/adapters/vertex)         | Gemini on Vertex AI with regional endpoints and Google Cloud credentials             |
+| [`@tanstack/ai-ollama`](https://tanstack.com/ai/latest/docs/adapters/ollama)         | Local Ollama models                                                                  |
+| [`@tanstack/ai-grok`](https://tanstack.com/ai/latest/docs/adapters/grok)             | xAI Grok chat, images, and realtime                                                  |
+| [`@tanstack/ai-groq`](https://tanstack.com/ai/latest/docs/adapters/groq)             | Groq low-latency inference                                                           |
+| [`@tanstack/ai-elevenlabs`](https://tanstack.com/ai/latest/docs/adapters/elevenlabs) | ElevenLabs realtime voice, speech, transcription, music, and sound effects           |
+| [`@tanstack/ai-byteplus`](https://tanstack.com/ai/latest/docs/adapters/byteplus)     | BytePlus Seed chat, Seedance video, Seedream image, and Seed Speech TTS/ASR          |
+| [`@tanstack/ai-fal`](https://tanstack.com/ai/latest/docs/adapters/fal)               | fal.ai image, video, audio, speech, and transcription models                         |
+| [`@tanstack/ai-cloudflare`](https://tanstack.com/ai/latest/docs/adapters/cloudflare) | Cloudflare Workers AI chat, embeddings, image, speech, transcription, and AI Gateway |
 
 The adapter system is tree-shakeable by activity. Import `openaiText` for chat,
 `openaiImage` for images, `falVideo` for video, `geminiSpeech` for TTS, and so

package/dist/esm/activities/chat/index.js

@@ -18,7 +18,7 @@ import { INTERRUPT_PAYLOAD_METADATA_KEY, createInterruptBinding, rehydrateInterr
 import { readGenericInterruptContinuation } from "../../generic-interrupt-continuation.js";
 import { normalizeToolResult } from "../../utilities/tool-result.js";
 import { isProviderExecutedToolCall } from "../../utilities/provider-executed.js";
-import { convertMessagesToModelMessages, generateMessageId, modelMessagesToUIMessages, safeJsonStringify } from "./messages.js";
+import { appendUiResourceToModelMessages, convertMessagesToModelMessages, generateMessageId, modelMessagesToUIMessages, safeJsonStringify, uiResourcePartFromCustomValue } from "./messages.js";
 import { uiMessagesToWire } from "../../utilities/ag-ui-wire.js";
 import { restorePublicUsage } from "../../utilities/restore-inbound-chunk.js";
 import { LazyToolManager } from "./tools/lazy-tool-manager.js";
@@ -361,14 +361,16 @@ var TextEngine = class {
 					if (this.shouldExecuteToolPhase()) {
 						this.deferredToolCallRunFinishedChunks.push(...this.deferredModelRunFinishedChunks);
 						this.deferredModelRunFinishedChunks = [];
-					} else yield* this.flushDeferredModelRunFinishedChunks();
+					} else if (!this.finalStructuredOutput?.nativeCombined) yield* this.flushDeferredModelRunFinishedChunks();
 				} else yield* this.processToolCalls();
 				this.endCycle();
 			} while (yield* this.runWhileYielding(this.shouldContinue()));
 			this.logger.agentLoop("run finished", { finishReason: this.lastFinishReason });
 			if (this.finalStructuredOutput && this.toolPhase !== "wait" && !this.isCancelled() && !this.finalizationError && !this.earlyTermination) {
-				if (this.finalStructuredOutput.nativeCombined === true) yield* this.harvestCombinedStructuredOutput();
-				else yield* this.runStructuredFinalization();
+				if (this.finalStructuredOutput.nativeCombined === true) {
+					yield* this.harvestCombinedStructuredOutput();
+					if (!this.finalizationError && !this.isCancelled()) yield* this.flushDeferredModelRunFinishedChunks();
+				} else yield* this.runStructuredFinalization();
 			}
 			if (!this.terminalHookCalled && this.toolPhase !== "wait" && !this.isCancelled()) {
 				if (this.finalizationError) {
@@ -1205,6 +1207,21 @@ var TextEngine = class {
 			}
 		};
 	}
+	/**
+	* Record a `ui-resource` CUSTOM chunk on the assistant ModelMessage owning
+	* its `toolCallId` so the resource survives later MESSAGES_SNAPSHOT chunks
+	* (e.g. the interrupt snapshot emitted when the run pauses on a client
+	* tool). Mirrors the anchor-preserving approach used for
+	* `toolCallMetadata` (#867). See #1397.
+	*/
+	recordEmittedUiResource(value) {
+		const part = uiResourcePartFromCustomValue(value);
+		if (!part) return;
+		const next = appendUiResourceToModelMessages(this.messages, part);
+		if (next === this.messages) return;
+		this.messages = next;
+		this.middlewareCtx.messages = this.messages;
+	}
 	buildMessagesSnapshotChunk() {
 		const withIds = this.messages.map((message, index) => ({
 			...message,
@@ -2118,6 +2135,7 @@ var TextEngine = class {
 				if (this.hasPublicRunStarted) continue;
 				this.hasPublicRunStarted = true;
 			}
+			if (spec.type === EventType.CUSTOM && spec.name === "ui-resource") this.recordEmittedUiResource(spec.value);
 			yield spec;
 			this.middlewareCtx.chunkIndex++;
 		}