ai 6.0.296 → 6.0.298

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.296" : "0.0.0-test";
+var VERSION = true ? "6.0.298" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5712,6 +5712,75 @@ function pipeTextStreamToResponse({
   });
 }
 
+// src/ui-message-stream/create-sse-stream-with-keep-alive.ts
+var STREAM_OPEN_COMMENT = ": stream-open\n\n";
+var KEEP_ALIVE_COMMENT = ": keep-alive\n\n";
+function createSseStreamWithKeepAlive({
+  stream,
+  keepAliveMs
+}) {
+  if (keepAliveMs == null) {
+    return stream;
+  }
+  if (!Number.isFinite(keepAliveMs) || keepAliveMs <= 0 || keepAliveMs > 2147483647) {
+    throw new Error(
+      "keepAliveMs must be a positive finite timer duration no greater than 2147483647"
+    );
+  }
+  const reader = stream.getReader();
+  let keepAliveTimeout;
+  let isCancelled = false;
+  const clearKeepAliveTimeout = () => {
+    clearTimeout(keepAliveTimeout);
+    keepAliveTimeout = void 0;
+  };
+  const scheduleKeepAlive = (controller) => {
+    clearKeepAliveTimeout();
+    keepAliveTimeout = setTimeout(() => {
+      if (isCancelled) {
+        return;
+      }
+      if (controller.desiredSize != null && controller.desiredSize > 0) {
+        controller.enqueue(KEEP_ALIVE_COMMENT);
+      }
+      scheduleKeepAlive(controller);
+    }, keepAliveMs);
+  };
+  return new ReadableStream({
+    start(controller) {
+      controller.enqueue(STREAM_OPEN_COMMENT);
+      scheduleKeepAlive(controller);
+    },
+    pull(controller) {
+      return reader.read().then(
+        (result) => {
+          clearKeepAliveTimeout();
+          if (isCancelled) {
+            return;
+          }
+          if (result.done) {
+            controller.close();
+          } else {
+            controller.enqueue(result.value);
+            scheduleKeepAlive(controller);
+          }
+        },
+        (error) => {
+          clearKeepAliveTimeout();
+          if (!isCancelled) {
+            throw error;
+          }
+        }
+      );
+    },
+    async cancel(reason) {
+      isCancelled = true;
+      clearKeepAliveTimeout();
+      await reader.cancel(reason);
+    }
+  });
+}
+
 // src/ui-message-stream/json-to-sse-transform-stream.ts
 var JsonToSseTransformStream = class extends TransformStream {
   constructor() {
@@ -5744,9 +5813,13 @@ function createUIMessageStreamResponse({
   statusText,
   headers,
   stream,
+  keepAliveMs,
   consumeSseStream
 }) {
-  let sseStream = stream.pipeThrough(new JsonToSseTransformStream());
+  let sseStream = createSseStreamWithKeepAlive({
+    stream: stream.pipeThrough(new JsonToSseTransformStream()),
+    keepAliveMs
+  });
   if (consumeSseStream) {
     const [stream1, stream2] = sseStream.tee();
     sseStream = stream1;
@@ -5998,16 +6071,43 @@ function createStreamingUIMessageState({
   lastMessage,
   messageId
 }) {
+  var _a23;
+  const message = (lastMessage == null ? void 0 : lastMessage.role) === "assistant" ? lastMessage : {
+    id: messageId,
+    metadata: void 0,
+    role: "assistant",
+    parts: []
+  };
+  const partialToolCalls = createIdMap();
+  const lastStepStartIndex = message.parts.findLastIndex(
+    (part) => part.type === "step-start"
+  );
+  let staticToolIndex = 0;
+  for (const part of message.parts.slice(lastStepStartIndex + 1)) {
+    if (!isToolUIPart(part)) {
+      continue;
+    }
+    const index = staticToolIndex;
+    if (isStaticToolUIPart(part)) {
+      staticToolIndex++;
+    }
+    if (part.state !== "input-streaming") {
+      continue;
+    }
+    partialToolCalls[part.toolCallId] = {
+      text: (_a23 = part.rawInput) != null ? _a23 : "",
+      index,
+      toolName: part.type === "dynamic-tool" ? part.toolName : getStaticToolName(part),
+      dynamic: part.type === "dynamic-tool",
+      title: part.title,
+      toolMetadata: part.toolMetadata
+    };
+  }
   return {
-    message: (lastMessage == null ? void 0 : lastMessage.role) === "assistant" ? lastMessage : {
-      id: messageId,
-      metadata: void 0,
-      role: "assistant",
-      parts: []
-    },
+    message,
     activeTextParts: createIdMap(),
     activeReasoningParts: createIdMap(),
-    partialToolCalls: createIdMap()
+    partialToolCalls
   };
 }
 function processUIMessageStream({
@@ -6108,7 +6208,7 @@ function processUIMessageStream({
             }
           }
           function updateDynamicToolPart(options, existingPart) {
-            var _a24, _b2;
+            var _a24;
             const part = existingPart != null ? existingPart : getCurrentStepParts().find(
               (part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId
             );
@@ -6120,7 +6220,7 @@ function processUIMessageStream({
               anyPart.input = anyOptions.input;
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
-              anyPart.rawInput = (_a24 = anyOptions.rawInput) != null ? _a24 : anyPart.rawInput;
+              anyPart.rawInput = anyOptions.rawInput;
               anyPart.preliminary = anyOptions.preliminary;
               if (options.title !== void 0) {
                 anyPart.title = options.title;
@@ -6128,7 +6228,7 @@ function processUIMessageStream({
               if (options.toolMetadata !== void 0) {
                 anyPart.toolMetadata = options.toolMetadata;
               }
-              anyPart.providerExecuted = (_b2 = anyOptions.providerExecuted) != null ? _b2 : part.providerExecuted;
+              anyPart.providerExecuted = (_a24 = anyOptions.providerExecuted) != null ? _a24 : part.providerExecuted;
               const providerMetadata = anyOptions.providerMetadata;
               if (providerMetadata != null) {
                 if (options.state === "output-available" || options.state === "output-error") {
@@ -6344,6 +6444,7 @@ function processUIMessageStream({
                   toolName: partialToolCall.toolName,
                   state: "input-streaming",
                   input: partialArgs,
+                  rawInput: partialToolCall.text,
                   title: partialToolCall.title,
                   toolMetadata: partialToolCall.toolMetadata
                 });
@@ -6353,6 +6454,7 @@ function processUIMessageStream({
                   toolName: partialToolCall.toolName,
                   state: "input-streaming",
                   input: partialArgs,
+                  rawInput: partialToolCall.text,
                   title: partialToolCall.title,
                   toolMetadata: partialToolCall.toolMetadata
                 });
@@ -6741,9 +6843,13 @@ function pipeUIMessageStreamToResponse({
   statusText,
   headers,
   stream,
+  keepAliveMs,
   consumeSseStream
 }) {
-  let sseStream = stream.pipeThrough(new JsonToSseTransformStream());
+  let sseStream = createSseStreamWithKeepAlive({
+    stream: stream.pipeThrough(new JsonToSseTransformStream()),
+    keepAliveMs
+  });
   if (consumeSseStream) {
     const [stream1, stream2] = sseStream.tee();
     sseStream = stream1;
@@ -8905,7 +9011,7 @@ var DefaultStreamTextResult = class {
     sendStart = true,
     sendFinish = true,
     onError = () => "An error occurred."
-    // prevent leaking server error details to the client by default
+    // masks errors except provider-executed tool execution errors
   } = {}) {
     let outcome = { status: "unknown" };
     let hasFatalFailure = false;
@@ -9943,6 +10049,7 @@ var uiMessagesSchema = (0, import_provider_utils24.lazySchema)(() => {
               toolMetadata: toolMetadataSchema2.optional(),
               state: import_v48.z.literal("input-streaming"),
               input: import_v48.z.unknown().optional(),
+              rawInput: import_v48.z.string().optional(),
               providerExecuted: import_v48.z.boolean().optional(),
               callProviderMetadata: providerMetadataSchema.optional(),
               output: import_v48.z.never().optional(),
@@ -10039,6 +10146,7 @@ var uiMessagesSchema = (0, import_provider_utils24.lazySchema)(() => {
               providerExecuted: import_v48.z.boolean().optional(),
               callProviderMetadata: providerMetadataSchema.optional(),
               input: import_v48.z.unknown().optional(),
+              rawInput: import_v48.z.string().optional(),
               output: import_v48.z.never().optional(),
               errorText: import_v48.z.never().optional(),
               approval: import_v48.z.never().optional()
@@ -10360,6 +10468,7 @@ async function createAgentUIStreamResponse({
   headers,
   status,
   statusText,
+  keepAliveMs,
   consumeSseStream,
   ...options
 }) {
@@ -10367,6 +10476,7 @@ async function createAgentUIStreamResponse({
     headers,
     status,
     statusText,
+    keepAliveMs,
     consumeSseStream,
     stream: await createAgentUIStream(options)
   });
@@ -10378,6 +10488,7 @@ async function pipeAgentUIStreamToResponse({
   headers,
   status,
   statusText,
+  keepAliveMs,
   consumeSseStream,
   ...options
 }) {
@@ -10386,6 +10497,7 @@ async function pipeAgentUIStreamToResponse({
     headers,
     status,
     statusText,
+    keepAliveMs,
     consumeSseStream,
     stream: await createAgentUIStream(options)
   });
@@ -10520,6 +10632,21 @@ var DefaultEmbedResult = class {
   }
 };
 
+// src/util/get-own.ts
+function getOwn(obj, key) {
+  return obj != null && Object.hasOwn(obj, key) ? obj[key] : void 0;
+}
+
+// src/util/set-own.ts
+function setOwn(object2, key, value) {
+  Object.defineProperty(object2, key, {
+    value,
+    enumerable: true,
+    configurable: true,
+    writable: true
+  });
+}
+
 // src/embed/embed-many.ts
 var import_provider_utils28 = require("@ai-sdk/provider-utils");
 
@@ -10769,10 +10896,10 @@ async function embedMany({
               for (const [providerName, metadata] of Object.entries(
                 result.providerMetadata
               )) {
-                providerMetadata[providerName] = {
-                  ...providerMetadata[providerName],
+                setOwn(providerMetadata, providerName, {
+                  ...getOwn(providerMetadata, providerName),
                   ...metadata
-                };
+                });
               }
             }
           }
@@ -10968,10 +11095,13 @@ async function generateImage({
             delete providerMetadata[providerName].images;
           }
         } else {
-          (_b = providerMetadata[providerName]) != null ? _b : providerMetadata[providerName] = { images: [] };
-          providerMetadata[providerName].images.push(
+          const currentEntry = (_b = getOwn(providerMetadata, providerName)) != null ? _b : {
+            images: []
+          };
+          currentEntry.images.push(
             ...result.providerMetadata[providerName].images
           );
+          setOwn(providerMetadata, providerName, currentEntry);
         }
       }
     }
@@ -12915,12 +13045,12 @@ async function experimental_generateVideo({
       for (const [providerName, metadata] of Object.entries(
         result.providerMetadata
       )) {
-        const existingMetadata = providerMetadata[providerName];
+        const existingMetadata = getOwn(providerMetadata, providerName);
         if (existingMetadata != null && typeof existingMetadata === "object") {
-          providerMetadata[providerName] = {
+          setOwn(providerMetadata, providerName, {
             ...existingMetadata,
             ...metadata
-          };
+          });
           if ("videos" in existingMetadata && Array.isArray(existingMetadata.videos) && "videos" in metadata && Array.isArray(metadata.videos)) {
             providerMetadata[providerName].videos = [
               ...existingMetadata.videos,
@@ -12928,7 +13058,7 @@ async function experimental_generateVideo({
             ];
           }
         } else {
-          providerMetadata[providerName] = metadata;
+          setOwn(providerMetadata, providerName, metadata);
         }
       }
     }
@@ -13799,7 +13929,7 @@ var DefaultProviderRegistry = class {
     languageModelMiddleware,
     imageModelMiddleware
   }) {
-    this.providers = {};
+    this.providers = /* @__PURE__ */ Object.create(null);
     this.separator = separator;
     this.languageModelMiddleware = languageModelMiddleware;
     this.imageModelMiddleware = imageModelMiddleware;
@@ -14308,6 +14438,10 @@ var import_provider_utils42 = require("@ai-sdk/provider-utils");
 
 // src/ui/http-chat-transport.ts
 var import_provider_utils41 = require("@ai-sdk/provider-utils");
+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",
@@ -14382,7 +14516,7 @@ var HttpChatTransport = class {
     return this.processResponseStream(response.body);
   }
   async reconnectToStream(options) {
-    var _a23, _b, _c, _d, _e;
+    var _a23, _b, _c, _d;
     const resolvedBody = await (0, import_provider_utils41.resolve)(this.body);
     const resolvedHeaders = await (0, import_provider_utils41.resolve)(this.headers);
     const resolvedCredentials = await (0, import_provider_utils41.resolve)(this.credentials);
@@ -14398,10 +14532,23 @@ var HttpChatTransport = class {
       credentials: resolvedCredentials,
       requestMetadata: options.metadata
     }));
-    const api = (_b = preparedRequest == null ? void 0 : preparedRequest.api) != null ? _b : `${this.api}/${options.chatId}/stream`;
+    let api = preparedRequest == null ? void 0 : preparedRequest.api;
+    if (api == null) {
+      if (options.chatId === "." || options.chatId === "..") {
+        throw new InvalidArgumentError({
+          parameter: "chatId",
+          value: options.chatId,
+          message: 'Chat IDs must not be "." or ".." when using the default reconnect URL.'
+        });
+      }
+      api = appendPathToUrl(
+        this.api,
+        `/${encodeURIComponent(options.chatId)}/stream`
+      );
+    }
     const headers = (preparedRequest == null ? void 0 : preparedRequest.headers) !== void 0 ? (0, import_provider_utils41.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;
+    const credentials = (_b = preparedRequest == null ? void 0 : preparedRequest.credentials) != null ? _b : resolvedCredentials;
+    const fetch2 = (_c = this.fetch) != null ? _c : globalThis.fetch;
     const response = await fetch2(api, {
       method: "GET",
       headers,
@@ -14413,7 +14560,7 @@ var HttpChatTransport = class {
     }
     if (!response.ok) {
       throw new Error(
-        (_e = await response.text()) != null ? _e : "Failed to fetch the chat response."
+        (_d = await response.text()) != null ? _d : "Failed to fetch the chat response."
       );
     }
     if (!response.body) {
@@ -14808,6 +14955,9 @@ var AbstractChat = class {
     const lastMessage = this.lastMessage;
     const responseMessageIndex = trigger === "submit-message" && messageId != null ? this.state.messages.findIndex((message) => message.id === messageId) : this.state.messages.length - 1;
     const responseMessage = responseMessageIndex === -1 ? lastMessage : this.state.messages[responseMessageIndex];
+    const resumableResponseMessage = trigger === "resume-stream" && (responseMessage == null ? void 0 : responseMessage.role) === "assistant" && responseMessage.parts.some(
+      (part) => isToolUIPart(part) && part.state === "input-streaming"
+    ) ? this.state.snapshot(responseMessage) : void 0;
     const usesEarlierAssistantMessage = responseMessageIndex !== -1 && responseMessageIndex < this.state.messages.length - 1 && (responseMessage == null ? void 0 : responseMessage.role) === "assistant";
     let isAbort = false;
     let isDisconnect = false;
@@ -14816,7 +14966,7 @@ var AbstractChat = class {
     try {
       const response = {
         state: trigger === "resume-stream" && this.resumableStreamState != null ? this.resumableStreamState : createStreamingUIMessageState({
-          lastMessage: trigger === "resume-stream" || trigger === "regenerate-message" ? void 0 : this.state.snapshot(responseMessage),
+          lastMessage: trigger === "resume-stream" ? resumableResponseMessage : trigger === "regenerate-message" ? void 0 : this.state.snapshot(responseMessage),
           messageId: this.generateId()
         }),
         abortController
@@ -15042,8 +15192,13 @@ function lastAssistantMessageIsCompleteWithToolCalls({
   const lastStepStartIndex = message.parts.reduce((lastIndex, part, index) => {
     return part.type === "step-start" ? index : lastIndex;
   }, -1);
-  const lastStepToolInvocations = message.parts.slice(lastStepStartIndex + 1).filter(isToolUIPart).filter((part) => !part.providerExecuted);
-  return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every(
+  const lastStepParts = message.parts.slice(lastStepStartIndex + 1);
+  const lastStepToolInvocations = lastStepParts.filter(isToolUIPart).filter((part) => !part.providerExecuted);
+  const lastToolInvocationIndex = lastStepParts.reduce(
+    (lastIndex, part, index) => isToolUIPart(part) && !part.providerExecuted ? index : lastIndex,
+    -1
+  );
+  return lastStepToolInvocations.length > 0 && !lastStepParts.slice(lastToolInvocationIndex + 1).some((part) => part.type === "text" && part.state !== "done") && lastStepToolInvocations.every(
     (part) => part.state === "output-available" && part.preliminary !== true || part.state === "output-error"
   );
 }