@mastra/ai-sdk 1.7.1-alpha.0 → 1.7.1-alpha.1

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @mastra/ai-sdk
 
+## 1.7.1-alpha.1
+
+### Patch Changes
+
+- Fixed the AI SDK stream announcing a response message id that matches no stored message when a processor rotates the response message id before the model runs (for example observational memory sealing a buffer chunk). The stream's `start` chunk now announces the id the assistant response is actually persisted under, so `useChat` and other AI SDK consumers can reconcile streamed messages with memory. Fixes [#19810](https://github.com/mastra-ai/mastra/issues/19810) ([#20084](https://github.com/mastra-ai/mastra/pull/20084))
+
+- Updated dependencies [[`4844167`](https://github.com/mastra-ai/mastra/commit/4844167cff2d5ec5004e94edd34970833040fa3f), [`5faf93f`](https://github.com/mastra-ai/mastra/commit/5faf93f03e19daea394b9e2a923f2e4f833407f2), [`80ad891`](https://github.com/mastra-ai/mastra/commit/80ad891f8cd10379aa5b5af7510c763783b2ab56), [`a1cb98d`](https://github.com/mastra-ai/mastra/commit/a1cb98d11990b560b98482292a1f34aa1a2d9092), [`598ad82`](https://github.com/mastra-ai/mastra/commit/598ad82d41c41389a686338a1d0e50b7400e1938), [`1fd6aad`](https://github.com/mastra-ai/mastra/commit/1fd6aad1ea4a9d32f65efa832307c35e981a4c0a)]:
+  - @mastra/core@1.56.0-alpha.4
+
 ## 1.7.1-alpha.0
 
 ### Patch Changes

package/dist/index.cjs

@@ -9437,52 +9437,88 @@ function createAgentStreamToAISDKTransformer(convertMastraChunkToAISDK, { lastMe
 	let bufferedSteps = /* @__PURE__ */ new Map();
 	let tripwireOccurred = false;
 	let finishEventSent = false;
+	let heldChunks = null;
+	let startIdResolved = false;
+	const processChunk = (chunk, controller) => {
+		if (chunk.type === "tripwire") tripwireOccurred = true;
+		if (chunk.type === "finish") finishEventSent = true;
+		if (chunk.type === "object-result") controller.enqueue({
+			type: "data-structured-output",
+			data: { object: chunk.object }
+		});
+		const part = convertMastraChunkToAISDK({
+			chunk,
+			mode: "stream"
+		});
+		const enqueueTransformedPart = (p) => {
+			const transformedChunk = convertFullStreamChunkToUIMessageStream({
+				part: p,
+				sendReasoning,
+				sendSources,
+				messageMetadataValue: p ? messageMetadata?.({ part: p }) : void 0,
+				sendStart,
+				sendFinish,
+				responseMessageId: lastMessageId,
+				onError(error) {
+					return onError ? onError(error) : safeParseErrorObject(error);
+				}
+			});
+			if (transformedChunk) if (transformedChunk.type === "tool-agent") {
+				const payload = transformedChunk.payload;
+				const agentTransformed = transformAgent(payload, bufferedSteps);
+				if (agentTransformed) controller.enqueue(agentTransformed);
+			} else if (transformedChunk.type === "tool-workflow") {
+				const payload = transformedChunk.payload;
+				const workflowChunk = transformWorkflow(payload, bufferedSteps, true, void 0, void 0, convertMastraChunkToAISDK);
+				if (workflowChunk) if (Array.isArray(workflowChunk)) for (const item of workflowChunk) controller.enqueue(item);
+				else controller.enqueue(workflowChunk);
+			} else if (transformedChunk.type === "tool-network") {
+				const payload = transformedChunk.payload;
+				const networkChunk = transformNetwork(payload, bufferedSteps, true);
+				if (Array.isArray(networkChunk)) {
+					for (const c of networkChunk) if (c) controller.enqueue(c);
+				} else if (networkChunk) controller.enqueue(networkChunk);
+			} else controller.enqueue(transformedChunk);
+		};
+		if (Array.isArray(part)) for (const p of part) enqueueTransformedPart(p);
+		else enqueueTransformedPart(part);
+	};
 	return new stream_web.TransformStream({
 		transform(chunk, controller) {
-			if (chunk.type === "tripwire") tripwireOccurred = true;
-			if (chunk.type === "finish") finishEventSent = true;
-			if (chunk.type === "object-result") controller.enqueue({
-				type: "data-structured-output",
-				data: { object: chunk.object }
-			});
-			const part = convertMastraChunkToAISDK({
-				chunk,
-				mode: "stream"
-			});
-			const enqueueTransformedPart = (p) => {
-				const transformedChunk = convertFullStreamChunkToUIMessageStream({
-					part: p,
-					sendReasoning,
-					sendSources,
-					messageMetadataValue: p ? messageMetadata?.({ part: p }) : void 0,
-					sendStart,
-					sendFinish,
-					responseMessageId: lastMessageId,
-					onError(error) {
-						return onError ? onError(error) : safeParseErrorObject(error);
+			if (!startIdResolved) {
+				if (sendStart && chunk.type === "start") {
+					heldChunks = [chunk];
+					return;
+				}
+				if (heldChunks) {
+					if (typeof chunk.type === "string" && chunk.type.startsWith("data-")) {
+						heldChunks.push(chunk);
+						return;
 					}
-				});
-				if (transformedChunk) if (transformedChunk.type === "tool-agent") {
-					const payload = transformedChunk.payload;
-					const agentTransformed = transformAgent(payload, bufferedSteps);
-					if (agentTransformed) controller.enqueue(agentTransformed);
-				} else if (transformedChunk.type === "tool-workflow") {
-					const payload = transformedChunk.payload;
-					const workflowChunk = transformWorkflow(payload, bufferedSteps, true, void 0, void 0, convertMastraChunkToAISDK);
-					if (workflowChunk) if (Array.isArray(workflowChunk)) for (const item of workflowChunk) controller.enqueue(item);
-					else controller.enqueue(workflowChunk);
-				} else if (transformedChunk.type === "tool-network") {
-					const payload = transformedChunk.payload;
-					const networkChunk = transformNetwork(payload, bufferedSteps, true);
-					if (Array.isArray(networkChunk)) {
-						for (const c of networkChunk) if (c) controller.enqueue(c);
-					} else if (networkChunk) controller.enqueue(networkChunk);
-				} else controller.enqueue(transformedChunk);
-			};
-			if (Array.isArray(part)) for (const p of part) enqueueTransformedPart(p);
-			else enqueueTransformedPart(part);
+					startIdResolved = true;
+					const held = heldChunks;
+					heldChunks = null;
+					if (chunk.type === "step-start") {
+						const startPayload = held[0]?.payload;
+						const stepMessageId = chunk.payload?.messageId;
+						if (held[0] && typeof stepMessageId === "string" && typeof startPayload?.messageId === "string" && stepMessageId !== startPayload.messageId) held[0] = {
+							...held[0],
+							payload: {
+								...startPayload,
+								messageId: stepMessageId
+							}
+						};
+					}
+					for (const heldChunk of held) processChunk(heldChunk, controller);
+				} else startIdResolved = true;
+			}
+			processChunk(chunk, controller);
 		},
 		flush(controller) {
+			if (heldChunks) {
+				for (const heldChunk of heldChunks) processChunk(heldChunk, controller);
+				heldChunks = null;
+			}
 			if (tripwireOccurred && !finishEventSent && sendFinish) controller.enqueue({
 				type: "finish",
 				finishReason: "other"