@tanstack/ai 0.54.0 → 0.55.0

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";
@@ -1207,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,
@@ -2120,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++;
 		}