ai 7.0.118 → 7.0.120

package/CHANGELOG.md

@@ -1,5 +1,37 @@
 # ai
 
+## 7.0.120
+
+### Patch Changes
+
+- b032d70: fix(ai): preserve tool metadata from tool output chunks
+- e21b98b: fix(ai): continue active UI message parts when resuming after a disconnect
+- Updated dependencies [e361d39]
+  - @ai-sdk/gateway@4.0.98
+
+## 7.0.119
+
+### Patch Changes
+
+- 33e94ba: fix(ai): cancel response streams when clients disconnect
+- 34d869e: fix(ai): prevent unhandled stream completion rejections in non-Node runtimes
+- def4df8: fix(ai): store static tool input errors in the current input field
+- 525efc5: feat(provider): advertise image model file and mask input support
+
+  Use confirmed model IDs for capability declarations so unrecognized model names
+  remain unknown. Include Together AI FLUX.2 Pro and Flex single-image editing, and
+  allow asynchronous capability lookups and middleware overrides to resolve to
+  unknown.
+
+  Advertise QuiverAI Arrow 2 and Arrow 2 Telos file-input support, and mark Together
+  AI Gemini image inputs unsupported by the current single-image request mapping.
+
+- Updated dependencies [e3605f6]
+- Updated dependencies [525efc5]
+  - @ai-sdk/provider-utils@5.0.50
+  - @ai-sdk/provider@4.0.19
+  - @ai-sdk/gateway@4.0.97
+
 ## 7.0.118
 
 ### Patch Changes

package/dist/index.d.ts

@@ -5816,6 +5816,7 @@ declare abstract class AbstractChat<UI_MESSAGE extends UIMessage> {
     private pendingApprovalMessageId;
     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

@@ -1328,7 +1328,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "7.0.118" : "0.0.0-test";
+var VERSION = true ? "7.0.120" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5056,6 +5056,8 @@ function openTelemetryChannelSpanContext({
   completion
 }) {
   if (!isNodeRuntime()) {
+    Promise.resolve(completion).catch(() => {
+    });
     return void 0;
   }
   const diagnosticsChannel = loadBuiltinModule(
@@ -7053,11 +7055,26 @@ function writeToServerResponse({
     response.writeHead(statusCode);
   }
   const reader = stream.getReader();
+  let clientDisconnected = false;
+  let onDisconnect;
+  const handleClose = () => {
+    if (response.writableFinished) {
+      return;
+    }
+    clientDisconnected = true;
+    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) break;
+        if (done || clientDisconnected || response.destroyed) break;
         const canContinue = response.write(value);
         const flush = response.flush;
         if (typeof flush === "function") {
@@ -7065,12 +7082,20 @@ function writeToServerResponse({
         }
         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();
@@ -7446,25 +7471,6 @@ function getToolName(part) {
 }
 var getToolOrDynamicToolName = getToolName;
 
-// src/ui/warn-if-ui-message-has-deprecated-raw-input.ts
-var rawInputDeprecationWarning = {
-  type: "deprecated",
-  setting: "rawInput in output-error UI message parts",
-  message: 'Use the "input" field instead. The "rawInput" field will be removed in the next major version.'
-};
-function warnIfUIMessageHasDeprecatedRawInput(messages) {
-  const hasDeprecatedRawInput = messages.some(
-    (message) => message.parts.some(
-      (part) => part != null && typeof part === "object" && "type" in part && typeof part.type === "string" && (part.type === "dynamic-tool" || part.type.startsWith("tool-")) && "state" in part && part.state === "output-error" && Object.prototype.hasOwnProperty.call(part, "rawInput") && "rawInput" in part && part.rawInput !== void 0
-    )
-  );
-  if (hasDeprecatedRawInput) {
-    logWarnings({
-      warnings: [rawInputDeprecationWarning]
-    });
-  }
-}
-
 // src/ui/process-ui-message-stream.ts
 function createStreamingUIMessageState({
   lastMessage,
@@ -7905,14 +7911,12 @@ function processUIMessageStream({
                   toolCallId: chunk.toolCallId,
                   toolName: chunk.toolName,
                   state: "output-error",
-                  input: void 0,
-                  rawInput: chunk.input,
+                  input: chunk.input,
                   errorText: chunk.errorText,
                   providerExecuted: chunk.providerExecuted,
                   providerMetadata: chunk.providerMetadata,
                   toolMetadata: chunk.toolMetadata
                 });
-                warnIfUIMessageHasDeprecatedRawInput([state.message]);
               }
               write();
               break;
@@ -7975,7 +7979,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -7991,7 +7995,7 @@ function processUIMessageStream({
                     preliminary: chunk.preliminary,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -8012,7 +8016,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -8028,7 +8032,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: chunk.toolMetadata ?? toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -12219,6 +12223,27 @@ function readUIMessageStream({
 import {
   isNonNullable
 } from "@ai-sdk/provider-utils";
+
+// src/ui/warn-if-ui-message-has-deprecated-raw-input.ts
+var rawInputDeprecationWarning = {
+  type: "deprecated",
+  setting: "rawInput in output-error UI message parts",
+  message: 'Use the "input" field instead. The "rawInput" field will be removed in the next major version.'
+};
+function warnIfUIMessageHasDeprecatedRawInput(messages) {
+  const hasDeprecatedRawInput = messages.some(
+    (message) => message.parts.some(
+      (part) => part != null && typeof part === "object" && "type" in part && typeof part.type === "string" && (part.type === "dynamic-tool" || part.type.startsWith("tool-")) && "state" in part && part.state === "output-error" && Object.prototype.hasOwnProperty.call(part, "rawInput") && "rawInput" in part && part.rawInput !== void 0
+    )
+  );
+  if (hasDeprecatedRawInput) {
+    logWarnings({
+      warnings: [rawInputDeprecationWarning]
+    });
+  }
+}
+
+// src/ui/convert-to-model-messages.ts
 async function convertToModelMessages(messages, options) {
   const modelMessages = [];
   warnIfUIMessageHasDeprecatedRawInput(messages);
@@ -17874,7 +17899,9 @@ var doWrap3 = ({
     wrapGenerate,
     overrideProvider,
     overrideModelId,
-    overrideMaxImagesPerCall
+    overrideMaxImagesPerCall,
+    overrideSupportsFileInputs,
+    overrideSupportsMaskInputs
   },
   modelId,
   providerId
@@ -17884,11 +17911,15 @@ var doWrap3 = ({
   }
   const maxImagesPerCallRaw = overrideMaxImagesPerCall?.({ model }) ?? model.maxImagesPerCall;
   const maxImagesPerCall = maxImagesPerCallRaw instanceof Function ? maxImagesPerCallRaw.bind(model) : maxImagesPerCallRaw;
+  const supportsFileInputs = overrideSupportsFileInputs !== void 0 ? overrideSupportsFileInputs({ model }) : model.supportsFileInputs;
+  const supportsMaskInputs = overrideSupportsMaskInputs !== void 0 ? overrideSupportsMaskInputs({ model }) : model.supportsMaskInputs;
   return {
     specificationVersion: "v4",
     provider: providerId ?? overrideProvider?.({ model }) ?? model.provider,
     modelId: modelId ?? overrideModelId?.({ model }) ?? model.modelId,
     maxImagesPerCall,
+    supportsFileInputs,
+    supportsMaskInputs,
     async doGenerate(params) {
       const transformedParams = await doTransform({ params });
       const doGenerate = async () => await model.doGenerate(transformedParams);
@@ -21976,6 +22007,7 @@ var AbstractChat = class {
     this.pendingMessagePreparations = /* @__PURE__ */ new Set();
     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
@@ -22270,6 +22302,9 @@ var AbstractChat = class {
     body,
     messageId
   }) {
+    if (trigger !== "resume-stream") {
+      this.resumableStreamState = void 0;
+    }
     const abortController = new AbortController();
     const activeResumeRequest = trigger === "resume-stream" ? { abortController } : void 0;
     if (activeResumeRequest) {
@@ -22297,12 +22332,14 @@ var AbstractChat = class {
           });
           if (isCurrentRequest()) {
             this.setStatus({ status: "ready" });
+            this.resumableStreamState = void 0;
           }
           clearActiveResumeRequest();
           return;
         }
         if (reconnect == null) {
           this.setStatus({ status: "ready" });
+          this.resumableStreamState = void 0;
           clearActiveResumeRequest();
           return;
         }
@@ -22311,6 +22348,7 @@ var AbstractChat = class {
         if (abortController.signal.aborted || err.name === "AbortError") {
           if (isCurrentRequest()) {
             this.setStatus({ status: "ready" });
+            this.resumableStreamState = void 0;
           }
           clearActiveResumeRequest();
           return;
@@ -22337,7 +22375,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" || trigger === "regenerate-message" ? void 0 : this.state.snapshot(responseMessage),
           messageId: this.generateId()
         }),
@@ -22415,17 +22453,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?.state) {
+            this.resumableStreamState = void 0;
+          }
         }
         return null;
       }
@@ -22436,6 +22483,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?.state;
+      } else if (this.resumableStreamState === activeResponse?.state) {
+        this.resumableStreamState = void 0;
+      }
       if (this.onError && err instanceof Error) {
         this.onError(err);
       }