ai 5.0.269 → 5.0.271

package/dist/index.mjs

@@ -727,7 +727,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.269" : "0.0.0-test";
+var VERSION = true ? "5.0.271" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -2999,6 +2999,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() {
@@ -3031,9 +3100,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;
@@ -3631,16 +3704,41 @@ function createStreamingUIMessageState({
   lastMessage,
   messageId
 }) {
+  var _a16;
+  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 (!isToolOrDynamicToolUIPart(part)) {
+      continue;
+    }
+    const index = staticToolIndex;
+    if (isToolUIPart(part)) {
+      staticToolIndex++;
+    }
+    if (part.state !== "input-streaming") {
+      continue;
+    }
+    partialToolCalls[part.toolCallId] = {
+      text: (_a16 = part.rawInput) != null ? _a16 : "",
+      index,
+      toolName: isDynamicToolUIPart(part) ? part.toolName : getToolName(part),
+      dynamic: isDynamicToolUIPart(part)
+    };
+  }
   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({
@@ -3748,7 +3846,7 @@ function processUIMessageStream({
             }
           }
           function updateDynamicToolPart(options, existingPart) {
-            var _a17, _b2;
+            var _a17;
             const part = existingPart != null ? existingPart : getCurrentStepParts().find(
               (part2) => part2.type === "dynamic-tool" && part2.toolCallId === options.toolCallId
             );
@@ -3760,9 +3858,9 @@ function processUIMessageStream({
               anyPart.input = anyOptions.input;
               anyPart.output = anyOptions.output;
               anyPart.errorText = anyOptions.errorText;
-              anyPart.rawInput = (_a17 = anyOptions.rawInput) != null ? _a17 : anyPart.rawInput;
+              anyPart.rawInput = anyOptions.rawInput;
               anyPart.preliminary = anyOptions.preliminary;
-              anyPart.providerExecuted = (_b2 = anyOptions.providerExecuted) != null ? _b2 : part.providerExecuted;
+              anyPart.providerExecuted = (_a17 = anyOptions.providerExecuted) != null ? _a17 : part.providerExecuted;
               if (anyOptions.providerMetadata != null && part.state === "input-available") {
                 part.callProviderMetadata = anyOptions.providerMetadata;
               }
@@ -3920,14 +4018,16 @@ function processUIMessageStream({
                   toolCallId: chunk.toolCallId,
                   toolName: partialToolCall.toolName,
                   state: "input-streaming",
-                  input: partialArgs
+                  input: partialArgs,
+                  rawInput: partialToolCall.text
                 });
               } else {
                 updateToolPart({
                   toolCallId: chunk.toolCallId,
                   toolName: partialToolCall.toolName,
                   state: "input-streaming",
-                  input: partialArgs
+                  input: partialArgs,
+                  rawInput: partialToolCall.text
                 });
               }
               write();
@@ -4241,9 +4341,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;
@@ -6573,6 +6677,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
 import {
   withUserAgentSuffix as withUserAgentSuffix4
@@ -6800,10 +6919,10 @@ async function embedMany({
               for (const [providerName, metadata] of Object.entries(
                 result.providerMetadata
               )) {
-                providerMetadata[providerName] = {
-                  ...providerMetadata[providerName],
+                setOwn(providerMetadata, providerName, {
+                  ...getOwn(providerMetadata, providerName),
                   ...metadata
-                };
+                });
               }
             }
           }
@@ -6983,10 +7102,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);
         }
       }
     }
@@ -9263,7 +9385,7 @@ var DefaultProviderRegistry = class {
     separator,
     languageModelMiddleware
   }) {
-    this.providers = {};
+    this.providers = /* @__PURE__ */ Object.create(null);
     this.separator = separator;
     this.languageModelMiddleware = languageModelMiddleware;
   }
@@ -9629,6 +9751,10 @@ import {
   normalizeHeaders,
   resolve
 } from "@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",
@@ -9703,7 +9829,7 @@ var HttpChatTransport = class {
     return this.processResponseStream(response.body);
   }
   async reconnectToStream(options) {
-    var _a16, _b, _c, _d, _e;
+    var _a16, _b, _c, _d;
     const resolvedBody = await resolve(this.body);
     const resolvedHeaders = await resolve(this.headers);
     const resolvedCredentials = await resolve(this.credentials);
@@ -9719,10 +9845,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 ? 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,
@@ -9734,7 +9873,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) {
@@ -9996,6 +10135,9 @@ var AbstractChat = class {
     };
     this.setStatus({ status: "submitted", error: void 0 });
     const lastMessage = this.lastMessage;
+    const resumableResponseMessage = trigger === "resume-stream" && (lastMessage == null ? void 0 : lastMessage.role) === "assistant" && lastMessage.parts.some(
+      (part) => isToolOrDynamicToolUIPart(part) && part.state === "input-streaming"
+    ) ? this.state.snapshot(lastMessage) : void 0;
     let isAbort = false;
     let isDisconnect = false;
     let isError = false;
@@ -10003,7 +10145,7 @@ var AbstractChat = class {
     try {
       const response = {
         state: trigger === "resume-stream" && this.resumableStreamState != null ? this.resumableStreamState : createStreamingUIMessageState({
-          lastMessage: trigger === "resume-stream" ? void 0 : this.state.snapshot(lastMessage),
+          lastMessage: trigger === "resume-stream" ? resumableResponseMessage : this.state.snapshot(lastMessage),
           messageId: this.generateId()
         }),
         abortController
@@ -10184,8 +10326,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(isToolOrDynamicToolUIPart).filter((part) => !part.providerExecuted);
-  return lastStepToolInvocations.length > 0 && lastStepToolInvocations.every(
+  const lastStepParts = message.parts.slice(lastStepStartIndex + 1);
+  const lastStepToolInvocations = lastStepParts.filter(isToolOrDynamicToolUIPart).filter((part) => !part.providerExecuted);
+  const lastToolInvocationIndex = lastStepParts.reduce(
+    (lastIndex, part, index) => isToolOrDynamicToolUIPart(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.state === "output-error"
   );
 }
@@ -10319,6 +10466,7 @@ var uiMessagesSchema = lazyValidator2(
               toolCallId: z8.string(),
               state: z8.literal("input-streaming"),
               input: z8.unknown().optional(),
+              rawInput: z8.string().optional(),
               providerExecuted: z8.boolean().optional(),
               output: z8.never().optional(),
               errorText: z8.never().optional()
@@ -10364,6 +10512,7 @@ var uiMessagesSchema = lazyValidator2(
               state: z8.literal("input-streaming"),
               providerExecuted: z8.boolean().optional(),
               input: z8.unknown().optional(),
+              rawInput: z8.string().optional(),
               output: z8.never().optional(),
               errorText: z8.never().optional(),
               approval: z8.never().optional()