@tanstack/ai 0.45.1 → 0.46.0

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

@@ -15,9 +15,10 @@ import { INTERRUPT_BINDING_METADATA_KEY, InterruptResumeValidationError, readUno
 import { normalizeToolResult } from "../../utilities/tool-result.js";
 import { isProviderExecutedToolCall } from "../../utilities/provider-executed.js";
 import { LazyToolManager } from "./tools/lazy-tool-manager.js";
+import { assertUniqueToolNames } from "./tools/unique-tool-names.js";
 import { MiddlewareAbortError, ToolCallManager, executeToolCalls } from "./tools/tool-calls.js";
 import { maxIterations } from "./agent-loop-strategies.js";
-import { convertMessagesToModelMessages, generateMessageId } from "./messages.js";
+import { convertMessagesToModelMessages, generateMessageId, modelMessageToUIMessage } from "./messages.js";
 import { MiddlewareRunner } from "./middleware/compose.js";
 import { provideSandboxRuntime } from "./middleware/sandbox-runtime.js";
 import { provideRunDisconnect } from "./middleware/run-disconnect.js";
@@ -148,6 +149,7 @@ var TextEngine = class {
 	accumulatedThinking = [];
 	currentThinkingContent = "";
 	currentThinkingSignature = "";
+	hasSeenReasoningEvents = false;
 	eventOptions;
 	eventToolNames;
 	finishedEvent = null;
@@ -202,6 +204,7 @@ var TextEngine = class {
 		this.initialApprovals = approvals;
 		this.initialClientToolResults = clientToolResults;
 		this.messages = convertMessagesToModelMessages(config.params.messages);
+		assertUniqueToolNames(config.params.tools || []);
 		this.lazyToolManager = new LazyToolManager(config.params.tools || [], this.messages, config.params.lazyToolsConfig);
 		this.tools = this.lazyToolManager.getActiveTools();
 		this.toolCallManager = new ToolCallManager(this.tools);
@@ -355,6 +358,7 @@ var TextEngine = class {
 						duration: Date.now() - this.streamStartTime
 					});
 				} else {
+					this.addTerminalReasoningMessage();
 					this.terminalHookCalled = true;
 					await this.middlewareRunner.runOnFinish(this.middlewareCtx, {
 						finishReason: this.lastFinishReason,
@@ -458,6 +462,7 @@ var TextEngine = class {
 		this.accumulatedThinking = [];
 		this.currentThinkingContent = "";
 		this.currentThinkingSignature = "";
+		this.hasSeenReasoningEvents = false;
 		this.finishedEvent = null;
 		this.streamedToolErrorResults.clear();
 		this.middlewareCtx.currentMessageId = this.currentMessageId;
@@ -593,7 +598,10 @@ var TextEngine = class {
 			case "STEP_STARTED":
 				this.handleStepStartedEvent();
 				break;
-			case "STEP_FINISHED": this.handleStepFinishedEvent(chunk);
+			case "STEP_FINISHED":
+				this.handleStepFinishedEvent(chunk);
+				break;
+			case "REASONING_MESSAGE_CONTENT": this.handleReasoningMessageContentEvent(chunk);
 		}
 	}
 	handleTextMessageContentEvent(chunk) {
@@ -657,9 +665,19 @@ var TextEngine = class {
 		this.finalizeCurrentThinkingStep();
 	}
 	handleStepFinishedEvent(chunk) {
-		if (chunk.delta) this.currentThinkingContent += chunk.delta;
+		if (!this.hasSeenReasoningEvents) {
+			if (chunk.delta) this.currentThinkingContent += chunk.delta;
+			else if (chunk.content) {
+				if (chunk.content.startsWith(this.currentThinkingContent)) this.currentThinkingContent = chunk.content;
+				else if (!this.currentThinkingContent.startsWith(chunk.content)) this.currentThinkingContent += chunk.content;
+			}
+		}
 		if (chunk.signature) this.currentThinkingSignature = chunk.signature;
 	}
+	handleReasoningMessageContentEvent(chunk) {
+		this.hasSeenReasoningEvents = true;
+		this.currentThinkingContent += chunk.delta;
+	}
 	/**
 	* Tools available for execution this turn. The discovery tool is dropped
 	* from the advertised set (`this.tools`) once every lazy tool is discovered,
@@ -871,6 +889,20 @@ var TextEngine = class {
 		}];
 		this.middlewareCtx.messages = this.messages;
 	}
+	addTerminalReasoningMessage() {
+		this.finalizeCurrentThinkingStep();
+		if (this.accumulatedThinking.length === 0) return;
+		const messages = this.middlewareCtx.messages;
+		if (messages.some((message) => message.role === "assistant" && message.id === this.currentMessageId)) return;
+		this.messages = [...messages, {
+			role: "assistant",
+			content: this.accumulatedContent || null,
+			id: this.currentMessageId ?? void 0,
+			createdAt: this.currentMessageCreatedAt ?? void 0,
+			thinking: this.accumulatedThinking
+		}];
+		this.middlewareCtx.messages = this.messages;
+	}
 	/**
 	* Extract client state (approvals and client tool results) from original messages.
 	* This is called in the constructor BEFORE converting to ModelMessage format,
@@ -980,10 +1012,13 @@ var TextEngine = class {
 	buildMessagesSnapshotChunk() {
 		const messages = this.messages.map((message, index) => {
 			const content = typeof message.content === "string" ? message.content : message.content === null ? void 0 : JSON.stringify(message.content);
+			const id = message.id || `snapshot_${this.runIdOverride ?? this.requestId}_${index}`;
+			const parts = message.role === "assistant" && message.thinking?.length ? modelMessageToUIMessage(message, id).parts : void 0;
 			return {
-				id: message.id || `snapshot_${this.runIdOverride ?? this.requestId}_${index}`,
+				id,
 				role: message.role,
 				...content !== void 0 ? { content } : {},
+				...parts ? { parts } : {},
 				..."toolCalls" in message && message.toolCalls ? { toolCalls: message.toolCalls } : {},
 				..."toolCallId" in message && message.toolCallId ? { toolCallId: message.toolCallId } : {}
 			};
@@ -1755,6 +1790,7 @@ var TextEngine = class {
 		this.applyResumeToolState(config.resumeToolState);
 		this.messages = config.messages;
 		this.systemPrompts = config.systemPrompts;
+		assertUniqueToolNames(config.tools);
 		this.tools = config.tools;
 		this.params = {
 			...this.params,
@@ -1879,6 +1915,7 @@ var TextEngine = class {
 */
 function chat(options) {
 	validateCapabilities(options.middleware ?? [], options.adapter);
+	if (options.tools) assertUniqueToolNames(options.tools);
 	const { outputSchema, stream } = options;
 	if (outputSchema && stream === true) return runStreamingStructuredOutput({
 		...options,