ai 5.0.0-canary.16 → 5.0.0-canary.18

package/dist/index.mjs

@@ -9,26 +9,14 @@ import { createIdGenerator as createIdGenerator5, generateId as generateId2 } fr
 
 // core/util/index.ts
 import {
+  asSchema,
   generateId,
-  jsonSchema,
-  asSchema
+  jsonSchema
 } from "@ai-sdk/provider-utils";
 
 // core/util/process-chat-response.ts
 import { generateId as generateIdFunction } from "@ai-sdk/provider-utils";
 
-// core/types/duplicated/usage.ts
-function calculateLanguageModelUsage({
-  promptTokens,
-  completionTokens
-}) {
-  return {
-    promptTokens,
-    completionTokens,
-    totalTokens: promptTokens + completionTokens
-  };
-}
-
 // core/ui/get-tool-invocations.ts
 function getToolInvocations(message) {
   return message.parts.filter(
@@ -494,10 +482,13 @@ var finishMessageStreamPart = {
     const result = {
       finishReason: value.finishReason
     };
-    if ("usage" in value && value.usage != null && typeof value.usage === "object" && "promptTokens" in value.usage && "completionTokens" in value.usage) {
+    if ("usage" in value && value.usage != null && typeof value.usage === "object") {
       result.usage = {
-        promptTokens: typeof value.usage.promptTokens === "number" ? value.usage.promptTokens : Number.NaN,
-        completionTokens: typeof value.usage.completionTokens === "number" ? value.usage.completionTokens : Number.NaN
+        inputTokens: "inputTokens" in value.usage && typeof value.usage.inputTokens === "number" ? value.usage.inputTokens : void 0,
+        outputTokens: "outputTokens" in value.usage && typeof value.usage.outputTokens === "number" ? value.usage.outputTokens : void 0,
+        totalTokens: "totalTokens" in value.usage && typeof value.usage.totalTokens === "number" ? value.usage.totalTokens : void 0,
+        reasoningTokens: "reasoningTokens" in value.usage && typeof value.usage.reasoningTokens === "number" ? value.usage.reasoningTokens : void 0,
+        cachedInputTokens: "cachedInputTokens" in value.usage && typeof value.usage.cachedInputTokens === "number" ? value.usage.cachedInputTokens : void 0
       };
     }
     return {
@@ -519,10 +510,13 @@ var finishStepStreamPart = {
       finishReason: value.finishReason,
       isContinued: false
     };
-    if ("usage" in value && value.usage != null && typeof value.usage === "object" && "promptTokens" in value.usage && "completionTokens" in value.usage) {
+    if ("usage" in value && value.usage != null && typeof value.usage === "object") {
       result.usage = {
-        promptTokens: typeof value.usage.promptTokens === "number" ? value.usage.promptTokens : Number.NaN,
-        completionTokens: typeof value.usage.completionTokens === "number" ? value.usage.completionTokens : Number.NaN
+        inputTokens: "inputTokens" in value.usage && typeof value.usage.inputTokens === "number" ? value.usage.inputTokens : void 0,
+        outputTokens: "outputTokens" in value.usage && typeof value.usage.outputTokens === "number" ? value.usage.outputTokens : void 0,
+        totalTokens: "totalTokens" in value.usage && typeof value.usage.totalTokens === "number" ? value.usage.totalTokens : void 0,
+        reasoningTokens: "reasoningTokens" in value.usage && typeof value.usage.reasoningTokens === "number" ? value.usage.reasoningTokens : void 0,
+        cachedInputTokens: "cachedInputTokens" in value.usage && typeof value.usage.cachedInputTokens === "number" ? value.usage.cachedInputTokens : void 0
       };
     }
     if ("isContinued" in value && typeof value.isContinued === "boolean") {
@@ -586,9 +580,9 @@ var fileStreamPart = {
   code: "k",
   name: "file",
   parse: (value) => {
-    if (value == null || typeof value !== "object" || !("data" in value) || typeof value.data !== "string" || !("mimeType" in value) || typeof value.mimeType !== "string") {
+    if (value == null || typeof value !== "object" || !("url" in value) || typeof value.url !== "string" || !("mediaType" in value) || typeof value.mediaType !== "string") {
       throw new Error(
-        '"file" parts expect an object with a "data" and "mimeType" property.'
+        '"file" parts expect an object with a "url" and "mediaType" property.'
       );
     }
     return { type: "file", value };
@@ -795,9 +789,9 @@ async function processChatResponse({
   let messageAnnotations = replaceLastMessage ? lastMessage == null ? void 0 : lastMessage.annotations : void 0;
   const partialToolCalls = {};
   let usage = {
-    completionTokens: NaN,
-    promptTokens: NaN,
-    totalTokens: NaN
+    inputTokens: void 0,
+    outputTokens: void 0,
+    totalTokens: void 0
   };
   let finishReason = "unknown";
   function execUpdate() {
@@ -859,8 +853,8 @@ async function processChatResponse({
     onFilePart(value) {
       message.parts.push({
         type: "file",
-        mediaType: value.mimeType,
-        data: value.data
+        mediaType: value.mediaType,
+        url: value.url
       });
       execUpdate();
     },
@@ -970,7 +964,7 @@ async function processChatResponse({
     onFinishMessagePart(value) {
       finishReason = value.finishReason;
       if (value.usage != null) {
-        usage = calculateLanguageModelUsage(value.usage);
+        usage = value.usage;
       }
     },
     onErrorPart(error) {
@@ -1027,7 +1021,11 @@ async function processChatTextResponse({
     }
   });
   onFinish == null ? void 0 : onFinish(resultMessage, {
-    usage: { completionTokens: NaN, promptTokens: NaN, totalTokens: NaN },
+    usage: {
+      inputTokens: void 0,
+      outputTokens: void 0,
+      totalTokens: void 0
+    },
     finishReason: "unknown"
   });
 }
@@ -1041,40 +1039,41 @@ async function callChatApi({
   credentials,
   headers,
   abortController,
-  restoreMessagesOnFailure,
   onResponse,
   onUpdate,
   onFinish,
   onToolCall,
   generateId: generateId3,
   fetch: fetch2 = getOriginalFetch(),
-  lastMessage
+  lastMessage,
+  getCurrentDate,
+  requestType = "generate"
 }) {
-  var _a17, _b;
-  const response = await fetch2(api, {
+  var _a17, _b, _c;
+  const response = requestType === "resume" ? await fetch2(`${api}?chatId=${body.id}`, {
+    method: "GET",
+    headers: {
+      "Content-Type": "application/json",
+      ...headers
+    },
+    signal: (_a17 = abortController == null ? void 0 : abortController()) == null ? void 0 : _a17.signal,
+    credentials
+  }) : await fetch2(api, {
     method: "POST",
     body: JSON.stringify(body),
     headers: {
       "Content-Type": "application/json",
       ...headers
     },
-    signal: (_a17 = abortController == null ? void 0 : abortController()) == null ? void 0 : _a17.signal,
+    signal: (_b = abortController == null ? void 0 : abortController()) == null ? void 0 : _b.signal,
     credentials
-  }).catch((err) => {
-    restoreMessagesOnFailure();
-    throw err;
   });
-  if (onResponse) {
-    try {
-      await onResponse(response);
-    } catch (err) {
-      throw err;
-    }
+  if (onResponse != null) {
+    await onResponse(response);
   }
   if (!response.ok) {
-    restoreMessagesOnFailure();
     throw new Error(
-      (_b = await response.text()) != null ? _b : "Failed to fetch the chat response."
+      (_c = await response.text()) != null ? _c : "Failed to fetch the chat response."
     );
   }
   if (!response.body) {
@@ -1086,7 +1085,8 @@ async function callChatApi({
         stream: response.body,
         update: onUpdate,
         onFinish,
-        generateId: generateId3
+        generateId: generateId3,
+        getCurrentDate
       });
       return;
     }
@@ -1101,7 +1101,8 @@ async function callChatApi({
             onFinish(message, { usage, finishReason });
           }
         },
-        generateId: generateId3
+        generateId: generateId3,
+        getCurrentDate
       });
       return;
     }
@@ -1222,6 +1223,36 @@ async function callCompletionApi({
   }
 }
 
+// core/util/convert-file-list-to-file-ui-parts.ts
+async function convertFileListToFileUIParts(files) {
+  if (files == null) {
+    return [];
+  }
+  if (!globalThis.FileList || !(files instanceof globalThis.FileList)) {
+    throw new Error("FileList is not supported in the current environment");
+  }
+  return Promise.all(
+    Array.from(files).map(async (file) => {
+      const { name: name17, type } = file;
+      const dataUrl = await new Promise((resolve, reject) => {
+        const reader = new FileReader();
+        reader.onload = (readerEvent) => {
+          var _a17;
+          resolve((_a17 = readerEvent.target) == null ? void 0 : _a17.result);
+        };
+        reader.onerror = (error) => reject(error);
+        reader.readAsDataURL(file);
+      });
+      return {
+        type: "file",
+        mediaType: type,
+        filename: name17,
+        url: dataUrl
+      };
+    })
+  );
+}
+
 // core/util/data-url.ts
 function getTextFromDataUrl(dataUrl) {
   const [header, base64Content] = dataUrl.split(",");
@@ -1271,58 +1302,6 @@ function isDeepEqualData(obj1, obj2) {
   return true;
 }
 
-// core/util/prepare-attachments-for-request.ts
-async function prepareAttachmentsForRequest(attachmentsFromOptions) {
-  if (!attachmentsFromOptions) {
-    return [];
-  }
-  if (globalThis.FileList && attachmentsFromOptions instanceof globalThis.FileList) {
-    return Promise.all(
-      Array.from(attachmentsFromOptions).map(async (attachment) => {
-        const { name: name17, type } = attachment;
-        const dataUrl = await new Promise((resolve, reject) => {
-          const reader = new FileReader();
-          reader.onload = (readerEvent) => {
-            var _a17;
-            resolve((_a17 = readerEvent.target) == null ? void 0 : _a17.result);
-          };
-          reader.onerror = (error) => reject(error);
-          reader.readAsDataURL(attachment);
-        });
-        return {
-          name: name17,
-          contentType: type,
-          url: dataUrl
-        };
-      })
-    );
-  }
-  if (Array.isArray(attachmentsFromOptions)) {
-    return attachmentsFromOptions;
-  }
-  throw new Error("Invalid attachments type");
-}
-
-// core/util/update-tool-call-result.ts
-function updateToolCallResult({
-  messages,
-  toolCallId,
-  toolResult: result
-}) {
-  const lastMessage = messages[messages.length - 1];
-  const invocationPart = lastMessage.parts.find(
-    (part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === toolCallId
-  );
-  if (invocationPart == null) {
-    return;
-  }
-  invocationPart.toolInvocation = {
-    ...invocationPart.toolInvocation,
-    state: "result",
-    result
-  };
-}
-
 // core/util/should-resubmit-messages.ts
 function shouldResubmitMessages({
   originalMaxToolInvocationStep,
@@ -1352,6 +1331,26 @@ function isAssistantMessageWithCompletedToolCalls(message) {
   return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every((part) => "result" in part.toolInvocation);
 }
 
+// core/util/update-tool-call-result.ts
+function updateToolCallResult({
+  messages,
+  toolCallId,
+  toolResult: result
+}) {
+  const lastMessage = messages[messages.length - 1];
+  const invocationPart = lastMessage.parts.find(
+    (part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === toolCallId
+  );
+  if (invocationPart == null) {
+    return;
+  }
+  invocationPart.toolInvocation = {
+    ...invocationPart.toolInvocation,
+    state: "result",
+    result
+  };
+}
+
 // core/data-stream/create-data-stream.ts
 function createDataStream({
   execute,
@@ -1982,6 +1981,7 @@ function splitArray(array, chunkSize) {
 async function embedMany({
   model,
   values,
+  maxParallelCalls = Infinity,
   maxRetries: maxRetriesArg,
   abortSignal,
   headers,
@@ -2011,7 +2011,10 @@ async function embedMany({
     }),
     tracer,
     fn: async (span) => {
-      const maxEmbeddingsPerCall = await model.maxEmbeddingsPerCall;
+      const [maxEmbeddingsPerCall, supportsParallelCalls] = await Promise.all([
+        model.maxEmbeddingsPerCall,
+        model.supportsParallelCalls
+      ]);
       if (maxEmbeddingsPerCall == null || maxEmbeddingsPerCall === Infinity) {
         const { embeddings: embeddings2, usage, response } = await retry(() => {
           return recordSpan({
@@ -2082,61 +2085,69 @@ async function embedMany({
       const embeddings = [];
       const responses = [];
       let tokens = 0;
-      for (const chunk of valueChunks) {
-        const {
-          embeddings: responseEmbeddings,
-          usage,
-          response
-        } = await retry(() => {
-          return recordSpan({
-            name: "ai.embedMany.doEmbed",
-            attributes: selectTelemetryAttributes({
-              telemetry,
-              attributes: {
-                ...assembleOperationName({
-                  operationId: "ai.embedMany.doEmbed",
-                  telemetry
-                }),
-                ...baseTelemetryAttributes,
-                // specific settings that only make sense on the outer level:
-                "ai.values": {
-                  input: () => chunk.map((value) => JSON.stringify(value))
-                }
-              }
-            }),
-            tracer,
-            fn: async (doEmbedSpan) => {
-              var _a17;
-              const modelResponse = await model.doEmbed({
-                values: chunk,
-                abortSignal,
-                headers,
-                providerOptions
-              });
-              const embeddings2 = modelResponse.embeddings;
-              const usage2 = (_a17 = modelResponse.usage) != null ? _a17 : { tokens: NaN };
-              doEmbedSpan.setAttributes(
-                selectTelemetryAttributes({
+      const parallelChunks = splitArray(
+        valueChunks,
+        supportsParallelCalls ? maxParallelCalls : 1
+      );
+      for (const parallelChunk of parallelChunks) {
+        const results = await Promise.all(
+          parallelChunk.map((chunk) => {
+            return retry(() => {
+              return recordSpan({
+                name: "ai.embedMany.doEmbed",
+                attributes: selectTelemetryAttributes({
                   telemetry,
                   attributes: {
-                    "ai.embeddings": {
-                      output: () => embeddings2.map((embedding) => JSON.stringify(embedding))
-                    },
-                    "ai.usage.tokens": usage2.tokens
+                    ...assembleOperationName({
+                      operationId: "ai.embedMany.doEmbed",
+                      telemetry
+                    }),
+                    ...baseTelemetryAttributes,
+                    // specific settings that only make sense on the outer level:
+                    "ai.values": {
+                      input: () => chunk.map((value) => JSON.stringify(value))
+                    }
                   }
-                })
-              );
-              return {
-                embeddings: embeddings2,
-                usage: usage2,
-                response: modelResponse.response
-              };
-            }
-          });
-        });
-        embeddings.push(...responseEmbeddings);
-        responses.push(response);
-        tokens += usage.tokens;
+                }),
+                tracer,
+                fn: async (doEmbedSpan) => {
+                  var _a17;
+                  const modelResponse = await model.doEmbed({
+                    values: chunk,
+                    abortSignal,
+                    headers,
+                    providerOptions
+                  });
+                  const embeddings2 = modelResponse.embeddings;
+                  const usage = (_a17 = modelResponse.usage) != null ? _a17 : { tokens: NaN };
+                  doEmbedSpan.setAttributes(
+                    selectTelemetryAttributes({
+                      telemetry,
+                      attributes: {
+                        "ai.embeddings": {
+                          output: () => embeddings2.map(
+                            (embedding) => JSON.stringify(embedding)
+                          )
+                        },
+                        "ai.usage.tokens": usage.tokens
+                      }
+                    })
+                  );
+                  return {
+                    embeddings: embeddings2,
+                    usage,
+                    response: modelResponse.response
+                  };
+                }
+              });
+            });
+          })
+        );
+        for (const result of results) {
+          embeddings.push(...result.embeddings);
+          responses.push(result.response);
+          tokens += result.usage.tokens;
+        }
       }
       span.setAttributes(
         selectTelemetryAttributes({
@@ -2372,7 +2383,7 @@ async function generateImage({
   abortSignal,
   headers
 }) {
-  var _a17;
+  var _a17, _b;
   const { retry } = prepareRetries({ maxRetries: maxRetriesArg });
   const maxImagesPerCall = (_a17 = model.maxImagesPerCall) != null ? _a17 : 1;
   const callCount = Math.ceil(n / maxImagesPerCall);
@@ -2402,6 +2413,7 @@ async function generateImage({
   const images = [];
   const warnings = [];
   const responses = [];
+  const providerMetadata = {};
   for (const result of results) {
     images.push(
       ...result.images.map(
@@ -2418,18 +2430,32 @@ async function generateImage({
       )
     );
     warnings.push(...result.warnings);
+    if (result.providerMetadata) {
+      for (const [providerName, metadata] of Object.entries(result.providerMetadata)) {
+        (_b = providerMetadata[providerName]) != null ? _b : providerMetadata[providerName] = { images: [] };
+        providerMetadata[providerName].images.push(
+          ...result.providerMetadata[providerName].images
+        );
+      }
+    }
     responses.push(result.response);
   }
   if (!images.length) {
     throw new NoImageGeneratedError({ responses });
   }
-  return new DefaultGenerateImageResult({ images, warnings, responses });
+  return new DefaultGenerateImageResult({
+    images,
+    warnings,
+    responses,
+    providerMetadata
+  });
 }
 var DefaultGenerateImageResult = class {
   constructor(options) {
     this.images = options.images;
     this.warnings = options.warnings;
     this.responses = options.responses;
+    this.providerMetadata = options.providerMetadata;
   }
   get image() {
     return this.images[0];
@@ -2482,6 +2508,9 @@ function extractContentText(content) {
   return parts.map((content2) => content2.text).join("");
 }
 
+// core/prompt/convert-to-language-model-prompt.ts
+import { isUrlSupported } from "@ai-sdk/provider-utils";
+
 // util/download-error.ts
 import { AISDKError as AISDKError5 } from "@ai-sdk/provider";
 var name5 = "AI_DownloadError";
@@ -2649,13 +2678,6 @@ function convertDataContentToUint8Array(content) {
   }
   throw new InvalidDataContentError({ content });
 }
-function convertUint8ArrayToText(uint8Array) {
-  try {
-    return new TextDecoder().decode(uint8Array);
-  } catch (error) {
-    throw new Error("Error decoding Uint8Array to text");
-  }
-}
 
 // core/prompt/invalid-message-role-error.ts
 import { AISDKError as AISDKError8 } from "@ai-sdk/provider";
@@ -2679,7 +2701,6 @@ var InvalidMessageRoleError = class extends AISDKError8 {
 _a7 = symbol7;
 
 // core/prompt/convert-to-language-model-prompt.ts
-import { isUrlSupported } from "@ai-sdk/provider-utils";
 async function convertToLanguageModelPrompt({
   prompt,
   supportedUrls,
@@ -2735,7 +2756,6 @@ function convertToLanguageModelMessage(message, downloadedAssets) {
           // remove empty text parts:
           (part) => part.type !== "text" || part.text !== ""
         ).map((part) => {
-          var _a17;
           const providerOptions = part.providerOptions;
           switch (part.type) {
             case "file": {
@@ -2746,7 +2766,7 @@ function convertToLanguageModelMessage(message, downloadedAssets) {
                 type: "file",
                 data,
                 filename: part.filename,
-                mediaType: (_a17 = mediaType != null ? mediaType : part.mediaType) != null ? _a17 : part.mimeType,
+                mediaType: mediaType != null ? mediaType : part.mediaType,
                 providerOptions
               };
             }
@@ -2805,8 +2825,8 @@ async function downloadAssets(messages, downloadImplementation, supportedUrls) {
   ).flat().filter(
     (part) => part.type === "image" || part.type === "file"
   ).map((part) => {
-    var _a17, _b;
-    const mediaType = (_b = (_a17 = part.mediaType) != null ? _a17 : part.mimeType) != null ? _b : part.type === "image" ? "image/*" : void 0;
+    var _a17;
+    const mediaType = (_a17 = part.mediaType) != null ? _a17 : part.type === "image" ? "image/*" : void 0;
     let data = part.type === "image" ? part.image : part.data;
     if (typeof data === "string") {
       try {
@@ -2833,7 +2853,7 @@ async function downloadAssets(messages, downloadImplementation, supportedUrls) {
   );
 }
 function convertPartToLanguageModelPart(part, downloadedAssets) {
-  var _a17, _b, _c;
+  var _a17, _b;
   if (part.type === "text") {
     return {
       type: "text",
@@ -2854,19 +2874,19 @@ function convertPartToLanguageModelPart(part, downloadedAssets) {
       throw new Error(`Unsupported part type: ${type}`);
   }
   const { data: convertedData, mediaType: convertedMediaType } = convertToLanguageModelV2DataContent(originalData);
-  let mediaType = (_a17 = convertedMediaType != null ? convertedMediaType : part.mediaType) != null ? _a17 : part.mimeType;
+  let mediaType = convertedMediaType != null ? convertedMediaType : part.mediaType;
   let data = convertedData;
   if (data instanceof URL) {
     const downloadedFile = downloadedAssets[data.toString()];
     if (downloadedFile) {
       data = downloadedFile.data;
-      mediaType = (_b = downloadedFile.mediaType) != null ? _b : mediaType;
+      mediaType = (_a17 = downloadedFile.mediaType) != null ? _a17 : mediaType;
     }
   }
   switch (type) {
     case "image": {
       if (data instanceof Uint8Array || typeof data === "string") {
-        mediaType = (_c = detectMediaType({ data, signatures: imageMediaTypeSignatures })) != null ? _c : mediaType;
+        mediaType = (_b = detectMediaType({ data, signatures: imageMediaTypeSignatures })) != null ? _b : mediaType;
       }
       return {
         type: "file",
@@ -2990,83 +3010,6 @@ import { InvalidPromptError } from "@ai-sdk/provider";
 import { safeValidateTypes } from "@ai-sdk/provider-utils";
 import { z as z7 } from "zod";
 
-// core/prompt/attachments-to-parts.ts
-function attachmentsToParts(attachments) {
-  var _a17, _b, _c;
-  const parts = [];
-  for (const attachment of attachments) {
-    let url;
-    try {
-      url = new URL(attachment.url);
-    } catch (error) {
-      throw new Error(`Invalid URL: ${attachment.url}`);
-    }
-    switch (url.protocol) {
-      case "http:":
-      case "https:": {
-        if ((_a17 = attachment.contentType) == null ? void 0 : _a17.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,
-            mediaType: attachment.contentType
-          });
-        }
-        break;
-      }
-      case "data:": {
-        let header;
-        let base64Content;
-        let mediaType;
-        try {
-          [header, base64Content] = attachment.url.split(",");
-          mediaType = header.split(";")[0].split(":")[1];
-        } catch (error) {
-          throw new Error(`Error processing data URL: ${attachment.url}`);
-        }
-        if (mediaType == null || base64Content == null) {
-          throw new Error(`Invalid data URL format: ${attachment.url}`);
-        }
-        if ((_b = attachment.contentType) == null ? void 0 : _b.startsWith("image/")) {
-          parts.push({
-            type: "image",
-            image: convertDataContentToUint8Array(base64Content)
-          });
-        } else if ((_c = attachment.contentType) == null ? void 0 : _c.startsWith("text/")) {
-          parts.push({
-            type: "text",
-            text: convertUint8ArrayToText(
-              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,
-            mediaType: attachment.contentType
-          });
-        }
-        break;
-      }
-      default: {
-        throw new Error(`Unsupported URL protocol: ${url.protocol}`);
-      }
-    }
-  }
-  return parts;
-}
-
 // core/prompt/message-conversion-error.ts
 import { AISDKError as AISDKError9 } from "@ai-sdk/provider";
 var name8 = "AI_MessageConversionError";
@@ -3088,15 +3031,15 @@ var MessageConversionError = class extends AISDKError9 {
 };
 _a8 = symbol8;
 
-// core/prompt/convert-to-core-messages.ts
-function convertToCoreMessages(messages, options) {
+// core/prompt/convert-to-model-messages.ts
+function convertToModelMessages(messages, options) {
   var _a17, _b;
   const tools = (_a17 = options == null ? void 0 : options.tools) != null ? _a17 : {};
   const coreMessages = [];
   for (let i = 0; i < messages.length; i++) {
     const message = messages[i];
     const isLastMessage = i === messages.length - 1;
-    const { role, content, experimental_attachments } = message;
+    const { role, content } = message;
     switch (role) {
       case "system": {
         coreMessages.push({
@@ -3106,30 +3049,24 @@ function convertToCoreMessages(messages, options) {
         break;
       }
       case "user": {
-        if (message.parts == null) {
-          coreMessages.push({
-            role: "user",
-            content: experimental_attachments ? [
-              { type: "text", text: content },
-              ...attachmentsToParts(experimental_attachments)
-            ] : content
-          });
-        } else {
-          const textParts = message.parts.filter((part) => part.type === "text").map((part) => ({
-            type: "text",
-            text: part.text
-          }));
-          coreMessages.push({
-            role: "user",
-            content: experimental_attachments ? [...textParts, ...attachmentsToParts(experimental_attachments)] : textParts
-          });
-        }
+        coreMessages.push({
+          role: "user",
+          content: message.parts.filter(
+            (part) => part.type === "text" || part.type === "file"
+          ).map(
+            (part) => part.type === "file" ? {
+              type: "file",
+              mediaType: part.mediaType,
+              filename: part.filename,
+              data: part.url
+            } : part
+          )
+        });
         break;
       }
       case "assistant": {
         if (message.parts != null) {
           let processBlock2 = function() {
-            var _a18;
             const content2 = [];
             for (const part of block) {
               switch (part.type) {
@@ -3140,9 +3077,8 @@ function convertToCoreMessages(messages, options) {
                 case "file": {
                   content2.push({
                     type: "file",
-                    data: part.data,
-                    mediaType: (_a18 = part.mediaType) != null ? _a18 : part.mimeType
-                    // TODO migration, remove
+                    mediaType: part.mediaType,
+                    data: part.url
                   });
                   break;
                 }
@@ -3255,6 +3191,7 @@ function convertToCoreMessages(messages, options) {
   }
   return coreMessages;
 }
+var convertToCoreMessages = convertToModelMessages;
 
 // core/prompt/detect-prompt-type.ts
 function detectPromptType(prompt) {
@@ -3343,7 +3280,6 @@ var imagePartSchema = z5.object({
   type: z5.literal("image"),
   image: z5.union([dataContentSchema, z5.instanceof(URL)]),
   mediaType: z5.string().optional(),
-  mimeType: z5.string().optional(),
   providerOptions: providerMetadataSchema.optional()
 });
 var filePartSchema = z5.object({
@@ -3351,7 +3287,6 @@ var filePartSchema = z5.object({
   data: z5.union([dataContentSchema, z5.instanceof(URL)]),
   filename: z5.string().optional(),
   mediaType: z5.string(),
-  mimeType: z5.string().optional(),
   providerOptions: providerMetadataSchema.optional()
 });
 var reasoningPartSchema = z5.object({
@@ -3377,12 +3312,15 @@ var toolResultPartSchema = z5.object({
 });
 
 // core/prompt/message.ts
-var coreSystemMessageSchema = z6.object({
-  role: z6.literal("system"),
-  content: z6.string(),
-  providerOptions: providerMetadataSchema.optional()
-});
-var coreUserMessageSchema = z6.object({
+var systemModelMessageSchema = z6.object(
+  {
+    role: z6.literal("system"),
+    content: z6.string(),
+    providerOptions: providerMetadataSchema.optional()
+  }
+);
+var coreSystemMessageSchema = systemModelMessageSchema;
+var userModelMessageSchema = z6.object({
   role: z6.literal("user"),
   content: z6.union([
     z6.string(),
@@ -3390,7 +3328,8 @@ var coreUserMessageSchema = z6.object({
   ]),
   providerOptions: providerMetadataSchema.optional()
 });
-var coreAssistantMessageSchema = z6.object({
+var coreUserMessageSchema = userModelMessageSchema;
+var assistantModelMessageSchema = z6.object({
   role: z6.literal("assistant"),
   content: z6.union([
     z6.string(),
@@ -3405,17 +3344,20 @@ var coreAssistantMessageSchema = z6.object({
   ]),
   providerOptions: providerMetadataSchema.optional()
 });
-var coreToolMessageSchema = z6.object({
+var coreAssistantMessageSchema = assistantModelMessageSchema;
+var toolModelMessageSchema = z6.object({
   role: z6.literal("tool"),
   content: z6.array(toolResultPartSchema),
   providerOptions: providerMetadataSchema.optional()
 });
-var coreMessageSchema = z6.union([
-  coreSystemMessageSchema,
-  coreUserMessageSchema,
-  coreAssistantMessageSchema,
-  coreToolMessageSchema
+var coreToolMessageSchema = toolModelMessageSchema;
+var modelMessageSchema = z6.union([
+  systemModelMessageSchema,
+  userModelMessageSchema,
+  assistantModelMessageSchema,
+  toolModelMessageSchema
 ]);
+var coreMessageSchema = modelMessageSchema;
 
 // core/prompt/standardize-prompt.ts
 async function standardizePrompt({
@@ -3462,10 +3404,10 @@ async function standardizePrompt({
     if (promptType === "other") {
       throw new InvalidPromptError({
         prompt,
-        message: "messages must be an array of CoreMessage or UIMessage"
+        message: "messages must be an array of ModelMessage or UIMessage"
       });
     }
-    const messages = promptType === "ui-messages" ? convertToCoreMessages(prompt.messages, {
+    const messages = promptType === "ui-messages" ? convertToModelMessages(prompt.messages, {
       tools
     }) : prompt.messages;
     if (messages.length === 0) {
@@ -3476,12 +3418,12 @@ async function standardizePrompt({
     }
     const validationResult = await safeValidateTypes({
       value: messages,
-      schema: z7.array(coreMessageSchema)
+      schema: z7.array(modelMessageSchema)
     });
     if (!validationResult.success) {
       throw new InvalidPromptError({
         prompt,
-        message: "messages must be an array of CoreMessage or UIMessage",
+        message: "messages must be an array of ModelMessage or UIMessage",
         cause: validationResult.error
       });
     }
@@ -3493,25 +3435,6 @@ async function standardizePrompt({
   throw new Error("unreachable");
 }
 
-// core/types/usage.ts
-function calculateLanguageModelUsage2({
-  inputTokens,
-  outputTokens
-}) {
-  return {
-    promptTokens: inputTokens != null ? inputTokens : NaN,
-    completionTokens: outputTokens != null ? outputTokens : NaN,
-    totalTokens: (inputTokens != null ? inputTokens : 0) + (outputTokens != null ? outputTokens : 0)
-  };
-}
-function addLanguageModelUsage(usage1, usage2) {
-  return {
-    promptTokens: usage1.promptTokens + usage2.promptTokens,
-    completionTokens: usage1.completionTokens + usage2.completionTokens,
-    totalTokens: usage1.totalTokens + usage2.totalTokens
-  };
-}
-
 // core/generate-object/output-strategy.ts
 import {
   isJSONArray,
@@ -4049,7 +3972,7 @@ async function generateObject(options) {
               throw new NoObjectGeneratedError({
                 message: "No object generated: the model did not return a response.",
                 response: responseData,
-                usage: calculateLanguageModelUsage2(result2.usage),
+                usage: result2.usage,
                 finishReason: result2.finishReason
               });
             }
@@ -4093,7 +4016,7 @@ async function generateObject(options) {
             cause: parseResult.error,
             text: result2,
             response,
-            usage: calculateLanguageModelUsage2(usage),
+            usage,
             finishReason
           });
         }
@@ -4102,7 +4025,7 @@ async function generateObject(options) {
           {
             text: result2,
             response,
-            usage: calculateLanguageModelUsage2(usage)
+            usage
           }
         );
         if (!validationResult.success) {
@@ -4111,7 +4034,7 @@ async function generateObject(options) {
             cause: validationResult.error,
             text: result2,
             response,
-            usage: calculateLanguageModelUsage2(usage),
+            usage,
             finishReason
           });
         }
@@ -4151,7 +4074,7 @@ async function generateObject(options) {
       return new DefaultGenerateObjectResult({
         object: object2,
         finishReason,
-        usage: calculateLanguageModelUsage2(usage),
+        usage,
         warnings,
         request,
         response,
@@ -4533,7 +4456,11 @@ var DefaultStreamObjectResult = class {
         );
         self.requestPromise.resolve(request != null ? request : {});
         let warnings;
-        let usage;
+        let usage = {
+          inputTokens: void 0,
+          outputTokens: void 0,
+          totalTokens: void 0
+        };
         let finishReason;
         let providerMetadata;
         let object2;
@@ -4613,7 +4540,7 @@ var DefaultStreamObjectResult = class {
                     controller.enqueue({ type: "text-delta", textDelta });
                   }
                   finishReason = chunk.finishReason;
-                  usage = calculateLanguageModelUsage2(chunk.usage);
+                  usage = chunk.usage;
                   providerMetadata = chunk.providerMetadata;
                   controller.enqueue({
                     ...chunk,
@@ -4675,14 +4602,17 @@ var DefaultStreamObjectResult = class {
                       "ai.response.id": fullResponse.id,
                       "ai.response.model": fullResponse.modelId,
                       "ai.response.timestamp": fullResponse.timestamp.toISOString(),
-                      "ai.usage.promptTokens": finalUsage.promptTokens,
-                      "ai.usage.completionTokens": finalUsage.completionTokens,
+                      "ai.usage.inputTokens": finalUsage.inputTokens,
+                      "ai.usage.outputTokens": finalUsage.outputTokens,
+                      "ai.usage.totalTokens": finalUsage.totalTokens,
+                      "ai.usage.reasoningTokens": finalUsage.reasoningTokens,
+                      "ai.usage.cachedInputTokens": finalUsage.cachedInputTokens,
                       // standardized gen-ai llm span attributes:
                       "gen_ai.response.finish_reasons": [finishReason],
                       "gen_ai.response.id": fullResponse.id,
                       "gen_ai.response.model": fullResponse.modelId,
-                      "gen_ai.usage.input_tokens": finalUsage.promptTokens,
-                      "gen_ai.usage.output_tokens": finalUsage.completionTokens
+                      "gen_ai.usage.input_tokens": finalUsage.inputTokens,
+                      "gen_ai.usage.output_tokens": finalUsage.outputTokens
                     }
                   })
                 );
@@ -4691,8 +4621,11 @@ var DefaultStreamObjectResult = class {
                   selectTelemetryAttributes({
                     telemetry,
                     attributes: {
-                      "ai.usage.promptTokens": finalUsage.promptTokens,
-                      "ai.usage.completionTokens": finalUsage.completionTokens,
+                      "ai.usage.inputTokens": finalUsage.inputTokens,
+                      "ai.usage.outputTokens": finalUsage.outputTokens,
+                      "ai.usage.totalTokens": finalUsage.totalTokens,
+                      "ai.usage.reasoningTokens": finalUsage.reasoningTokens,
+                      "ai.usage.cachedInputTokens": finalUsage.cachedInputTokens,
                       "ai.response.object": {
                         output: () => JSON.stringify(object2)
                       }
@@ -4922,6 +4855,26 @@ function prepareToolsAndToolChoice({
   };
 }
 
+// core/types/usage.ts
+function addLanguageModelUsage(usage1, usage2) {
+  return {
+    inputTokens: addTokenCounts(usage1.inputTokens, usage2.inputTokens),
+    outputTokens: addTokenCounts(usage1.outputTokens, usage2.outputTokens),
+    totalTokens: addTokenCounts(usage1.totalTokens, usage2.totalTokens),
+    reasoningTokens: addTokenCounts(
+      usage1.reasoningTokens,
+      usage2.reasoningTokens
+    ),
+    cachedInputTokens: addTokenCounts(
+      usage1.cachedInputTokens,
+      usage2.cachedInputTokens
+    )
+  };
+}
+function addTokenCounts(tokenCount1, tokenCount2) {
+  return tokenCount1 == null && tokenCount2 == null ? void 0 : (tokenCount1 != null ? tokenCount1 : 0) + (tokenCount2 != null ? tokenCount2 : 0);
+}
+
 // core/util/split-on-last-whitespace.ts
 var lastWhitespaceRegexp = /^([\s\S]*?)(\s+)(\S*)$/;
 function splitOnLastWhitespace(text2) {
@@ -5246,9 +5199,9 @@ async function generateText({
       const sources = [];
       const steps = [];
       let usage = {
-        completionTokens: 0,
-        promptTokens: 0,
-        totalTokens: 0
+        inputTokens: void 0,
+        outputTokens: void 0,
+        totalTokens: void 0
       };
       let stepType = "initial";
       do {
@@ -5389,10 +5342,7 @@ async function generateText({
           messages: stepInputMessages,
           abortSignal
         });
-        const currentUsage = calculateLanguageModelUsage2(
-          currentModelResponse.usage
-        );
-        usage = addLanguageModelUsage(usage, currentUsage);
+        usage = addLanguageModelUsage(usage, currentModelResponse.usage);
         let nextStepType = "done";
         if (++stepCount < maxSteps) {
           if (continueSteps && currentModelResponse.finishReason === "length" && // only use continue when there are no tool calls:
@@ -5457,7 +5407,7 @@ async function generateText({
           toolCalls: currentToolCalls,
           toolResults: currentToolResults,
           finishReason: currentModelResponse.finishReason,
-          usage: currentUsage,
+          usage: currentModelResponse.usage,
           warnings: currentModelResponse.warnings,
           request: (_e = currentModelResponse.request) != null ? _e : {},
           response: {
@@ -5995,6 +5945,7 @@ function runToolsTransformation({
       const chunkType = chunk.type;
       switch (chunkType) {
         case "stream-start":
+        case "finish":
         case "text":
         case "reasoning":
         case "reasoning-part-finish":
@@ -6115,15 +6066,6 @@ function runToolsTransformation({
           }
           break;
         }
-        case "finish": {
-          finishChunk = {
-            type: "finish",
-            finishReason: chunk.finishReason,
-            usage: calculateLanguageModelUsage2(chunk.usage),
-            providerMetadata: chunk.providerMetadata
-          };
-          break;
-        }
         default: {
           const _exhaustiveCheck = chunkType;
           throw new Error(`Unhandled chunk type: ${_exhaustiveCheck}`);
@@ -6486,9 +6428,9 @@ var DefaultStreamTextResult = class {
           self.reasoningDetailsPromise.resolve(lastStep.reasoning);
           const finishReason = recordedFinishReason != null ? recordedFinishReason : "unknown";
           const usage = recordedUsage != null ? recordedUsage : {
-            completionTokens: NaN,
-            promptTokens: NaN,
-            totalTokens: NaN
+            inputTokens: void 0,
+            outputTokens: void 0,
+            totalTokens: void 0
           };
           self.finishReasonPromise.resolve(finishReason);
           self.usagePromise.resolve(usage);
@@ -6524,8 +6466,11 @@ var DefaultStreamTextResult = class {
                     return ((_a18 = lastStep.toolCalls) == null ? void 0 : _a18.length) ? JSON.stringify(lastStep.toolCalls) : void 0;
                   }
                 },
-                "ai.usage.promptTokens": usage.promptTokens,
-                "ai.usage.completionTokens": usage.completionTokens
+                "ai.usage.inputTokens": usage.inputTokens,
+                "ai.usage.outputTokens": usage.outputTokens,
+                "ai.usage.totalTokens": usage.totalTokens,
+                "ai.usage.reasoningTokens": usage.reasoningTokens,
+                "ai.usage.cachedInputTokens": usage.cachedInputTokens
               }
             })
           );
@@ -6690,9 +6635,9 @@ var DefaultStreamTextResult = class {
           let activeReasoningPart2 = void 0;
           let stepFinishReason = "unknown";
           let stepUsage = {
-            promptTokens: 0,
-            completionTokens: 0,
-            totalTokens: 0
+            inputTokens: void 0,
+            outputTokens: void 0,
+            totalTokens: void 0
           };
           let stepProviderMetadata;
           let stepFirstChunk = true;
@@ -6721,7 +6666,7 @@ var DefaultStreamTextResult = class {
             transformedStream.pipeThrough(
               new TransformStream({
                 async transform(chunk, controller) {
-                  var _a17, _b, _c;
+                  var _a17, _b, _c, _d;
                   if (chunk.type === "stream-start") {
                     warnings = chunk.warnings;
                     return;
@@ -6817,7 +6762,7 @@ var DefaultStreamTextResult = class {
                       doStreamSpan.addEvent("ai.stream.finish");
                       doStreamSpan.setAttributes({
                         "ai.response.msToFinish": msToFinish,
-                        "ai.response.avgCompletionTokensPerSecond": 1e3 * stepUsage.completionTokens / msToFinish
+                        "ai.response.avgOutputTokensPerSecond": 1e3 * ((_d = stepUsage.outputTokens) != null ? _d : 0) / msToFinish
                       });
                       break;
                     }
@@ -6880,14 +6825,17 @@ var DefaultStreamTextResult = class {
                           "ai.response.id": stepResponse.id,
                           "ai.response.model": stepResponse.modelId,
                           "ai.response.timestamp": stepResponse.timestamp.toISOString(),
-                          "ai.usage.promptTokens": stepUsage.promptTokens,
-                          "ai.usage.completionTokens": stepUsage.completionTokens,
+                          "ai.usage.inputTokens": stepUsage.inputTokens,
+                          "ai.usage.outputTokens": stepUsage.outputTokens,
+                          "ai.usage.totalTokens": stepUsage.totalTokens,
+                          "ai.usage.reasoningTokens": stepUsage.reasoningTokens,
+                          "ai.usage.cachedInputTokens": stepUsage.cachedInputTokens,
                           // standardized gen-ai llm span attributes:
                           "gen_ai.response.finish_reasons": [stepFinishReason],
                           "gen_ai.response.id": stepResponse.id,
                           "gen_ai.response.model": stepResponse.modelId,
-                          "gen_ai.usage.input_tokens": stepUsage.promptTokens,
-                          "gen_ai.usage.output_tokens": stepUsage.completionTokens
+                          "gen_ai.usage.input_tokens": stepUsage.inputTokens,
+                          "gen_ai.usage.output_tokens": stepUsage.outputTokens
                         }
                       })
                     );
@@ -6969,9 +6917,9 @@ var DefaultStreamTextResult = class {
           currentStep: 0,
           responseMessages: [],
           usage: {
-            promptTokens: 0,
-            completionTokens: 0,
-            totalTokens: 0
+            inputTokens: void 0,
+            outputTokens: void 0,
+            totalTokens: void 0
           },
           previousStepText: "",
           stepType: "initial",
@@ -7132,8 +7080,8 @@ var DefaultStreamTextResult = class {
               controller.enqueue(
                 // TODO update protocol to v2 or replace with event stream
                 formatDataStreamPart("file", {
-                  mimeType: chunk.file.mediaType,
-                  data: chunk.file.base64
+                  mediaType: chunk.file.mediaType,
+                  url: `data:${chunk.file.mediaType};base64,${chunk.file.base64}`
                 })
               );
               break;
@@ -7200,8 +7148,11 @@ var DefaultStreamTextResult = class {
                 formatDataStreamPart("finish_step", {
                   finishReason: chunk.finishReason,
                   usage: sendUsage ? {
-                    promptTokens: chunk.usage.promptTokens,
-                    completionTokens: chunk.usage.completionTokens
+                    inputTokens: chunk.usage.inputTokens,
+                    outputTokens: chunk.usage.outputTokens,
+                    totalTokens: chunk.usage.totalTokens,
+                    reasoningTokens: chunk.usage.reasoningTokens,
+                    cachedInputTokens: chunk.usage.cachedInputTokens
                   } : void 0,
                   isContinued: chunk.isContinued
                 })
@@ -7214,8 +7165,11 @@ var DefaultStreamTextResult = class {
                   formatDataStreamPart("finish_message", {
                     finishReason: chunk.finishReason,
                     usage: sendUsage ? {
-                      promptTokens: chunk.usage.promptTokens,
-                      completionTokens: chunk.usage.completionTokens
+                      inputTokens: chunk.usage.inputTokens,
+                      outputTokens: chunk.usage.outputTokens,
+                      totalTokens: chunk.usage.totalTokens,
+                      reasoningTokens: chunk.usage.reasoningTokens,
+                      cachedInputTokens: chunk.usage.cachedInputTokens
                     } : void 0
                   })
                 );
@@ -7355,16 +7309,16 @@ var DefaultGeneratedAudioFile = class extends DefaultGeneratedFile {
     super({ data, mediaType });
     let format = "mp3";
     if (mediaType) {
-      const mimeTypeParts = mediaType.split("/");
-      if (mimeTypeParts.length === 2) {
+      const mediaTypeParts = mediaType.split("/");
+      if (mediaTypeParts.length === 2) {
         if (mediaType !== "audio/mpeg") {
-          format = mimeTypeParts[1];
+          format = mediaTypeParts[1];
         }
       }
     }
     if (!format) {
       throw new Error(
-        "Audio format must be provided or determinable from mimeType"
+        "Audio format must be provided or determinable from media type"
       );
     }
     this.format = format;
@@ -7793,7 +7747,7 @@ function appendResponseMessages({
   responseMessages,
   _internal: { currentDate = () => /* @__PURE__ */ new Date() } = {}
 }) {
-  var _a17, _b, _c;
+  var _a17, _b;
   const clonedMessages = structuredClone(messages);
   for (const message of responseMessages) {
     const role = message.role;
@@ -7857,8 +7811,8 @@ function appendResponseMessages({
                 }
                 parts.push({
                   type: "file",
-                  mediaType: (_a17 = part.mediaType) != null ? _a17 : part.mimeType,
-                  data: convertDataContentToBase64String(part.data)
+                  mediaType: part.mediaType,
+                  url: `data:${part.mediaType};base64,${convertDataContentToBase64String(part.data)}`
                 });
                 break;
             }
@@ -7867,9 +7821,8 @@ function appendResponseMessages({
         if (isLastMessageAssistant) {
           const maxStep = extractMaxToolInvocationStep(
             getToolInvocations(lastMessage)
-            // TODO remove once Message is removed
           );
-          (_b = lastMessage.parts) != null ? _b : lastMessage.parts = [];
+          (_a17 = lastMessage.parts) != null ? _a17 : lastMessage.parts = [];
           lastMessage.content = textContent;
           lastMessage.parts.push(...parts);
           getToolInvocationsForStep2(maxStep === void 0 ? 0 : maxStep + 1).map((call) => ({
@@ -7902,12 +7855,11 @@ function appendResponseMessages({
             `Tool result must follow an assistant message: ${lastMessage.role}`
           );
         }
-        (_c = lastMessage.parts) != null ? _c : lastMessage.parts = [];
+        (_b = lastMessage.parts) != null ? _b : lastMessage.parts = [];
         for (const contentPart of message.content) {
-          const toolCall = getToolInvocations(
-            lastMessage
-            // TODO remove once Message is removed
-          ).find((call) => call.toolCallId === contentPart.toolCallId);
+          const toolCall = getToolInvocations(lastMessage).find(
+            (call) => call.toolCallId === contentPart.toolCallId
+          );
           const toolCallPart = lastMessage.parts.find(
             (part) => part.type === "tool-invocation" && part.toolInvocation.toolCallId === contentPart.toolCallId
           );
@@ -8795,9 +8747,12 @@ export {
   appendClientMessage,
   appendResponseMessages,
   asSchema,
+  assistantModelMessageSchema,
   callChatApi,
   callCompletionApi,
+  convertFileListToFileUIParts,
   convertToCoreMessages,
+  convertToModelMessages,
   coreAssistantMessageSchema,
   coreMessageSchema,
   coreSystemMessageSchema,
@@ -8829,10 +8784,10 @@ export {
   isAssistantMessageWithCompletedToolCalls,
   isDeepEqualData,
   jsonSchema,
+  modelMessageSchema,
   parseDataStreamPart,
   parsePartialJson,
   pipeDataStreamToResponse,
-  prepareAttachmentsForRequest,
   processDataStream,
   processTextStream,
   shouldResubmitMessages,
@@ -8841,8 +8796,11 @@ export {
   smoothStream,
   streamObject,
   streamText,
+  systemModelMessageSchema,
   tool,
+  toolModelMessageSchema,
   updateToolCallResult,
+  userModelMessageSchema,
   wrapLanguageModel
 };
 //# sourceMappingURL=index.mjs.map