ai 7.0.106 → 7.0.107

package/CHANGELOG.md

@@ -1,5 +1,25 @@
 # ai
 
+## 7.0.107
+
+### Patch Changes
+
+- 79681c4: fix(ai): preserve provider file and skill upload APIs in wrapProvider
+- 98c7275: fix(ai): preserve query parameters in chat reconnect URLs
+- a105059: fix(workflow): support deferred tool discovery in WorkflowAgent
+- 31532f3: fix(ai): prevent preliminary tool outputs from completing chats
+- e61cbd8: fix(ai): preserve raw speech audio format metadata
+- 8ade040: fix(ai): pass tool-specific context to input callbacks
+- 970a01e: fix(ai): enforce polling timeouts for in-flight video status requests
+- 85539c5: fix(ai): preserve multiple Set-Cookie headers in Node stream responses
+- 611d301: fix(ai): prevent duplicate content types in chat transport requests
+- c415657: fix(ai): decode base64 text data URLs using their declared charset
+- Updated dependencies [2973485]
+- Updated dependencies [a4db5ea]
+- Updated dependencies [2937ea2]
+  - @ai-sdk/provider-utils@5.0.45
+  - @ai-sdk/gateway@4.0.87
+
 ## 7.0.106
 
 ### Patch Changes

package/dist/index.js

@@ -1211,7 +1211,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.106" : "0.0.0-test";
+var VERSION = true ? "7.0.107" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -6265,22 +6265,29 @@ async function generateText({
                 if (tool3 == null) {
                   continue;
                 }
-                if (tool3.onInputStart != null) {
-                  await tool3.onInputStart({
-                    toolCallId: toolCall.toolCallId,
-                    messages: stepMessages,
-                    abortSignal: mergedAbortSignal,
-                    context: runtimeContext
-                  });
-                }
-                if ((tool3 == null ? void 0 : tool3.onInputAvailable) != null) {
-                  await tool3.onInputAvailable({
-                    input: toolCall.input,
-                    toolCallId: toolCall.toolCallId,
-                    messages: stepMessages,
-                    abortSignal: mergedAbortSignal,
-                    context: runtimeContext
+                if (tool3.onInputStart != null || tool3.onInputAvailable != null) {
+                  const context = await validateToolContext({
+                    toolName: toolCall.toolName,
+                    context: getOwn(toolsContext, toolCall.toolName),
+                    contextSchema: tool3.contextSchema
                   });
+                  if (tool3.onInputStart != null) {
+                    await tool3.onInputStart({
+                      toolCallId: toolCall.toolCallId,
+                      messages: stepMessages,
+                      abortSignal: mergedAbortSignal,
+                      context
+                    });
+                  }
+                  if (tool3.onInputAvailable != null) {
+                    await tool3.onInputAvailable({
+                      input: toolCall.input,
+                      toolCallId: toolCall.toolCallId,
+                      messages: stepMessages,
+                      abortSignal: mergedAbortSignal,
+                      context
+                    });
+                  }
                 }
                 const toolApprovalStatus = await resolveToolApproval({
                   tools: stepExecutionTools,
@@ -6783,10 +6790,13 @@ function writeToServerResponse({
   stream
 }) {
   const statusCode = status != null ? status : 200;
+  if (headers != null) {
+    response.setHeaders(headers);
+  }
   if (statusText !== void 0) {
-    response.writeHead(statusCode, statusText, headers);
+    response.writeHead(statusCode, statusText);
   } else {
-    response.writeHead(statusCode, headers);
+    response.writeHead(statusCode);
   }
   const reader = stream.getReader();
   const read = async () => {
@@ -6826,11 +6836,9 @@ function pipeTextStreamToResponse({
     response,
     status,
     statusText,
-    headers: Object.fromEntries(
-      prepareHeaders(headers, {
-        "content-type": "text/plain; charset=utf-8"
-      }).entries()
-    ),
+    headers: prepareHeaders(headers, {
+      "content-type": "text/plain; charset=utf-8"
+    }),
     stream: stream.pipeThrough(new TextEncoderStream())
   });
 }
@@ -6916,9 +6924,7 @@ function pipeUIMessageStreamToResponse({
     response,
     status,
     statusText,
-    headers: Object.fromEntries(
-      prepareHeaders(headers, UI_MESSAGE_STREAM_HEADERS).entries()
-    ),
+    headers: prepareHeaders(headers, UI_MESSAGE_STREAM_HEADERS),
     stream: sseStream.pipeThrough(new TextEncoderStream())
   });
 }
@@ -8796,33 +8802,60 @@ function invokeToolCallbacksFromStream({
   tools,
   stepInputMessages,
   abortSignal,
-  runtimeContext
+  toolsContext
 }) {
   if (tools == null) return stream;
-  const ongoingToolCallToolNames = createIdMap();
+  let ongoingToolCalls = createIdMap();
+  const getValidatedContext = ({
+    toolCallId,
+    toolName
+  }) => {
+    const ongoingToolCall = ongoingToolCalls[toolCallId];
+    if ((ongoingToolCall == null ? void 0 : ongoingToolCall.validatedContext) != null) {
+      return ongoingToolCall.validatedContext;
+    }
+    const tool3 = getOwn(tools, toolName);
+    const validatedContext = validateToolContext({
+      toolName,
+      context: getOwn(toolsContext, toolName),
+      contextSchema: tool3 == null ? void 0 : tool3.contextSchema
+    });
+    if (ongoingToolCall != null) {
+      ongoingToolCall.validatedContext = validatedContext;
+    }
+    return validatedContext;
+  };
   return stream.pipeThrough(
     new TransformStream({
       async transform(chunk, controller) {
+        var _a25, _b25;
         controller.enqueue(chunk);
         if (isStreamRetryAttemptBoundaryPart(chunk)) {
+          ongoingToolCalls = createIdMap();
           return;
         }
         switch (chunk.type) {
           case "tool-input-start": {
-            ongoingToolCallToolNames[chunk.id] = chunk.toolName;
+            ongoingToolCalls[chunk.id] = {
+              toolName: chunk.toolName,
+              validatedContext: void 0
+            };
             const tool3 = getOwn(tools, chunk.toolName);
             if ((tool3 == null ? void 0 : tool3.onInputStart) != null) {
               await tool3.onInputStart({
                 toolCallId: chunk.id,
                 messages: stepInputMessages,
                 abortSignal,
-                context: runtimeContext
+                context: await getValidatedContext({
+                  toolCallId: chunk.id,
+                  toolName: chunk.toolName
+                })
               });
             }
             break;
           }
           case "tool-input-delta": {
-            const toolName = ongoingToolCallToolNames[chunk.id];
+            const toolName = (_a25 = ongoingToolCalls[chunk.id]) == null ? void 0 : _a25.toolName;
             const tool3 = getOwn(tools, toolName);
             if ((tool3 == null ? void 0 : tool3.onInputDelta) != null) {
               await tool3.onInputDelta({
@@ -8830,23 +8863,32 @@ function invokeToolCallbacksFromStream({
                 toolCallId: chunk.id,
                 messages: stepInputMessages,
                 abortSignal,
-                context: runtimeContext
+                context: await getValidatedContext({
+                  toolCallId: chunk.id,
+                  toolName
+                })
               });
             }
             break;
           }
           case "tool-call": {
-            const toolName = ongoingToolCallToolNames[chunk.toolCallId];
+            const toolName = (_b25 = ongoingToolCalls[chunk.toolCallId]) == null ? void 0 : _b25.toolName;
             const tool3 = getOwn(tools, toolName);
-            delete ongoingToolCallToolNames[chunk.toolCallId];
             if (!chunk.invalid && (tool3 == null ? void 0 : tool3.onInputAvailable) != null) {
+              const validatedContext = getValidatedContext({
+                toolCallId: chunk.toolCallId,
+                toolName
+              });
+              delete ongoingToolCalls[chunk.toolCallId];
               await tool3.onInputAvailable({
                 input: chunk.input,
                 toolCallId: chunk.toolCallId,
                 messages: stepInputMessages,
                 abortSignal,
-                context: runtimeContext
+                context: await validatedContext
               });
+            } else {
+              delete ongoingToolCalls[chunk.toolCallId];
             }
           }
         }
@@ -10881,7 +10923,7 @@ var DefaultStreamTextResult = class {
             tools: stepExecutionTools,
             stepInputMessages: stepMessages,
             abortSignal,
-            runtimeContext
+            toolsContext
           });
           const runInTracingChannelSpanInStep = telemetryDispatcher.runInTracingChannelSpan == null ? void 0 : (options) => runInStepTracingChannelContext(
             () => telemetryDispatcher.runInTracingChannelSpan(options)
@@ -15331,8 +15373,13 @@ function createDownload(options) {
 // src/util/data-url.ts
 var { atob: atob2 } = globalThis;
 function getTextFromDataUrl(dataUrl) {
+  var _a25;
   const [header, base64Content] = dataUrl.split(",");
   const mediaType = header.split(";")[0].split(":")[1];
+  const charsetMatch = /(?:^|;)\s*charset\s*=\s*(?:"([^"]+)"|([^;\s]+))/i.exec(
+    header
+  );
+  const charset = (_a25 = charsetMatch == null ? void 0 : charsetMatch[1]) != null ? _a25 : charsetMatch == null ? void 0 : charsetMatch[2];
   if (mediaType == null || base64Content == null) {
     throw new InvalidArgumentError({
       parameter: "dataUrl",
@@ -15341,7 +15388,13 @@ function getTextFromDataUrl(dataUrl) {
     });
   }
   try {
-    return atob2(base64Content);
+    const byteString = atob2(base64Content);
+    if (charset == null) {
+      return byteString;
+    }
+    return new TextDecoder(charset).decode(
+      Uint8Array.from(byteString, (byte) => byte.codePointAt(0))
+    );
   } catch (e) {
     throw new InvalidArgumentError({
       parameter: "dataUrl",
@@ -16010,7 +16063,7 @@ async function generateSpeech({
   abortSignal,
   headers
 }) {
-  var _a25;
+  var _a25, _b25;
   const resolvedModel = resolveSpeechModel(model);
   if (!resolvedModel) {
     throw new Error("Model could not be resolved");
@@ -16044,19 +16097,41 @@ async function generateSpeech({
     provider: resolvedModel.provider,
     model: resolvedModel.modelId
   });
+  const detectedMediaType = detectMediaType4({
+    data: result.audio,
+    topLevelType: "audio"
+  });
   return new DefaultSpeechResult({
     audio: new DefaultGeneratedAudioFile({
       data: result.audio,
-      mediaType: (_a25 = detectMediaType4({
-        data: result.audio,
-        topLevelType: "audio"
-      })) != null ? _a25 : "audio/mp3"
+      mediaType: (_b25 = (_a25 = detectedMediaType != null ? detectedMediaType : getResponseAudioMediaType(result.response.headers)) != null ? _a25 : getOutputFormatMediaType(outputFormat)) != null ? _b25 : "audio/mp3"
     }),
     warnings: result.warnings,
     responses: [result.response],
     providerMetadata: result.providerMetadata
   });
 }
+function getResponseAudioMediaType(headers) {
+  var _a25;
+  const mediaType = (_a25 = Object.entries(headers != null ? headers : {}).find(
+    ([name25]) => name25.toLowerCase() === "content-type"
+  )) == null ? void 0 : _a25[1];
+  if (mediaType == null) {
+    return void 0;
+  }
+  const normalizedMediaType = mediaType.split(";", 1)[0].trim().toLowerCase();
+  if (normalizedMediaType.length === 0) {
+    return void 0;
+  }
+  return normalizedMediaType.startsWith("audio/") ? normalizedMediaType : void 0;
+}
+function getOutputFormatMediaType(outputFormat) {
+  if (outputFormat == null) {
+    return void 0;
+  }
+  const normalizedOutputFormat = outputFormat.trim().toLowerCase();
+  return normalizedOutputFormat === "pcm" || normalizedOutputFormat === "audio/pcm" ? "audio/pcm" : void 0;
+}
 var DefaultSpeechResult = class {
   constructor(options) {
     var _a25;
@@ -16364,7 +16439,7 @@ async function experimental_generateVideo({
     headers != null ? headers : {},
     `ai/${VERSION}`
   );
-  const { retry } = prepareRetries({
+  const { maxRetries, retry } = prepareRetries({
     maxRetries: maxRetriesArg,
     abortSignal
   });
@@ -16424,6 +16499,7 @@ async function experimental_generateVideo({
           callOptions,
           poll,
           webhook,
+          maxRetries,
           retry
         });
       }
@@ -16512,6 +16588,7 @@ async function executeStartStatusFlow({
   callOptions,
   poll: pollConfig,
   webhook: webhookFactory,
+  maxRetries,
   retry
 }) {
   var _a25, _b25, _c, _d;
@@ -16553,6 +16630,9 @@ async function executeStartStatusFlow({
   const timeoutMs = (_c = pollConfig == null ? void 0 : pollConfig.timeoutMs) != null ? _c : 6e5;
   const delay = (_d = pollConfig == null ? void 0 : pollConfig.delay) != null ? _d : defaultDelay;
   const startTime = Date.now();
+  const pollingTimeoutError = new Error(
+    `Video generation timed out after ${timeoutMs}ms.`
+  );
   if (webhookReceived != null) {
     await waitForWebhook({
       received: webhookReceived,
@@ -16565,22 +16645,65 @@ async function executeStartStatusFlow({
     if (webhookReceived == null) {
       const elapsedMs = Date.now() - startTime;
       if (elapsedMs >= timeoutMs) {
-        throw new Error(`Video generation timed out after ${timeoutMs}ms.`);
+        throw pollingTimeoutError;
       }
       await delay(Math.min(intervalMs, timeoutMs - elapsedMs), {
         abortSignal: callOptions.abortSignal
       });
       if (Date.now() - startTime >= timeoutMs) {
-        throw new Error(`Video generation timed out after ${timeoutMs}ms.`);
+        throw pollingTimeoutError;
+      }
+    }
+    let statusResult;
+    if (webhookReceived != null) {
+      statusResult = await retry(
+        () => model.doStatus({
+          operation: startResult.operation,
+          abortSignal: callOptions.abortSignal,
+          headers: callOptions.headers
+        })
+      );
+    } else {
+      const statusTimeoutController = new AbortController();
+      const statusAbortSignal = mergeAbortSignals(
+        callOptions.abortSignal,
+        statusTimeoutController.signal
+      );
+      const statusTimeoutId = setTimeout(
+        () => statusTimeoutController.abort(pollingTimeoutError),
+        timeoutMs - (Date.now() - startTime)
+      );
+      const statusTimeoutPromise = new Promise((_, reject) => {
+        statusTimeoutController.signal.addEventListener(
+          "abort",
+          () => reject(pollingTimeoutError),
+          { once: true }
+        );
+      });
+      const { retry: statusRetry } = prepareRetries({
+        maxRetries,
+        abortSignal: statusAbortSignal
+      });
+      try {
+        statusResult = await Promise.race([
+          statusRetry(
+            () => model.doStatus({
+              operation: startResult.operation,
+              abortSignal: statusAbortSignal,
+              headers: callOptions.headers
+            })
+          ),
+          statusTimeoutPromise
+        ]);
+      } catch (error) {
+        if (statusTimeoutController.signal.aborted) {
+          throw pollingTimeoutError;
+        }
+        throw error;
+      } finally {
+        clearTimeout(statusTimeoutId);
       }
     }
-    const statusResult = await retry(
-      () => model.doStatus({
-        operation: startResult.operation,
-        abortSignal: callOptions.abortSignal,
-        headers: callOptions.headers
-      })
-    );
     if (statusResult.status === "error") {
       throw new Error(statusResult.error);
     }
@@ -17551,7 +17674,9 @@ function wrapProvider({
     },
     transcriptionModel: providerV4.transcriptionModel,
     speechModel: providerV4.speechModel,
-    rerankingModel: providerV4.rerankingModel
+    rerankingModel: providerV4.rerankingModel,
+    ...providerV4.files != null ? { files: providerV4.files } : {},
+    ...providerV4.skills != null ? { skills: providerV4.skills } : {}
   };
 }
 
@@ -21532,6 +21657,10 @@ import {
   resolve as resolve2
 } from "@ai-sdk/provider-utils";
 import { EmptyResponseBodyError as EmptyResponseBodyError3 } from "@ai-sdk/provider";
+function appendPathToUrl(url, path) {
+  const queryOrFragmentStart = url.search(/[?#]/);
+  return queryOrFragmentStart === -1 ? `${url}${path}` : `${url.slice(0, queryOrFragmentStart)}${path}${url.slice(queryOrFragmentStart)}`;
+}
 var HttpChatTransport = class {
   constructor({
     api = "/api/chat",
@@ -21588,7 +21717,7 @@ var HttpChatTransport = class {
     const response = await fetch2(api, {
       method: "POST",
       headers: {
-        "Content-Type": "application/json",
+        "content-type": "application/json",
         ...headers
       },
       body: JSON.stringify(body),
@@ -21626,7 +21755,7 @@ var HttpChatTransport = class {
       credentials: resolvedCredentials,
       requestMetadata: options.metadata
     }));
-    const api = (_b25 = preparedRequest == null ? void 0 : preparedRequest.api) != null ? _b25 : `${this.api}/${options.chatId}/stream`;
+    const api = (_b25 = preparedRequest == null ? void 0 : preparedRequest.api) != null ? _b25 : appendPathToUrl(this.api, `/${options.chatId}/stream`);
     const headers = (preparedRequest == null ? void 0 : preparedRequest.headers) !== void 0 ? normalizeHeaders(preparedRequest.headers) : baseHeaders;
     const credentials = (_c = preparedRequest == null ? void 0 : preparedRequest.credentials) != null ? _c : resolvedCredentials;
     const fetch2 = (_d = this.fetch) != null ? _d : globalThis.fetch;
@@ -22212,7 +22341,7 @@ function lastAssistantMessageIsCompleteWithApprovalResponses({
     // has at least one tool approval response
     lastStepToolInvocations.filter((part) => part.state === "approval-responded").length > 0 && // all tool approvals must have a response
     lastStepToolInvocations.every(
-      (part) => part.state === "output-available" || part.state === "output-error" || part.state === "output-denied" || part.state === "approval-responded"
+      (part) => part.state === "output-available" && part.preliminary !== true || part.state === "output-error" || part.state === "output-denied" || part.state === "approval-responded"
     )
   );
 }
@@ -22233,7 +22362,7 @@ function lastAssistantMessageIsCompleteWithToolCalls({
   }, -1);
   const lastStepToolInvocations = message.parts.slice(lastStepStartIndex + 1).filter(isToolUIPart).filter((part) => !part.providerExecuted);
   return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every(
-    (part) => part.state === "output-available" || part.state === "output-error"
+    (part) => part.state === "output-available" && part.preliminary !== true || part.state === "output-error"
   );
 }