ai 6.0.285 → 6.0.287

package/CHANGELOG.md

@@ -1,5 +1,30 @@
 # ai
 
+## 6.0.287
+
+### Patch Changes
+
+- 9af6cfa: fix(ai): preserve provider metadata when converting failed tool calls
+- Updated dependencies [5f2d078]
+- Updated dependencies [1096a27]
+- Updated dependencies [2a82d35]
+- Updated dependencies [82e18b0]
+- Updated dependencies [eea6869]
+  - @ai-sdk/gateway@3.0.197
+  - @ai-sdk/provider-utils@4.0.52
+
+## 6.0.286
+
+### Patch Changes
+
+- 8a9722e: fix(ai): preserve provider metadata when simulating text streams
+- 6a0f2d9: fix(ai): report the prepareStep model in streamed step results
+- db02971: fix(ai): prevent unhandled rejections when UI message stream reading stops early
+- 4d91999: fix(ai): preserve tool calls required by retained pending approvals
+- b2ad994: fix(ai): cancel prompt attachment downloads when model calls are aborted or time out
+- 9c85bd4: fix(ai): stream null and empty string JSON partial outputs
+- becaa32: fix(ai): stream structured output from the final tool-loop step
+
 ## 6.0.285
 
 ### Patch Changes

package/dist/index.js

@@ -1334,7 +1334,7 @@ function detectMediaType({
 var import_provider_utils3 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "6.0.285" : "0.0.0-test";
+var VERSION = true ? "6.0.287" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1381,9 +1381,9 @@ var download = async ({
 };
 
 // src/util/download/download-function.ts
-var createDefaultDownloadFunction = (download2 = download) => (requestedDownloads) => Promise.all(
+var createDefaultDownloadFunction = (download2 = download, abortSignal) => (requestedDownloads) => Promise.all(
   requestedDownloads.map(
-    async (requestedDownload) => requestedDownload.isUrlSupportedByModel ? null : download2(requestedDownload)
+    async (requestedDownload) => requestedDownload.isUrlSupportedByModel ? null : await download2({ ...requestedDownload, abortSignal })
   )
 );
 
@@ -1519,11 +1519,12 @@ function convertDataContentToUint8Array(content) {
 async function convertToLanguageModelPrompt({
   prompt,
   supportedUrls,
-  download: download2 = createDefaultDownloadFunction()
+  download: download2,
+  abortSignal
 }) {
   const downloadedAssets = await downloadAssets(
     prompt.messages,
-    download2,
+    download2 != null ? download2 : createDefaultDownloadFunction(void 0, abortSignal),
     supportedUrls
   );
   const approvalIdToToolCallId = /* @__PURE__ */ new Map();
@@ -4815,7 +4816,8 @@ async function generateText({
                 messages: (_c = prepareStepResult == null ? void 0 : prepareStepResult.messages) != null ? _c : stepInputMessages
               },
               supportedUrls: await stepModel.supportedUrls,
-              download: download2
+              download: download2,
+              abortSignal: mergedAbortSignal
             });
             experimental_context = (_d = prepareStepResult == null ? void 0 : prepareStepResult.experimental_context) != null ? _d : experimental_context;
             const stepActiveTools = (_e = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _e : activeTools;
@@ -7294,7 +7296,15 @@ function createOutputTransformStream(output) {
   let text2 = "";
   let textChunk = "";
   let textProviderMetadata = void 0;
-  let lastPublishedValue = "";
+  let lastPublishedValue = void 0;
+  let hasPublishedValue = false;
+  function resetOutputState() {
+    firstTextChunkId = void 0;
+    text2 = "";
+    textChunk = "";
+    textProviderMetadata = void 0;
+    lastPublishedValue = "";
+  }
   function publishTextChunk({
     controller,
     partialOutput = void 0
@@ -7313,6 +7323,9 @@ function createOutputTransformStream(output) {
   return new TransformStream({
     async transform(chunk, controller) {
       var _a23;
+      if (chunk.type === "start-step") {
+        resetOutputState();
+      }
       if (chunk.type === "finish-step" && textChunk.length > 0) {
         publishTextChunk({ controller });
       }
@@ -7347,9 +7360,10 @@ function createOutputTransformStream(output) {
       const result = await output.parsePartialOutput({ text: text2 });
       if (result !== void 0) {
         const currentValue = typeof result.partial === "string" ? result.partial : JSON.stringify(result.partial);
-        if (currentValue !== lastPublishedValue) {
+        if (!hasPublishedValue || currentValue !== lastPublishedValue) {
           publishTextChunk({ controller, partialOutput: result.partial });
           lastPublishedValue = currentValue;
+          hasPublishedValue = true;
         }
       }
     }
@@ -7420,6 +7434,7 @@ var DefaultStreamTextResult = class {
     const recordedSteps = [];
     let recordedNoOutputError;
     let currentStepToolSet = tools;
+    let currentStepModel = model;
     const createPartIdReserver = () => {
       const usedIds = /* @__PURE__ */ new Set();
       return (id) => {
@@ -7576,7 +7591,10 @@ var DefaultStreamTextResult = class {
           });
           const currentStepResult = new DefaultStepResult({
             stepNumber: recordedSteps.length,
-            model: modelInfo,
+            model: {
+              provider: currentStepModel.provider,
+              modelId: currentStepModel.modelId
+            },
             ...callbackTelemetryProps,
             experimental_context,
             content: recordedContent,
@@ -7597,8 +7615,8 @@ var DefaultStreamTextResult = class {
           });
           logWarnings({
             warnings: recordedWarnings,
-            provider: modelInfo.provider,
-            model: modelInfo.modelId
+            provider: currentStepModel.provider,
+            model: currentStepModel.modelId
           });
           recordedSteps.push(currentStepResult);
           recordedResponseMessages.push(...stepMessages);
@@ -8037,6 +8055,7 @@ var DefaultStreamTextResult = class {
             const stepModel = resolveLanguageModel(
               (_a23 = prepareStepResult == null ? void 0 : prepareStepResult.model) != null ? _a23 : model
             );
+            currentStepModel = stepModel;
             const stepModelInfo = {
               provider: stepModel.provider,
               modelId: stepModel.modelId
@@ -8047,7 +8066,8 @@ var DefaultStreamTextResult = class {
                 messages: (_c = prepareStepResult == null ? void 0 : prepareStepResult.messages) != null ? _c : stepInputMessages
               },
               supportedUrls: await stepModel.supportedUrls,
-              download: download2
+              download: download2,
+              abortSignal
             });
             const stepActiveTools = (_d = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _d : activeTools;
             const stepToolSet = filterActiveTools({
@@ -8196,7 +8216,7 @@ var DefaultStreamTextResult = class {
             let stepResponse = {
               id: generateId2(),
               timestamp: /* @__PURE__ */ new Date(),
-              modelId: modelInfo.modelId
+              modelId: stepModelInfo.modelId
             };
             let activeText = "";
             const textPartIds = /* @__PURE__ */ new Map();
@@ -8718,7 +8738,7 @@ var DefaultStreamTextResult = class {
       this.teeStream().pipeThrough(
         new TransformStream({
           transform({ partialOutput }, controller) {
-            if (partialOutput != null) {
+            if (partialOutput !== void 0) {
               controller.enqueue(partialOutput);
             }
           }
@@ -9410,9 +9430,15 @@ function readUIMessageStream({
   var _a23;
   let controller;
   let hasErrored = false;
+  let isCancelled = false;
+  const abortController = new AbortController();
   const outputStream = new ReadableStream({
     start(controllerParam) {
       controller = controllerParam;
+    },
+    cancel() {
+      isCancelled = true;
+      abortController.abort();
     }
   });
   const state = createStreamingUIMessageState({
@@ -9433,15 +9459,18 @@ function readUIMessageStream({
         return job({
           state,
           write: () => {
-            controller == null ? void 0 : controller.enqueue(createUIMessageSnapshot(state.message));
+            if (!isCancelled) {
+              controller == null ? void 0 : controller.enqueue(createUIMessageSnapshot(state.message));
+            }
           }
         });
       },
       onError: handleError
     }),
-    onError: handleError
+    onError: handleError,
+    abortSignal: abortController.signal
   }).finally(() => {
-    if (!hasErrored) {
+    if (!hasErrored && !isCancelled) {
       controller == null ? void 0 : controller.close();
     }
   });
@@ -9514,7 +9543,7 @@ async function convertToModelMessages(messages, options) {
         if (message.parts != null) {
           let block = [];
           async function processBlock() {
-            var _a23, _b, _c, _d, _e, _f, _g, _h;
+            var _a23, _b, _c, _d, _e, _f, _g, _h, _i;
             if (block.length === 0) {
               return;
             }
@@ -9543,13 +9572,14 @@ async function convertToModelMessages(messages, options) {
               } else if (isToolUIPart(part)) {
                 const toolName = getToolName(part);
                 if (part.state !== "input-streaming") {
+                  const callProviderMetadata = (_a23 = part.callProviderMetadata) != null ? _a23 : part.state === "output-error" ? part.resultProviderMetadata : void 0;
                   content.push({
                     type: "tool-call",
                     toolCallId: part.toolCallId,
                     toolName,
-                    input: part.state === "output-error" ? (_a23 = part.input) != null ? _a23 : "rawInput" in part ? part.rawInput : void 0 : part.input,
+                    input: part.state === "output-error" ? (_b = part.input) != null ? _b : "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({
@@ -9560,7 +9590,7 @@ async function convertToModelMessages(messages, options) {
                     });
                   }
                   if (part.providerExecuted === true && part.state !== "approval-responded" && (part.state === "output-available" || part.state === "output-error")) {
-                    const resultProviderMetadata = (_b = part.resultProviderMetadata) != null ? _b : part.callProviderMetadata;
+                    const resultProviderMetadata = (_c = part.resultProviderMetadata) != null ? _c : part.callProviderMetadata;
                     content.push({
                       type: "tool-result",
                       toolCallId: part.toolCallId,
@@ -9569,7 +9599,7 @@ async function convertToModelMessages(messages, options) {
                         toolCallId: part.toolCallId,
                         input: part.input,
                         output: part.state === "output-error" ? part.errorText : part.output,
-                        tool: (_c = options == null ? void 0 : options.tools) == null ? void 0 : _c[toolName],
+                        tool: (_d = options == null ? void 0 : options.tools) == null ? void 0 : _d[toolName],
                         errorMode: part.state === "output-error" ? "json" : "none"
                       }),
                       ...resultProviderMetadata != null ? { providerOptions: resultProviderMetadata } : {}
@@ -9577,7 +9607,7 @@ async function convertToModelMessages(messages, options) {
                   }
                 }
               } else if (isDataUIPart(part)) {
-                const dataPart = (_d = options == null ? void 0 : options.convertDataPart) == null ? void 0 : _d.call(
+                const dataPart = (_e = options == null ? void 0 : options.convertDataPart) == null ? void 0 : _e.call(
                   options,
                   part
                 );
@@ -9605,7 +9635,7 @@ async function convertToModelMessages(messages, options) {
               {
                 const content2 = [];
                 for (const toolPart of toolParts) {
-                  if (((_e = toolPart.approval) == null ? void 0 : _e.approved) != null) {
+                  if (((_f = toolPart.approval) == null ? void 0 : _f.approved) != null) {
                     content2.push({
                       type: "tool-approval-response",
                       approvalId: toolPart.approval.id,
@@ -9625,7 +9655,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 } : {}
                       });
@@ -9642,7 +9672,7 @@ async function convertToModelMessages(messages, options) {
                           toolCallId: toolPart.toolCallId,
                           input: toolPart.input,
                           output: toolPart.state === "output-error" ? toolPart.errorText : toolPart.output,
-                          tool: (_h = options == null ? void 0 : options.tools) == null ? void 0 : _h[toolName],
+                          tool: (_i = options == null ? void 0 : options.tools) == null ? void 0 : _i[toolName],
                           errorMode: toolPart.state === "output-error" ? "text" : "none"
                         }),
                         ...toolPart.callProviderMetadata != null ? { providerOptions: toolPart.callProviderMetadata } : {}
@@ -11460,7 +11490,8 @@ async function generateObject(options) {
         const promptMessages = await convertToLanguageModelPrompt({
           prompt: standardizedPrompt,
           supportedUrls: await model.supportedUrls,
-          download: download2
+          download: download2,
+          abortSignal
         });
         const generateResult = await retry(
           () => recordSpan({
@@ -11949,7 +11980,8 @@ var DefaultStreamObjectResult = class {
           prompt: await convertToLanguageModelPrompt({
             prompt: standardizedPrompt,
             supportedUrls: await model.supportedUrls,
-            download: download2
+            download: download2,
+            abortSignal
           }),
           providerOptions,
           abortSignal,
@@ -12509,11 +12541,13 @@ function pruneMessages({
         }
       }
     }
+    const approvalIdToToolCallId = /* @__PURE__ */ new Map();
     const approvalIdToToolName = /* @__PURE__ */ new Map();
     for (const message of messages) {
       if ((message.role === "assistant" || message.role === "tool") && typeof message.content !== "string") {
         for (const part of message.content) {
           if (part.type === "tool-approval-request") {
+            approvalIdToToolCallId.set(part.approvalId, part.toolCallId);
             const toolName = toolCallIdToToolName.get(part.toolCallId);
             if (toolName != null) {
               approvalIdToToolName.set(part.approvalId, toolName);
@@ -12522,6 +12556,12 @@ function pruneMessages({
         }
       }
     }
+    for (const approvalId of keptApprovalIds) {
+      const toolCallId = approvalIdToToolCallId.get(approvalId);
+      if (toolCallId != null) {
+        keptToolCallIds.add(toolCallId);
+      }
+    }
     messages = messages.map((message, messageIndex) => {
       if (message.role !== "assistant" && message.role !== "tool" || typeof message.content === "string" || keepLastMessagesCount && messageIndex >= messages.length - keepLastMessagesCount) {
         return message;
@@ -13253,7 +13293,11 @@ function simulateStreamingMiddleware() {
             switch (part.type) {
               case "text": {
                 if (part.text.length > 0) {
-                  controller.enqueue({ type: "text-start", id: String(id) });
+                  controller.enqueue({
+                    type: "text-start",
+                    id: String(id),
+                    ...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
+                  });
                   controller.enqueue({
                     type: "text-delta",
                     id: String(id),