@mastra/core 1.0.0-beta.4 → 1.0.0-beta.5

package/dist/{chunk-VSM3NLUX.cjs → chunk-DQIZ5FFX.cjs}

@@ -230,14 +230,137 @@ function convertImageFilePart(part, downloadedAssets) {
     }
   }
 }
+zod.z.union([
+  zod.z.string(),
+  zod.z.instanceof(Uint8Array),
+  zod.z.instanceof(ArrayBuffer),
+  zod.z.custom(
+    // Buffer might not be available in some environments such as CloudFlare:
+    (value) => globalThis.Buffer?.isBuffer(value) ?? false,
+    { message: "Must be a Buffer" }
+  )
+]);
+function convertDataContentToBase64String(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (content instanceof ArrayBuffer) {
+    return chunkSVLMF4UZ_cjs.convertUint8ArrayToBase64(new Uint8Array(content));
+  }
+  return chunkSVLMF4UZ_cjs.convertUint8ArrayToBase64(content);
+}
+
+// src/agent/message-list/prompt/image-utils.ts
+function parseDataUri(dataUri) {
+  if (!dataUri.startsWith("data:")) {
+    return {
+      isDataUri: false,
+      base64Content: dataUri
+    };
+  }
+  const base64Index = dataUri.indexOf(",");
+  if (base64Index === -1) {
+    return {
+      isDataUri: true,
+      base64Content: dataUri
+    };
+  }
+  const header = dataUri.substring(5, base64Index);
+  const base64Content = dataUri.substring(base64Index + 1);
+  const semicolonIndex = header.indexOf(";");
+  const mimeType = semicolonIndex !== -1 ? header.substring(0, semicolonIndex) : header;
+  return {
+    isDataUri: true,
+    mimeType: mimeType || void 0,
+    base64Content
+  };
+}
+function createDataUri(base64Content, mimeType = "application/octet-stream") {
+  if (base64Content.startsWith("data:")) {
+    return base64Content;
+  }
+  return `data:${mimeType};base64,${base64Content}`;
+}
+function imageContentToString(image, fallbackMimeType) {
+  if (typeof image === "string") {
+    return image;
+  }
+  if (image instanceof URL) {
+    return image.toString();
+  }
+  if (image instanceof Uint8Array || image instanceof ArrayBuffer || globalThis.Buffer && Buffer.isBuffer(image)) {
+    const base64 = convertDataContentToBase64String(image);
+    return base64;
+  }
+  return String(image);
+}
+function getImageCacheKey(image) {
+  if (image instanceof URL) {
+    return image.toString();
+  }
+  if (typeof image === "string") {
+    return image.length;
+  }
+  if (image instanceof Uint8Array) {
+    return image.byteLength;
+  }
+  if (image instanceof ArrayBuffer) {
+    return image.byteLength;
+  }
+  return image;
+}
+function isValidUrl(str) {
+  try {
+    new URL(str);
+    return true;
+  } catch {
+    if (str.startsWith("//")) {
+      try {
+        new URL(`https:${str}`);
+        return true;
+      } catch {
+        return false;
+      }
+    }
+    return false;
+  }
+}
+function categorizeFileData(data, fallbackMimeType) {
+  const parsed = parseDataUri(data);
+  const mimeType = parsed.isDataUri && parsed.mimeType ? parsed.mimeType : fallbackMimeType;
+  if (parsed.isDataUri) {
+    return {
+      type: "dataUri",
+      mimeType,
+      data
+    };
+  }
+  if (isValidUrl(data)) {
+    return {
+      type: "url",
+      mimeType,
+      data
+    };
+  }
+  return {
+    type: "raw",
+    mimeType,
+    data
+  };
+}
 
 // src/agent/message-list/prompt/attachments-to-parts.ts
 function attachmentsToParts(attachments) {
   const parts = [];
   for (const attachment of attachments) {
+    const categorized = categorizeFileData(attachment.url, attachment.contentType);
+    let urlString = attachment.url;
+    if (categorized.type === "raw") {
+      urlString = createDataUri(attachment.url, attachment.contentType || "application/octet-stream");
+    }
     let url;
     try {
-      url = new URL(attachment.url);
+      url = new URL(urlString);
     } catch {
       throw new Error(`Invalid URL: ${attachment.url}`);
     }
@@ -262,13 +385,13 @@ function attachmentsToParts(attachments) {
         if (attachment.contentType?.startsWith("image/")) {
           parts.push({
             type: "image",
-            image: attachment.url,
+            image: urlString,
             mimeType: attachment.contentType
           });
         } else if (attachment.contentType?.startsWith("text/")) {
           parts.push({
             type: "file",
-            data: attachment.url,
+            data: urlString,
             mimeType: attachment.contentType
           });
         } else {
@@ -277,7 +400,7 @@ function attachmentsToParts(attachments) {
           }
           parts.push({
             type: "file",
-            data: attachment.url,
+            data: urlString,
             mimeType: attachment.contentType
           });
         }
@@ -588,25 +711,6 @@ function convertToV1Messages(messages) {
   }
   return v1Messages;
 }
-zod.z.union([
-  zod.z.string(),
-  zod.z.instanceof(Uint8Array),
-  zod.z.instanceof(ArrayBuffer),
-  zod.z.custom(
-    // Buffer might not be available in some environments such as CloudFlare:
-    (value) => globalThis.Buffer?.isBuffer(value) ?? false,
-    { message: "Must be a Buffer" }
-  )
-]);
-function convertDataContentToBase64String(content) {
-  if (typeof content === "string") {
-    return content;
-  }
-  if (content instanceof ArrayBuffer) {
-    return chunkSVLMF4UZ_cjs.convertUint8ArrayToBase64(new Uint8Array(content));
-  }
-  return chunkSVLMF4UZ_cjs.convertUint8ArrayToBase64(content);
-}
 
 // src/utils/fetchWithRetry.ts
 async function fetchWithRetry(url, options = {}, maxRetries = 3) {
@@ -728,105 +832,6 @@ async function downloadAssetsFromMessages({
   return Object.fromEntries(downloadFileList);
 }
 
-// src/agent/message-list/prompt/image-utils.ts
-function parseDataUri(dataUri) {
-  if (!dataUri.startsWith("data:")) {
-    return {
-      isDataUri: false,
-      base64Content: dataUri
-    };
-  }
-  const base64Index = dataUri.indexOf(",");
-  if (base64Index === -1) {
-    return {
-      isDataUri: true,
-      base64Content: dataUri
-    };
-  }
-  const header = dataUri.substring(5, base64Index);
-  const base64Content = dataUri.substring(base64Index + 1);
-  const semicolonIndex = header.indexOf(";");
-  const mimeType = semicolonIndex !== -1 ? header.substring(0, semicolonIndex) : header;
-  return {
-    isDataUri: true,
-    mimeType: mimeType || void 0,
-    base64Content
-  };
-}
-function createDataUri(base64Content, mimeType = "application/octet-stream") {
-  if (base64Content.startsWith("data:")) {
-    return base64Content;
-  }
-  return `data:${mimeType};base64,${base64Content}`;
-}
-function imageContentToString(image, fallbackMimeType) {
-  if (typeof image === "string") {
-    return image;
-  }
-  if (image instanceof URL) {
-    return image.toString();
-  }
-  if (image instanceof Uint8Array || image instanceof ArrayBuffer || globalThis.Buffer && Buffer.isBuffer(image)) {
-    const base64 = convertDataContentToBase64String(image);
-    return base64;
-  }
-  return String(image);
-}
-function getImageCacheKey(image) {
-  if (image instanceof URL) {
-    return image.toString();
-  }
-  if (typeof image === "string") {
-    return image.length;
-  }
-  if (image instanceof Uint8Array) {
-    return image.byteLength;
-  }
-  if (image instanceof ArrayBuffer) {
-    return image.byteLength;
-  }
-  return image;
-}
-function isValidUrl(str) {
-  try {
-    new URL(str);
-    return true;
-  } catch {
-    if (str.startsWith("//")) {
-      try {
-        new URL(`https:${str}`);
-        return true;
-      } catch {
-        return false;
-      }
-    }
-    return false;
-  }
-}
-function categorizeFileData(data, fallbackMimeType) {
-  const parsed = parseDataUri(data);
-  const mimeType = parsed.isDataUri && parsed.mimeType ? parsed.mimeType : fallbackMimeType;
-  if (parsed.isDataUri) {
-    return {
-      type: "dataUri",
-      mimeType,
-      data
-    };
-  }
-  if (isValidUrl(data)) {
-    return {
-      type: "url",
-      mimeType,
-      data
-    };
-  }
-  return {
-    type: "raw",
-    mimeType,
-    data
-  };
-}
-
 // src/agent/message-list/utils/ai-v5/gemini-compatibility.ts
 function ensureGeminiCompatibleMessages(messages) {
   const result = [...messages];
@@ -1426,9 +1431,20 @@ var MessageList = class _MessageList {
     if (m.content.parts.length) {
       for (const part of m.content.parts) {
         if (part.type === `file`) {
+          let normalizedUrl;
+          if (typeof part.data === "string") {
+            const categorized = categorizeFileData(part.data, part.mimeType);
+            if (categorized.type === "raw") {
+              normalizedUrl = createDataUri(part.data, part.mimeType || "application/octet-stream");
+            } else {
+              normalizedUrl = part.data;
+            }
+          } else {
+            normalizedUrl = part.data;
+          }
           experimentalAttachments.push({
             contentType: part.mimeType,
-            url: part.data
+            url: normalizedUrl
           });
         } else if (part.type === "tool-invocation" && (part.toolInvocation.state === "call" || part.toolInvocation.state === "partial-call")) {
           continue;
@@ -2110,6 +2126,9 @@ var MessageList = class _MessageList {
     if (coreMessage.providerOptions) {
       content.providerMetadata = coreMessage.providerOptions;
     }
+    if ("metadata" in coreMessage && coreMessage.metadata !== null && coreMessage.metadata !== void 0) {
+      content.metadata = coreMessage.metadata;
+    }
     return {
       id,
       role: _MessageList.getRole(coreMessage),
@@ -2312,9 +2331,23 @@ var MessageList = class _MessageList {
           if (role === `tool` || role === `assistant`) {
             throw new Error(incompatibleMessage);
           }
+          let processedImage;
+          if (part.image instanceof URL || part.image instanceof Uint8Array) {
+            processedImage = part.image;
+          } else if (Buffer.isBuffer(part.image) || part.image instanceof ArrayBuffer) {
+            processedImage = new Uint8Array(part.image);
+          } else {
+            const categorized = categorizeFileData(part.image, part.mimeType);
+            if (categorized.type === "raw") {
+              const dataUri = createDataUri(part.image, part.mimeType || "image/png");
+              processedImage = new URL(dataUri);
+            } else {
+              processedImage = new URL(part.image);
+            }
+          }
           roleContent[role].push({
             ...part,
-            image: part.image instanceof URL || part.image instanceof Uint8Array ? part.image : Buffer.isBuffer(part.image) || part.image instanceof ArrayBuffer ? new Uint8Array(part.image) : new URL(part.image)
+            image: processedImage
           });
           break;
         }
@@ -2964,7 +2997,7 @@ var MessageList = class _MessageList {
       }
     }
     const contentString = mastraDBParts.filter((p) => p.type === "text").map((p) => p.text).join("\n");
-    const metadata = {};
+    const metadata = "metadata" in modelMsg && modelMsg.metadata !== null && modelMsg.metadata !== void 0 ? modelMsg.metadata : {};
     const id = `id` in modelMsg && typeof modelMsg.id === `string` ? modelMsg.id : `msg-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
     const message = {
       id,
@@ -2977,7 +3010,7 @@ var MessageList = class _MessageList {
         reasoning: reasoningParts.length > 0 ? reasoningParts.join("\n") : void 0,
         experimental_attachments: experimental_attachments.length > 0 ? experimental_attachments : void 0,
         content: contentString || void 0,
-        metadata
+        metadata: Object.keys(metadata).length > 0 ? metadata : void 0
       }
     };
     if (modelMsg.providerOptions) {
@@ -3010,7 +3043,8 @@ var MessageList = class _MessageList {
       if (message.role !== `assistant`) continue;
       for (const [index, part] of message.parts.entries()) {
         if (!AIV5__namespace.isToolUIPart(part)) continue;
-        if (message.parts.at(index + 1)?.type !== `step-start`) {
+        const nextPart = message.parts.at(index + 1);
+        if (nextPart && nextPart.type !== `step-start` && !AIV5__namespace.isToolUIPart(nextPart)) {
           message.parts.splice(index + 1, 0, { type: "step-start" });
         }
       }
@@ -3139,5 +3173,5 @@ exports.DefaultGeneratedFile = DefaultGeneratedFile;
 exports.DefaultGeneratedFileWithType = DefaultGeneratedFileWithType;
 exports.MessageList = MessageList;
 exports.convertMessages = convertMessages;
-//# sourceMappingURL=chunk-VSM3NLUX.cjs.map
-//# sourceMappingURL=chunk-VSM3NLUX.cjs.map
+//# sourceMappingURL=chunk-DQIZ5FFX.cjs.map
+//# sourceMappingURL=chunk-DQIZ5FFX.cjs.map