ai 5.0.216 → 5.0.218

package/dist/index.js

@@ -738,7 +738,17 @@ var audioMediaTypeSignatures = [
   },
   {
     mediaType: "audio/mp4",
-    bytesPrefix: [102, 116, 121, 112]
+    bytesPrefix: [
+      0,
+      0,
+      0,
+      null,
+      102,
+      116,
+      121,
+      112
+      // ftyp
+    ]
   },
   {
     mediaType: "audio/webm",
@@ -778,7 +788,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "5.0.216" : "0.0.0-test";
+var VERSION = true ? "5.0.218" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1922,6 +1932,21 @@ function extractTextContent(content) {
   return parts.map((content2) => content2.text).join("");
 }
 
+// src/generate-text/filter-active-tools.ts
+function filterActiveTools({
+  tools,
+  activeTools
+}) {
+  if (tools == null || activeTools == null) {
+    return tools;
+  }
+  return Object.fromEntries(
+    Object.entries(tools).filter(
+      ([name16]) => activeTools.includes(name16)
+    )
+  );
+}
+
 // src/generate-text/generated-file.ts
 var import_provider_utils8 = require("@ai-sdk/provider-utils");
 var DefaultGeneratedFile = class {
@@ -2334,6 +2359,9 @@ async function generateText({
         const responseMessages = [];
         const steps = [];
         do {
+          if (steps.length > 0) {
+            abortSignal == null ? void 0 : abortSignal.throwIfAborted();
+          }
           const stepInputMessages = [
             ...initialPrompt.messages,
             ...responseMessages
@@ -2355,10 +2383,15 @@ async function generateText({
             supportedUrls: await stepModel.supportedUrls,
             download: download2
           });
+          const stepActiveTools = (_d = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _d : activeTools;
+          const stepToolSet = filterActiveTools({
+            tools,
+            activeTools: stepActiveTools
+          });
           const { toolChoice: stepToolChoice, tools: stepTools } = prepareToolsAndToolChoice({
             tools,
-            toolChoice: (_d = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _d : toolChoice,
-            activeTools: (_e = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _e : activeTools
+            toolChoice: (_e = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _e : toolChoice,
+            activeTools: stepActiveTools
           });
           currentModelResponse = await retry(
             () => {
@@ -2462,7 +2495,7 @@ async function generateText({
             ).map(
               (toolCall) => parseToolCall({
                 toolCall,
-                tools,
+                tools: stepToolSet,
                 repairToolCall,
                 system,
                 messages: stepInputMessages
@@ -2473,7 +2506,7 @@ async function generateText({
             if (toolCall.invalid) {
               continue;
             }
-            const tool2 = tools[toolCall.toolName];
+            const tool2 = stepToolSet[toolCall.toolName];
             if (tool2.onInputStart != null) {
               await tool2.onInputStart({
                 toolCallId: toolCall.toolCallId,
@@ -2509,13 +2542,13 @@ async function generateText({
           clientToolCalls = stepToolCalls.filter(
             (toolCall) => !toolCall.providerExecuted
           );
-          if (tools != null) {
+          if (stepToolSet != null) {
             clientToolOutputs.push(
               ...await executeTools({
                 toolCalls: clientToolCalls.filter(
                   (toolCall) => !toolCall.invalid
                 ),
-                tools,
+                tools: stepToolSet,
                 tracer,
                 telemetry,
                 messages: stepInputMessages,
@@ -2532,7 +2565,7 @@ async function generateText({
           responseMessages.push(
             ...toResponseMessages({
               content: stepContent,
-              tools
+              tools: stepToolSet
             })
           );
           const currentStepResult = new DefaultStepResult({
@@ -3001,39 +3034,39 @@ var import_provider_utils11 = require("@ai-sdk/provider-utils");
 var uiMessageChunkSchema = (0, import_provider_utils11.lazyValidator)(
   () => (0, import_provider_utils11.zodSchema)(
     import_v47.z.union([
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("text-start"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("text-delta"),
         id: import_v47.z.string(),
         delta: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("text-end"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("error"),
         errorText: import_v47.z.string()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-start"),
         toolCallId: import_v47.z.string(),
         toolName: import_v47.z.string(),
         providerExecuted: import_v47.z.boolean().optional(),
         dynamic: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-delta"),
         toolCallId: import_v47.z.string(),
         inputTextDelta: import_v47.z.string()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-available"),
         toolCallId: import_v47.z.string(),
         toolName: import_v47.z.string(),
@@ -3042,7 +3075,7 @@ var uiMessageChunkSchema = (0, import_provider_utils11.lazyValidator)(
         providerMetadata: providerMetadataSchema.optional(),
         dynamic: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-input-error"),
         toolCallId: import_v47.z.string(),
         toolName: import_v47.z.string(),
@@ -3052,7 +3085,7 @@ var uiMessageChunkSchema = (0, import_provider_utils11.lazyValidator)(
         dynamic: import_v47.z.boolean().optional(),
         errorText: import_v47.z.string()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-output-available"),
         toolCallId: import_v47.z.string(),
         output: import_v47.z.unknown(),
@@ -3060,37 +3093,37 @@ var uiMessageChunkSchema = (0, import_provider_utils11.lazyValidator)(
         dynamic: import_v47.z.boolean().optional(),
         preliminary: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("tool-output-error"),
         toolCallId: import_v47.z.string(),
         errorText: import_v47.z.string(),
         providerExecuted: import_v47.z.boolean().optional(),
         dynamic: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("reasoning-start"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("reasoning-delta"),
         id: import_v47.z.string(),
         delta: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("reasoning-end"),
         id: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("source-url"),
         sourceId: import_v47.z.string(),
         url: import_v47.z.string(),
         title: import_v47.z.string().optional(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("source-document"),
         sourceId: import_v47.z.string(),
         mediaType: import_v47.z.string(),
@@ -3098,13 +3131,13 @@ var uiMessageChunkSchema = (0, import_provider_utils11.lazyValidator)(
         filename: import_v47.z.string().optional(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("file"),
         url: import_v47.z.string(),
         mediaType: import_v47.z.string(),
         providerMetadata: providerMetadataSchema.optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.custom(
           (value) => typeof value === "string" && value.startsWith("data-"),
           { message: 'Type must start with "data-"' }
@@ -3113,18 +3146,18 @@ var uiMessageChunkSchema = (0, import_provider_utils11.lazyValidator)(
         data: import_v47.z.unknown(),
         transient: import_v47.z.boolean().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("start-step")
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("finish-step")
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("start"),
         messageId: import_v47.z.string().optional(),
         messageMetadata: import_v47.z.unknown().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("finish"),
         finishReason: import_v47.z.enum([
           "stop",
@@ -3137,10 +3170,10 @@ var uiMessageChunkSchema = (0, import_provider_utils11.lazyValidator)(
         ]).optional(),
         messageMetadata: import_v47.z.unknown().optional()
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("abort")
       }),
-      import_v47.z.strictObject({
+      import_v47.z.looseObject({
         type: import_v47.z.literal("message-metadata"),
         messageMetadata: import_v47.z.unknown()
       })
@@ -3589,11 +3622,32 @@ function processUIMessageStream({
       async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
           var _a16, _b, _c, _d;
+          function getCurrentStepParts() {
+            const parts = state.message.parts;
+            let currentStepStartIndex = parts.length - 1;
+            while (currentStepStartIndex >= 0 && parts[currentStepStartIndex].type !== "step-start") {
+              currentStepStartIndex--;
+            }
+            return parts.slice(currentStepStartIndex + 1);
+          }
+          function getCurrentStepToolInvocations() {
+            return getCurrentStepParts().filter(isToolUIPart);
+          }
           function getToolInvocation(toolCallId) {
-            const toolInvocations = state.message.parts.filter(isToolUIPart);
-            const toolInvocation = toolInvocations.find(
+            const toolInvocations = getCurrentStepToolInvocations();
+            let toolInvocation = toolInvocations.find(
               (invocation) => invocation.toolCallId === toolCallId
             );
+            if (toolInvocation == null) {
+              const parts = state.message.parts;
+              for (let i = parts.length - 1; i >= 0; i--) {
+                const part = parts[i];
+                if (isToolUIPart(part) && part.toolCallId === toolCallId) {
+                  toolInvocation = part;
+                  break;
+                }
+              }
+            }
             if (toolInvocation == null) {
               throw new Error(
                 "tool-output-error must be preceded by a tool-input-available"
@@ -3602,12 +3656,22 @@ function processUIMessageStream({
             return toolInvocation;
           }
           function getDynamicToolInvocation(toolCallId) {
-            const toolInvocations = state.message.parts.filter(
+            const toolInvocations = getCurrentStepParts().filter(
               (part) => part.type === "dynamic-tool"
             );
-            const toolInvocation = toolInvocations.find(
+            let toolInvocation = toolInvocations.find(
               (invocation) => invocation.toolCallId === toolCallId
             );
+            if (toolInvocation == null) {
+              const parts = state.message.parts;
+              for (let i = parts.length - 1; i >= 0; i--) {
+                const part = parts[i];
+                if (part.type === "dynamic-tool" && part.toolCallId === toolCallId) {
+                  toolInvocation = part;
+                  break;
+                }
+              }
+            }
             if (toolInvocation == null) {
               throw new Error(
                 "tool-output-error must be preceded by a tool-input-available"
@@ -3615,9 +3679,9 @@ function processUIMessageStream({
             }
             return toolInvocation;
           }
-          function updateToolPart(options) {
+          function updateToolPart(options, existingPart) {
             var _a17;
-            const part = state.message.parts.find(
+            const part = existingPart != null ? existingPart : getCurrentStepParts().find(
               (part2) => isToolUIPart(part2) && part2.toolCallId === options.toolCallId
             );
             const anyOptions = options;
@@ -3648,9 +3712,9 @@ function processUIMessageStream({
               });
             }
           }
-          function updateDynamicToolPart(options) {
+          function updateDynamicToolPart(options, existingPart) {
             var _a17, _b2;
-            const part = state.message.parts.find(
+            const part = existingPart != null ? existingPart : getCurrentStepParts().find(
               (part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId
             );
             const anyOptions = options;
@@ -3782,7 +3846,7 @@ function processUIMessageStream({
               break;
             }
             case "tool-input-start": {
-              const toolInvocations = state.message.parts.filter(isToolUIPart);
+              const toolInvocations = getCurrentStepParts().filter(isToolUIPart);
               state.partialToolCalls[chunk.toolCallId] = {
                 text: "",
                 toolName: chunk.toolName,
@@ -3892,25 +3956,31 @@ function processUIMessageStream({
                 const toolInvocation = getDynamicToolInvocation(
                   chunk.toolCallId
                 );
-                updateDynamicToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: toolInvocation.toolName,
-                  state: "output-available",
-                  input: toolInvocation.input,
-                  output: chunk.output,
-                  preliminary: chunk.preliminary
-                });
+                updateDynamicToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: toolInvocation.toolName,
+                    state: "output-available",
+                    input: toolInvocation.input,
+                    output: chunk.output,
+                    preliminary: chunk.preliminary
+                  },
+                  toolInvocation
+                );
               } else {
                 const toolInvocation = getToolInvocation(chunk.toolCallId);
-                updateToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: getToolName(toolInvocation),
-                  state: "output-available",
-                  input: toolInvocation.input,
-                  output: chunk.output,
-                  providerExecuted: chunk.providerExecuted,
-                  preliminary: chunk.preliminary
-                });
+                updateToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: getToolName(toolInvocation),
+                    state: "output-available",
+                    input: toolInvocation.input,
+                    output: chunk.output,
+                    providerExecuted: chunk.providerExecuted,
+                    preliminary: chunk.preliminary
+                  },
+                  toolInvocation
+                );
               }
               write();
               break;
@@ -3920,25 +3990,31 @@ function processUIMessageStream({
                 const toolInvocation = getDynamicToolInvocation(
                   chunk.toolCallId
                 );
-                updateDynamicToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: toolInvocation.toolName,
-                  state: "output-error",
-                  input: toolInvocation.input,
-                  errorText: chunk.errorText,
-                  providerExecuted: chunk.providerExecuted
-                });
+                updateDynamicToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: toolInvocation.toolName,
+                    state: "output-error",
+                    input: toolInvocation.input,
+                    errorText: chunk.errorText,
+                    providerExecuted: chunk.providerExecuted
+                  },
+                  toolInvocation
+                );
               } else {
                 const toolInvocation = getToolInvocation(chunk.toolCallId);
-                updateToolPart({
-                  toolCallId: chunk.toolCallId,
-                  toolName: getToolName(toolInvocation),
-                  state: "output-error",
-                  input: toolInvocation.input,
-                  rawInput: toolInvocation.rawInput,
-                  errorText: chunk.errorText,
-                  providerExecuted: chunk.providerExecuted
-                });
+                updateToolPart(
+                  {
+                    toolCallId: chunk.toolCallId,
+                    toolName: getToolName(toolInvocation),
+                    state: "output-error",
+                    input: toolInvocation.input,
+                    rawInput: toolInvocation.rawInput,
+                    errorText: chunk.errorText,
+                    providerExecuted: chunk.providerExecuted
+                  },
+                  toolInvocation
+                );
               }
               write();
               break;
@@ -4759,6 +4835,7 @@ var DefaultStreamTextResult = class {
     let recordedRequest = {};
     let recordedWarnings = [];
     const recordedSteps = [];
+    let currentStepToolSet = tools;
     let rootSpan;
     let activeTextContent = createIdMap();
     let activeReasoningContent = createIdMap();
@@ -4874,7 +4951,7 @@ var DefaultStreamTextResult = class {
         if (part.type === "finish-step") {
           const stepMessages = toResponseMessages({
             content: recordedContent,
-            tools
+            tools: currentStepToolSet
           });
           const currentStepResult = new DefaultStepResult({
             content: recordedContent,
@@ -5088,10 +5165,16 @@ var DefaultStreamTextResult = class {
             supportedUrls: await stepModel.supportedUrls,
             download: download2
           });
+          const stepActiveTools = (_d = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _d : activeTools;
+          const stepToolSet = filterActiveTools({
+            tools,
+            activeTools: stepActiveTools
+          });
+          currentStepToolSet = stepToolSet;
           const { toolChoice: stepToolChoice, tools: stepTools } = prepareToolsAndToolChoice({
             tools,
-            toolChoice: (_d = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _d : toolChoice,
-            activeTools: (_e = prepareStepResult == null ? void 0 : prepareStepResult.activeTools) != null ? _e : activeTools
+            toolChoice: (_e = prepareStepResult == null ? void 0 : prepareStepResult.toolChoice) != null ? _e : toolChoice,
+            activeTools: stepActiveTools
           });
           const {
             result: { stream: stream2, response, request },
@@ -5157,7 +5240,7 @@ var DefaultStreamTextResult = class {
             })
           );
           const streamWithToolResults = runToolsTransformation({
-            tools,
+            tools: stepToolSet,
             generatorStream: stream2,
             tracer,
             telemetry,
@@ -5290,7 +5373,7 @@ var DefaultStreamTextResult = class {
                     }
                     case "tool-input-start": {
                       activeToolCallToolNames[chunk.id] = chunk.toolName;
-                      const tool2 = tools == null ? void 0 : tools[chunk.toolName];
+                      const tool2 = stepToolSet == null ? void 0 : stepToolSet[chunk.toolName];
                       if ((tool2 == null ? void 0 : tool2.onInputStart) != null) {
                         await tool2.onInputStart({
                           toolCallId: chunk.id,
@@ -5312,7 +5395,7 @@ var DefaultStreamTextResult = class {
                     }
                     case "tool-input-delta": {
                       const toolName = activeToolCallToolNames[chunk.id];
-                      const tool2 = tools == null ? void 0 : tools[toolName];
+                      const tool2 = stepToolSet == null ? void 0 : stepToolSet[toolName];
                       if ((tool2 == null ? void 0 : tool2.onInputDelta) != null) {
                         await tool2.onInputDelta({
                           inputTextDelta: chunk.delta,
@@ -5409,7 +5492,7 @@ var DefaultStreamTextResult = class {
                           // use transformed content to create the messages for the next step:
                           recordedSteps[recordedSteps.length - 1].content
                         ),
-                        tools
+                        tools: stepToolSet
                       })
                     );
                     try {
@@ -9468,24 +9551,26 @@ var AbstractChat = class {
     body,
     messageId
   }) {
-    var _a16, _b, _c, _d;
+    var _a16, _b, _c;
     this.setStatus({ status: "submitted", error: void 0 });
     const lastMessage = this.lastMessage;
     let isAbort = false;
     let isDisconnect = false;
     let isError = false;
+    let activeResponse;
     try {
-      const activeResponse = {
+      const response = {
         state: createStreamingUIMessageState({
           lastMessage: this.state.snapshot(lastMessage),
           messageId: this.generateId()
         }),
         abortController: new AbortController()
       };
-      activeResponse.abortController.signal.addEventListener("abort", () => {
+      activeResponse = response;
+      response.abortController.signal.addEventListener("abort", () => {
         isAbort = true;
       });
-      this.activeResponse = activeResponse;
+      this.activeResponse = response;
       let stream;
       if (trigger === "resume-stream") {
         const reconnect = await this.transport.reconnectToStream({
@@ -9503,7 +9588,7 @@ var AbstractChat = class {
         stream = await this.transport.sendMessages({
           chatId: this.id,
           messages: this.state.messages,
-          abortSignal: activeResponse.abortController.signal,
+          abortSignal: response.abortController.signal,
           metadata,
           headers,
           body,
@@ -9515,18 +9600,18 @@ var AbstractChat = class {
         // serialize the job execution to avoid race conditions:
         this.jobExecutor.run(
           () => job({
-            state: activeResponse.state,
+            state: response.state,
             write: () => {
               var _a17;
               this.setStatus({ status: "streaming" });
-              const replaceLastMessage = activeResponse.state.message.id === ((_a17 = this.lastMessage) == null ? void 0 : _a17.id);
+              const replaceLastMessage = response.state.message.id === ((_a17 = this.lastMessage) == null ? void 0 : _a17.id);
               if (replaceLastMessage) {
                 this.state.replaceMessage(
                   this.state.messages.length - 1,
-                  activeResponse.state.message
+                  response.state.message
                 );
               } else {
-                this.state.pushMessage(activeResponse.state.message);
+                this.state.pushMessage(response.state.message);
               }
             }
           })
@@ -9565,23 +9650,27 @@ var AbstractChat = class {
       this.setStatus({ status: "error", error: err });
     } finally {
       try {
-        (_b = this.onFinish) == null ? void 0 : _b.call(this, {
-          message: this.activeResponse.state.message,
-          messages: this.state.messages,
-          isAbort,
-          isDisconnect,
-          isError,
-          finishReason: (_a16 = this.activeResponse) == null ? void 0 : _a16.state.finishReason
-        });
+        if (activeResponse) {
+          (_a16 = this.onFinish) == null ? void 0 : _a16.call(this, {
+            message: activeResponse.state.message,
+            messages: this.state.messages,
+            isAbort,
+            isDisconnect,
+            isError,
+            finishReason: activeResponse.state.finishReason
+          });
+        }
       } catch (err) {
         console.error(err);
       }
-      this.activeResponse = void 0;
+      if (this.activeResponse === activeResponse) {
+        this.activeResponse = void 0;
+      }
     }
-    if (((_c = this.sendAutomaticallyWhen) == null ? void 0 : _c.call(this, { messages: this.state.messages })) && !isError) {
+    if (((_b = this.sendAutomaticallyWhen) == null ? void 0 : _b.call(this, { messages: this.state.messages })) && !isError) {
       await this.makeRequest({
         trigger: "submit-message",
-        messageId: (_d = this.lastMessage) == null ? void 0 : _d.id,
+        messageId: (_c = this.lastMessage) == null ? void 0 : _c.id,
         metadata,
         headers,
         body