ai 3.4.4 → 3.4.6

package/dist/index.mjs

@@ -743,6 +743,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;
@@ -864,98 +873,119 @@ 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;
+            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 (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
+                    };
+                  }
+                }
+                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 "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 [header, base64Content] = part.image.split(",");
+                          const mimeType = header.split(";")[0].split(":")[1];
+                          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) {
+                      }
+                      default: {
                         throw new Error(
-                          `Error processing data URL: ${getErrorMessage2(
-                            message
-                          )}`
+                          `Unsupported URL protocol: ${url.protocol}`
                         );
                       }
                     }
-                    default: {
-                      throw new Error(
-                        `Unsupported URL protocol: ${url.protocol}`
-                      );
-                    }
+                  } catch (_ignored) {
                   }
-                } 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) {
+                  return {
+                    type: "file",
+                    data: part.data,
+                    mimeType: part.mimeType,
+                    providerMetadata: part.experimental_providerMetadata
+                  };
+                }
+                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
       };
     }
@@ -1216,6 +1246,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 +1277,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()
 });
@@ -3203,7 +3239,8 @@ async function generateText({
             ...currentModelResponse.response,
             headers: (_d = currentModelResponse.rawResponse) == null ? void 0 : _d.headers
           },
-          experimental_providerMetadata: currentModelResponse.providerMetadata
+          experimental_providerMetadata: currentModelResponse.providerMetadata,
+          isContinued: nextStepType === "continue"
         };
         steps.push(currentStep);
         await (onStepFinish == null ? void 0 : onStepFinish(currentStep));
@@ -4143,7 +4180,8 @@ var DefaultStreamTextResult = class {
                 usage: stepUsage,
                 experimental_providerMetadata: stepProviderMetadata,
                 logprobs: stepLogProbs,
-                response: stepResponse
+                response: stepResponse,
+                isContinued: nextStepType === "continue"
               });
               const stepResult = {
                 stepType,
@@ -4156,7 +4194,8 @@ var DefaultStreamTextResult = class {
                 logprobs: stepLogProbs,
                 response: stepResponse,
                 rawResponse: self.rawResponse,
-                experimental_providerMetadata: stepProviderMetadata
+                experimental_providerMetadata: stepProviderMetadata,
+                isContinued: nextStepType === "continue"
               };
               stepResults.push(stepResult);
               await (onStepFinish == null ? void 0 : onStepFinish(stepResult));
@@ -4432,7 +4471,8 @@ var DefaultStreamTextResult = class {
                 usage: sendUsage ? {
                   promptTokens: chunk.usage.promptTokens,
                   completionTokens: chunk.usage.completionTokens
-                } : void 0
+                } : void 0,
+                isContinued: chunk.isContinued
               })
             );
             break;