ai 4.3.8 → 4.3.10

package/dist/index.mjs

@@ -862,6 +862,7 @@ var DefaultGeneratedFileWithType = class extends DefaultGeneratedFile {
 };
 
 // core/util/detect-mimetype.ts
+import { convertBase64ToUint8Array as convertBase64ToUint8Array2 } from "@ai-sdk/provider-utils";
 var imageMimeTypeSignatures = [
   {
     mimeType: "image/gif",
@@ -967,12 +968,26 @@ var audioMimeTypeSignatures = [
     base64Prefix: "ZnR5cA"
   }
 ];
+var stripID3 = (data) => {
+  const bytes = typeof data === "string" ? convertBase64ToUint8Array2(data) : data;
+  const id3Size = (bytes[6] & 127) << 21 | (bytes[7] & 127) << 14 | (bytes[8] & 127) << 7 | bytes[9] & 127;
+  return bytes.slice(id3Size + 10);
+};
+function stripID3TagsIfPresent(data) {
+  const hasId3 = typeof data === "string" && data.startsWith("SUQz") || typeof data !== "string" && data.length > 10 && data[0] === 73 && // 'I'
+  data[1] === 68 && // 'D'
+  data[2] === 51;
+  return hasId3 ? stripID3(data) : data;
+}
 function detectMimeType({
   data,
   signatures
 }) {
+  const processedData = stripID3TagsIfPresent(data);
   for (const signature of signatures) {
-    if (typeof data === "string" ? data.startsWith(signature.base64Prefix) : data.length >= signature.bytesPrefix.length && signature.bytesPrefix.every((byte, index) => data[index] === byte)) {
+    if (typeof processedData === "string" ? processedData.startsWith(signature.base64Prefix) : processedData.length >= signature.bytesPrefix.length && signature.bytesPrefix.every(
+      (byte, index) => processedData[index] === byte
+    )) {
       return signature.mimeType;
     }
   }
@@ -1144,7 +1159,7 @@ async function download({ url }) {
 
 // core/prompt/data-content.ts
 import {
-  convertBase64ToUint8Array as convertBase64ToUint8Array2,
+  convertBase64ToUint8Array as convertBase64ToUint8Array3,
   convertUint8ArrayToBase64 as convertUint8ArrayToBase642
 } from "@ai-sdk/provider-utils";
 
@@ -1200,7 +1215,7 @@ function convertDataContentToUint8Array(content) {
   }
   if (typeof content === "string") {
     try {
-      return convertBase64ToUint8Array2(content);
+      return convertBase64ToUint8Array3(content);
     } catch (error) {
       throw new InvalidDataContentError({
         message: "Invalid data content. Content string is not a base64-encoded media.",
@@ -4091,6 +4106,7 @@ async function generateText({
   experimental_providerMetadata,
   providerOptions = experimental_providerMetadata,
   experimental_activeTools: activeTools,
+  experimental_prepareStep: prepareStep,
   experimental_repairToolCall: repairToolCall,
   _internal: {
     generateId: generateId3 = originalGenerateId3,
@@ -4133,6 +4149,9 @@ async function generateText({
           telemetry
         }),
         ...baseTelemetryAttributes,
+        // model:
+        "ai.model.provider": model.provider,
+        "ai.model.id": model.modelId,
         // specific settings that only make sense on the outer level:
         "ai.prompt": {
           input: () => JSON.stringify({ system, prompt, messages })
@@ -4142,11 +4161,7 @@ async function generateText({
     }),
     tracer,
     fn: async (span) => {
-      var _a18, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k;
-      const mode = {
-        type: "regular",
-        ...prepareToolsAndToolChoice({ tools, toolChoice, activeTools })
-      };
+      var _a18, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n;
       const callSettings = prepareCallSettings(settings);
       let currentModelResponse;
       let currentToolCalls = [];
@@ -4169,16 +4184,33 @@ async function generateText({
           ...initialPrompt.messages,
           ...responseMessages
         ];
+        const prepareStepResult = await (prepareStep == null ? void 0 : prepareStep({
+          model,
+          steps,
+          maxSteps,
+          stepNumber: stepCount
+        }));
+        const stepToolChoice = (_a18 = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _a18 : toolChoice;
+        const stepActiveTools = (_b = prepareStepResult == null ? void 0 : prepareStepResult.experimental_activeTools) != null ? _b : activeTools;
+        const stepModel = (_c = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _c : model;
         const promptMessages = await convertToLanguageModelPrompt({
           prompt: {
             type: promptFormat,
             system: initialPrompt.system,
             messages: stepInputMessages
           },
-          modelSupportsImageUrls: model.supportsImageUrls,
-          modelSupportsUrl: (_a18 = model.supportsUrl) == null ? void 0 : _a18.bind(model)
+          modelSupportsImageUrls: stepModel.supportsImageUrls,
+          modelSupportsUrl: (_d = stepModel.supportsUrl) == null ? void 0 : _d.bind(stepModel)
           // support 'this' context
         });
+        const mode = {
+          type: "regular",
+          ...prepareToolsAndToolChoice({
+            tools,
+            toolChoice: stepToolChoice,
+            activeTools: stepActiveTools
+          })
+        };
         currentModelResponse = await retry(
           () => recordSpan({
             name: "ai.generateText.doGenerate",
@@ -4190,6 +4222,10 @@ async function generateText({
                   telemetry
                 }),
                 ...baseTelemetryAttributes,
+                // model:
+                "ai.model.provider": stepModel.provider,
+                "ai.model.id": stepModel.modelId,
+                // prompt:
                 "ai.prompt.format": { input: () => promptFormat },
                 "ai.prompt.messages": {
                   input: () => JSON.stringify(promptMessages)
@@ -4205,8 +4241,8 @@ async function generateText({
                   input: () => mode.toolChoice != null ? JSON.stringify(mode.toolChoice) : void 0
                 },
                 // standardized gen-ai llm span attributes:
-                "gen_ai.system": model.provider,
-                "gen_ai.request.model": model.modelId,
+                "gen_ai.system": stepModel.provider,
+                "gen_ai.request.model": stepModel.modelId,
                 "gen_ai.request.frequency_penalty": settings.frequencyPenalty,
                 "gen_ai.request.max_tokens": settings.maxTokens,
                 "gen_ai.request.presence_penalty": settings.presencePenalty,
@@ -4219,7 +4255,7 @@ async function generateText({
             tracer,
             fn: async (span2) => {
               var _a19, _b2, _c2, _d2, _e2, _f2;
-              const result = await model.doGenerate({
+              const result = await stepModel.doGenerate({
                 mode,
                 ...callSettings,
                 inputFormat: promptFormat,
@@ -4232,7 +4268,7 @@ async function generateText({
               const responseData = {
                 id: (_b2 = (_a19 = result.response) == null ? void 0 : _a19.id) != null ? _b2 : generateId3(),
                 timestamp: (_d2 = (_c2 = result.response) == null ? void 0 : _c2.timestamp) != null ? _d2 : currentDate(),
-                modelId: (_f2 = (_e2 = result.response) == null ? void 0 : _e2.modelId) != null ? _f2 : model.modelId
+                modelId: (_f2 = (_e2 = result.response) == null ? void 0 : _e2.modelId) != null ? _f2 : stepModel.modelId
               };
               span2.setAttributes(
                 selectTelemetryAttributes({
@@ -4264,7 +4300,7 @@ async function generateText({
           })
         );
         currentToolCalls = await Promise.all(
-          ((_b = currentModelResponse.toolCalls) != null ? _b : []).map(
+          ((_e = currentModelResponse.toolCalls) != null ? _e : []).map(
             (toolCall) => parseToolCall({
               toolCall,
               tools,
@@ -4299,7 +4335,7 @@ async function generateText({
             nextStepType = "tool-result";
           }
         }
-        const originalText = (_c = currentModelResponse.text) != null ? _c : "";
+        const originalText = (_f = currentModelResponse.text) != null ? _f : "";
         const stepTextLeadingWhitespaceTrimmed = stepType === "continue" && // only for continue steps
         text2.trimEnd() !== text2 ? originalText.trimStart() : originalText;
         const stepText = nextStepType === "continue" ? removeTextAfterLastWhitespace(stepTextLeadingWhitespaceTrimmed) : stepTextLeadingWhitespaceTrimmed;
@@ -4307,7 +4343,7 @@ async function generateText({
         currentReasoningDetails = asReasoningDetails(
           currentModelResponse.reasoning
         );
-        sources.push(...(_d = currentModelResponse.sources) != null ? _d : []);
+        sources.push(...(_g = currentModelResponse.sources) != null ? _g : []);
         if (stepType === "continue") {
           const lastMessage = responseMessages[responseMessages.length - 1];
           if (typeof lastMessage.content === "string") {
@@ -4339,18 +4375,18 @@ async function generateText({
           reasoning: asReasoningText(currentReasoningDetails),
           reasoningDetails: currentReasoningDetails,
           files: asFiles(currentModelResponse.files),
-          sources: (_e = currentModelResponse.sources) != null ? _e : [],
+          sources: (_h = currentModelResponse.sources) != null ? _h : [],
           toolCalls: currentToolCalls,
           toolResults: currentToolResults,
           finishReason: currentModelResponse.finishReason,
           usage: currentUsage,
           warnings: currentModelResponse.warnings,
           logprobs: currentModelResponse.logprobs,
-          request: (_f = currentModelResponse.request) != null ? _f : {},
+          request: (_i = currentModelResponse.request) != null ? _i : {},
           response: {
             ...currentModelResponse.response,
-            headers: (_g = currentModelResponse.rawResponse) == null ? void 0 : _g.headers,
-            body: (_h = currentModelResponse.rawResponse) == null ? void 0 : _h.body,
+            headers: (_j = currentModelResponse.rawResponse) == null ? void 0 : _j.headers,
+            body: (_k = currentModelResponse.rawResponse) == null ? void 0 : _k.body,
             // deep clone msgs to avoid mutating past messages in multi-step:
             messages: structuredClone(responseMessages)
           },
@@ -4402,11 +4438,11 @@ async function generateText({
         finishReason: currentModelResponse.finishReason,
         usage,
         warnings: currentModelResponse.warnings,
-        request: (_i = currentModelResponse.request) != null ? _i : {},
+        request: (_l = currentModelResponse.request) != null ? _l : {},
         response: {
           ...currentModelResponse.response,
-          headers: (_j = currentModelResponse.rawResponse) == null ? void 0 : _j.headers,
-          body: (_k = currentModelResponse.rawResponse) == null ? void 0 : _k.body,
+          headers: (_m = currentModelResponse.rawResponse) == null ? void 0 : _m.headers,
+          body: (_n = currentModelResponse.rawResponse) == null ? void 0 : _n.body,
           messages: responseMessages
         },
         logprobs: currentModelResponse.logprobs,