@tanstack/ai 0.57.0 → 0.58.0

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

@@ -6,6 +6,7 @@ import { publishRunDisconnectHandler } from "../../delivery-disconnect.js";
 import { EventType } from "../../types.js";
 import { rebuildTokenUsage } from "../../utilities/ag-ui-usage.js";
 import { tanstackMetadata, withTanstackMetadata } from "../../utilities/merge-metadata.js";
+import { withDurabilityBatchHint } from "../../utilities/durability-batch.js";
 import { normalizeStreamChunk } from "../../utilities/normalize-stream-chunk.js";
 import { resolveDebugOption } from "../../logger/resolve.js";
 import { streamToText } from "../../stream-to-response.js";
@@ -228,8 +229,8 @@ var TextEngine = class {
 				this.abortReason = reason;
 				this.middlewareAbortController?.abort(reason);
 			},
-			emitCustomEvent: (name, value) => {
-				this.middlewareCustomQueue.push(this.createCustomEventChunk(name, value));
+			emitCustomEvent: (name, value, options) => {
+				this.middlewareCustomQueue.push(this.createCustomEventChunk(name, value, options));
 				const waiters = this.middlewareCustomWaiters;
 				this.middlewareCustomWaiters = [];
 				for (const waiter of waiters) waiter();
@@ -796,7 +797,7 @@ var TextEngine = class {
 			return "wait";
 		}
 		const { approvals, clientToolResults } = this.collectClientState();
-		const generator = executeToolCalls(executablePendingCalls, this.resolveExecutableTools(executablePendingCalls), approvals, clientToolResults, (eventName, data) => this.createCustomEventChunk(eventName, data), {
+		const generator = executeToolCalls(executablePendingCalls, this.resolveExecutableTools(executablePendingCalls), approvals, clientToolResults, (eventName, data, options) => this.createCustomEventChunk(eventName, data, options), {
 			onBeforeToolCall: async (toolCall, tool, args) => {
 				this.logger.tools(`phase=before name=${toolCall.function.name}`, {
 					name: toolCall.function.name,
@@ -888,7 +889,7 @@ var TextEngine = class {
 			return;
 		}
 		const { approvals, clientToolResults } = this.collectClientState();
-		const generator = executeToolCalls(executableToolCalls, this.resolveExecutableTools(executableToolCalls), approvals, clientToolResults, (eventName, data) => this.createCustomEventChunk(eventName, data), {
+		const generator = executeToolCalls(executableToolCalls, this.resolveExecutableTools(executableToolCalls), approvals, clientToolResults, (eventName, data, options) => this.createCustomEventChunk(eventName, data, options), {
 			onBeforeToolCall: async (toolCall, tool, args) => {
 				this.logger.tools(`phase=before name=${toolCall.function.name}`, {
 					name: toolCall.function.name,
@@ -2220,13 +2221,14 @@ var TextEngine = class {
 			pending = generator.next();
 		}
 	}
-	createCustomEventChunk(eventName, value) {
-		return {
+	createCustomEventChunk(eventName, value, options) {
+		const chunk = {
 			type: EventType.CUSTOM,
 			timestamp: Date.now(),
 			name: eventName,
 			value
 		};
+		return options?.batch ? withDurabilityBatchHint(chunk) : chunk;
 	}
 	createId(prefix) {
 		return `${prefix}-${Date.now()}-${Math.random().toString(36).slice(2, 9)}`;