ai 7.0.92 → 7.0.94

package/dist/index.js

@@ -7,7 +7,7 @@ var __export = (target, all) => {
 // src/index.ts
 import {
   createGateway,
-  gateway as gateway2
+  gateway as gateway3
 } from "@ai-sdk/gateway";
 import {
   asSchema as asSchema8,
@@ -222,10 +222,12 @@ var NoImageGeneratedError = class extends (_b8 = AISDKError8, _a8 = symbol8, _b8
   constructor({
     message = "No image generated.",
     cause,
+    calls,
     responses
   }) {
     super({ name: name8, message, cause });
     this[_a8] = true;
+    this.calls = calls;
     this.responses = responses;
   }
   static isInstance(error) {
@@ -1176,7 +1178,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.92" : "0.0.0-test";
+var VERSION = true ? "7.0.94" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -2749,7 +2751,7 @@ function getOwn(obj, key) {
 import { filterNullable } from "@ai-sdk/provider-utils";
 function mergeAbortSignals(...signals) {
   const validSignals = filterNullable(...signals).map(
-    (signal) => signal instanceof AbortSignal ? signal : AbortSignal.timeout(signal)
+    (signal) => typeof signal === "number" ? AbortSignal.timeout(signal) : signal
   );
   return validSignals.length === 0 ? void 0 : validSignals.length === 1 ? validSignals[0] : AbortSignal.any(validSignals);
 }
@@ -2811,13 +2813,14 @@ var retryWithExponentialBackoffRespectingRetryHeaders = ({
   maxRetries = 2,
   initialDelayInMs = 2e3,
   backoffFactor = 2,
-  abortSignal
+  abortSignal,
+  additionalRetryableError
 } = {}) => retryWithExponentialBackoff({
   maxRetries,
   initialDelayInMs,
   backoffFactor,
   abortSignal,
-  shouldRetry: (error) => error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true),
+  shouldRetry: async (error) => error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true) || additionalRetryableError != null && await additionalRetryableError(error),
   getDelayInMs: ({ error, exponentialBackoffDelay }) => getRetryDelayInMs({
     error,
     exponentialBackoffDelay
@@ -2829,6 +2832,7 @@ var retryWithExponentialBackoffRespectingRetryHeaders = ({
 function prepareRetries({
   maxRetries,
   abortSignal,
+  additionalRetryableError,
   parameter = "maxRetries",
   defaultMaxRetries = 2
 }) {
@@ -2853,7 +2857,8 @@ function prepareRetries({
     maxRetries: maxRetriesResult,
     retry: retryWithExponentialBackoffRespectingRetryHeaders({
       maxRetries: maxRetriesResult,
-      abortSignal
+      abortSignal,
+      additionalRetryableError
     })
   };
 }
@@ -3780,9 +3785,20 @@ var object2 = ({
 };
 var array2 = ({
   element: inputElementSchema,
+  minItems,
+  maxItems,
   name: name25,
   description
 }) => {
+  validateArrayBound({ name: "minItems", value: minItems });
+  validateArrayBound({ name: "maxItems", value: maxItems });
+  if (minItems != null && maxItems != null && minItems > maxItems) {
+    throw new InvalidArgumentError({
+      parameter: "minItems",
+      value: minItems,
+      message: "minItems must be less than or equal to maxItems"
+    });
+  }
   const elementSchema = asSchema2(inputElementSchema);
   return {
     name: "array",
@@ -3802,7 +3818,12 @@ var array2 = ({
           ...$defs != null && { $defs },
           type: "object",
           properties: {
-            elements: { type: "array", items: itemSchema }
+            elements: {
+              type: "array",
+              items: itemSchema,
+              ...minItems != null && { minItems },
+              ...maxItems != null && { maxItems }
+            }
           },
           required: ["elements"],
           additionalProperties: false
@@ -3837,6 +3858,21 @@ var array2 = ({
           finishReason: context.finishReason
         });
       }
+      const lengthValidationError = getArrayLengthValidationError({
+        value: outerValue.elements,
+        minItems,
+        maxItems
+      });
+      if (lengthValidationError != null) {
+        throw new NoObjectGeneratedError({
+          message: "No object generated: response did not match schema.",
+          cause: lengthValidationError,
+          text: text2,
+          response: context.response,
+          usage: context.usage,
+          finishReason: context.finishReason
+        });
+      }
       const validatedElements = [];
       for (const element of outerValue.elements) {
         const validationResult = await safeValidateTypes2({
@@ -3891,6 +3927,15 @@ var array2 = ({
         transform({ partialOutput }, controller) {
           if (partialOutput != null) {
             for (; publishedElements < partialOutput.length; publishedElements++) {
+              if (maxItems != null && publishedElements >= maxItems) {
+                controller.error(
+                  getArrayLengthValidationError({
+                    value: partialOutput,
+                    maxItems
+                  })
+                );
+                return;
+              }
               controller.enqueue(partialOutput[publishedElements]);
             }
           }
@@ -3899,6 +3944,47 @@ var array2 = ({
     }
   };
 };
+function validateArrayBound({
+  name: name25,
+  value
+}) {
+  if (value == null) {
+    return;
+  }
+  if (!Number.isInteger(value)) {
+    throw new InvalidArgumentError({
+      parameter: name25,
+      value,
+      message: `${name25} must be an integer`
+    });
+  }
+  if (value < 0) {
+    throw new InvalidArgumentError({
+      parameter: name25,
+      value,
+      message: `${name25} must be greater than or equal to 0`
+    });
+  }
+}
+function getArrayLengthValidationError({
+  value,
+  minItems,
+  maxItems
+}) {
+  if (minItems != null && value.length < minItems) {
+    return new TypeValidationError2({
+      value,
+      cause: `elements array must contain at least ${minItems} items`
+    });
+  }
+  if (maxItems != null && value.length > maxItems) {
+    return new TypeValidationError2({
+      value,
+      cause: `elements array must contain at most ${maxItems} items`
+    });
+  }
+  return void 0;
+}
 var choice = ({
   options: choiceOptions,
   name: name25,
@@ -8709,6 +8795,7 @@ async function streamLanguageModelCall({
       callId: effectiveCallId,
       provider: resolvedModel.provider,
       modelId: resolvedModel.modelId,
+      toolChoice: stepToolChoice,
       generateId: generateId4,
       now: now2,
       callStartTimestampMs,
@@ -8730,6 +8817,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
   callId,
   provider,
   modelId,
+  toolChoice,
   generateId: generateId4,
   now: now2,
   callStartTimestampMs,
@@ -8737,8 +8825,11 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
 }) {
   const toolCallsByToolCallId = /* @__PURE__ */ new Map();
   const modelCallContent = [];
+  const rawModelCallContent = [];
   const textPartIndexes = /* @__PURE__ */ new Map();
   const reasoningPartIndexes = /* @__PURE__ */ new Map();
+  const rawTextPartIndexes = /* @__PURE__ */ new Map();
+  const rawReasoningPartIndexes = /* @__PURE__ */ new Map();
   let responseId = generateId4();
   let responseModelId = modelId;
   let timeToFirstOutputMs;
@@ -8768,7 +8859,9 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
         case "text-start":
           upsertTextContentPart({
             content: modelCallContent,
+            rawContent: rawModelCallContent,
             partIndexes: textPartIndexes,
+            rawPartIndexes: rawTextPartIndexes,
             id: chunk.id,
             type: "text",
             providerMetadata: chunk.providerMetadata
@@ -8778,7 +8871,9 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
         case "text-delta":
           upsertTextContentPart({
             content: modelCallContent,
+            rawContent: rawModelCallContent,
             partIndexes: textPartIndexes,
+            rawPartIndexes: rawTextPartIndexes,
             id: chunk.id,
             type: "text",
             textDelta: chunk.delta,
@@ -8794,18 +8889,23 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
         case "text-end":
           upsertTextContentPart({
             content: modelCallContent,
+            rawContent: rawModelCallContent,
             partIndexes: textPartIndexes,
+            rawPartIndexes: rawTextPartIndexes,
             id: chunk.id,
             type: "text",
             providerMetadata: chunk.providerMetadata
           });
           textPartIndexes.delete(chunk.id);
+          rawTextPartIndexes.delete(chunk.id);
           controller.enqueue(chunk);
           break;
         case "reasoning-start":
           upsertTextContentPart({
             content: modelCallContent,
+            rawContent: rawModelCallContent,
             partIndexes: reasoningPartIndexes,
+            rawPartIndexes: rawReasoningPartIndexes,
             id: chunk.id,
             type: "reasoning",
             providerMetadata: chunk.providerMetadata
@@ -8815,7 +8915,9 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
         case "reasoning-delta":
           upsertTextContentPart({
             content: modelCallContent,
+            rawContent: rawModelCallContent,
             partIndexes: reasoningPartIndexes,
+            rawPartIndexes: rawReasoningPartIndexes,
             id: chunk.id,
             type: "reasoning",
             textDelta: chunk.delta,
@@ -8831,12 +8933,15 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
         case "reasoning-end":
           upsertTextContentPart({
             content: modelCallContent,
+            rawContent: rawModelCallContent,
             partIndexes: reasoningPartIndexes,
+            rawPartIndexes: rawReasoningPartIndexes,
             id: chunk.id,
             type: "reasoning",
             providerMetadata: chunk.providerMetadata
           });
           reasoningPartIndexes.delete(chunk.id);
+          rawReasoningPartIndexes.delete(chunk.id);
           controller.enqueue(chunk);
           break;
         case "file":
@@ -8850,6 +8955,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
             file,
             ...chunk.providerMetadata != null ? { providerMetadata: chunk.providerMetadata } : {}
           });
+          rawModelCallContent.push(chunk);
           controller.enqueue({
             type: chunk.type,
             file,
@@ -8903,9 +9009,26 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
             providerMetadata: chunk.providerMetadata,
             performance
           });
+          const enforcedToolChoice = toolChoice.type === "required" || toolChoice.type === "tool" ? toolChoice : void 0;
+          if (enforcedToolChoice != null && ![...toolCallsByToolCallId.values()].some(
+            (toolCall) => enforcedToolChoice.type === "required" || toolCall.toolName === enforcedToolChoice.toolName
+          )) {
+            controller.enqueue({
+              type: "error",
+              error: new ToolChoiceViolationError({
+                toolChoice: enforcedToolChoice,
+                finishReason: chunk.finishReason.unified,
+                provider,
+                modelId,
+                content: rawModelCallContent
+              })
+            });
+            break;
+          }
           break;
         }
         case "tool-call": {
+          rawModelCallContent.push(chunk);
           try {
             const toolCall = await parseToolCall({
               toolCall: chunk,
@@ -8939,6 +9062,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
           break;
         }
         case "tool-approval-request": {
+          rawModelCallContent.push(chunk);
           const toolCall = toolCallsByToolCallId.get(chunk.toolCallId);
           if (toolCall == null) {
             controller.enqueue({
@@ -8960,6 +9084,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
           break;
         }
         case "tool-result": {
+          rawModelCallContent.push(chunk);
           const toolName = chunk.toolName;
           const toolCall = toolCallsByToolCallId.get(chunk.toolCallId);
           const toolResultPart = chunk.isError ? {
@@ -9018,6 +9143,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({
         default:
           if (chunk.type === "custom" || chunk.type === "source") {
             modelCallContent.push(chunk);
+            rawModelCallContent.push(chunk);
           }
           controller.enqueue(chunk);
           break;
@@ -9045,7 +9171,9 @@ function calculateNearestRankPercentile(sortedValues, percentile) {
 }
 function upsertTextContentPart({
   content,
+  rawContent,
   partIndexes,
+  rawPartIndexes,
   id,
   type,
   textDelta,
@@ -9060,12 +9188,24 @@ function upsertTextContentPart({
     }) - 1;
     partIndexes.set(id, partIndex);
   }
+  let rawPartIndex = rawPartIndexes.get(id);
+  if (rawPartIndex == null) {
+    rawPartIndex = rawContent.push({
+      type,
+      text: "",
+      ...providerMetadata != null ? { providerMetadata } : {}
+    }) - 1;
+    rawPartIndexes.set(id, rawPartIndex);
+  }
   const part = content[partIndex];
+  const rawPart = rawContent[rawPartIndex];
   if (textDelta != null) {
     part.text += textDelta;
+    rawPart.text += textDelta;
   }
   if (providerMetadata != null) {
     part.providerMetadata = providerMetadata;
+    rawPart.providerMetadata = providerMetadata;
   }
 }
 
@@ -10409,14 +10549,15 @@ var DefaultStreamTextResult = class {
                   callbacks: onChunk
                 });
                 const error = wrapGatewayError(value.error);
+                const isToolChoiceViolation = ToolChoiceViolationError.isInstance(error);
                 let onErrorResult;
                 try {
                   onErrorResult = await onError({ error });
                 } catch (e) {
                 }
                 const callbackRequestedRetry = canRetryStreamViaOnError && typeof onErrorResult === "object" && onErrorResult != null && "retry" in onErrorResult && onErrorResult.retry === true;
-                const automaticRetry = automaticStreamRetryCount < streamRetries;
-                const callbackRetry = !automaticRetry && callbackRequestedRetry && callbackStreamRetryCount < 1;
+                const automaticRetry = !isToolChoiceViolation && automaticStreamRetryCount < streamRetries;
+                const callbackRetry = !isToolChoiceViolation && !automaticRetry && callbackRequestedRetry && callbackStreamRetryCount < 1;
                 if (!automaticRetry && !callbackRetry) {
                   flushBufferedAttemptParts();
                   errorsHandledForStreamRetry.add(value.error);
@@ -11589,7 +11730,7 @@ async function convertToModelMessages(messages, options) {
         if (message.parts != null) {
           let block = [];
           async function processBlock() {
-            var _a25, _b25, _c, _d, _e, _f, _g;
+            var _a25, _b25, _c, _d, _e, _f, _g, _h;
             if (block.length === 0) {
               return;
             }
@@ -11634,13 +11775,14 @@ async function convertToModelMessages(messages, options) {
               } else if (isToolUIPart(part)) {
                 const toolName = getToolName(part);
                 if (part.state !== "input-streaming") {
+                  const callProviderMetadata = (_a25 = part.callProviderMetadata) != null ? _a25 : part.state === "output-error" ? part.resultProviderMetadata : void 0;
                   content.push({
                     type: "tool-call",
                     toolCallId: part.toolCallId,
                     toolName,
-                    input: part.state === "output-error" ? (_a25 = part.input) != null ? _a25 : "rawInput" in part ? part.rawInput : void 0 : part.input,
+                    input: part.state === "output-error" ? (_b25 = part.input) != null ? _b25 : "rawInput" in part ? part.rawInput : void 0 : part.input,
                     providerExecuted: part.providerExecuted,
-                    ...part.callProviderMetadata != null ? { providerOptions: part.callProviderMetadata } : {}
+                    ...callProviderMetadata != null ? { providerOptions: callProviderMetadata } : {}
                   });
                   if (part.approval != null) {
                     content.push({
@@ -11653,7 +11795,7 @@ async function convertToModelMessages(messages, options) {
                     });
                   }
                   if (part.providerExecuted === true && part.state !== "approval-responded" && (part.state === "output-available" || part.state === "output-error")) {
-                    const resultProviderMetadata = (_b25 = part.resultProviderMetadata) != null ? _b25 : part.callProviderMetadata;
+                    const resultProviderMetadata = (_c = part.resultProviderMetadata) != null ? _c : part.callProviderMetadata;
                     content.push({
                       type: "tool-result",
                       toolCallId: part.toolCallId,
@@ -11670,7 +11812,7 @@ async function convertToModelMessages(messages, options) {
                   }
                 }
               } else if (isDataUIPart(part)) {
-                const dataPart = (_c = options == null ? void 0 : options.convertDataPart) == null ? void 0 : _c.call(
+                const dataPart = (_d = options == null ? void 0 : options.convertDataPart) == null ? void 0 : _d.call(
                   options,
                   part
                 );
@@ -11698,7 +11840,7 @@ async function convertToModelMessages(messages, options) {
               {
                 const content2 = [];
                 for (const toolPart of toolParts) {
-                  if (((_d = toolPart.approval) == null ? void 0 : _d.approved) != null) {
+                  if (((_e = toolPart.approval) == null ? void 0 : _e.approved) != null) {
                     content2.push({
                       type: "tool-approval-response",
                       approvalId: toolPart.approval.id,
@@ -11707,7 +11849,7 @@ async function convertToModelMessages(messages, options) {
                       providerExecuted: toolPart.providerExecuted
                     });
                   }
-                  if (toolPart.state === "approval-responded" && ((_e = toolPart.approval) == null ? void 0 : _e.approved) === false) {
+                  if (toolPart.state === "approval-responded" && ((_f = toolPart.approval) == null ? void 0 : _f.approved) === false) {
                     content2.push({
                       type: "tool-result",
                       toolCallId: toolPart.toolCallId,
@@ -11730,7 +11872,7 @@ async function convertToModelMessages(messages, options) {
                         toolName: getToolName(toolPart),
                         output: {
                           type: "error-text",
-                          value: (_g = (_f = toolPart.approval) == null ? void 0 : _f.reason) != null ? _g : "Tool call execution denied."
+                          value: (_h = (_g = toolPart.approval) == null ? void 0 : _g.reason) != null ? _h : "Tool call execution denied."
                         },
                         ...toolPart.callProviderMetadata != null ? { providerOptions: toolPart.callProviderMetadata } : {}
                       });
@@ -11882,6 +12024,7 @@ var uiMessagesSchema = lazySchema2(
               type: z.literal("dynamic-tool"),
               toolName: z.string(),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("input-streaming"),
               input: z.unknown().optional(),
@@ -11895,6 +12038,7 @@ var uiMessagesSchema = lazySchema2(
               type: z.literal("dynamic-tool"),
               toolName: z.string(),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("input-available"),
               input: z.unknown(),
@@ -11908,6 +12052,7 @@ var uiMessagesSchema = lazySchema2(
               type: z.literal("dynamic-tool"),
               toolName: z.string(),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("approval-requested"),
               input: z.unknown(),
@@ -11929,6 +12074,7 @@ var uiMessagesSchema = lazySchema2(
               type: z.literal("dynamic-tool"),
               toolName: z.string(),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("approval-responded"),
               input: z.unknown(),
@@ -11950,6 +12096,7 @@ var uiMessagesSchema = lazySchema2(
               type: z.literal("dynamic-tool"),
               toolName: z.string(),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("output-available"),
               input: z.unknown(),
@@ -11973,6 +12120,7 @@ var uiMessagesSchema = lazySchema2(
               type: z.literal("dynamic-tool"),
               toolName: z.string(),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("output-error"),
               input: z.unknown().optional(),
@@ -11996,6 +12144,7 @@ var uiMessagesSchema = lazySchema2(
               type: z.literal("dynamic-tool"),
               toolName: z.string(),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("output-denied"),
               input: z.unknown(),
@@ -12016,6 +12165,7 @@ var uiMessagesSchema = lazySchema2(
             z.object({
               type: z.string().startsWith("tool-"),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("input-streaming"),
               providerExecuted: z.boolean().optional(),
@@ -12028,6 +12178,7 @@ var uiMessagesSchema = lazySchema2(
             z.object({
               type: z.string().startsWith("tool-"),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("input-available"),
               providerExecuted: z.boolean().optional(),
@@ -12040,6 +12191,7 @@ var uiMessagesSchema = lazySchema2(
             z.object({
               type: z.string().startsWith("tool-"),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("approval-requested"),
               input: z.unknown(),
@@ -12060,6 +12212,7 @@ var uiMessagesSchema = lazySchema2(
             z.object({
               type: z.string().startsWith("tool-"),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("approval-responded"),
               input: z.unknown(),
@@ -12080,6 +12233,7 @@ var uiMessagesSchema = lazySchema2(
             z.object({
               type: z.string().startsWith("tool-"),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("output-available"),
               providerExecuted: z.boolean().optional(),
@@ -12102,6 +12256,7 @@ var uiMessagesSchema = lazySchema2(
             z.object({
               type: z.string().startsWith("tool-"),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("output-error"),
               providerExecuted: z.boolean().optional(),
@@ -12124,6 +12279,7 @@ var uiMessagesSchema = lazySchema2(
             z.object({
               type: z.string().startsWith("tool-"),
               toolCallId: z.string(),
+              title: z.string().optional(),
               toolMetadata: toolMetadataSchema2.optional(),
               state: z.literal("output-denied"),
               providerExecuted: z.boolean().optional(),
@@ -12424,14 +12580,40 @@ async function pipeAgentUIStreamToResponse({
 import {
   UnsupportedFunctionalityError as UnsupportedFunctionalityError3
 } from "@ai-sdk/provider";
+import { gateway as gateway2 } from "@ai-sdk/gateway";
 import { withUserAgentSuffix as withUserAgentSuffix4 } from "@ai-sdk/provider-utils";
-async function startTextBatch({
-  model: modelArg,
+
+// src/util/is-deep-equal-data.ts
+function isDeepEqualData(obj1, obj2) {
+  if (obj1 === obj2) return true;
+  if (obj1 == null || obj2 == null) return false;
+  if (typeof obj1 !== "object" && typeof obj2 !== "object")
+    return obj1 === obj2;
+  if (obj1.constructor !== obj2.constructor) return false;
+  if (obj1 instanceof Date && obj2 instanceof Date) {
+    return obj1.getTime() === obj2.getTime();
+  }
+  if (Array.isArray(obj1)) {
+    if (obj1.length !== obj2.length) return false;
+    for (let i = 0; i < obj1.length; i++) {
+      if (!isDeepEqualData(obj1[i], obj2[i])) return false;
+    }
+    return true;
+  }
+  const keys1 = Object.keys(obj1);
+  const keys2 = Object.keys(obj2);
+  if (keys1.length !== keys2.length) return false;
+  for (const key of keys1) {
+    if (!keys2.includes(key)) return false;
+    if (!isDeepEqualData(obj1[key], obj2[key])) return false;
+  }
+  return true;
+}
+
+// src/batch/batch.ts
+async function startBatch({
+  provider,
   requests,
-  tools,
-  toolChoice,
-  toolOrder,
-  toolsContext,
   providerOptions,
   webhookUrl,
   abortSignal,
@@ -12439,37 +12621,49 @@ async function startTextBatch({
   timeout
 }) {
   validateRequests(requests);
-  const model = resolveBatchLanguageModel(modelArg);
+  const batchApi = resolveBatchApi(provider);
   const operationAbortSignal = mergeAbortSignals(
     abortSignal,
     getTotalTimeoutMs(timeout)
   );
-  const supportedUrls = await model.supportedUrls;
-  const preparedTools = await prepareTools({
-    tools,
-    toolOrder,
-    toolsContext
-  });
-  const preparedToolChoice = prepareToolChoice({ toolChoice });
+  const supportedUrls = await batchApi.supportedUrls;
   operationAbortSignal == null ? void 0 : operationAbortSignal.throwIfAborted();
   const normalizedRequests = [];
+  const toolsByName = /* @__PURE__ */ new Map();
   for (const request of requests) {
-    const standardizedPrompt = await standardizePrompt(request);
-    normalizedRequests.push({
-      id: request.id,
-      options: {
-        ...prepareLanguageModelCallOptions(request),
-        prompt: await convertToLanguageModelPrompt({
-          prompt: standardizedPrompt,
-          supportedUrls,
-          download: void 0,
-          provider: model.provider.split(".")[0]
-        }),
-        tools: preparedTools,
-        toolChoice: preparedToolChoice,
-        providerOptions: request.providerOptions
+    switch (request.type) {
+      case "text": {
+        const standardizedPrompt = await standardizePrompt(request);
+        const preparedTools = await prepareTools({
+          tools: request.tools,
+          toolOrder: request.toolOrder,
+          toolsContext: request.toolsContext
+        });
+        validateCompatibleTools({
+          requestId: request.id,
+          tools: preparedTools,
+          toolsByName
+        });
+        normalizedRequests.push({
+          id: request.id,
+          type: request.type,
+          modelId: request.model,
+          options: {
+            ...prepareLanguageModelCallOptions(request),
+            prompt: await convertToLanguageModelPrompt({
+              prompt: standardizedPrompt,
+              supportedUrls,
+              download: void 0,
+              provider: batchApi.provider.split(".")[0]
+            }),
+            tools: preparedTools,
+            toolChoice: prepareToolChoice({ toolChoice: request.toolChoice }),
+            providerOptions: request.providerOptions
+          }
+        });
+        break;
       }
-    });
+    }
     operationAbortSignal == null ? void 0 : operationAbortSignal.throwIfAborted();
   }
   const headersWithUserAgent = withUserAgentSuffix4(
@@ -12477,7 +12671,7 @@ async function startTextBatch({
     `ai/${VERSION}`
   );
   try {
-    const result = await model.experimental_doStartBatch({
+    const result = await batchApi.doStartBatch({
       requests: normalizedRequests,
       providerOptions,
       abortSignal: operationAbortSignal,
@@ -12485,17 +12679,20 @@ async function startTextBatch({
       ...webhookUrl != null && { webhookUrl }
     });
     const { batchId, warnings, ...status } = result;
-    logWarnings({
-      warnings: warnings.map(({ warning }) => warning),
-      provider: model.provider,
-      model: model.modelId
-    });
+    const modelByRequestId = new Map(
+      normalizedRequests.map((request) => [request.id, request.modelId])
+    );
+    for (const { requestId, warning } of warnings) {
+      logWarnings({
+        warnings: [warning],
+        provider: batchApi.provider,
+        model: requestId == null ? void 0 : modelByRequestId.get(requestId)
+      });
+    }
     return {
-      version: 1,
-      type: "text",
+      version: 2,
       id: batchId,
-      provider: model.provider,
-      modelId: model.modelId,
+      provider: batchApi.provider,
       ...status,
       warnings
     };
@@ -12503,8 +12700,25 @@ async function startTextBatch({
     throw wrapGatewayError(error);
   }
 }
+function validateCompatibleTools({
+  requestId,
+  tools,
+  toolsByName
+}) {
+  for (const tool2 of tools != null ? tools : []) {
+    const previousTool = toolsByName.get(tool2.name);
+    if (previousTool != null && !isDeepEqualData(previousTool, tool2)) {
+      throw new InvalidArgumentError({
+        parameter: "requests",
+        value: requestId,
+        message: `tool "${tool2.name}" must have the same definition in every batch request`
+      });
+    }
+    toolsByName.set(tool2.name, tool2);
+  }
+}
 async function getBatchStatus({
-  model: modelArg,
+  provider,
   batch,
   providerOptions,
   maxRetries,
@@ -12512,8 +12726,8 @@ async function getBatchStatus({
   headers,
   timeout
 }) {
-  const model = resolveBatchLanguageModel(modelArg);
-  validateBatchReference({ model, batch });
+  const batchApi = resolveBatchApi(provider);
+  validateBatchReference({ batchApi, batch });
   const operationAbortSignal = mergeAbortSignals(
     abortSignal,
     getTotalTimeoutMs(timeout)
@@ -12524,7 +12738,7 @@ async function getBatchStatus({
   });
   try {
     const status = await retry(
-      () => model.experimental_doGetBatchStatus({
+      () => batchApi.doGetBatchStatus({
         batchId: batch.id,
         providerOptions,
         abortSignal: operationAbortSignal,
@@ -12537,7 +12751,7 @@ async function getBatchStatus({
   }
 }
 function getBatchResults({
-  model: modelArg,
+  provider,
   batch,
   tools,
   providerOptions,
@@ -12546,8 +12760,8 @@ function getBatchResults({
   headers,
   timeout
 }) {
-  const model = resolveBatchLanguageModel(modelArg);
-  validateBatchReference({ model, batch });
+  const batchApi = resolveBatchApi(provider);
+  validateBatchReference({ batchApi, batch });
   const streamAbortController = new AbortController();
   const operationAbortSignal = mergeAbortSignals(
     abortSignal,
@@ -12572,7 +12786,7 @@ function getBatchResults({
   void (async () => {
     try {
       const stream = await retry(
-        () => model.experimental_doGetBatchResults({
+        () => batchApi.doGetBatchResults({
           batchId: batch.id,
           providerOptions,
           abortSignal: operationAbortSignal,
@@ -12589,19 +12803,26 @@ function getBatchResults({
   })();
   return asAsyncIterableStream(transform.readable);
 }
-function resolveBatchLanguageModel(modelArg) {
-  const model = resolveLanguageModel(modelArg);
-  if (!isBatchLanguageModel(model)) {
+function resolveBatchApi(provider) {
+  var _a25;
+  provider != null ? provider : provider = asProviderV4((_a25 = globalThis.AI_SDK_DEFAULT_PROVIDER) != null ? _a25 : gateway2);
+  if (isBatchApi(provider)) {
+    return provider;
+  }
+  if (!hasBatchFactory(provider)) {
     throw new UnsupportedFunctionalityError3({
       functionality: "batch processing",
-      message: `The ${model.provider} model "${model.modelId}" does not support batch processing.`
+      message: "The provider does not support batch processing. Make sure it exposes an experimental_batch() method."
     });
   }
-  return model;
+  return provider.experimental_batch();
+}
+function hasBatchFactory(provider) {
+  return typeof provider.experimental_batch === "function";
 }
-function isBatchLanguageModel(model) {
-  const candidate = model;
-  return typeof candidate.experimental_doStartBatch === "function" && typeof candidate.experimental_doGetBatchStatus === "function" && typeof candidate.experimental_doGetBatchResults === "function";
+function isBatchApi(provider) {
+  const candidate = provider;
+  return typeof candidate.doStartBatch === "function" && typeof candidate.doGetBatchStatus === "function" && typeof candidate.doGetBatchResults === "function";
 }
 function validateRequests(requests) {
   if (requests.length === 0) {
@@ -12631,21 +12852,21 @@ function validateRequests(requests) {
   }
 }
 function validateBatchReference({
-  model,
+  batchApi,
   batch
 }) {
-  if (batch.version !== 1 || batch.type !== "text") {
+  if (batch.version !== 2) {
     throw new InvalidArgumentError({
       parameter: "batch",
       value: batch,
-      message: "batch must be a supported text batch reference"
+      message: "batch must be a supported batch reference"
     });
   }
-  if (batch.provider !== model.provider || batch.modelId !== model.modelId) {
+  if (batch.provider !== batchApi.provider) {
     throw new InvalidArgumentError({
-      parameter: "model",
-      value: model,
-      message: `model ${model.provider}:${model.modelId} is not compatible with batch ${batch.provider}:${batch.modelId}`
+      parameter: "provider",
+      value: batchApi,
+      message: `provider ${batchApi.provider} is not compatible with batch provider ${batch.provider}`
     });
   }
 }
@@ -12653,14 +12874,32 @@ async function convertBatchItemResult({
   item,
   tools
 }) {
-  if (item.status !== "succeeded") {
-    return item;
+  switch (item.type) {
+    case "text":
+      switch (item.status) {
+        case "succeeded":
+          return {
+            id: item.id,
+            status: item.status,
+            ...await convertGenerateResult({ result: item.result, tools })
+          };
+        case "failed":
+          return {
+            id: item.id,
+            status: item.status,
+            error: item.error,
+            providerMetadata: item.providerMetadata
+          };
+        case "cancelled":
+        case "expired":
+          return {
+            id: item.id,
+            status: item.status,
+            error: item.error,
+            providerMetadata: item.providerMetadata
+          };
+      }
   }
-  return {
-    id: item.id,
-    status: "succeeded",
-    ...await convertGenerateResult({ result: item.result, tools })
-  };
 }
 async function convertGenerateResult({
   result,
@@ -12859,6 +13098,7 @@ var DefaultEmbedResult = class {
 };
 
 // src/embed/embed-many.ts
+import { InvalidResponseDataError as InvalidResponseDataError2 } from "@ai-sdk/provider";
 import {
   createIdGenerator as createIdGenerator5,
   withUserAgentSuffix as withUserAgentSuffix6
@@ -12995,6 +13235,7 @@ async function embedMany({
               response: modelResponse.response
             };
           });
+          validateEmbeddingCount({ embeddings: embeddings2, values });
           logWarnings({
             warnings: warnings2,
             provider: model.provider,
@@ -13040,8 +13281,8 @@ async function embedMany({
         );
         for (const parallelChunk of parallelChunks) {
           const results = await Promise.all(
-            parallelChunk.map((chunk) => {
-              return retry(async () => {
+            parallelChunk.map(async (chunk) => {
+              const result = await retry(async () => {
                 var _a27, _b26;
                 const embedCallId = generateCallId();
                 await notify({
@@ -13084,6 +13325,11 @@ async function embedMany({
                   response: modelResponse.response
                 };
               });
+              validateEmbeddingCount({
+                embeddings: result.embeddings,
+                values: chunk
+              });
+              return result;
             })
           );
           for (const result of results) {
@@ -13142,6 +13388,17 @@ async function embedMany({
     }
   });
 }
+function validateEmbeddingCount({
+  embeddings,
+  values
+}) {
+  if (embeddings.length !== values.length) {
+    throw new InvalidResponseDataError2({
+      data: embeddings,
+      message: `Expected ${values.length} embeddings, but received ${embeddings.length}.`
+    });
+  }
+}
 var textEncoder = new TextEncoder();
 function splitByEmbeddingLimits({
   values,
@@ -13239,6 +13496,12 @@ var gatewayCostMetadataKeys = [
   "outputInferenceCost",
   "surchargeCost"
 ];
+var RetryableNoImageResultError = class extends Error {
+  constructor() {
+    super("No image generated.");
+    this.name = "RetryableNoImageResultError";
+  }
+};
 async function generateImage({
   model: modelArg,
   prompt: promptArg,
@@ -13260,7 +13523,8 @@ async function generateImage({
   );
   const { retry } = prepareRetries({
     maxRetries: maxRetriesArg,
-    abortSignal
+    abortSignal,
+    additionalRetryableError: (error) => error instanceof RetryableNoImageResultError
   });
   const maxImagesPerCallWithDefault = (_a25 = maxImagesPerCall != null ? maxImagesPerCall : await invokeModelMaxImagesPerCall(model)) != null ? _a25 : 1;
   const callCount = Math.ceil(n / maxImagesPerCallWithDefault);
@@ -13271,25 +13535,41 @@ async function generateImage({
     const remainder = n % maxImagesPerCallWithDefault;
     return remainder === 0 ? maxImagesPerCallWithDefault : remainder;
   });
-  const results = await Promise.all(
-    callImageCounts.map(
-      async (callImageCount) => await retry(() => {
-        const { prompt, files, mask } = normalizePrompt(promptArg);
-        return model.doGenerate({
-          prompt,
-          files,
-          mask,
-          n: callImageCount,
-          abortSignal,
-          headers: headersWithUserAgent,
-          size,
-          aspectRatio,
-          seed,
-          providerOptions: providerOptions != null ? providerOptions : {}
+  const resultGroups = await Promise.all(
+    callImageCounts.map(async (callImageCount) => {
+      const callResults = [];
+      try {
+        await retry(async () => {
+          const { prompt, files, mask } = normalizePrompt(promptArg);
+          const result = await model.doGenerate({
+            prompt,
+            files,
+            mask,
+            n: callImageCount,
+            abortSignal,
+            headers: headersWithUserAgent,
+            size,
+            aspectRatio,
+            seed,
+            providerOptions: providerOptions != null ? providerOptions : {}
+          });
+          callResults.push(result);
+          if (result.images.length === 0 && result.isRetryable !== false) {
+            throw new RetryableNoImageResultError();
+          }
+          return result;
         });
-      })
-    )
+        return callResults;
+      } catch (error) {
+        const noImageResultError = error instanceof RetryableNoImageResultError ? error : RetryError.isInstance(error) && error.lastError instanceof RetryableNoImageResultError ? error.lastError : void 0;
+        if (noImageResultError != null) {
+          return callResults;
+        }
+        throw error;
+      }
+    })
   );
+  const results = resultGroups.flat();
   const images = [];
   const calls = [];
   const warnings = [];
@@ -13370,7 +13650,7 @@ async function generateImage({
   }
   logWarnings({ warnings, provider: model.provider, model: model.modelId });
   if (!images.length) {
-    throw new NoImageGeneratedError({ responses });
+    throw new NoImageGeneratedError({ calls, responses });
   }
   return new DefaultGenerateImageResult({
     images,
@@ -14240,39 +14520,12 @@ function getTextFromDataUrl(dataUrl) {
     throw new Error("Invalid data URL format");
   }
   try {
-    return window.atob(base64Content);
+    return globalThis.atob(base64Content);
   } catch (e) {
     throw new Error(`Error decoding data URL`);
   }
 }
 
-// src/util/is-deep-equal-data.ts
-function isDeepEqualData(obj1, obj2) {
-  if (obj1 === obj2) return true;
-  if (obj1 == null || obj2 == null) return false;
-  if (typeof obj1 !== "object" && typeof obj2 !== "object")
-    return obj1 === obj2;
-  if (obj1.constructor !== obj2.constructor) return false;
-  if (obj1 instanceof Date && obj2 instanceof Date) {
-    return obj1.getTime() === obj2.getTime();
-  }
-  if (Array.isArray(obj1)) {
-    if (obj1.length !== obj2.length) return false;
-    for (let i = 0; i < obj1.length; i++) {
-      if (!isDeepEqualData(obj1[i], obj2[i])) return false;
-    }
-    return true;
-  }
-  const keys1 = Object.keys(obj1);
-  const keys2 = Object.keys(obj2);
-  if (keys1.length !== keys2.length) return false;
-  for (const key of keys1) {
-    if (!keys2.includes(key)) return false;
-    if (!isDeepEqualData(obj1[key], obj2[key])) return false;
-  }
-  return true;
-}
-
 // src/util/serial-job-executor.ts
 var SerialJobExecutor = class {
   constructor() {
@@ -18815,8 +19068,8 @@ var AbstractChat = class {
         }
         this.state.messages = this.state.messages.slice(0, messageIndex + 1);
         this.state.replaceMessage(messageIndex, {
-          ...uiMessage,
           id: message.messageId,
+          ...uiMessage,
           role: (_b25 = uiMessage.role) != null ? _b25 : "user",
           metadata: message.metadata
         });
@@ -19523,7 +19776,7 @@ export {
   getRealtimeToolDefinitions as experimental_getRealtimeToolDefinitions,
   experimental_getVideoStatus,
   resampleAudio as experimental_resampleAudio,
-  startTextBatch as experimental_startTextBatch,
+  startBatch as experimental_startBatch,
   experimental_startVideo,
   streamLanguageModelCall as experimental_streamLanguageModelCall,
   streamTranscribe as experimental_streamTranscribe,
@@ -19533,7 +19786,7 @@ export {
   extractJsonMiddleware,
   extractReasoningMiddleware,
   fingerprintTools,
-  gateway2 as gateway,
+  gateway3 as gateway,
   generateId3 as generateId,
   generateImage,
   generateObject,