ai 6.0.293 → 6.0.294

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # ai
 
+## 6.0.294
+
+### Patch Changes
+
+- 16a04d7: fix(ai): cancel response streams when clients disconnect
+- Updated dependencies [29dc427]
+  - @ai-sdk/provider-utils@4.0.56
+  - @ai-sdk/gateway@3.0.204
+
 ## 6.0.293
 
 ### Patch Changes

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.293" : "0.0.0-test";
+var VERSION = true ? "6.0.294" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5648,21 +5648,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((resolve3) => {
+            onDisconnect = resolve3;
             response.once("drain", resolve3);
           });
+          onDisconnect = void 0;
         }
       }
     } finally {
-      response.end();
+      response.off("close", handleClose);
+      if (clientDisconnected || response.destroyed) {
+        reader.cancel().catch(() => {
+        });
+      } else {
+        response.end();
+      }
     }
   };
   return read();