ai 3.4.5 → 3.4.7

package/dist/index.mjs

@@ -591,7 +591,10 @@ var DefaultEmbedManyResult = class {
 import { createIdGenerator, safeParseJSON } from "@ai-sdk/provider-utils";
 
 // core/prompt/convert-to-language-model-prompt.ts
-import { getErrorMessage as getErrorMessage2 } from "@ai-sdk/provider-utils";
+import {
+  convertUint8ArrayToBase64 as convertUint8ArrayToBase642,
+  getErrorMessage as getErrorMessage2
+} from "@ai-sdk/provider-utils";
 
 // util/download-error.ts
 import { AISDKError as AISDKError2 } from "@ai-sdk/provider";
@@ -743,6 +746,15 @@ var dataContentSchema = z.union([
     { message: "Must be a Buffer" }
   )
 ]);
+function convertDataContentToBase64String(content) {
+  if (typeof content === "string") {
+    return content;
+  }
+  if (content instanceof ArrayBuffer) {
+    return convertUint8ArrayToBase64(new Uint8Array(content));
+  }
+  return convertUint8ArrayToBase64(content);
+}
 function convertDataContentToUint8Array(content) {
   if (content instanceof Uint8Array) {
     return content;
@@ -809,6 +821,22 @@ var InvalidMessageRoleError = class extends AISDKError4 {
 };
 _a4 = symbol4;
 
+// 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,
@@ -819,7 +847,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": {
@@ -832,7 +860,7 @@ async function convertToLanguageModelPrompt({
     case "messages": {
       languageModelMessages.push(
         ...prompt.messages.map(
-          (message) => convertToLanguageModelMessage(message, downloadedImages)
+          (message) => convertToLanguageModelMessage(message, downloadedAssets)
         )
       );
       break;
@@ -844,7 +872,7 @@ async function convertToLanguageModelPrompt({
   }
   return languageModelMessages;
 }
-function convertToLanguageModelMessage(message, downloadedImages) {
+function convertToLanguageModelMessage(message, downloadedAssets) {
   const role = message.role;
   switch (role) {
     case "system": {
@@ -864,98 +892,178 @@ function convertToLanguageModelMessage(message, downloadedImages) {
       }
       return {
         role: "user",
-        content: message.content.map((part) => {
-          var _a11, _b, _c;
-          switch (part.type) {
-            case "text": {
-              return {
-                type: "text",
-                text: part.text,
-                providerMetadata: part.experimental_providerMetadata
-              };
-            }
-            case "image": {
-              if (part.image instanceof URL) {
-                if (downloadedImages == null) {
-                  return {
-                    type: "image",
-                    image: part.image,
-                    mimeType: part.mimeType,
-                    providerMetadata: part.experimental_providerMetadata
-                  };
-                } else {
-                  const downloadedImage = downloadedImages[part.image.toString()];
-                  return {
-                    type: "image",
-                    image: downloadedImage.data,
-                    mimeType: (_a11 = part.mimeType) != null ? _a11 : downloadedImage.mimeType,
-                    providerMetadata: part.experimental_providerMetadata
-                  };
-                }
+        content: message.content.map(
+          (part) => {
+            var _a11, _b, _c, _d, _e;
+            switch (part.type) {
+              case "text": {
+                return {
+                  type: "text",
+                  text: part.text,
+                  providerMetadata: part.experimental_providerMetadata
+                };
               }
-              if (typeof part.image === "string") {
-                try {
-                  const url = new URL(part.image);
-                  switch (url.protocol) {
-                    case "http:":
-                    case "https:": {
-                      if (downloadedImages == null) {
-                        return {
-                          type: "image",
-                          image: url,
-                          mimeType: part.mimeType,
-                          providerMetadata: part.experimental_providerMetadata
-                        };
-                      } else {
-                        const downloadedImage = downloadedImages[part.image];
-                        return {
-                          type: "image",
-                          image: downloadedImage.data,
-                          mimeType: (_b = part.mimeType) != null ? _b : downloadedImage.mimeType,
-                          providerMetadata: part.experimental_providerMetadata
-                        };
+              case "image": {
+                if (part.image instanceof URL) {
+                  if (downloadedAssets == null) {
+                    return {
+                      type: "image",
+                      image: part.image,
+                      mimeType: part.mimeType,
+                      providerMetadata: part.experimental_providerMetadata
+                    };
+                  } else {
+                    const downloadedImage = downloadedAssets[part.image.toString()];
+                    return {
+                      type: "image",
+                      image: downloadedImage.data,
+                      mimeType: (_a11 = part.mimeType) != null ? _a11 : downloadedImage.mimeType,
+                      providerMetadata: part.experimental_providerMetadata
+                    };
+                  }
+                }
+                if (typeof part.image === "string") {
+                  try {
+                    const url = new URL(part.image);
+                    switch (url.protocol) {
+                      case "http:":
+                      case "https:": {
+                        if (downloadedAssets == null) {
+                          return {
+                            type: "image",
+                            image: url,
+                            mimeType: part.mimeType,
+                            providerMetadata: part.experimental_providerMetadata
+                          };
+                        } else {
+                          const downloadedImage = downloadedAssets[url.toString()];
+                          return {
+                            type: "image",
+                            image: downloadedImage.data,
+                            mimeType: (_b = part.mimeType) != null ? _b : downloadedImage.mimeType,
+                            providerMetadata: part.experimental_providerMetadata
+                          };
+                        }
                       }
-                    }
-                    case "data:": {
-                      try {
-                        const [header, base64Content] = part.image.split(",");
-                        const mimeType = header.split(";")[0].split(":")[1];
-                        if (mimeType == null || base64Content == null) {
-                          throw new Error("Invalid data URL format");
+                      case "data:": {
+                        try {
+                          const { mimeType, base64Content } = splitDataUrl(
+                            part.image
+                          );
+                          if (mimeType == null || base64Content == null) {
+                            throw new Error("Invalid data URL format");
+                          }
+                          return {
+                            type: "image",
+                            image: convertDataContentToUint8Array(base64Content),
+                            mimeType,
+                            providerMetadata: part.experimental_providerMetadata
+                          };
+                        } catch (error) {
+                          throw new Error(
+                            `Error processing data URL: ${getErrorMessage2(
+                              message
+                            )}`
+                          );
                         }
-                        return {
-                          type: "image",
-                          image: convertDataContentToUint8Array(base64Content),
-                          mimeType,
-                          providerMetadata: part.experimental_providerMetadata
-                        };
-                      } catch (error) {
-                        throw new Error(
-                          `Error processing data URL: ${getErrorMessage2(
-                            message
-                          )}`
-                        );
                       }
                     }
-                    default: {
-                      throw new Error(
-                        `Unsupported URL protocol: ${url.protocol}`
-                      );
+                  } catch (_ignored) {
+                  }
+                }
+                const imageUint8 = convertDataContentToUint8Array(part.image);
+                return {
+                  type: "image",
+                  image: imageUint8,
+                  mimeType: (_c = part.mimeType) != null ? _c : detectImageMimeType(imageUint8),
+                  providerMetadata: part.experimental_providerMetadata
+                };
+              }
+              case "file": {
+                if (part.data instanceof URL) {
+                  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: ${getErrorMessage2(
+                              message
+                            )}`
+                          );
+                        }
+                      }
                     }
+                  } catch (_ignored) {
                   }
-                } catch (_ignored) {
                 }
+                const imageBase64 = convertDataContentToBase64String(
+                  part.data
+                );
+                return {
+                  type: "file",
+                  data: imageBase64,
+                  mimeType: part.mimeType,
+                  providerMetadata: part.experimental_providerMetadata
+                };
               }
-              const imageUint8 = convertDataContentToUint8Array(part.image);
-              return {
-                type: "image",
-                image: imageUint8,
-                mimeType: (_c = part.mimeType) != null ? _c : detectImageMimeType(imageUint8),
-                providerMetadata: part.experimental_providerMetadata
-              };
             }
           }
-        }).filter((part) => part.type !== "text" || part.text !== ""),
+        ).filter((part) => part.type !== "text" || part.text !== ""),
         providerMetadata: message.experimental_providerMetadata
       };
     }
@@ -1001,12 +1109,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);
@@ -1216,6 +1326,12 @@ var imagePartSchema = z4.object({
   mimeType: z4.string().optional(),
   experimental_providerMetadata: providerMetadataSchema.optional()
 });
+var filePartSchema = z4.object({
+  type: z4.literal("file"),
+  data: z4.union([dataContentSchema, z4.instanceof(URL)]),
+  mimeType: z4.string(),
+  experimental_providerMetadata: providerMetadataSchema.optional()
+});
 var toolCallPartSchema = z4.object({
   type: z4.literal("tool-call"),
   toolCallId: z4.string(),
@@ -1241,7 +1357,7 @@ var coreUserMessageSchema = z5.object({
   role: z5.literal("user"),
   content: z5.union([
     z5.string(),
-    z5.array(z5.union([textPartSchema, imagePartSchema]))
+    z5.array(z5.union([textPartSchema, imagePartSchema, filePartSchema]))
   ]),
   experimental_providerMetadata: providerMetadataSchema.optional()
 });
@@ -4589,6 +4705,17 @@ function attachmentsToParts(attachments) {
       case "https:": {
         if ((_a11 = attachment.contentType) == null ? void 0 : _a11.startsWith("image/")) {
           parts.push({ type: "image", image: url });
+        } else {
+          if (!attachment.contentType) {
+            throw new Error(
+              "If the attachment is not an image, it must specify a content type"
+            );
+          }
+          parts.push({
+            type: "file",
+            data: url,
+            mimeType: attachment.contentType
+          });
         }
         break;
       }
@@ -4617,6 +4744,17 @@ function attachmentsToParts(attachments) {
               convertDataContentToUint8Array(base64Content)
             )
           });
+        } else {
+          if (!attachment.contentType) {
+            throw new Error(
+              "If the attachment is not an image or text, it must specify a content type"
+            );
+          }
+          parts.push({
+            type: "file",
+            data: base64Content,
+            mimeType: attachment.contentType
+          });
         }
         break;
       }