ai 5.0.267 → 5.0.269

package/CHANGELOG.md

@@ -1,5 +1,22 @@
 # ai
 
+## 5.0.269
+
+### Patch Changes
+
+- 11f5cda: Preserve provider metadata on corresponding `smoothStream` chunks without carrying it into subsequent metadata-free deltas.
+
+## 5.0.268
+
+### Patch Changes
+
+- dfffb7e: fix(ai): cancel response streams when clients disconnect
+- 1d0096a: fix(ai): continue active UI message parts when resuming after a disconnect
+- Updated dependencies [6dc04c5]
+- Updated dependencies [14810cb]
+  - @ai-sdk/gateway@2.0.160
+  - @ai-sdk/provider-utils@3.0.40
+
 ## 5.0.267
 
 ### Patch Changes

package/dist/index.d.mts

@@ -4158,6 +4158,7 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
     private sendAutomaticallyWhen?;
     private activeResponse;
     private activeResumeRequest;
+    private resumableStreamState;
     private jobExecutor;
     constructor({ generateId, id, transport, messageMetadataSchema, dataPartSchemas, state, onError, onToolCall, onFinish, onData, sendAutomaticallyWhen, }: Omit<ChatInit<UI_MESSAGE>, 'messages'> & {
         state: ChatState<UI_MESSAGE>;

package/dist/index.d.ts

@@ -4158,6 +4158,7 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
     private sendAutomaticallyWhen?;
     private activeResponse;
     private activeResumeRequest;
+    private resumableStreamState;
     private jobExecutor;
     constructor({ generateId, id, transport, messageMetadataSchema, dataPartSchemas, state, onError, onToolCall, onFinish, onData, sendAutomaticallyWhen, }: Omit<ChatInit<UI_MESSAGE>, 'messages'> & {
         state: ChatState<UI_MESSAGE>;

package/dist/index.js

@@ -816,7 +816,7 @@ function detectMediaType({
 var import_provider_utils2 = require("@ai-sdk/provider-utils");
 
 // src/version.ts
-var VERSION = true ? "5.0.267" : "0.0.0-test";
+var VERSION = true ? "5.0.269" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -3006,21 +3006,44 @@ function writeToServerResponse({
     response.writeHead(statusCode, headers);
   }
   const reader = stream.getReader();
+  let clientDisconnected = false;
+  let onDisconnect;
+  const handleClose = () => {
+    if (response.writableFinished) {
+      return;
+    }
+    clientDisconnected = true;
+    onDisconnect == null ? void 0 : onDisconnect();
+    reader.cancel(new Error("Client disconnected.")).catch(() => {
+    });
+  };
+  response.once("close", handleClose);
+  if (response.destroyed) {
+    handleClose();
+  }
   const read = async () => {
     try {
       while (true) {
         const { done, value } = await reader.read();
-        if (done)
+        if (done || clientDisconnected || response.destroyed)
           break;
         const canContinue = response.write(value);
         if (!canContinue) {
           await new Promise((resolve2) => {
+            onDisconnect = resolve2;
             response.once("drain", resolve2);
           });
+          onDisconnect = void 0;
         }
       }
     } finally {
-      response.end();
+      response.off("close", handleClose);
+      if (clientDisconnected || response.destroyed) {
+        reader.cancel().catch(() => {
+        });
+      } else {
+        response.end();
+      }
     }
   };
   return read();
@@ -8787,28 +8810,51 @@ function smoothStream({
   return () => {
     let buffer = "";
     let id = "";
+    let providerMetadata;
+    function flushBuffer(controller) {
+      if (buffer.length > 0 || providerMetadata != null) {
+        controller.enqueue({
+          type: "text-delta",
+          text: buffer,
+          id,
+          ...providerMetadata != null ? { providerMetadata } : {}
+        });
+        buffer = "";
+        providerMetadata = void 0;
+      }
+    }
     return new TransformStream({
       async transform(chunk, controller) {
         if (chunk.type !== "text-delta") {
-          if (buffer.length > 0) {
-            controller.enqueue({ type: "text-delta", text: buffer, id });
-            buffer = "";
-          }
+          flushBuffer(controller);
           controller.enqueue(chunk);
           return;
         }
-        if (chunk.id !== id && buffer.length > 0) {
-          controller.enqueue({ type: "text-delta", text: buffer, id });
-          buffer = "";
+        if (chunk.text.length === 0 && chunk.providerMetadata != null) {
+          flushBuffer(controller);
+          controller.enqueue(chunk);
+          return;
+        }
+        if (buffer.length > 0 && (chunk.id !== id || providerMetadata != null || chunk.providerMetadata != null)) {
+          flushBuffer(controller);
         }
         buffer += chunk.text;
         id = chunk.id;
+        providerMetadata = chunk.providerMetadata;
         let match;
         while ((match = detectChunk(buffer)) != null) {
-          controller.enqueue({ type: "text-delta", text: match, id });
+          controller.enqueue({
+            type: "text-delta",
+            text: match,
+            id,
+            ...providerMetadata != null ? { providerMetadata } : {}
+          });
           buffer = buffer.slice(match.length);
           await delay2(isDocumentHidden() ? null : delayInMs);
         }
+        if (buffer.length === 0) {
+          providerMetadata = void 0;
+        }
       }
     });
   };
@@ -9761,6 +9807,7 @@ var AbstractChat = class {
   }) {
     this.activeResponse = void 0;
     this.activeResumeRequest = void 0;
+    this.resumableStreamState = void 0;
     this.jobExecutor = new SerialJobExecutor();
     /**
      * Appends or replaces a user message to the chat list. This triggers the API call to fetch
@@ -9956,6 +10003,9 @@ var AbstractChat = class {
     messageId
   }) {
     var _a16, _b, _c, _d;
+    if (trigger !== "resume-stream") {
+      this.resumableStreamState = void 0;
+    }
     const abortController = new AbortController();
     const activeResumeRequest = trigger === "resume-stream" ? { abortController } : void 0;
     if (activeResumeRequest) {
@@ -9976,7 +10026,7 @@ var AbstractChat = class {
     let activeResponse;
     try {
       const response = {
-        state: createStreamingUIMessageState({
+        state: trigger === "resume-stream" && this.resumableStreamState != null ? this.resumableStreamState : createStreamingUIMessageState({
           lastMessage: trigger === "resume-stream" ? void 0 : this.state.snapshot(lastMessage),
           messageId: this.generateId()
         }),
@@ -10001,12 +10051,14 @@ var AbstractChat = class {
           }));
           if (isCurrentRequest()) {
             this.setStatus({ status: "ready" });
+            this.resumableStreamState = void 0;
           }
           return;
         }
         if (reconnect == null) {
           if (isCurrentRequest()) {
             this.setStatus({ status: "ready" });
+            this.resumableStreamState = void 0;
           }
           return;
         }
@@ -10072,17 +10124,26 @@ var AbstractChat = class {
       if (isAbort) {
         if (isCurrentRequest()) {
           this.setStatus({ status: "ready" });
+          if (this.resumableStreamState === response.state) {
+            this.resumableStreamState = void 0;
+          }
         }
         return null;
       }
       if (isCurrentRequest()) {
         this.setStatus({ status: "ready" });
+        if (this.resumableStreamState === response.state) {
+          this.resumableStreamState = void 0;
+        }
       }
     } catch (err) {
       if (isAbort || err.name === "AbortError") {
         isAbort = true;
         if (isCurrentRequest()) {
           this.setStatus({ status: "ready" });
+          if (this.resumableStreamState === (activeResponse == null ? void 0 : activeResponse.state)) {
+            this.resumableStreamState = void 0;
+          }
         }
         return null;
       }
@@ -10093,6 +10154,11 @@ var AbstractChat = class {
       if (err instanceof TypeError && (err.message.toLowerCase().includes("fetch") || err.message.toLowerCase().includes("network"))) {
         isDisconnect = true;
       }
+      if (isDisconnect) {
+        this.resumableStreamState = activeResponse == null ? void 0 : activeResponse.state;
+      } else if (this.resumableStreamState === (activeResponse == null ? void 0 : activeResponse.state)) {
+        this.resumableStreamState = void 0;
+      }
       if (this.onError && err instanceof Error) {
         this.onError(err);
       }