ai 5.1.0-beta.0 → 5.1.0-beta.10

package/dist/index.js

@@ -103,10 +103,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 +118,7 @@ __export(src_exports, {
   systemModelMessageSchema: () => systemModelMessageSchema,
   tool: () => import_provider_utils35.tool,
   toolModelMessageSchema: () => toolModelMessageSchema,
+  uiMessageChunkSchema: () => uiMessageChunkSchema,
   userModelMessageSchema: () => userModelMessageSchema,
   validateUIMessages: () => validateUIMessages,
   wrapLanguageModel: () => wrapLanguageModel,
@@ -555,7 +558,7 @@ function resolveLanguageModel(model) {
 }
 function resolveEmbeddingModel(model) {
   if (typeof model !== "string") {
-    if (model.specificationVersion !== "v2") {
+    if (model.specificationVersion !== "v3") {
       throw new UnsupportedModelVersionError({
         version: model.specificationVersion,
         provider: model.provider,
@@ -765,7 +768,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "5.1.0-beta.0" : "0.0.0-test";
+var VERSION = true ? "5.1.0-beta.10" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -1055,12 +1058,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) {
@@ -2987,11 +2990,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 +3006,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(),
@@ -3033,7 +3028,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()
@@ -6383,7 +6381,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,
@@ -9723,9 +9721,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;
@@ -9733,16 +9733,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
         );
       }
@@ -9951,7 +9951,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
@@ -10137,82 +10139,118 @@ 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
@@ -10433,10 +10471,12 @@ function readUIMessageStream({
   jsonSchema,
   lastAssistantMessageIsCompleteWithToolCalls,
   modelMessageSchema,
+  parseJsonEventStream,
   parsePartialJson,
   pipeTextStreamToResponse,
   pipeUIMessageStreamToResponse,
   readUIMessageStream,
+  safeValidateUIMessages,
   simulateReadableStream,
   simulateStreamingMiddleware,
   smoothStream,
@@ -10446,6 +10486,7 @@ function readUIMessageStream({
   systemModelMessageSchema,
   tool,
   toolModelMessageSchema,
+  uiMessageChunkSchema,
   userModelMessageSchema,
   validateUIMessages,
   wrapLanguageModel,