@zenning/ai 5.1.0 → 5.2.0

package/dist/index.js

@@ -20,43 +20,45 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
-  AISDKError: () => import_provider17.AISDKError,
-  APICallError: () => import_provider17.APICallError,
+  AISDKError: () => import_provider18.AISDKError,
+  APICallError: () => import_provider18.APICallError,
   AbstractChat: () => AbstractChat,
+  Agent: () => Agent,
   DefaultChatTransport: () => DefaultChatTransport,
   DownloadError: () => DownloadError,
-  EmptyResponseBodyError: () => import_provider17.EmptyResponseBodyError,
+  EmptyResponseBodyError: () => import_provider18.EmptyResponseBodyError,
   Experimental_Agent: () => Agent,
   HttpChatTransport: () => HttpChatTransport,
   InvalidArgumentError: () => InvalidArgumentError,
   InvalidDataContentError: () => InvalidDataContentError,
   InvalidMessageRoleError: () => InvalidMessageRoleError,
-  InvalidPromptError: () => import_provider17.InvalidPromptError,
-  InvalidResponseDataError: () => import_provider17.InvalidResponseDataError,
+  InvalidPromptError: () => import_provider18.InvalidPromptError,
+  InvalidResponseDataError: () => import_provider18.InvalidResponseDataError,
   InvalidStreamPartError: () => InvalidStreamPartError,
   InvalidToolInputError: () => InvalidToolInputError,
-  JSONParseError: () => import_provider17.JSONParseError,
+  JSONParseError: () => import_provider18.JSONParseError,
   JsonToSseTransformStream: () => JsonToSseTransformStream,
-  LoadAPIKeyError: () => import_provider17.LoadAPIKeyError,
+  LoadAPIKeyError: () => import_provider18.LoadAPIKeyError,
   MCPClientError: () => MCPClientError,
   MessageConversionError: () => MessageConversionError,
-  NoContentGeneratedError: () => import_provider17.NoContentGeneratedError,
+  NoContentGeneratedError: () => import_provider18.NoContentGeneratedError,
   NoImageGeneratedError: () => NoImageGeneratedError,
   NoObjectGeneratedError: () => NoObjectGeneratedError,
   NoOutputGeneratedError: () => NoOutputGeneratedError,
   NoOutputSpecifiedError: () => NoOutputSpecifiedError,
-  NoSuchModelError: () => import_provider17.NoSuchModelError,
+  NoSpeechGeneratedError: () => NoSpeechGeneratedError,
+  NoSuchModelError: () => import_provider18.NoSuchModelError,
   NoSuchProviderError: () => NoSuchProviderError,
   NoSuchToolError: () => NoSuchToolError,
   Output: () => output_exports,
   RetryError: () => RetryError,
   SerialJobExecutor: () => SerialJobExecutor,
   TextStreamChatTransport: () => TextStreamChatTransport,
-  TooManyEmbeddingValuesForCallError: () => import_provider17.TooManyEmbeddingValuesForCallError,
+  TooManyEmbeddingValuesForCallError: () => import_provider18.TooManyEmbeddingValuesForCallError,
   ToolCallRepairError: () => ToolCallRepairError,
-  TypeValidationError: () => import_provider17.TypeValidationError,
+  TypeValidationError: () => import_provider18.TypeValidationError,
   UI_MESSAGE_STREAM_HEADERS: () => UI_MESSAGE_STREAM_HEADERS,
-  UnsupportedFunctionalityError: () => import_provider17.UnsupportedFunctionalityError,
+  UnsupportedFunctionalityError: () => import_provider18.UnsupportedFunctionalityError,
   UnsupportedModelVersionError: () => UnsupportedModelVersionError,
   asSchema: () => import_provider_utils35.asSchema,
   assistantModelMessageSchema: () => assistantModelMessageSchema,
@@ -103,10 +105,12 @@ __export(src_exports, {
   jsonSchema: () => import_provider_utils35.jsonSchema,
   lastAssistantMessageIsCompleteWithToolCalls: () => lastAssistantMessageIsCompleteWithToolCalls,
   modelMessageSchema: () => modelMessageSchema,
+  parseJsonEventStream: () => import_provider_utils35.parseJsonEventStream,
   parsePartialJson: () => parsePartialJson,
   pipeTextStreamToResponse: () => pipeTextStreamToResponse,
   pipeUIMessageStreamToResponse: () => pipeUIMessageStreamToResponse,
   readUIMessageStream: () => readUIMessageStream,
+  safeValidateUIMessages: () => safeValidateUIMessages,
   simulateReadableStream: () => simulateReadableStream,
   simulateStreamingMiddleware: () => simulateStreamingMiddleware,
   smoothStream: () => smoothStream,
@@ -116,6 +120,7 @@ __export(src_exports, {
   systemModelMessageSchema: () => systemModelMessageSchema,
   tool: () => import_provider_utils35.tool,
   toolModelMessageSchema: () => toolModelMessageSchema,
+  uiMessageChunkSchema: () => uiMessageChunkSchema,
   userModelMessageSchema: () => userModelMessageSchema,
   validateUIMessages: () => validateUIMessages,
   wrapLanguageModel: () => wrapLanguageModel,
@@ -124,10 +129,10 @@ __export(src_exports, {
 });
 module.exports = __toCommonJS(src_exports);
 var import_gateway3 = require("@ai-sdk/gateway");
-var import_provider_utils35 = require("@ai-sdk/provider-utils");
+var import_provider_utils35 = require("@zenning/provider-utils");
 
 // src/generate-text/generate-text.ts
-var import_provider_utils10 = require("@ai-sdk/provider-utils");
+var import_provider_utils10 = require("@zenning/provider-utils");
 
 // src/error/no-output-specified-error.ts
 var import_provider = require("@zenning/provider");
@@ -201,7 +206,7 @@ var logWarnings = (warnings) => {
 var import_gateway = require("@ai-sdk/gateway");
 
 // src/error/index.ts
-var import_provider17 = require("@zenning/provider");
+var import_provider18 = require("@zenning/provider");
 
 // src/error/invalid-argument-error.ts
 var import_provider2 = require("@zenning/provider");
@@ -366,13 +371,25 @@ var NoOutputGeneratedError = class extends import_provider8.AISDKError {
 };
 _a8 = symbol8;
 
-// src/error/no-such-tool-error.ts
+// src/error/no-speech-generated-error.ts
 var import_provider9 = require("@zenning/provider");
+var NoSpeechGeneratedError = class extends import_provider9.AISDKError {
+  constructor(options) {
+    super({
+      name: "AI_NoSpeechGeneratedError",
+      message: "No speech audio generated."
+    });
+    this.responses = options.responses;
+  }
+};
+
+// src/error/no-such-tool-error.ts
+var import_provider10 = require("@zenning/provider");
 var name9 = "AI_NoSuchToolError";
 var marker9 = `vercel.ai.error.${name9}`;
 var symbol9 = Symbol.for(marker9);
 var _a9;
-var NoSuchToolError = class extends import_provider9.AISDKError {
+var NoSuchToolError = class extends import_provider10.AISDKError {
   constructor({
     toolName,
     availableTools = void 0,
@@ -384,36 +401,36 @@ var NoSuchToolError = class extends import_provider9.AISDKError {
     this.availableTools = availableTools;
   }
   static isInstance(error) {
-    return import_provider9.AISDKError.hasMarker(error, marker9);
+    return import_provider10.AISDKError.hasMarker(error, marker9);
   }
 };
 _a9 = symbol9;
 
 // src/error/tool-call-repair-error.ts
-var import_provider10 = require("@zenning/provider");
+var import_provider11 = require("@zenning/provider");
 var name10 = "AI_ToolCallRepairError";
 var marker10 = `vercel.ai.error.${name10}`;
 var symbol10 = Symbol.for(marker10);
 var _a10;
-var ToolCallRepairError = class extends import_provider10.AISDKError {
+var ToolCallRepairError = class extends import_provider11.AISDKError {
   constructor({
     cause,
     originalError,
-    message = `Error repairing tool call: ${(0, import_provider10.getErrorMessage)(cause)}`
+    message = `Error repairing tool call: ${(0, import_provider11.getErrorMessage)(cause)}`
   }) {
     super({ name: name10, message, cause });
     this[_a10] = true;
     this.originalError = originalError;
   }
   static isInstance(error) {
-    return import_provider10.AISDKError.hasMarker(error, marker10);
+    return import_provider11.AISDKError.hasMarker(error, marker10);
   }
 };
 _a10 = symbol10;
 
 // src/error/unsupported-model-version-error.ts
-var import_provider11 = require("@zenning/provider");
-var UnsupportedModelVersionError = class extends import_provider11.AISDKError {
+var import_provider12 = require("@zenning/provider");
+var UnsupportedModelVersionError = class extends import_provider12.AISDKError {
   constructor(options) {
     super({
       name: "AI_UnsupportedModelVersionError",
@@ -426,12 +443,12 @@ var UnsupportedModelVersionError = class extends import_provider11.AISDKError {
 };
 
 // src/prompt/invalid-data-content-error.ts
-var import_provider12 = require("@zenning/provider");
+var import_provider13 = require("@zenning/provider");
 var name11 = "AI_InvalidDataContentError";
 var marker11 = `vercel.ai.error.${name11}`;
 var symbol11 = Symbol.for(marker11);
 var _a11;
-var InvalidDataContentError = class extends import_provider12.AISDKError {
+var InvalidDataContentError = class extends import_provider13.AISDKError {
   constructor({
     content,
     cause,
@@ -442,18 +459,18 @@ var InvalidDataContentError = class extends import_provider12.AISDKError {
     this.content = content;
   }
   static isInstance(error) {
-    return import_provider12.AISDKError.hasMarker(error, marker11);
+    return import_provider13.AISDKError.hasMarker(error, marker11);
   }
 };
 _a11 = symbol11;
 
 // src/prompt/invalid-message-role-error.ts
-var import_provider13 = require("@zenning/provider");
+var import_provider14 = require("@zenning/provider");
 var name12 = "AI_InvalidMessageRoleError";
 var marker12 = `vercel.ai.error.${name12}`;
 var symbol12 = Symbol.for(marker12);
 var _a12;
-var InvalidMessageRoleError = class extends import_provider13.AISDKError {
+var InvalidMessageRoleError = class extends import_provider14.AISDKError {
   constructor({
     role,
     message = `Invalid message role: '${role}'. Must be one of: "system", "user", "assistant", "tool".`
@@ -463,18 +480,18 @@ var InvalidMessageRoleError = class extends import_provider13.AISDKError {
     this.role = role;
   }
   static isInstance(error) {
-    return import_provider13.AISDKError.hasMarker(error, marker12);
+    return import_provider14.AISDKError.hasMarker(error, marker12);
   }
 };
 _a12 = symbol12;
 
 // src/prompt/message-conversion-error.ts
-var import_provider14 = require("@zenning/provider");
+var import_provider15 = require("@zenning/provider");
 var name13 = "AI_MessageConversionError";
 var marker13 = `vercel.ai.error.${name13}`;
 var symbol13 = Symbol.for(marker13);
 var _a13;
-var MessageConversionError = class extends import_provider14.AISDKError {
+var MessageConversionError = class extends import_provider15.AISDKError {
   constructor({
     originalMessage,
     message
@@ -484,18 +501,18 @@ var MessageConversionError = class extends import_provider14.AISDKError {
     this.originalMessage = originalMessage;
   }
   static isInstance(error) {
-    return import_provider14.AISDKError.hasMarker(error, marker13);
+    return import_provider15.AISDKError.hasMarker(error, marker13);
   }
 };
 _a13 = symbol13;
 
 // src/util/download/download-error.ts
-var import_provider15 = require("@zenning/provider");
+var import_provider16 = require("@zenning/provider");
 var name14 = "AI_DownloadError";
 var marker14 = `vercel.ai.error.${name14}`;
 var symbol14 = Symbol.for(marker14);
 var _a14;
-var DownloadError = class extends import_provider15.AISDKError {
+var DownloadError = class extends import_provider16.AISDKError {
   constructor({
     url,
     statusCode,
@@ -510,18 +527,18 @@ var DownloadError = class extends import_provider15.AISDKError {
     this.statusText = statusText;
   }
   static isInstance(error) {
-    return import_provider15.AISDKError.hasMarker(error, marker14);
+    return import_provider16.AISDKError.hasMarker(error, marker14);
   }
 };
 _a14 = symbol14;
 
 // src/util/retry-error.ts
-var import_provider16 = require("@zenning/provider");
+var import_provider17 = require("@zenning/provider");
 var name15 = "AI_RetryError";
 var marker15 = `vercel.ai.error.${name15}`;
 var symbol15 = Symbol.for(marker15);
 var _a15;
-var RetryError = class extends import_provider16.AISDKError {
+var RetryError = class extends import_provider17.AISDKError {
   constructor({
     message,
     reason,
@@ -534,34 +551,54 @@ var RetryError = class extends import_provider16.AISDKError {
     this.lastError = errors[errors.length - 1];
   }
   static isInstance(error) {
-    return import_provider16.AISDKError.hasMarker(error, marker15);
+    return import_provider17.AISDKError.hasMarker(error, marker15);
   }
 };
 _a15 = symbol15;
 
 // src/model/resolve-model.ts
+function transformToV3LanguageModel(model) {
+  return {
+    ...model,
+    specificationVersion: "v3"
+  };
+}
+function transformToV3EmbeddingModel(model) {
+  return {
+    ...model,
+    specificationVersion: "v3"
+  };
+}
 function resolveLanguageModel(model) {
   if (typeof model !== "string") {
-    if (model.specificationVersion !== "v2") {
+    if (model.specificationVersion !== "v3" && model.specificationVersion !== "v2") {
+      const unsupportedModel = model;
       throw new UnsupportedModelVersionError({
-        version: model.specificationVersion,
-        provider: model.provider,
-        modelId: model.modelId
+        version: unsupportedModel.specificationVersion,
+        provider: unsupportedModel.provider,
+        modelId: unsupportedModel.modelId
       });
     }
+    if (model.specificationVersion === "v2") {
+      return transformToV3LanguageModel(model);
+    }
     return model;
   }
   return getGlobalProvider().languageModel(model);
 }
 function resolveEmbeddingModel(model) {
   if (typeof model !== "string") {
-    if (model.specificationVersion !== "v2") {
+    if (model.specificationVersion !== "v3" && model.specificationVersion !== "v2") {
+      const unsupportedModel = model;
       throw new UnsupportedModelVersionError({
-        version: model.specificationVersion,
-        provider: model.provider,
-        modelId: model.modelId
+        version: unsupportedModel.specificationVersion,
+        provider: unsupportedModel.provider,
+        modelId: unsupportedModel.modelId
       });
     }
+    if (model.specificationVersion === "v2") {
+      return transformToV3EmbeddingModel(model);
+    }
     return model;
   }
   return getGlobalProvider().textEmbeddingModel(
@@ -574,10 +611,10 @@ function getGlobalProvider() {
 }
 
 // src/prompt/convert-to-language-model-prompt.ts
-var import_provider_utils4 = require("@ai-sdk/provider-utils");
+var import_provider_utils4 = require("@zenning/provider-utils");
 
 // src/util/detect-media-type.ts
-var import_provider_utils = require("@ai-sdk/provider-utils");
+var import_provider_utils = require("@zenning/provider-utils");
 var imageMediaTypeSignatures = [
   {
     mediaType: "image/gif",
@@ -762,10 +799,10 @@ function detectMediaType({
 }
 
 // src/util/download/download.ts
-var import_provider_utils2 = require("@ai-sdk/provider-utils");
+var import_provider_utils2 = require("@zenning/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "5.1.0" : "0.0.0-test";
+var VERSION = true ? "5.2.0" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -806,8 +843,8 @@ var createDefaultDownloadFunction = (download2 = download) => (requestedDownload
 );
 
 // src/prompt/data-content.ts
-var import_provider18 = require("@zenning/provider");
-var import_provider_utils3 = require("@ai-sdk/provider-utils");
+var import_provider19 = require("@zenning/provider");
+var import_provider_utils3 = require("@zenning/provider-utils");
 var import_v4 = require("zod/v4");
 
 // src/prompt/split-data-url.ts
@@ -840,7 +877,7 @@ var dataContentSchema = import_v4.z.union([
     { message: "Must be a Buffer" }
   )
 ]);
-function convertToLanguageModelV2DataContent(content) {
+function convertToLanguageModelV3DataContent(content) {
   if (content instanceof Uint8Array) {
     return { data: content, mediaType: void 0 };
   }
@@ -858,7 +895,7 @@ function convertToLanguageModelV2DataContent(content) {
       content.toString()
     );
     if (dataUrlMediaType == null || base64Content == null) {
-      throw new import_provider18.AISDKError({
+      throw new import_provider19.AISDKError({
         name: "InvalidDataContentError",
         message: `Invalid data URL format in content ${content.toString()}`
       });
@@ -959,7 +996,7 @@ function convertToLanguageModelMessage({
           const providerOptions = part.providerOptions;
           switch (part.type) {
             case "file": {
-              const { data, mediaType } = convertToLanguageModelV2DataContent(
+              const { data, mediaType } = convertToLanguageModelV3DataContent(
                 part.data
               );
               return {
@@ -1055,12 +1092,12 @@ async function downloadAssets(messages, download2, supportedUrls) {
   }));
   const downloadedFiles = await download2(plannedDownloads);
   return Object.fromEntries(
-    downloadedFiles.filter(
-      (downloadedFile) => (downloadedFile == null ? void 0 : downloadedFile.data) != null
-    ).map(({ data, mediaType }, index) => [
-      plannedDownloads[index].url.toString(),
-      { data, mediaType }
-    ])
+    downloadedFiles.map(
+      (file, index) => file == null ? null : [
+        plannedDownloads[index].url.toString(),
+        { data: file.data, mediaType: file.mediaType }
+      ]
+    ).filter((file) => file != null)
   );
 }
 function convertPartToLanguageModelPart(part, downloadedAssets) {
@@ -1084,7 +1121,7 @@ function convertPartToLanguageModelPart(part, downloadedAssets) {
     default:
       throw new Error(`Unsupported part type: ${type}`);
   }
-  const { data: convertedData, mediaType: convertedMediaType } = convertToLanguageModelV2DataContent(originalData);
+  const { data: convertedData, mediaType: convertedMediaType } = convertToLanguageModelV3DataContent(originalData);
   let mediaType = convertedMediaType != null ? convertedMediaType : part.mediaType;
   let data = convertedData;
   if (data instanceof URL) {
@@ -1217,7 +1254,7 @@ function prepareCallSettings({
 }
 
 // src/prompt/prepare-tools-and-tool-choice.ts
-var import_provider_utils5 = require("@ai-sdk/provider-utils");
+var import_provider_utils5 = require("@zenning/provider-utils");
 
 // src/util/is-non-empty-object.ts
 function isNonEmptyObject(object2) {
@@ -1271,8 +1308,8 @@ function prepareToolsAndToolChoice({
 }
 
 // src/prompt/standardize-prompt.ts
-var import_provider19 = require("@zenning/provider");
-var import_provider_utils6 = require("@ai-sdk/provider-utils");
+var import_provider20 = require("@zenning/provider");
+var import_provider_utils6 = require("@zenning/provider-utils");
 var import_v46 = require("zod/v4");
 
 // src/prompt/message.ts
@@ -1427,19 +1464,19 @@ var coreMessageSchema = modelMessageSchema;
 // src/prompt/standardize-prompt.ts
 async function standardizePrompt(prompt) {
   if (prompt.prompt == null && prompt.messages == null) {
-    throw new import_provider19.InvalidPromptError({
+    throw new import_provider20.InvalidPromptError({
       prompt,
       message: "prompt or messages must be defined"
     });
   }
   if (prompt.prompt != null && prompt.messages != null) {
-    throw new import_provider19.InvalidPromptError({
+    throw new import_provider20.InvalidPromptError({
       prompt,
       message: "prompt and messages cannot be defined at the same time"
     });
   }
   if (prompt.system != null && typeof prompt.system !== "string") {
-    throw new import_provider19.InvalidPromptError({
+    throw new import_provider20.InvalidPromptError({
       prompt,
       message: "system must be a string"
     });
@@ -1452,13 +1489,13 @@ async function standardizePrompt(prompt) {
   } else if (prompt.messages != null) {
     messages = prompt.messages;
   } else {
-    throw new import_provider19.InvalidPromptError({
+    throw new import_provider20.InvalidPromptError({
       prompt,
       message: "prompt or messages must be defined"
     });
   }
   if (messages.length === 0) {
-    throw new import_provider19.InvalidPromptError({
+    throw new import_provider20.InvalidPromptError({
       prompt,
       message: "messages must not be empty"
     });
@@ -1468,7 +1505,7 @@ async function standardizePrompt(prompt) {
     schema: import_v46.z.array(modelMessageSchema)
   });
   if (!validationResult.success) {
-    throw new import_provider19.InvalidPromptError({
+    throw new import_provider20.InvalidPromptError({
       prompt,
       message: "The messages must be a ModelMessage[]. If you have passed a UIMessage[], you can use convertToModelMessages to convert them.",
       cause: validationResult.error
@@ -1482,12 +1519,12 @@ async function standardizePrompt(prompt) {
 
 // src/prompt/wrap-gateway-error.ts
 var import_gateway2 = require("@ai-sdk/gateway");
-var import_provider20 = require("@zenning/provider");
+var import_provider21 = require("@zenning/provider");
 function wrapGatewayError(error) {
   if (import_gateway2.GatewayAuthenticationError.isInstance(error) || import_gateway2.GatewayModelNotFoundError.isInstance(error)) {
-    return new import_provider20.AISDKError({
+    return new import_provider21.AISDKError({
       name: "GatewayError",
-      message: "Vercel AI Gateway access failed. If you want to use AI SDK providers directly, use the providers, e.g. @ai-sdk/openai, or register a different global default provider.",
+      message: "Vercel AI Gateway access failed. If you want to use AI SDK providers directly, use the providers, e.g. @zenning/openai, or register a different global default provider.",
       cause: error
     });
   }
@@ -1731,8 +1768,8 @@ function asArray(value) {
 }
 
 // src/util/retry-with-exponential-backoff.ts
-var import_provider21 = require("@zenning/provider");
-var import_provider_utils7 = require("@ai-sdk/provider-utils");
+var import_provider22 = require("@zenning/provider");
+var import_provider_utils7 = require("@zenning/provider-utils");
 function getRetryDelayInMs({
   error,
   exponentialBackoffDelay
@@ -1798,7 +1835,7 @@ async function _retryWithExponentialBackoff(f, {
         errors: newErrors
       });
     }
-    if (error instanceof Error && import_provider21.APICallError.isInstance(error) && error.isRetryable === true && tryNumber <= maxRetries) {
+    if (error instanceof Error && import_provider22.APICallError.isInstance(error) && error.isRetryable === true && tryNumber <= maxRetries) {
       await (0, import_provider_utils7.delay)(
         getRetryDelayInMs({
           error,
@@ -1871,7 +1908,7 @@ function extractTextContent(content) {
 }
 
 // src/generate-text/generated-file.ts
-var import_provider_utils8 = require("@ai-sdk/provider-utils");
+var import_provider_utils8 = require("@zenning/provider-utils");
 var DefaultGeneratedFile = class {
   constructor({
     data,
@@ -1905,7 +1942,7 @@ var DefaultGeneratedFileWithType = class extends DefaultGeneratedFile {
 };
 
 // src/generate-text/parse-tool-call.ts
-var import_provider_utils9 = require("@ai-sdk/provider-utils");
+var import_provider_utils9 = require("@zenning/provider-utils");
 async function parseToolCall({
   toolCall,
   tools,
@@ -2082,14 +2119,14 @@ async function isStopConditionMet({
 }
 
 // src/prompt/create-tool-model-output.ts
-var import_provider22 = require("@zenning/provider");
+var import_provider23 = require("@zenning/provider");
 function createToolModelOutput({
   output,
   tool: tool3,
   errorMode
 }) {
   if (errorMode === "text") {
-    return { type: "error-text", value: (0, import_provider22.getErrorMessage)(output) };
+    return { type: "error-text", value: (0, import_provider23.getErrorMessage)(output) };
   } else if (errorMode === "json") {
     return { type: "error-json", value: toJSONValue(output) };
   }
@@ -2778,8 +2815,8 @@ function asContent({
 }
 
 // src/generate-text/stream-text.ts
-var import_provider23 = require("@zenning/provider");
-var import_provider_utils14 = require("@ai-sdk/provider-utils");
+var import_provider24 = require("@zenning/provider");
+var import_provider_utils14 = require("@zenning/provider-utils");
 
 // src/util/prepare-headers.ts
 function prepareHeaders(headers, defaultHeaders) {
@@ -2916,7 +2953,7 @@ function getResponseUIMessageId({
 }
 
 // src/ui/process-ui-message-stream.ts
-var import_provider_utils12 = require("@ai-sdk/provider-utils");
+var import_provider_utils12 = require("@zenning/provider-utils");
 
 // src/ui-message-stream/ui-message-chunks.ts
 var import_v47 = require("zod/v4");
@@ -2987,11 +3024,6 @@ var uiMessageChunkSchema = import_v47.z.union([
     providerExecuted: import_v47.z.boolean().optional(),
     dynamic: import_v47.z.boolean().optional()
   }),
-  import_v47.z.strictObject({
-    type: import_v47.z.literal("reasoning"),
-    text: import_v47.z.string(),
-    providerMetadata: providerMetadataSchema.optional()
-  }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("reasoning-start"),
     id: import_v47.z.string(),
@@ -3008,9 +3040,6 @@ var uiMessageChunkSchema = import_v47.z.union([
     id: import_v47.z.string(),
     providerMetadata: providerMetadataSchema.optional()
   }),
-  import_v47.z.strictObject({
-    type: import_v47.z.literal("reasoning-part-finish")
-  }),
   import_v47.z.strictObject({
     type: import_v47.z.literal("source-url"),
     sourceId: import_v47.z.string(),
@@ -3036,7 +3065,10 @@ var uiMessageChunkSchema = import_v47.z.union([
     providerMetadata: providerMetadataSchema.optional()
   }),
   import_v47.z.strictObject({
-    type: import_v47.z.string().startsWith("data-"),
+    type: import_v47.z.custom(
+      (value) => typeof value === "string" && value.startsWith("data-"),
+      { message: 'Type must start with "data-"' }
+    ),
     id: import_v47.z.string().optional(),
     data: import_v47.z.unknown(),
     transient: import_v47.z.boolean().optional()
@@ -3102,7 +3134,7 @@ function mergeObjects(base, overrides) {
 }
 
 // src/util/parse-partial-json.ts
-var import_provider_utils11 = require("@ai-sdk/provider-utils");
+var import_provider_utils11 = require("@zenning/provider-utils");
 
 // src/util/fix-json.ts
 function fixJson(input) {
@@ -4263,7 +4295,7 @@ function now() {
 }
 
 // src/generate-text/run-tools-transformation.ts
-var import_provider_utils13 = require("@ai-sdk/provider-utils");
+var import_provider_utils13 = require("@zenning/provider-utils");
 function runToolsTransformation({
   tools,
   generatorStream,
@@ -4695,7 +4727,7 @@ var DefaultStreamTextResult = class {
     let activeReasoningContent = {};
     const eventProcessor = new TransformStream({
       async transform(chunk, controller) {
-        var _a17, _b, _c;
+        var _a17, _b, _c, _d;
         controller.enqueue(chunk);
         const { part } = chunk;
         if (part.type === "text-delta" || part.type === "reasoning-delta" || part.type === "source" || part.type === "tool-call" || part.type === "tool-result" || part.type === "tool-input-start" || part.type === "tool-input-delta" || part.type === "raw") {
@@ -4728,6 +4760,18 @@ var DefaultStreamTextResult = class {
           activeText.providerMetadata = (_a17 = part.providerMetadata) != null ? _a17 : activeText.providerMetadata;
         }
         if (part.type === "text-end") {
+          const activeText = activeTextContent[part.id];
+          if (activeText == null) {
+            controller.enqueue({
+              part: {
+                type: "error",
+                error: `text part ${part.id} not found`
+              },
+              partialOutput: void 0
+            });
+            return;
+          }
+          activeText.providerMetadata = (_b = part.providerMetadata) != null ? _b : activeText.providerMetadata;
           delete activeTextContent[part.id];
         }
         if (part.type === "reasoning-start") {
@@ -4751,7 +4795,7 @@ var DefaultStreamTextResult = class {
             return;
           }
           activeReasoning.text += part.text;
-          activeReasoning.providerMetadata = (_b = part.providerMetadata) != null ? _b : activeReasoning.providerMetadata;
+          activeReasoning.providerMetadata = (_c = part.providerMetadata) != null ? _c : activeReasoning.providerMetadata;
         }
         if (part.type === "reasoning-end") {
           const activeReasoning = activeReasoningContent[part.id];
@@ -4765,7 +4809,7 @@ var DefaultStreamTextResult = class {
             });
             return;
           }
-          activeReasoning.providerMetadata = (_c = part.providerMetadata) != null ? _c : activeReasoning.providerMetadata;
+          activeReasoning.providerMetadata = (_d = part.providerMetadata) != null ? _d : activeReasoning.providerMetadata;
           delete activeReasoningContent[part.id];
         }
         if (part.type === "file") {
@@ -5514,7 +5558,7 @@ var DefaultStreamTextResult = class {
     sendSources = false,
     sendStart = true,
     sendFinish = true,
-    onError = import_provider23.getErrorMessage
+    onError = import_provider24.getErrorMessage
   } = {}) {
     const responseMessageId = generateMessageId != null ? getResponseUIMessageId({
       originalMessages,
@@ -5610,6 +5654,9 @@ var DefaultStreamTextResult = class {
                   mediaType: part.mediaType,
                   title: part.title,
                   filename: part.filename,
+                  fileId: part.fileId,
+                  startIndex: part.startIndex,
+                  endIndex: part.endIndex,
                   ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
                 });
               }
@@ -6017,14 +6064,33 @@ var Agent = class {
   constructor(settings) {
     this.settings = settings;
   }
+  /**
+   * The name of the agent.
+   */
+  get name() {
+    return this.settings.name;
+  }
+  /**
+   * The tools that the agent can use.
+   */
   get tools() {
     return this.settings.tools;
   }
   async generate(options) {
-    return generateText({ ...this.settings, ...options });
+    var _a17;
+    return generateText({
+      ...this.settings,
+      stopWhen: (_a17 = this.settings.stopWhen) != null ? _a17 : stepCountIs(20),
+      ...options
+    });
   }
   stream(options) {
-    return streamText({ ...this.settings, ...options });
+    var _a17;
+    return streamText({
+      ...this.settings,
+      stopWhen: (_a17 = this.settings.stopWhen) != null ? _a17 : stepCountIs(20),
+      ...options
+    });
   }
   /**
    * Creates a response object that streams UI messages to the client.
@@ -6037,7 +6103,7 @@ var Agent = class {
 };
 
 // src/embed/embed.ts
-var import_provider_utils15 = require("@ai-sdk/provider-utils");
+var import_provider_utils15 = require("@zenning/provider-utils");
 async function embed({
   model: modelArg,
   value,
@@ -6156,7 +6222,7 @@ var DefaultEmbedResult = class {
 };
 
 // src/embed/embed-many.ts
-var import_provider_utils16 = require("@ai-sdk/provider-utils");
+var import_provider_utils16 = require("@zenning/provider-utils");
 
 // src/util/split-array.ts
 function splitArray(array, chunkSize) {
@@ -6405,7 +6471,7 @@ var DefaultEmbedManyResult = class {
 };
 
 // src/generate-image/generate-image.ts
-var import_provider_utils17 = require("@ai-sdk/provider-utils");
+var import_provider_utils17 = require("@zenning/provider-utils");
 async function generateImage({
   model,
   prompt,
@@ -6420,7 +6486,7 @@ async function generateImage({
   headers
 }) {
   var _a17, _b;
-  if (model.specificationVersion !== "v2") {
+  if (model.specificationVersion !== "v3") {
     throw new UnsupportedModelVersionError({
       version: model.specificationVersion,
       provider: model.provider,
@@ -6523,7 +6589,7 @@ async function invokeModelMaxImagesPerCall(model) {
 }
 
 // src/generate-object/generate-object.ts
-var import_provider_utils20 = require("@ai-sdk/provider-utils");
+var import_provider_utils20 = require("@zenning/provider-utils");
 
 // src/generate-text/extract-reasoning-content.ts
 function extractReasoningContent(content) {
@@ -6534,8 +6600,8 @@ function extractReasoningContent(content) {
 }
 
 // src/generate-object/output-strategy.ts
-var import_provider24 = require("@zenning/provider");
-var import_provider_utils18 = require("@ai-sdk/provider-utils");
+var import_provider25 = require("@zenning/provider");
+var import_provider_utils18 = require("@zenning/provider-utils");
 var noSchemaOutputStrategy = {
   type: "no-schema",
   jsonSchema: void 0,
@@ -6555,7 +6621,7 @@ var noSchemaOutputStrategy = {
     } : { success: true, value };
   },
   createElementStream() {
-    throw new import_provider24.UnsupportedFunctionalityError({
+    throw new import_provider25.UnsupportedFunctionalityError({
       functionality: "element streams in no-schema mode"
     });
   }
@@ -6577,7 +6643,7 @@ var objectOutputStrategy = (schema) => ({
     return (0, import_provider_utils18.safeValidateTypes)({ value, schema });
   },
   createElementStream() {
-    throw new import_provider24.UnsupportedFunctionalityError({
+    throw new import_provider25.UnsupportedFunctionalityError({
       functionality: "element streams in object mode"
     });
   }
@@ -6605,10 +6671,10 @@ var arrayOutputStrategy = (schema) => {
       isFinalDelta
     }) {
       var _a17;
-      if (!(0, import_provider24.isJSONObject)(value) || !(0, import_provider24.isJSONArray)(value.elements)) {
+      if (!(0, import_provider25.isJSONObject)(value) || !(0, import_provider25.isJSONArray)(value.elements)) {
         return {
           success: false,
-          error: new import_provider24.TypeValidationError({
+          error: new import_provider25.TypeValidationError({
             value,
             cause: "value must be an object that contains an array of elements"
           })
@@ -6648,10 +6714,10 @@ var arrayOutputStrategy = (schema) => {
       };
     },
     async validateFinalResult(value) {
-      if (!(0, import_provider24.isJSONObject)(value) || !(0, import_provider24.isJSONArray)(value.elements)) {
+      if (!(0, import_provider25.isJSONObject)(value) || !(0, import_provider25.isJSONArray)(value.elements)) {
         return {
           success: false,
-          error: new import_provider24.TypeValidationError({
+          error: new import_provider25.TypeValidationError({
             value,
             cause: "value must be an object that contains an array of elements"
           })
@@ -6714,10 +6780,10 @@ var enumOutputStrategy = (enumValues) => {
       additionalProperties: false
     },
     async validateFinalResult(value) {
-      if (!(0, import_provider24.isJSONObject)(value) || typeof value.result !== "string") {
+      if (!(0, import_provider25.isJSONObject)(value) || typeof value.result !== "string") {
         return {
           success: false,
-          error: new import_provider24.TypeValidationError({
+          error: new import_provider25.TypeValidationError({
             value,
             cause: 'value must be an object that contains a string in the "result" property.'
           })
@@ -6726,17 +6792,17 @@ var enumOutputStrategy = (enumValues) => {
       const result = value.result;
       return enumValues.includes(result) ? { success: true, value: result } : {
         success: false,
-        error: new import_provider24.TypeValidationError({
+        error: new import_provider25.TypeValidationError({
           value,
           cause: "value must be a string in the enum"
         })
       };
     },
     async validatePartialResult({ value, textDelta }) {
-      if (!(0, import_provider24.isJSONObject)(value) || typeof value.result !== "string") {
+      if (!(0, import_provider25.isJSONObject)(value) || typeof value.result !== "string") {
         return {
           success: false,
-          error: new import_provider24.TypeValidationError({
+          error: new import_provider25.TypeValidationError({
             value,
             cause: 'value must be an object that contains a string in the "result" property.'
           })
@@ -6749,7 +6815,7 @@ var enumOutputStrategy = (enumValues) => {
       if (value.result.length === 0 || possibleEnumValues.length === 0) {
         return {
           success: false,
-          error: new import_provider24.TypeValidationError({
+          error: new import_provider25.TypeValidationError({
             value,
             cause: "value must be a string in the enum"
           })
@@ -6764,7 +6830,7 @@ var enumOutputStrategy = (enumValues) => {
       };
     },
     createElementStream() {
-      throw new import_provider24.UnsupportedFunctionalityError({
+      throw new import_provider25.UnsupportedFunctionalityError({
         functionality: "element streams in enum mode"
       });
     }
@@ -6792,8 +6858,8 @@ function getOutputStrategy({
 }
 
 // src/generate-object/parse-and-validate-object-result.ts
-var import_provider25 = require("@zenning/provider");
-var import_provider_utils19 = require("@ai-sdk/provider-utils");
+var import_provider26 = require("@zenning/provider");
+var import_provider_utils19 = require("@zenning/provider-utils");
 async function parseAndValidateObjectResult(result, outputStrategy, context) {
   const parseResult = await (0, import_provider_utils19.safeParseJSON)({ text: result });
   if (!parseResult.success) {
@@ -6830,7 +6896,7 @@ async function parseAndValidateObjectResultWithRepair(result, outputStrategy, re
   try {
     return await parseAndValidateObjectResult(result, outputStrategy, context);
   } catch (error) {
-    if (repairText != null && NoObjectGeneratedError.isInstance(error) && (import_provider25.JSONParseError.isInstance(error.cause) || import_provider25.TypeValidationError.isInstance(error.cause))) {
+    if (repairText != null && NoObjectGeneratedError.isInstance(error) && (import_provider26.JSONParseError.isInstance(error.cause) || import_provider26.TypeValidationError.isInstance(error.cause))) {
       const repairedText = await repairText({
         text: result,
         error: error.cause
@@ -7231,7 +7297,7 @@ var DefaultGenerateObjectResult = class {
 };
 
 // src/generate-object/stream-object.ts
-var import_provider_utils22 = require("@ai-sdk/provider-utils");
+var import_provider_utils22 = require("@zenning/provider-utils");
 
 // src/util/cosine-similarity.ts
 function cosineSimilarity(vector1, vector2) {
@@ -7341,7 +7407,7 @@ var SerialJobExecutor = class {
 };
 
 // src/util/simulate-readable-stream.ts
-var import_provider_utils21 = require("@ai-sdk/provider-utils");
+var import_provider_utils21 = require("@zenning/provider-utils");
 function simulateReadableStream({
   chunks,
   initialDelayInMs = 0,
@@ -7885,19 +7951,7 @@ var DefaultStreamObjectResult = class {
 };
 
 // src/generate-speech/generate-speech.ts
-var import_provider_utils23 = require("@ai-sdk/provider-utils");
-
-// src/error/no-speech-generated-error.ts
-var import_provider26 = require("@zenning/provider");
-var NoSpeechGeneratedError = class extends import_provider26.AISDKError {
-  constructor(options) {
-    super({
-      name: "AI_NoSpeechGeneratedError",
-      message: "No speech audio generated."
-    });
-    this.responses = options.responses;
-  }
-};
+var import_provider_utils23 = require("@zenning/provider-utils");
 
 // src/generate-speech/generated-audio-file.ts
 var DefaultGeneratedAudioFile = class extends DefaultGeneratedFile {
@@ -8000,7 +8054,7 @@ __export(output_exports, {
   object: () => object,
   text: () => text
 });
-var import_provider_utils24 = require("@ai-sdk/provider-utils");
+var import_provider_utils24 = require("@zenning/provider-utils");
 var text = () => ({
   type: "text",
   responseFormat: { type: "text" },
@@ -8071,7 +8125,7 @@ var object = ({
 };
 
 // src/generate-text/smooth-stream.ts
-var import_provider_utils25 = require("@ai-sdk/provider-utils");
+var import_provider_utils25 = require("@zenning/provider-utils");
 var import_provider27 = require("@zenning/provider");
 var CHUNKING_REGEXPS = {
   word: /\S+\s+/m,
@@ -8150,7 +8204,7 @@ function defaultSettingsMiddleware({
   settings
 }) {
   return {
-    middlewareVersion: "v2",
+    middlewareVersion: "v3",
     transformParams: async ({ params }) => {
       return mergeObjects(settings, params);
     }
@@ -8184,7 +8238,7 @@ function extractReasoningMiddleware({
   const openingTag = `<${tagName}>`;
   const closingTag = `</${tagName}>`;
   return {
-    middlewareVersion: "v2",
+    middlewareVersion: "v3",
     wrapGenerate: async ({ doGenerate }) => {
       const { content, ...rest } = await doGenerate();
       const transformedContent = [];
@@ -8330,7 +8384,7 @@ function extractReasoningMiddleware({
 // src/middleware/simulate-streaming-middleware.ts
 function simulateStreamingMiddleware() {
   return {
-    middlewareVersion: "v2",
+    middlewareVersion: "v3",
     wrapStream: async ({ doGenerate }) => {
       const result = await doGenerate();
       let id = 0;
@@ -8427,7 +8481,7 @@ var doWrap = ({
     return transformParams ? await transformParams({ params, type, model }) : params;
   }
   return {
-    specificationVersion: "v2",
+    specificationVersion: "v3",
     provider: (_a17 = providerId != null ? providerId : overrideProvider == null ? void 0 : overrideProvider({ model })) != null ? _a17 : model.provider,
     modelId: (_b = modelId != null ? modelId : overrideModelId == null ? void 0 : overrideModelId({ model })) != null ? _b : model.modelId,
     supportedUrls: (_c = overrideSupportedUrls == null ? void 0 : overrideSupportedUrls({ model })) != null ? _c : model.supportedUrls,
@@ -8679,10 +8733,10 @@ var DefaultProviderRegistry = class {
 };
 
 // src/tool/mcp/mcp-client.ts
-var import_provider_utils27 = require("@ai-sdk/provider-utils");
+var import_provider_utils27 = require("@zenning/provider-utils");
 
 // src/tool/mcp/mcp-sse-transport.ts
-var import_provider_utils26 = require("@ai-sdk/provider-utils");
+var import_provider_utils26 = require("@zenning/provider-utils");
 
 // src/tool/mcp/json-rpc-message.ts
 var import_v49 = require("zod/v4");
@@ -9254,7 +9308,7 @@ var DefaultMCPClient = class {
 };
 
 // src/transcribe/transcribe.ts
-var import_provider_utils28 = require("@ai-sdk/provider-utils");
+var import_provider_utils28 = require("@zenning/provider-utils");
 
 // src/error/no-transcript-generated-error.ts
 var import_provider31 = require("@zenning/provider");
@@ -9336,7 +9390,7 @@ var DefaultTranscriptionResult = class {
 };
 
 // src/ui/call-completion-api.ts
-var import_provider_utils29 = require("@ai-sdk/provider-utils");
+var import_provider_utils29 = require("@zenning/provider-utils");
 
 // src/ui/process-text-stream.ts
 async function processTextStream({
@@ -9470,7 +9524,7 @@ async function callCompletionApi({
 }
 
 // src/ui/chat.ts
-var import_provider_utils32 = require("@ai-sdk/provider-utils");
+var import_provider_utils32 = require("@zenning/provider-utils");
 
 // src/ui/convert-file-list-to-file-ui-parts.ts
 async function convertFileListToFileUIParts(files) {
@@ -9503,10 +9557,10 @@ async function convertFileListToFileUIParts(files) {
 }
 
 // src/ui/default-chat-transport.ts
-var import_provider_utils31 = require("@ai-sdk/provider-utils");
+var import_provider_utils31 = require("@zenning/provider-utils");
 
 // src/ui/http-chat-transport.ts
-var import_provider_utils30 = require("@ai-sdk/provider-utils");
+var import_provider_utils30 = require("@zenning/provider-utils");
 var HttpChatTransport = class {
   constructor({
     api = "/api/chat",
@@ -9760,9 +9814,11 @@ var AbstractChat = class {
       }
     };
     this.addToolResult = async ({
+      state = "output-available",
       tool: tool3,
       toolCallId,
-      output
+      output,
+      errorText
     }) => this.jobExecutor.run(async () => {
       var _a17, _b;
       const messages = this.state.messages;
@@ -9770,16 +9826,16 @@ var AbstractChat = class {
       this.state.replaceMessage(messages.length - 1, {
         ...lastMessage,
         parts: lastMessage.parts.map(
-          (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? { ...part, state: "output-available", output } : part
+          (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? { ...part, state, output, errorText } : part
         )
       });
       if (this.activeResponse) {
         this.activeResponse.state.message.parts = this.activeResponse.state.message.parts.map(
           (part) => isToolOrDynamicToolUIPart(part) && part.toolCallId === toolCallId ? {
             ...part,
-            state: "output-available",
+            state,
             output,
-            errorText: void 0
+            errorText
           } : part
         );
       }
@@ -10037,7 +10093,9 @@ function lastAssistantMessageIsCompleteWithToolCalls({
     return part.type === "step-start" ? index : lastIndex;
   }, -1);
   const lastStepToolInvocations = message.parts.slice(lastStepStartIndex + 1).filter(isToolOrDynamicToolUIPart);
-  return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every((part) => part.state === "output-available");
+  return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every(
+    (part) => part.state === "output-available" || part.state === "output-error"
+  );
 }
 
 // src/ui/transform-text-to-ui-message-stream.ts
@@ -10077,7 +10135,7 @@ var TextStreamChatTransport = class extends HttpChatTransport {
 
 // src/ui/validate-ui-messages.ts
 var import_provider32 = require("@zenning/provider");
-var import_provider_utils33 = require("@ai-sdk/provider-utils");
+var import_provider_utils33 = require("@zenning/provider-utils");
 var import_v410 = require("zod/v4");
 var textUIPartSchema = import_v410.z.object({
   type: import_v410.z.literal("text"),
@@ -10226,86 +10284,122 @@ var uiMessageSchema = import_v410.z.object({
     ])
   )
 });
-async function validateUIMessages({
+async function safeValidateUIMessages({
   messages,
   metadataSchema,
   dataSchemas,
   tools
 }) {
-  if (messages == null) {
-    throw new InvalidArgumentError({
-      parameter: "messages",
+  try {
+    if (messages == null) {
+      return {
+        success: false,
+        error: new InvalidArgumentError({
+          parameter: "messages",
+          value: messages,
+          message: "messages parameter must be provided"
+        })
+      };
+    }
+    const validatedMessages = await (0, import_provider_utils33.validateTypes)({
       value: messages,
-      message: "messages parameter must be provided"
+      schema: import_v410.z.array(uiMessageSchema)
     });
-  }
-  const validatedMessages = await (0, import_provider_utils33.validateTypes)({
-    value: messages,
-    schema: import_v410.z.array(uiMessageSchema)
-  });
-  if (metadataSchema) {
-    for (const message of validatedMessages) {
-      await (0, import_provider_utils33.validateTypes)({
-        value: message.metadata,
-        schema: metadataSchema
-      });
-    }
-  }
-  if (dataSchemas) {
-    for (const message of validatedMessages) {
-      const dataParts = message.parts.filter(
-        (part) => part.type.startsWith("data-")
-      );
-      for (const dataPart of dataParts) {
-        const dataName = dataPart.type.slice(5);
-        const dataSchema = dataSchemas[dataName];
-        if (!dataSchema) {
-          throw new import_provider32.TypeValidationError({
-            value: dataPart.data,
-            cause: `No data schema found for data part ${dataName}`
-          });
-        }
+    if (metadataSchema) {
+      for (const message of validatedMessages) {
         await (0, import_provider_utils33.validateTypes)({
-          value: dataPart.data,
-          schema: dataSchema
+          value: message.metadata,
+          schema: metadataSchema
         });
       }
     }
-  }
-  if (tools) {
-    for (const message of validatedMessages) {
-      const toolParts = message.parts.filter(
-        (part) => part.type.startsWith("tool-")
-      );
-      for (const toolPart of toolParts) {
-        const toolName = toolPart.type.slice(5);
-        const tool3 = tools[toolName];
-        if (!tool3) {
-          throw new import_provider32.TypeValidationError({
-            value: toolPart.input,
-            cause: `No tool schema found for tool part ${toolName}`
-          });
-        }
-        if (toolPart.state === "input-available" || toolPart.state === "output-available" || toolPart.state === "output-error") {
+    if (dataSchemas) {
+      for (const message of validatedMessages) {
+        const dataParts = message.parts.filter(
+          (part) => part.type.startsWith("data-")
+        );
+        for (const dataPart of dataParts) {
+          const dataName = dataPart.type.slice(5);
+          const dataSchema = dataSchemas[dataName];
+          if (!dataSchema) {
+            return {
+              success: false,
+              error: new import_provider32.TypeValidationError({
+                value: dataPart.data,
+                cause: `No data schema found for data part ${dataName}`
+              })
+            };
+          }
           await (0, import_provider_utils33.validateTypes)({
-            value: toolPart.input,
-            schema: tool3.inputSchema
+            value: dataPart.data,
+            schema: dataSchema
           });
         }
-        if (toolPart.state === "output-available" && tool3.outputSchema) {
-          await (0, import_provider_utils33.validateTypes)({
-            value: toolPart.output,
-            schema: tool3.outputSchema
-          });
+      }
+    }
+    if (tools) {
+      for (const message of validatedMessages) {
+        const toolParts = message.parts.filter(
+          (part) => part.type.startsWith("tool-")
+        );
+        for (const toolPart of toolParts) {
+          const toolName = toolPart.type.slice(5);
+          const tool3 = tools[toolName];
+          if (!tool3) {
+            return {
+              success: false,
+              error: new import_provider32.TypeValidationError({
+                value: toolPart.input,
+                cause: `No tool schema found for tool part ${toolName}`
+              })
+            };
+          }
+          if (toolPart.state === "input-available" || toolPart.state === "output-available" || toolPart.state === "output-error") {
+            await (0, import_provider_utils33.validateTypes)({
+              value: toolPart.input,
+              schema: tool3.inputSchema
+            });
+          }
+          if (toolPart.state === "output-available" && tool3.outputSchema) {
+            await (0, import_provider_utils33.validateTypes)({
+              value: toolPart.output,
+              schema: tool3.outputSchema
+            });
+          }
         }
       }
     }
+    return {
+      success: true,
+      data: validatedMessages
+    };
+  } catch (error) {
+    const err = error;
+    return {
+      success: false,
+      error: err
+    };
   }
-  return validatedMessages;
+}
+async function validateUIMessages({
+  messages,
+  metadataSchema,
+  dataSchemas,
+  tools
+}) {
+  const response = await safeValidateUIMessages({
+    messages,
+    metadataSchema,
+    dataSchemas,
+    tools
+  });
+  if (!response.success)
+    throw response.error;
+  return response.data;
 }
 
 // src/ui-message-stream/create-ui-message-stream.ts
-var import_provider_utils34 = require("@ai-sdk/provider-utils");
+var import_provider_utils34 = require("@zenning/provider-utils");
 function createUIMessageStream({
   execute,
   onError = import_provider_utils34.getErrorMessage,
@@ -10442,6 +10536,7 @@ function readUIMessageStream({
   AISDKError,
   APICallError,
   AbstractChat,
+  Agent,
   DefaultChatTransport,
   DownloadError,
   EmptyResponseBodyError,
@@ -10464,6 +10559,7 @@ function readUIMessageStream({
   NoObjectGeneratedError,
   NoOutputGeneratedError,
   NoOutputSpecifiedError,
+  NoSpeechGeneratedError,
   NoSuchModelError,
   NoSuchProviderError,
   NoSuchToolError,
@@ -10522,10 +10618,12 @@ function readUIMessageStream({
   jsonSchema,
   lastAssistantMessageIsCompleteWithToolCalls,
   modelMessageSchema,
+  parseJsonEventStream,
   parsePartialJson,
   pipeTextStreamToResponse,
   pipeUIMessageStreamToResponse,
   readUIMessageStream,
+  safeValidateUIMessages,
   simulateReadableStream,
   simulateStreamingMiddleware,
   smoothStream,
@@ -10535,6 +10633,7 @@ function readUIMessageStream({
   systemModelMessageSchema,
   tool,
   toolModelMessageSchema,
+  uiMessageChunkSchema,
   userModelMessageSchema,
   validateUIMessages,
   wrapLanguageModel,