ai 3.4.6 → 3.4.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "3.4.6",
+  "version": "3.4.7",
   "description": "AI SDK by Vercel - The AI Toolkit for TypeScript and JavaScript",
   "license": "Apache-2.0",
   "sideEffects": false,

package/rsc/dist/rsc-server.mjs

@@ -214,7 +214,10 @@ function createAI({
 import { safeParseJSON } from "@ai-sdk/provider-utils";
 
 // core/prompt/convert-to-language-model-prompt.ts
-import { getErrorMessage } from "@ai-sdk/provider-utils";
+import {
+  convertUint8ArrayToBase64 as convertUint8ArrayToBase642,
+  getErrorMessage
+} from "@ai-sdk/provider-utils";
 
 // util/download-error.ts
 import { AISDKError } from "@ai-sdk/provider";
@@ -434,6 +437,22 @@ var InvalidMessageRoleError = class extends AISDKError3 {
 };
 _a3 = symbol3;
 
+// core/prompt/split-data-url.ts
+function splitDataUrl(dataUrl) {
+  try {
+    const [header, base64Content] = dataUrl.split(",");
+    return {
+      mimeType: header.split(";")[0].split(":")[1],
+      base64Content
+    };
+  } catch (error) {
+    return {
+      mimeType: void 0,
+      base64Content: void 0
+    };
+  }
+}
+
 // core/prompt/convert-to-language-model-prompt.ts
 async function convertToLanguageModelPrompt({
   prompt,
@@ -444,7 +463,7 @@ async function convertToLanguageModelPrompt({
   if (prompt.system != null) {
     languageModelMessages.push({ role: "system", content: prompt.system });
   }
-  const downloadedImages = modelSupportsImageUrls || prompt.messages == null ? null : await downloadImages(prompt.messages, downloadImplementation);
+  const downloadedAssets = modelSupportsImageUrls || prompt.messages == null ? null : await downloadAssets(prompt.messages, downloadImplementation);
   const promptType = prompt.type;
   switch (promptType) {
     case "prompt": {
@@ -457,7 +476,7 @@ async function convertToLanguageModelPrompt({
     case "messages": {
       languageModelMessages.push(
         ...prompt.messages.map(
-          (message) => convertToLanguageModelMessage(message, downloadedImages)
+          (message) => convertToLanguageModelMessage(message, downloadedAssets)
         )
       );
       break;
@@ -469,7 +488,7 @@ async function convertToLanguageModelPrompt({
   }
   return languageModelMessages;
 }
-function convertToLanguageModelMessage(message, downloadedImages) {
+function convertToLanguageModelMessage(message, downloadedAssets) {
   const role = message.role;
   switch (role) {
     case "system": {
@@ -491,7 +510,7 @@ function convertToLanguageModelMessage(message, downloadedImages) {
         role: "user",
         content: message.content.map(
           (part) => {
-            var _a8, _b, _c;
+            var _a8, _b, _c, _d, _e;
             switch (part.type) {
               case "text": {
                 return {
@@ -502,7 +521,7 @@ function convertToLanguageModelMessage(message, downloadedImages) {
               }
               case "image": {
                 if (part.image instanceof URL) {
-                  if (downloadedImages == null) {
+                  if (downloadedAssets == null) {
                     return {
                       type: "image",
                       image: part.image,
@@ -510,7 +529,7 @@ function convertToLanguageModelMessage(message, downloadedImages) {
                       providerMetadata: part.experimental_providerMetadata
                     };
                   } else {
-                    const downloadedImage = downloadedImages[part.image.toString()];
+                    const downloadedImage = downloadedAssets[part.image.toString()];
                     return {
                       type: "image",
                       image: downloadedImage.data,
@@ -525,7 +544,7 @@ function convertToLanguageModelMessage(message, downloadedImages) {
                     switch (url.protocol) {
                       case "http:":
                       case "https:": {
-                        if (downloadedImages == null) {
+                        if (downloadedAssets == null) {
                           return {
                             type: "image",
                             image: url,
@@ -533,7 +552,7 @@ function convertToLanguageModelMessage(message, downloadedImages) {
                             providerMetadata: part.experimental_providerMetadata
                           };
                         } else {
-                          const downloadedImage = downloadedImages[part.image];
+                          const downloadedImage = downloadedAssets[url.toString()];
                           return {
                             type: "image",
                             image: downloadedImage.data,
@@ -544,8 +563,9 @@ function convertToLanguageModelMessage(message, downloadedImages) {
                       }
                       case "data:": {
                         try {
-                          const [header, base64Content] = part.image.split(",");
-                          const mimeType = header.split(";")[0].split(":")[1];
+                          const { mimeType, base64Content } = splitDataUrl(
+                            part.image
+                          );
                           if (mimeType == null || base64Content == null) {
                             throw new Error("Invalid data URL format");
                           }
@@ -563,11 +583,6 @@ function convertToLanguageModelMessage(message, downloadedImages) {
                           );
                         }
                       }
-                      default: {
-                        throw new Error(
-                          `Unsupported URL protocol: ${url.protocol}`
-                        );
-                      }
                     }
                   } catch (_ignored) {
                   }
@@ -582,12 +597,75 @@ function convertToLanguageModelMessage(message, downloadedImages) {
               }
               case "file": {
                 if (part.data instanceof URL) {
-                  return {
-                    type: "file",
-                    data: part.data,
-                    mimeType: part.mimeType,
-                    providerMetadata: part.experimental_providerMetadata
-                  };
+                  if (downloadedAssets == null) {
+                    return {
+                      type: "file",
+                      data: part.data,
+                      mimeType: part.mimeType,
+                      providerMetadata: part.experimental_providerMetadata
+                    };
+                  } else {
+                    const downloadedImage = downloadedAssets[part.data.toString()];
+                    return {
+                      type: "file",
+                      data: convertUint8ArrayToBase642(downloadedImage.data),
+                      mimeType: (_d = part.mimeType) != null ? _d : downloadedImage.mimeType,
+                      providerMetadata: part.experimental_providerMetadata
+                    };
+                  }
+                }
+                if (typeof part.data === "string") {
+                  try {
+                    const url = new URL(part.data);
+                    switch (url.protocol) {
+                      case "http:":
+                      case "https:": {
+                        if (downloadedAssets == null) {
+                          return {
+                            type: "file",
+                            data: url,
+                            mimeType: part.mimeType,
+                            providerMetadata: part.experimental_providerMetadata
+                          };
+                        } else {
+                          const downloadedImage = downloadedAssets[url.toString()];
+                          return {
+                            type: "file",
+                            data: convertUint8ArrayToBase642(
+                              downloadedImage.data
+                            ),
+                            mimeType: (_e = part.mimeType) != null ? _e : downloadedImage.mimeType,
+                            providerMetadata: part.experimental_providerMetadata
+                          };
+                        }
+                      }
+                      case "data:": {
+                        try {
+                          const { mimeType, base64Content } = splitDataUrl(
+                            part.data
+                          );
+                          if (mimeType == null || base64Content == null) {
+                            throw new Error("Invalid data URL format");
+                          }
+                          return {
+                            type: "file",
+                            data: convertDataContentToBase64String(
+                              base64Content
+                            ),
+                            mimeType,
+                            providerMetadata: part.experimental_providerMetadata
+                          };
+                        } catch (error) {
+                          throw new Error(
+                            `Error processing data URL: ${getErrorMessage(
+                              message
+                            )}`
+                          );
+                        }
+                      }
+                    }
+                  } catch (_ignored) {
+                  }
                 }
                 const imageBase64 = convertDataContentToBase64String(
                   part.data
@@ -647,12 +725,14 @@ function convertToLanguageModelMessage(message, downloadedImages) {
     }
   }
 }
-async function downloadImages(messages, downloadImplementation) {
+async function downloadAssets(messages, downloadImplementation) {
   const urls = messages.filter((message) => message.role === "user").map((message) => message.content).filter(
     (content) => Array.isArray(content)
-  ).flat().filter((part) => part.type === "image").map((part) => part.image).map(
+  ).flat().filter(
+    (part) => part.type === "image" || part.type === "file"
+  ).map((part) => part.type === "image" ? part.image : part.data).map(
     (part) => (
-      // support string urls in image parts:
+      // support string urls:
       typeof part === "string" && (part.startsWith("http:") || part.startsWith("https:")) ? new URL(part) : part
     )
   ).filter((image) => image instanceof URL);