@mastra/core 0.15.3-alpha.7 → 0.15.3-alpha.9

package/dist/{chunk-BGE27UEX.js → chunk-HF7HXOCC.js}

@@ -661,6 +661,79 @@ 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);
+    if (fallbackMimeType && !base64.startsWith("data:")) {
+      return `data:${fallbackMimeType};base64,${base64}`;
+    }
+    return base64;
+  }
+  return String(image);
+}
+function imageContentToDataUri(image, mimeType = "image/png") {
+  const imageStr = imageContentToString(image, mimeType);
+  if (imageStr.startsWith("data:")) {
+    return imageStr;
+  }
+  if (imageStr.startsWith("http://") || imageStr.startsWith("https://")) {
+    return imageStr;
+  }
+  return `data:${mimeType};base64,${imageStr}`;
+}
+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;
+}
+
 // src/agent/message-list/utils/ai-v5/tool.ts
 function getToolName(type) {
   if (typeof type === "object" && type && "type" in type) {
@@ -970,7 +1043,7 @@ var MessageList = class _MessageList {
             return {
               type: "file",
               file: new DefaultGeneratedFileWithType({
-                data: typeof c.data === `string` ? c.data : c.data instanceof URL ? c.data.toString() : convertDataContentToBase64String(c.data),
+                data: typeof c.data === `string` ? parseDataUri(c.data).base64Content : c.data instanceof URL ? c.data.toString() : convertDataContentToBase64String(c.data),
                 mediaType: c.mediaType
               })
             };
@@ -978,7 +1051,7 @@ var MessageList = class _MessageList {
             return {
               type: "file",
               file: new DefaultGeneratedFileWithType({
-                data: typeof c.image === `string` ? c.image : c.image instanceof URL ? c.image.toString() : convertDataContentToBase64String(c.image),
+                data: typeof c.image === `string` ? parseDataUri(c.image).base64Content : c.image instanceof URL ? c.image.toString() : convertDataContentToBase64String(c.image),
                 mediaType: c.mediaType || "unknown"
               })
             };
@@ -1628,7 +1701,11 @@ var MessageList = class _MessageList {
             });
             break;
           case "image":
-            parts.push({ type: "file", data: part.image.toString(), mimeType: part.mimeType });
+            parts.push({
+              type: "file",
+              data: imageContentToString(part.image),
+              mimeType: part.mimeType
+            });
             break;
           case "file":
             if (part.data instanceof URL) {
@@ -1761,7 +1838,7 @@ var MessageList = class _MessageList {
         key += part.mimeType;
       }
       if (part.type === `image`) {
-        key += part.image instanceof URL ? part.image.toString() : part.image.toString().length;
+        key += getImageCacheKey(part.image);
         key += part.mimeType;
       }
       if (part.type === `redacted-reasoning`) {
@@ -2236,15 +2313,39 @@ var MessageList = class _MessageList {
             });
           }
           break;
-        case "file":
-          parts.push({
-            type: "file",
-            url: part.data,
-            mediaType: part.mimeType,
-            providerMetadata: part.providerMetadata
-          });
+        case "file": {
+          if (typeof part.data === "string" && (part.data.startsWith("http://") || part.data.startsWith("https://"))) {
+            parts.push({
+              type: "file",
+              url: part.data,
+              mediaType: part.mimeType || "image/png",
+              providerMetadata: part.providerMetadata
+            });
+          } else {
+            let filePartData;
+            let extractedMimeType = part.mimeType;
+            if (typeof part.data === "string") {
+              const parsed = parseDataUri(part.data);
+              filePartData = parsed.base64Content;
+              if (parsed.isDataUri && parsed.mimeType) {
+                extractedMimeType = extractedMimeType || parsed.mimeType;
+              }
+            } else {
+              filePartData = part.data;
+            }
+            const finalMimeType = extractedMimeType || "image/png";
+            const dataUri = createDataUri(filePartData, finalMimeType);
+            parts.push({
+              type: "file",
+              url: dataUri,
+              // Use url field with data URI
+              mediaType: finalMimeType,
+              providerMetadata: part.providerMetadata
+            });
+          }
           fileUrls.add(part.data);
           break;
+        }
       }
     }
     if (v2Msg.content.content && !v3Msg.content.parts?.some((p) => p.type === `text`)) {
@@ -2415,35 +2516,102 @@ var MessageList = class _MessageList {
               providerMetadata: part.providerOptions
             });
             break;
-          case "image":
+          case "image": {
+            let imageData;
+            let extractedMimeType = part.mediaType;
+            const imageStr = imageContentToDataUri(part.image, extractedMimeType || "image/png");
+            const parsed = parseDataUri(imageStr);
+            if (parsed.isDataUri) {
+              imageData = parsed.base64Content;
+              if (!extractedMimeType && parsed.mimeType) {
+                extractedMimeType = parsed.mimeType;
+              }
+            } else if (imageStr.startsWith("http://") || imageStr.startsWith("https://")) {
+              parts.push({
+                type: "file",
+                url: imageStr,
+                mediaType: part.mediaType || "image/jpeg",
+                // Default to image/jpeg for URLs
+                providerMetadata: part.providerOptions
+              });
+              break;
+            } else {
+              imageData = imageStr;
+            }
+            const finalMimeType = extractedMimeType || "image/jpeg";
+            const dataUri = imageData.startsWith("data:") ? imageData : createDataUri(imageData, finalMimeType);
             parts.push({
               type: "file",
-              url: part.image.toString(),
-              mediaType: part.mediaType || "unknown",
+              url: dataUri,
+              mediaType: finalMimeType,
               providerMetadata: part.providerOptions
             });
             break;
-          case "file":
+          }
+          case "file": {
             if (part.data instanceof URL) {
-              parts.push({
-                type: "file",
-                url: part.data.toString(),
-                mediaType: part.mediaType,
-                providerMetadata: part.providerOptions
-              });
-            } else {
-              try {
+              const urlStr = part.data.toString();
+              let extractedMimeType = part.mediaType;
+              const parsed = parseDataUri(urlStr);
+              if (parsed.isDataUri) {
+                if (!extractedMimeType && parsed.mimeType) {
+                  extractedMimeType = parsed.mimeType;
+                }
+                if (parsed.base64Content !== urlStr) {
+                  const dataUri = createDataUri(parsed.base64Content, extractedMimeType || "image/png");
+                  parts.push({
+                    type: "file",
+                    url: dataUri,
+                    mediaType: extractedMimeType || "image/png",
+                    providerMetadata: part.providerOptions
+                  });
+                } else {
+                  parts.push({
+                    type: "file",
+                    url: urlStr,
+                    mediaType: part.mediaType || "image/png",
+                    providerMetadata: part.providerOptions
+                  });
+                }
+              } else {
                 parts.push({
                   type: "file",
-                  mediaType: part.mediaType,
-                  url: convertDataContentToBase64String(part.data),
+                  url: urlStr,
+                  mediaType: part.mediaType || "application/octet-stream",
                   providerMetadata: part.providerOptions
                 });
+              }
+            } else {
+              try {
+                const base64Data = convertDataContentToBase64String(part.data);
+                let extractedMimeType = part.mediaType;
+                const parsed = parseDataUri(base64Data);
+                if (parsed.isDataUri) {
+                  if (!extractedMimeType && parsed.mimeType) {
+                    extractedMimeType = parsed.mimeType;
+                  }
+                  const dataUri = createDataUri(parsed.base64Content, extractedMimeType || "image/png");
+                  parts.push({
+                    type: "file",
+                    url: dataUri,
+                    mediaType: extractedMimeType || "image/png",
+                    providerMetadata: part.providerOptions
+                  });
+                } else {
+                  const dataUri = createDataUri(base64Data, part.mediaType || "image/png");
+                  parts.push({
+                    type: "file",
+                    url: dataUri,
+                    mediaType: part.mediaType || "image/png",
+                    providerMetadata: part.providerOptions
+                  });
+                }
               } catch (error) {
                 console.error(`Failed to convert binary data to base64 in CoreMessage file part: ${error}`, error);
               }
             }
             break;
+          }
         }
       }
     }
@@ -2549,7 +2717,7 @@ var MessageList = class _MessageList {
         key += part.mediaType;
       }
       if (part.type === `image`) {
-        key += part.image instanceof URL ? part.image.toString() : part.image.toString().length;
+        key += getImageCacheKey(part.image);
         key += part.mediaType;
       }
     }
@@ -2586,5 +2754,5 @@ function convertMessages(messages) {
 }
 
 export { DefaultGeneratedFile, DefaultGeneratedFileWithType, MessageList, convertMessages };
-//# sourceMappingURL=chunk-BGE27UEX.js.map
-//# sourceMappingURL=chunk-BGE27UEX.js.map
+//# sourceMappingURL=chunk-HF7HXOCC.js.map
+//# sourceMappingURL=chunk-HF7HXOCC.js.map