ai 6.0.292 → 6.0.294

package/dist/index.mjs

@@ -1224,7 +1224,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.292" : "0.0.0-test";
+var VERSION = true ? "6.0.294" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -1696,7 +1696,7 @@ async function downloadAssets(messages, download2, supportedUrls) {
           if (contentPart.type === "image-url" || contentPart.type === "file-url") {
             downloadableFiles.push({
               data: new URL(contentPart.url),
-              mediaType: contentPart.type === "image-url" ? "image/*" : void 0
+              mediaType: contentPart.type === "image-url" ? "image/*" : contentPart.mediaType
             });
           }
         }
@@ -2139,6 +2139,7 @@ var outputSchema = z4.discriminatedUnion(
           z4.object({
             type: z4.literal("file-url"),
             url: z4.string(),
+            mediaType: z4.string().optional(),
             providerOptions: providerMetadataSchema.optional()
           }),
           z4.object({
@@ -5574,21 +5575,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();