@ai-sdk/harness 1.0.149 → 1.0.150

package/CHANGELOG.md

@@ -1,5 +1,16 @@
 # @ai-sdk/harness
 
+## 1.0.150
+
+### Patch Changes
+
+- 6617f11: fix(harness): preserve interleaved text and reasoning blocks by stream ID
+- aae69cb: fix(harness): invoke UI stream step callbacks for HarnessAgent results
+- a7a66f5: fix(harness): expose final-step reasoning through stream result accessors
+- 52e3f53: fix(harness): preserve response message order across steps
+- 6c0295c: fix(harness): preserve native tool result input and execution origin
+  - ai@7.0.139
+
 ## 1.0.149
 
 ### Patch Changes

package/dist/agent/index.js

@@ -251,6 +251,8 @@ var HarnessStreamTextResult = class {
 		this._providerMetadata = new DelayedPromise();
 		this.stepsBuffer = [];
 		this.currentStepContent = [];
+		this.currentStepTextContent = /* @__PURE__ */ new Map();
+		this.currentStepReasoningContent = /* @__PURE__ */ new Map();
 		this.currentStepWarnings = [];
 		this.stepNumber = 0;
 		this.stepStarted = false;
@@ -319,6 +321,8 @@ var HarnessStreamTextResult = class {
 	*/
 	discardCurrentStepContent() {
 		this.currentStepContent = [];
+		this.currentStepTextContent.clear();
+		this.currentStepReasoningContent.clear();
 		this.currentStepWarnings = [];
 		this.stepStarted = false;
 	}
@@ -372,6 +376,8 @@ var HarnessStreamTextResult = class {
 		if (this.currentStepWarnings.length > 0) this.aggregateWarnings.push(...this.currentStepWarnings);
 		this.stepNumber += 1;
 		this.currentStepContent = [];
+		this.currentStepTextContent.clear();
+		this.currentStepReasoningContent.clear();
 		this.currentStepWarnings = [];
 		this.stepStarted = false;
 		return step;
@@ -426,8 +432,8 @@ var HarnessStreamTextResult = class {
 		const aggregatedContent = this.stepsBuffer.flatMap((s) => s.content);
 		this._content.resolve(aggregatedContent);
 		this._text.resolve(finalStep.text);
-		this._reasoning.resolve([]);
-		this._reasoningText.resolve(void 0);
+		this._reasoning.resolve(finalStep.content.filter((part) => part.type === "reasoning" || part.type === "reasoning-file"));
+		this._reasoningText.resolve(finalStep.reasoningText);
 		this._files.resolve(this.stepsBuffer.flatMap((s) => s.files));
 		this._sources.resolve(this.stepsBuffer.flatMap((s) => s.sources));
 		this._toolCalls.resolve(this.stepsBuffer.flatMap((s) => s.toolCalls));
@@ -445,10 +451,11 @@ var HarnessStreamTextResult = class {
 		this._request.resolve(finalStep.request);
 		this._response.resolve(finalStep.response);
 		this._providerMetadata.resolve(this.finalProviderMetadata);
-		const responseMessages = await toResponseMessages({
-			content: aggregatedContent,
+		const responseMessages = [];
+		for (const step of this.stepsBuffer) responseMessages.push(...await toResponseMessages({
+			content: step.content,
 			tools: this.tools
-		});
+		}));
 		this._responseMessages.resolve(responseMessages);
 		this.fullStreamController.enqueue({
 			type: "finish",
@@ -640,12 +647,14 @@ var HarnessStreamTextResult = class {
 			reader.releaseLock();
 		}
 	}
-	toUIMessageStream({ originalMessages, generateMessageId, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendStart, sendFinish, onError } = {}) {
+	toUIMessageStream({ originalMessages, generateMessageId, onStepEnd, onStepFinish, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendStart, sendFinish, onError } = {}) {
 		return createAsyncIterableStream(toUIMessageStream({
 			stream: this.stream,
 			tools: this.tools,
 			originalMessages,
 			generateMessageId,
+			onStepEnd,
+			onStepFinish,
 			onEnd,
 			onFinish,
 			messageMetadata,
@@ -662,11 +671,13 @@ var HarnessStreamTextResult = class {
 	pipeTextStreamToResponse() {
 		throw notSupportedYet("pipeTextStreamToResponse");
 	}
-	toUIMessageStreamResponse({ originalMessages, generateMessageId, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendStart, sendFinish, onError, ...init } = {}) {
+	toUIMessageStreamResponse({ originalMessages, generateMessageId, onStepEnd, onStepFinish, onEnd, onFinish, messageMetadata, sendReasoning, sendSources, sendStart, sendFinish, onError, ...init } = {}) {
 		return createUIMessageStreamResponse({
 			stream: this.toUIMessageStream({
 				originalMessages,
 				generateMessageId,
+				onStepEnd,
+				onStepFinish,
 				onEnd,
 				onFinish,
 				messageMetadata,
@@ -684,22 +695,71 @@ var HarnessStreamTextResult = class {
 	}
 	appendToCurrentStepContent(part) {
 		switch (part.type) {
-			case "text-delta": {
-				const last = this.currentStepContent[this.currentStepContent.length - 1];
-				if (last && last.type === "text") last.text += part.text;
-				else this.currentStepContent.push({
+			case "text-start": {
+				const content = {
 					type: "text",
-					text: part.text
-				});
+					text: "",
+					...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+				};
+				this.currentStepTextContent.set(part.id, content);
+				this.currentStepContent.push(content);
 				return;
 			}
-			case "reasoning-delta": {
-				const last = this.currentStepContent[this.currentStepContent.length - 1];
-				if (last && last.type === "reasoning") last.text += part.text;
-				else this.currentStepContent.push({
+			case "text-delta": {
+				let content = this.currentStepTextContent.get(part.id);
+				if (content == null) {
+					content = {
+						type: "text",
+						text: "",
+						...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+					};
+					this.currentStepTextContent.set(part.id, content);
+					this.currentStepContent.push(content);
+				}
+				content.text += part.text;
+				if (part.providerMetadata != null) content.providerMetadata = part.providerMetadata;
+				return;
+			}
+			case "text-end": {
+				const content = this.currentStepTextContent.get(part.id);
+				if (content != null) {
+					if (part.providerMetadata != null) content.providerMetadata = part.providerMetadata;
+					if (part.citations != null) content.citations = part.citations;
+					this.currentStepTextContent.delete(part.id);
+				}
+				return;
+			}
+			case "reasoning-start": {
+				const content = {
 					type: "reasoning",
-					text: part.text
-				});
+					text: "",
+					...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+				};
+				this.currentStepReasoningContent.set(part.id, content);
+				this.currentStepContent.push(content);
+				return;
+			}
+			case "reasoning-delta": {
+				let content = this.currentStepReasoningContent.get(part.id);
+				if (content == null) {
+					content = {
+						type: "reasoning",
+						text: "",
+						...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+					};
+					this.currentStepReasoningContent.set(part.id, content);
+					this.currentStepContent.push(content);
+				}
+				content.text += part.text;
+				if (part.providerMetadata != null) content.providerMetadata = part.providerMetadata;
+				return;
+			}
+			case "reasoning-end": {
+				const content = this.currentStepReasoningContent.get(part.id);
+				if (content != null) {
+					if (part.providerMetadata != null) content.providerMetadata = part.providerMetadata;
+					this.currentStepReasoningContent.delete(part.id);
+				}
 				return;
 			}
 			case "tool-call":
@@ -869,11 +929,12 @@ function translateStreamPart(event, options = {}) {
 		case "tool-call": return [];
 		case "tool-approval-request": return [];
 		case "tool-result":
-			if (event.isError === true && (options.isProviderExecuted?.(event.toolCallId) ?? true)) return [{
+			const toolCall = options.getToolCall?.(event.toolCallId);
+			if (event.isError === true && (toolCall?.providerExecuted ?? options.isProviderExecuted?.(event.toolCallId) ?? true)) return [{
 				type: "tool-error",
 				toolCallId: event.toolCallId,
 				toolName: event.toolName,
-				input: void 0,
+				input: toolCall?.input,
 				error: event.result,
 				providerExecuted: true,
 				...event.dynamic !== void 0 ? { dynamic: event.dynamic } : {},
@@ -883,8 +944,9 @@ function translateStreamPart(event, options = {}) {
 				type: "tool-result",
 				toolCallId: event.toolCallId,
 				toolName: event.toolName,
-				input: void 0,
+				input: toolCall?.input,
 				output: event.result,
+				...toolCall?.providerExecuted !== void 0 ? { providerExecuted: toolCall.providerExecuted } : {},
 				...event.preliminary !== void 0 ? { preliminary: event.preliminary } : {},
 				...event.dynamic !== void 0 ? { dynamic: event.dynamic } : {},
 				...event.providerMetadata !== void 0 ? { providerMetadata: event.providerMetadata } : {}
@@ -1565,10 +1627,13 @@ function runPrompt(input) {
 		const stripToolInputWorkDir = createToolInputWorkDirStripper({ sessionWorkDir: input.sessionWorkDir });
 		const toolCallsByToolCallId = /* @__PURE__ */ new Map();
 		const rawToolCallsByToolCallId = /* @__PURE__ */ new Map();
-		const translateOptions = { isProviderExecuted: (toolCallId) => {
-			const rawToolCall = rawToolCallsByToolCallId.get(toolCallId);
-			return rawToolCall == null || rawToolCall.providerExecuted === true;
-		} };
+		const translateOptions = {
+			getToolCall: (toolCallId) => toolCallsByToolCallId.get(toolCallId),
+			isProviderExecuted: (toolCallId) => {
+				const rawToolCall = rawToolCallsByToolCallId.get(toolCallId);
+				return rawToolCall == null || rawToolCall.providerExecuted === true;
+			}
+		};
 		const pendingApprovalsByApprovalId = new Map(pendingToolApprovals.map((approval) => [approval.approvalId, approval]));
 		const pendingApprovalsByToolCallId = new Map(pendingToolApprovals.map((approval) => [approval.toolCallId, approval]));
 		const continuationsByApprovalId = new Map((input.toolApprovalContinuations ?? []).map((continuation) => [continuation.approvalId, continuation]));