ai 5.0.249 → 5.0.251

package/dist/index.mjs

@@ -727,7 +727,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.249" : "0.0.0-test";
+var VERSION = true ? "5.0.251" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1744,12 +1744,13 @@ function asArray(value) {
 
 // src/util/retry-with-exponential-backoff.ts
 import { APICallError as APICallError2 } from "@ai-sdk/provider";
+import { GatewayError } from "@ai-sdk/gateway";
 import { delay, getErrorMessage as getErrorMessage3, isAbortError } from "@ai-sdk/provider-utils";
 function getRetryDelayInMs({
   error,
   exponentialBackoffDelay
 }) {
-  const headers = error.responseHeaders;
+  const headers = APICallError2.isInstance(error) ? error.responseHeaders : APICallError2.isInstance(error.cause) ? error.cause.responseHeaders : void 0;
   if (!headers)
     return exponentialBackoffDelay;
   let ms;
@@ -1810,7 +1811,7 @@ async function _retryWithExponentialBackoff(f, {
         errors: newErrors
       });
     }
-    if (error instanceof Error && APICallError2.isInstance(error) && error.isRetryable === true && tryNumber <= maxRetries) {
+    if (error instanceof Error && (APICallError2.isInstance(error) && error.isRetryable === true || GatewayError.isInstance(error) && error.isRetryable === true) && tryNumber <= maxRetries) {
       await delay(
         getRetryDelayInMs({
           error,
@@ -10081,10 +10082,38 @@ var TextStreamChatTransport = class extends HttpChatTransport {
 import { TypeValidationError as TypeValidationError4 } from "@ai-sdk/provider";
 import {
   lazyValidator as lazyValidator2,
+  safeValidateTypes as safeValidateTypes5,
   validateTypes as validateTypes2,
   zodSchema as zodSchema2
 } from "@ai-sdk/provider-utils";
 import { z as z8 } from "zod/v4";
+function isEmptyObject(value) {
+  return value != null && typeof value === "object" && !Array.isArray(value) && Object.keys(value).length === 0;
+}
+function asDynamicToolPart(toolPart) {
+  const common = {
+    type: "dynamic-tool",
+    toolName: toolPart.type.slice(5),
+    toolCallId: toolPart.toolCallId,
+    ...toolPart.providerExecuted === void 0 ? {} : { providerExecuted: toolPart.providerExecuted },
+    ...toolPart.callProviderMetadata === void 0 ? {} : { callProviderMetadata: toolPart.callProviderMetadata }
+  };
+  if (toolPart.state === "output-available") {
+    return {
+      ...common,
+      state: "output-available",
+      input: toolPart.input,
+      output: toolPart.output,
+      ...toolPart.preliminary === void 0 ? {} : { preliminary: toolPart.preliminary }
+    };
+  }
+  return {
+    ...common,
+    state: "output-error",
+    input: toolPart.input,
+    errorText: toolPart.errorText
+  };
+}
 var uiMessagesSchema = lazyValidator2(
   () => zodSchema2(
     z8.array(
@@ -10175,7 +10204,8 @@ var uiMessagesSchema = lazyValidator2(
               toolName: z8.string(),
               toolCallId: z8.string(),
               state: z8.literal("output-error"),
-              input: z8.unknown(),
+              input: z8.unknown().optional(),
+              rawInput: z8.unknown().optional(),
               providerExecuted: z8.boolean().optional(),
               output: z8.never().optional(),
               errorText: z8.string(),
@@ -10253,7 +10283,8 @@ var uiMessagesSchema = lazyValidator2(
               toolCallId: z8.string(),
               state: z8.literal("output-error"),
               providerExecuted: z8.boolean().optional(),
-              input: z8.unknown(),
+              input: z8.unknown().optional(),
+              rawInput: z8.unknown().optional(),
               output: z8.never().optional(),
               errorText: z8.string(),
               callProviderMetadata: providerMetadataSchema.optional(),
@@ -10317,6 +10348,17 @@ async function safeValidateUIMessages({
       value: messages,
       schema: uiMessagesSchema
     });
+    for (const message of validatedMessages) {
+      for (const part of message.parts) {
+        if (part.type !== "dynamic-tool" && !part.type.startsWith("tool-")) {
+          continue;
+        }
+        const toolPart = part;
+        if (toolPart.state === "output-error" && !("input" in toolPart)) {
+          toolPart.input = void 0;
+        }
+      }
+    }
     if (metadataSchema) {
       for (const message of validatedMessages) {
         await validateTypes2({
@@ -10351,12 +10393,17 @@ async function safeValidateUIMessages({
     }
     if (tools) {
       for (const message of validatedMessages) {
-        const toolParts = message.parts.filter(
-          (part) => part.type.startsWith("tool-")
-        );
-        for (const toolPart of toolParts) {
+        for (const [partIdx, part] of message.parts.entries()) {
+          if (!part.type.startsWith("tool-")) {
+            continue;
+          }
+          const toolPart = part;
           const toolName = toolPart.type.slice(5);
           const tool2 = tools[toolName];
+          if (!tool2 && (toolPart.state === "output-available" || toolPart.state === "output-error")) {
+            message.parts[partIdx] = asDynamicToolPart(toolPart);
+            continue;
+          }
           if (!tool2) {
             return {
               success: false,
@@ -10366,7 +10413,30 @@ async function safeValidateUIMessages({
               })
             };
           }
-          if (toolPart.state === "input-available") {
+          let dynamicToolPart;
+          if (toolPart.state === "output-error") {
+            if (toolPart.input !== void 0) {
+              const result = await safeValidateTypes5({
+                value: toolPart.input,
+                schema: tool2.inputSchema
+              });
+              if (!result.success) {
+                dynamicToolPart = asDynamicToolPart(toolPart);
+              }
+            }
+          } else if (toolPart.state === "output-available") {
+            const result = await safeValidateTypes5({
+              value: toolPart.input,
+              schema: tool2.inputSchema
+            });
+            if (!result.success) {
+              if (isEmptyObject(toolPart.input)) {
+                dynamicToolPart = asDynamicToolPart(toolPart);
+              } else {
+                throw result.error;
+              }
+            }
+          } else if (toolPart.state === "input-available") {
             await validateTypes2({
               value: toolPart.input,
               schema: tool2.inputSchema
@@ -10378,6 +10448,9 @@ async function safeValidateUIMessages({
               schema: tool2.outputSchema
             });
           }
+          if (dynamicToolPart) {
+            message.parts[partIdx] = dynamicToolPart;
+          }
         }
       }
     }
@@ -10393,18 +10466,8 @@ async function safeValidateUIMessages({
     };
   }
 }
-async function validateUIMessages({
-  messages,
-  metadataSchema,
-  dataSchemas,
-  tools
-}) {
-  const response = await safeValidateUIMessages({
-    messages,
-    metadataSchema,
-    dataSchemas,
-    tools
-  });
+async function validateUIMessages(options) {
+  const response = await safeValidateUIMessages(options);
   if (!response.success)
     throw response.error;
   return response.data;