@mastra/ai-sdk 1.7.1 → 1.7.2

package/dist/index.js

@@ -9,7 +9,7 @@ import { MASTRA_RESOURCE_ID_KEY } from "@mastra/core/request-context";
 import { MessageList, TripWire, aiV5ModelMessageToV2PromptMessage } from "@mastra/core/agent";
 import { RequestContext } from "@mastra/core/di";
 import { MessageHistory, SemanticRecall, WorkingMemory } from "@mastra/core/processors";
-//#region ../../packages/_vendored/ai_v5/dist/dist-Qv-F35RT.js
+//#region ../../packages/_vendored/ai_v5/dist/dist-BuEMdYEn.js
 var marker$1$4 = "vercel.ai.error";
 var symbol$1$5 = Symbol.for(marker$1$4);
 var _a$1$5;
@@ -533,6 +533,26 @@ async function cancelResponseBody$1(response) {
 		await ((_a2 = response.body) == null ? void 0 : _a2.cancel());
 	} catch (e) {}
 }
+var initialGlobalFetch$1 = globalThis.fetch;
+isNodeDefaultFetch$1(initialGlobalFetch$1);
+function isNodeDefaultFetch$1(fetch) {
+	const source = Function.prototype.toString.call(fetch);
+	return source.includes("internal/deps/undici") || source.includes("lazy loading of undici");
+}
+function getCurrentModulePath$1() {
+	const originalPrepareStackTrace = Error.prepareStackTrace;
+	try {
+		Error.prepareStackTrace = (_error, callSites) => callSites;
+		const error = /* @__PURE__ */ new Error("Capture current module path");
+		Error.captureStackTrace(error, getCurrentModulePath$1);
+		const [caller] = error.stack;
+		const fileName = caller == null ? void 0 : caller.getFileName();
+		if (fileName == null) throw new Error("Unable to determine the current module path");
+		return fileName;
+	} finally {
+		Error.prepareStackTrace = originalPrepareStackTrace;
+	}
+}
 var DEFAULT_MAX_DOWNLOAD_SIZE$1 = 2 * 1024 * 1024 * 1024;
 async function readResponseWithSizeLimit$1({ response, url, maxBytes = DEFAULT_MAX_DOWNLOAD_SIZE$1 }) {
 	const contentLength = response.headers.get("content-length");
@@ -636,11 +656,11 @@ function withUserAgentSuffix$1(headers, ...userAgentSuffixParts) {
 	normalizedHeaders.set("user-agent", [currentUserAgentHeader, ...userAgentSuffixParts].filter(Boolean).join(" "));
 	return Object.fromEntries(normalizedHeaders.entries());
 }
-var VERSION$4 = "3.0.30";
+var VERSION$4 = "3.0.31";
 var getOriginalFetch = () => globalThis.fetch;
-var getFromApi = async ({ url, headers = {}, successfulResponseHandler, failedResponseHandler, abortSignal, fetch: fetch2 = getOriginalFetch() }) => {
+var getFromApi = async ({ url, headers = {}, successfulResponseHandler, failedResponseHandler, abortSignal, fetch = getOriginalFetch() }) => {
 	try {
-		const response = await fetch2(url, {
+		const response = await fetch(url, {
 			method: "GET",
 			headers: withUserAgentSuffix$1(headers, `ai-sdk/provider-utils/${VERSION$4}`, getRuntimeEnvironmentUserAgent$1()),
 			signal: abortSignal
@@ -866,7 +886,7 @@ function parseJsonEventStream({ stream, schema }) {
 	} }));
 }
 var getOriginalFetch2 = () => globalThis.fetch;
-var postJsonToApi = async ({ url, headers, body, failedResponseHandler, successfulResponseHandler, abortSignal, fetch: fetch2 }) => postToApi({
+var postJsonToApi = async ({ url, headers, body, failedResponseHandler, successfulResponseHandler, abortSignal, fetch }) => postToApi({
 	url,
 	headers: {
 		"Content-Type": "application/json",
@@ -879,11 +899,11 @@ var postJsonToApi = async ({ url, headers, body, failedResponseHandler, successf
 	failedResponseHandler,
 	successfulResponseHandler,
 	abortSignal,
-	fetch: fetch2
+	fetch
 });
-var postToApi = async ({ url, headers = {}, body, successfulResponseHandler, failedResponseHandler, abortSignal, fetch: fetch2 = getOriginalFetch2() }) => {
+var postToApi = async ({ url, headers = {}, body, successfulResponseHandler, failedResponseHandler, abortSignal, fetch = getOriginalFetch2() }) => {
 	try {
-		const response = await fetch2(url, {
+		const response = await fetch(url, {
 			method: "POST",
 			headers: withUserAgentSuffix$1(headers, `ai-sdk/provider-utils/${VERSION$4}`, getRuntimeEnvironmentUserAgent$1()),
 			body: body.content,
@@ -2945,7 +2965,7 @@ async function getVercelRequestId() {
 	var _a10;
 	return (_a10 = getContext().headers) == null ? void 0 : _a10["x-vercel-id"];
 }
-var VERSION$2 = "2.0.115";
+var VERSION$2 = "2.0.125";
 var AI_GATEWAY_PROTOCOL_VERSION = "0.0.1";
 function createGatewayProvider(options = {}) {
 	var _a10, _b10;
@@ -4395,19 +4415,48 @@ function processUIMessageStream$1({ stream, messageMetadataSchema, dataPartSchem
 	return stream.pipeThrough(new TransformStream({ async transform(chunk, controller) {
 		await runUpdateMessageJob(async ({ state, write }) => {
 			var _a16, _b, _c, _d;
+			function getCurrentStepParts() {
+				const parts = state.message.parts;
+				let currentStepStartIndex = parts.length - 1;
+				while (currentStepStartIndex >= 0 && parts[currentStepStartIndex].type !== "step-start") currentStepStartIndex--;
+				return parts.slice(currentStepStartIndex + 1);
+			}
+			function getCurrentStepToolInvocations() {
+				return getCurrentStepParts().filter(isToolUIPart$1);
+			}
 			function getToolInvocation(toolCallId) {
-				const toolInvocation = state.message.parts.filter(isToolUIPart$1).find((invocation) => invocation.toolCallId === toolCallId);
+				let toolInvocation = getCurrentStepToolInvocations().find((invocation) => invocation.toolCallId === toolCallId);
+				if (toolInvocation == null) {
+					const parts = state.message.parts;
+					for (let i = parts.length - 1; i >= 0; i--) {
+						const part = parts[i];
+						if (isToolUIPart$1(part) && part.toolCallId === toolCallId) {
+							toolInvocation = part;
+							break;
+						}
+					}
+				}
 				if (toolInvocation == null) throw new Error("tool-output-error must be preceded by a tool-input-available");
 				return toolInvocation;
 			}
 			function getDynamicToolInvocation(toolCallId) {
-				const toolInvocation = state.message.parts.filter((part) => part.type === "dynamic-tool").find((invocation) => invocation.toolCallId === toolCallId);
+				let toolInvocation = getCurrentStepParts().filter((part) => part.type === "dynamic-tool").find((invocation) => invocation.toolCallId === toolCallId);
+				if (toolInvocation == null) {
+					const parts = state.message.parts;
+					for (let i = parts.length - 1; i >= 0; i--) {
+						const part = parts[i];
+						if (part.type === "dynamic-tool" && part.toolCallId === toolCallId) {
+							toolInvocation = part;
+							break;
+						}
+					}
+				}
 				if (toolInvocation == null) throw new Error("tool-output-error must be preceded by a tool-input-available");
 				return toolInvocation;
 			}
-			function updateToolPart(options) {
+			function updateToolPart(options, existingPart) {
 				var _a17;
-				const part = state.message.parts.find((part2) => isToolUIPart$1(part2) && part2.toolCallId === options.toolCallId);
+				const part = existingPart != null ? existingPart : getCurrentStepParts().find((part2) => isToolUIPart$1(part2) && part2.toolCallId === options.toolCallId);
 				const anyOptions = options;
 				const anyPart = part;
 				if (part != null) {
@@ -4432,9 +4481,9 @@ function processUIMessageStream$1({ stream, messageMetadataSchema, dataPartSchem
 					...anyOptions.providerMetadata != null ? { callProviderMetadata: anyOptions.providerMetadata } : {}
 				});
 			}
-			function updateDynamicToolPart(options) {
+			function updateDynamicToolPart(options, existingPart) {
 				var _a17, _b2;
-				const part = state.message.parts.find((part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId);
+				const part = existingPart != null ? existingPart : getCurrentStepParts().find((part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId);
 				const anyOptions = options;
 				const anyPart = part;
 				if (part != null) {
@@ -4555,7 +4604,7 @@ function processUIMessageStream$1({ stream, messageMetadataSchema, dataPartSchem
 					write();
 					break;
 				case "tool-input-start": {
-					const toolInvocations = state.message.parts.filter(isToolUIPart$1);
+					const toolInvocations = getCurrentStepParts().filter(isToolUIPart$1);
 					state.partialToolCalls[chunk.toolCallId] = {
 						text: "",
 						toolName: chunk.toolName,
@@ -4650,7 +4699,7 @@ function processUIMessageStream$1({ stream, messageMetadataSchema, dataPartSchem
 							input: toolInvocation.input,
 							output: chunk.output,
 							preliminary: chunk.preliminary
-						});
+						}, toolInvocation);
 					} else {
 						const toolInvocation = getToolInvocation(chunk.toolCallId);
 						updateToolPart({
@@ -4661,7 +4710,7 @@ function processUIMessageStream$1({ stream, messageMetadataSchema, dataPartSchem
 							output: chunk.output,
 							providerExecuted: chunk.providerExecuted,
 							preliminary: chunk.preliminary
-						});
+						}, toolInvocation);
 					}
 					write();
 					break;
@@ -4675,7 +4724,7 @@ function processUIMessageStream$1({ stream, messageMetadataSchema, dataPartSchem
 							input: toolInvocation.input,
 							errorText: chunk.errorText,
 							providerExecuted: chunk.providerExecuted
-						});
+						}, toolInvocation);
 					} else {
 						const toolInvocation = getToolInvocation(chunk.toolCallId);
 						updateToolPart({
@@ -4686,7 +4735,7 @@ function processUIMessageStream$1({ stream, messageMetadataSchema, dataPartSchem
 							rawInput: toolInvocation.rawInput,
 							errorText: chunk.errorText,
 							providerExecuted: chunk.providerExecuted
-						});
+						}, toolInvocation);
 					}
 					write();
 					break;
@@ -4977,7 +5026,7 @@ function createUIMessageStream$1({ execute, onError = () => "An error occurred."
 	});
 }
 //#endregion
-//#region ../../packages/_vendored/ai_v6/dist/dist-BomuqvJJ.js
+//#region ../../packages/_vendored/ai_v6/dist/dist-BTWHzT8H.js
 var marker$1 = "vercel.ai.error";
 var symbol$1 = Symbol.for(marker$1);
 var _a$1;
@@ -5317,6 +5366,26 @@ var _b;
 		return AISDKError.hasMarker(error, marker$2);
 	}
 });
+var initialGlobalFetch = globalThis.fetch;
+isNodeDefaultFetch(initialGlobalFetch);
+function isNodeDefaultFetch(fetch) {
+	const source = Function.prototype.toString.call(fetch);
+	return source.includes("internal/deps/undici") || source.includes("lazy loading of undici");
+}
+function getCurrentModulePath() {
+	const originalPrepareStackTrace = Error.prepareStackTrace;
+	try {
+		Error.prepareStackTrace = (_error, callSites) => callSites;
+		const error = /* @__PURE__ */ new Error("Capture current module path");
+		Error.captureStackTrace(error, getCurrentModulePath);
+		const [caller] = error.stack;
+		const fileName = caller == null ? void 0 : caller.getFileName();
+		if (fileName == null) throw new Error("Unable to determine the current module path");
+		return fileName;
+	} finally {
+		Error.prepareStackTrace = originalPrepareStackTrace;
+	}
+}
 var createIdGenerator = ({ prefix, size = 16, alphabet = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz", separator = "-" } = {}) => {
 	const generator = () => {
 		const alphabetLength = alphabet.length;
@@ -7267,6 +7336,25 @@ _a16 = symbol16;
 function asArray(value) {
 	return value === void 0 ? [] : Array.isArray(value) ? value : [value];
 }
+function mergeObjects(base, overrides) {
+	if (base === void 0 && overrides === void 0) return;
+	if (base === void 0) return overrides;
+	if (overrides === void 0) return base;
+	const result = { ...base };
+	for (const key in overrides) {
+		if (key === "__proto__" || key === "constructor" || key === "prototype") continue;
+		if (Object.prototype.hasOwnProperty.call(overrides, key)) {
+			const overridesValue = overrides[key];
+			if (overridesValue === void 0) continue;
+			const baseValue = key in base ? base[key] : void 0;
+			const isSourceObject = overridesValue !== null && typeof overridesValue === "object" && !Array.isArray(overridesValue) && !(overridesValue instanceof Date) && !(overridesValue instanceof RegExp);
+			const isTargetObject = baseValue !== null && baseValue !== void 0 && typeof baseValue === "object" && !Array.isArray(baseValue) && !(baseValue instanceof Date) && !(baseValue instanceof RegExp);
+			if (isSourceObject && isTargetObject) result[key] = mergeObjects(baseValue, overridesValue);
+			else result[key] = overridesValue;
+		}
+	}
+	return result;
+}
 var dataContentSchema = z.union([
 	z.string(),
 	z.instanceof(Uint8Array),
@@ -7450,25 +7538,6 @@ z.union([
 	assistantModelMessageSchema,
 	toolModelMessageSchema
 ]);
-function mergeObjects(base, overrides) {
-	if (base === void 0 && overrides === void 0) return;
-	if (base === void 0) return overrides;
-	if (overrides === void 0) return base;
-	const result = { ...base };
-	for (const key in overrides) {
-		if (key === "__proto__" || key === "constructor" || key === "prototype") continue;
-		if (Object.prototype.hasOwnProperty.call(overrides, key)) {
-			const overridesValue = overrides[key];
-			if (overridesValue === void 0) continue;
-			const baseValue = key in base ? base[key] : void 0;
-			const isSourceObject = overridesValue !== null && typeof overridesValue === "object" && !Array.isArray(overridesValue) && !(overridesValue instanceof Date) && !(overridesValue instanceof RegExp);
-			const isTargetObject = baseValue !== null && baseValue !== void 0 && typeof baseValue === "object" && !Array.isArray(baseValue) && !(baseValue instanceof Date) && !(baseValue instanceof RegExp);
-			if (isSourceObject && isTargetObject) result[key] = mergeObjects(baseValue, overridesValue);
-			else result[key] = overridesValue;
-		}
-	}
-	return result;
-}
 new TextEncoder();
 __export({}, {
 	array: () => array,
@@ -8080,8 +8149,27 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 	return stream.pipeThrough(new TransformStream({ async transform(chunk, controller) {
 		await runUpdateMessageJob(async ({ state, write }) => {
 			var _a22, _b, _c, _d;
+			function getCurrentStepParts() {
+				const parts = state.message.parts;
+				let currentStepStartIndex = parts.length - 1;
+				while (currentStepStartIndex >= 0 && parts[currentStepStartIndex].type !== "step-start") currentStepStartIndex--;
+				return parts.slice(currentStepStartIndex + 1);
+			}
+			function getCurrentStepToolInvocations() {
+				return getCurrentStepParts().filter(isToolUIPart);
+			}
 			function getToolInvocation(toolCallId) {
-				const toolInvocation = state.message.parts.filter(isToolUIPart).find((invocation) => invocation.toolCallId === toolCallId);
+				let toolInvocation = getCurrentStepToolInvocations().find((invocation) => invocation.toolCallId === toolCallId);
+				if (toolInvocation == null) {
+					const parts = state.message.parts;
+					for (let i = parts.length - 1; i >= 0; i--) {
+						const part = parts[i];
+						if (isToolUIPart(part) && part.toolCallId === toolCallId) {
+							toolInvocation = part;
+							break;
+						}
+					}
+				}
 				if (toolInvocation == null) throw new UIMessageStreamError({
 					chunkType: "tool-invocation",
 					chunkId: toolCallId,
@@ -8089,9 +8177,9 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 				});
 				return toolInvocation;
 			}
-			function updateToolPart(options) {
+			function updateToolPart(options, existingPart) {
 				var _a23;
-				const part = state.message.parts.find((part2) => isStaticToolUIPart(part2) && part2.toolCallId === options.toolCallId);
+				const part = existingPart != null ? existingPart : getCurrentStepParts().find((part2) => isStaticToolUIPart(part2) && part2.toolCallId === options.toolCallId);
 				const anyOptions = options;
 				const anyPart = part;
 				if (part != null) {
@@ -8125,9 +8213,9 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 					...anyOptions.providerMetadata != null && !(options.state === "output-available" || options.state === "output-error") ? { callProviderMetadata: anyOptions.providerMetadata } : {}
 				});
 			}
-			function updateDynamicToolPart(options) {
+			function updateDynamicToolPart(options, existingPart) {
 				var _a23, _b2;
-				const part = state.message.parts.find((part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId);
+				const part = existingPart != null ? existingPart : getCurrentStepParts().find((part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId);
 				const anyOptions = options;
 				const anyPart = part;
 				if (part != null) {
@@ -8282,7 +8370,7 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 					write();
 					break;
 				case "tool-input-start": {
-					const toolInvocations = state.message.parts.filter(isStaticToolUIPart);
+					const toolInvocations = getCurrentStepParts().filter(isStaticToolUIPart);
 					state.partialToolCalls[chunk.toolCallId] = {
 						text: "",
 						toolName: chunk.toolName,
@@ -8367,7 +8455,7 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 					if (onToolCall && !chunk.providerExecuted) await onToolCall({ toolCall: chunk });
 					break;
 				case "tool-input-error": {
-					const existingPart = state.message.parts.filter(isToolUIPart).find((p) => p.toolCallId === chunk.toolCallId);
+					const existingPart = getCurrentStepParts().filter(isToolUIPart).find((p) => p.toolCallId === chunk.toolCallId);
 					if (existingPart != null ? existingPart.type === "dynamic-tool" : !!chunk.dynamic) updateDynamicToolPart({
 						toolCallId: chunk.toolCallId,
 						toolName: chunk.toolName,
@@ -8421,7 +8509,7 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 						providerMetadata: chunk.providerMetadata,
 						title: toolInvocation.title,
 						toolMetadata: toolInvocation.toolMetadata
-					});
+					}, toolInvocation);
 					else updateToolPart({
 						toolCallId: chunk.toolCallId,
 						toolName: getStaticToolName(toolInvocation),
@@ -8433,7 +8521,7 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 						providerMetadata: chunk.providerMetadata,
 						title: toolInvocation.title,
 						toolMetadata: toolInvocation.toolMetadata
-					});
+					}, toolInvocation);
 					write();
 					break;
 				}
@@ -8449,7 +8537,7 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 						providerMetadata: chunk.providerMetadata,
 						title: toolInvocation.title,
 						toolMetadata: toolInvocation.toolMetadata
-					});
+					}, toolInvocation);
 					else updateToolPart({
 						toolCallId: chunk.toolCallId,
 						toolName: getStaticToolName(toolInvocation),
@@ -8461,7 +8549,7 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 						providerMetadata: chunk.providerMetadata,
 						title: toolInvocation.title,
 						toolMetadata: toolInvocation.toolMetadata
-					});
+					}, toolInvocation);
 					write();
 					break;
 				}