ai 5.0.51 → 5.0.53

package/CHANGELOG.md

@@ -1,5 +1,20 @@
 # ai
 
+## 5.0.53
+
+### Patch Changes
+
+- d335191: Extend addToolResult to support error results
+
+## 5.0.52
+
+### Patch Changes
+
+- c56822d: fix(ai): update `uiMessageChunkSchema` to satisfy the `UIMessageChunk` type
+- 930399b: fix(ai): download files when intermediate file cannot be downloaded
+- Updated dependencies [7ca78f1]
+  - @ai-sdk/gateway@1.0.29
+
 ## 5.0.51
 
 ### Patch Changes

package/dist/index.d.mts

@@ -1650,10 +1650,6 @@ declare const uiMessageChunkSchema: z.ZodUnion<readonly [z.ZodObject<{
     errorText: z.ZodString;
     providerExecuted: z.ZodOptional<z.ZodBoolean>;
     dynamic: z.ZodOptional<z.ZodBoolean>;
-}, z.core.$strict>, z.ZodObject<{
-    type: z.ZodLiteral<"reasoning">;
-    text: z.ZodString;
-    providerMetadata: z.ZodOptional<z.ZodType<_ai_sdk_provider.SharedV2ProviderMetadata, unknown, z.core.$ZodTypeInternals<_ai_sdk_provider.SharedV2ProviderMetadata, unknown>>>;
 }, z.core.$strict>, z.ZodObject<{
     type: z.ZodLiteral<"reasoning-start">;
     id: z.ZodString;
@@ -1667,8 +1663,6 @@ declare const uiMessageChunkSchema: z.ZodUnion<readonly [z.ZodObject<{
     type: z.ZodLiteral<"reasoning-end">;
     id: z.ZodString;
     providerMetadata: z.ZodOptional<z.ZodType<_ai_sdk_provider.SharedV2ProviderMetadata, unknown, z.core.$ZodTypeInternals<_ai_sdk_provider.SharedV2ProviderMetadata, unknown>>>;
-}, z.core.$strict>, z.ZodObject<{
-    type: z.ZodLiteral<"reasoning-part-finish">;
 }, z.core.$strict>, z.ZodObject<{
     type: z.ZodLiteral<"source-url">;
     sourceId: z.ZodString;
@@ -1688,7 +1682,7 @@ declare const uiMessageChunkSchema: z.ZodUnion<readonly [z.ZodObject<{
     mediaType: z.ZodString;
     providerMetadata: z.ZodOptional<z.ZodType<_ai_sdk_provider.SharedV2ProviderMetadata, unknown, z.core.$ZodTypeInternals<_ai_sdk_provider.SharedV2ProviderMetadata, unknown>>>;
 }, z.core.$strict>, z.ZodObject<{
-    type: z.ZodString;
+    type: z.ZodCustom<`data-${string}`, `data-${string}`>;
     id: z.ZodOptional<z.ZodString>;
     data: z.ZodUnknown;
     transient: z.ZodOptional<z.ZodBoolean>;
@@ -1803,6 +1797,7 @@ type UIMessageChunk<METADATA = unknown, DATA_TYPES extends UIDataTypes = UIDataT
     type: 'file';
     url: string;
     mediaType: string;
+    providerMetadata?: ProviderMetadata;
 } | DataUIMessageChunk<DATA_TYPES> | {
     type: 'start-step';
 } | {
@@ -4332,10 +4327,18 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
      * Clear the error state and set the status to ready if the chat is in an error state.
      */
     clearError: () => void;
-    addToolResult: <TOOL extends keyof InferUIMessageTools<UI_MESSAGE>>({ tool, toolCallId, output, }: {
+    addToolResult: <TOOL extends keyof InferUIMessageTools<UI_MESSAGE>>({ state, tool, toolCallId, output, errorText, }: {
+        state?: "output-available";
         tool: TOOL;
         toolCallId: string;
         output: InferUIMessageTools<UI_MESSAGE>[TOOL]["output"];
+        errorText?: never;
+    } | {
+        state: "output-error";
+        tool: TOOL;
+        toolCallId: string;
+        output?: never;
+        errorText: string;
     }) => Promise<void>;
     /**
      * Abort the current request immediately, keep the generated tokens if any.

package/dist/index.d.ts

@@ -1650,10 +1650,6 @@ declare const uiMessageChunkSchema: z.ZodUnion<readonly [z.ZodObject<{
     errorText: z.ZodString;
     providerExecuted: z.ZodOptional<z.ZodBoolean>;
     dynamic: z.ZodOptional<z.ZodBoolean>;
-}, z.core.$strict>, z.ZodObject<{
-    type: z.ZodLiteral<"reasoning">;
-    text: z.ZodString;
-    providerMetadata: z.ZodOptional<z.ZodType<_ai_sdk_provider.SharedV2ProviderMetadata, unknown, z.core.$ZodTypeInternals<_ai_sdk_provider.SharedV2ProviderMetadata, unknown>>>;
 }, z.core.$strict>, z.ZodObject<{
     type: z.ZodLiteral<"reasoning-start">;
     id: z.ZodString;
@@ -1667,8 +1663,6 @@ declare const uiMessageChunkSchema: z.ZodUnion<readonly [z.ZodObject<{
     type: z.ZodLiteral<"reasoning-end">;
     id: z.ZodString;
     providerMetadata: z.ZodOptional<z.ZodType<_ai_sdk_provider.SharedV2ProviderMetadata, unknown, z.core.$ZodTypeInternals<_ai_sdk_provider.SharedV2ProviderMetadata, unknown>>>;
-}, z.core.$strict>, z.ZodObject<{
-    type: z.ZodLiteral<"reasoning-part-finish">;
 }, z.core.$strict>, z.ZodObject<{
     type: z.ZodLiteral<"source-url">;
     sourceId: z.ZodString;
@@ -1688,7 +1682,7 @@ declare const uiMessageChunkSchema: z.ZodUnion<readonly [z.ZodObject<{
     mediaType: z.ZodString;
     providerMetadata: z.ZodOptional<z.ZodType<_ai_sdk_provider.SharedV2ProviderMetadata, unknown, z.core.$ZodTypeInternals<_ai_sdk_provider.SharedV2ProviderMetadata, unknown>>>;
 }, z.core.$strict>, z.ZodObject<{
-    type: z.ZodString;
+    type: z.ZodCustom<`data-${string}`, `data-${string}`>;
     id: z.ZodOptional<z.ZodString>;
     data: z.ZodUnknown;
     transient: z.ZodOptional<z.ZodBoolean>;
@@ -1803,6 +1797,7 @@ type UIMessageChunk<METADATA = unknown, DATA_TYPES extends UIDataTypes = UIDataT
     type: 'file';
     url: string;
     mediaType: string;
+    providerMetadata?: ProviderMetadata;
 } | DataUIMessageChunk<DATA_TYPES> | {
     type: 'start-step';
 } | {
@@ -4332,10 +4327,18 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
      * Clear the error state and set the status to ready if the chat is in an error state.
      */
     clearError: () => void;
-    addToolResult: <TOOL extends keyof InferUIMessageTools<UI_MESSAGE>>({ tool, toolCallId, output, }: {
+    addToolResult: <TOOL extends keyof InferUIMessageTools<UI_MESSAGE>>({ state, tool, toolCallId, output, errorText, }: {
+        state?: "output-available";
         tool: TOOL;
         toolCallId: string;
         output: InferUIMessageTools<UI_MESSAGE>[TOOL]["output"];
+        errorText?: never;
+    } | {
+        state: "output-error";
+        tool: TOOL;
+        toolCallId: string;
+        output?: never;
+        errorText: string;
     }) => Promise<void>;
     /**
      * Abort the current request immediately, keep the generated tokens if any.

package/dist/index.js

@@ -768,7 +768,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "5.0.51" : "0.0.0-test";
+var VERSION = true ? "5.0.53" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({ url }) => {
@@ -1058,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) {
@@ -2990,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(),
@@ -3011,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(),
@@ -3036,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()
@@ -9726,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;
@@ -9736,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
         );
       }
@@ -9954,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