ai 6.0.198 → 6.0.200

package/dist/index.mjs

@@ -1144,7 +1144,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.198" : "0.0.0-test";
+var VERSION = true ? "6.0.200" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5324,6 +5324,11 @@ function isDataUIMessageChunk(chunk) {
   return chunk.type.startsWith("data-");
 }
 
+// src/util/create-id-map.ts
+function createIdMap() {
+  return /* @__PURE__ */ Object.create(null);
+}
+
 // src/ui/ui-messages.ts
 function isDataUIPart(part) {
   return part.type.startsWith("data-");
@@ -5367,9 +5372,9 @@ function createStreamingUIMessageState({
       role: "assistant",
       parts: []
     },
-    activeTextParts: {},
-    activeReasoningParts: {},
-    partialToolCalls: {}
+    activeTextParts: createIdMap(),
+    activeReasoningParts: createIdMap(),
+    partialToolCalls: createIdMap()
   };
 }
 function processUIMessageStream({
@@ -5843,8 +5848,8 @@ function processUIMessageStream({
               break;
             }
             case "finish-step": {
-              state.activeTextParts = {};
-              state.activeReasoningParts = {};
+              state.activeTextParts = createIdMap();
+              state.activeReasoningParts = createIdMap();
               break;
             }
             case "start": {
@@ -6485,6 +6490,28 @@ var originalGenerateId2 = createIdGenerator2({
   prefix: "aitxt",
   size: 24
 });
+var isOutputChunkType = {
+  file: true,
+  source: true,
+  "text-start": true,
+  "text-end": true,
+  "text-delta": true,
+  "reasoning-start": true,
+  "reasoning-end": true,
+  "reasoning-delta": true,
+  "tool-input-start": true,
+  "tool-input-end": true,
+  "tool-input-delta": true,
+  "tool-approval-request": true,
+  "tool-call": true,
+  "tool-result": true,
+  "tool-error": true,
+  "stream-start": false,
+  "response-metadata": false,
+  finish: false,
+  error: false,
+  raw: false
+};
 function streamText({
   model,
   tools,
@@ -6703,6 +6730,7 @@ var DefaultStreamTextResult = class {
     let recordedRequest = {};
     let recordedWarnings = [];
     const recordedSteps = [];
+    let recordedNoOutputError;
     const pendingDeferredToolCalls = /* @__PURE__ */ new Map();
     let rootSpan;
     let activeTextContent = {};
@@ -6716,7 +6744,11 @@ var DefaultStreamTextResult = class {
           await (onChunk == null ? void 0 : onChunk({ chunk: part }));
         }
         if (part.type === "error") {
-          await onError({ error: wrapGatewayError(part.error) });
+          const error = wrapGatewayError(part.error);
+          if (NoOutputGeneratedError.isInstance(error)) {
+            recordedNoOutputError = error;
+          }
+          await onError({ error });
         }
         if (part.type === "text-start") {
           activeTextContent[part.id] = {
@@ -6867,8 +6899,8 @@ var DefaultStreamTextResult = class {
       async flush(controller) {
         var _a21, _b, _c, _d, _e, _f, _g;
         try {
-          if (recordedSteps.length === 0) {
-            const error = (abortSignal == null ? void 0 : abortSignal.aborted) ? abortSignal.reason : new NoOutputGeneratedError({
+          if (recordedSteps.length === 0 || recordedNoOutputError != null) {
+            const error = (abortSignal == null ? void 0 : abortSignal.aborted) ? abortSignal.reason : recordedNoOutputError != null ? recordedNoOutputError : new NoOutputGeneratedError({
               message: "No output generated. Check the stream for errors."
             });
             self._finishReason.reject(error);
@@ -7367,6 +7399,8 @@ var DefaultStreamTextResult = class {
             const activeToolCallToolNames = {};
             let stepFinishReason = "other";
             let stepRawFinishReason = void 0;
+            let hasReceivedTerminalChunk = false;
+            let hasReceivedOutputChunk = false;
             let stepUsage = createNullLanguageModelUsage();
             let stepProviderMetadata;
             let stepFirstChunk = true;
@@ -7402,6 +7436,9 @@ var DefaultStreamTextResult = class {
                       });
                     }
                     const chunkType = chunk.type;
+                    if (isOutputChunkType[chunkType]) {
+                      hasReceivedOutputChunk = true;
+                    }
                     switch (chunkType) {
                       case "tool-approval-request":
                       case "text-start":
@@ -7461,6 +7498,7 @@ var DefaultStreamTextResult = class {
                         break;
                       }
                       case "finish": {
+                        hasReceivedTerminalChunk = true;
                         stepUsage = chunk.usage;
                         stepFinishReason = chunk.finishReason;
                         stepRawFinishReason = chunk.rawFinishReason;
@@ -7520,6 +7558,7 @@ var DefaultStreamTextResult = class {
                         break;
                       }
                       case "error": {
+                        hasReceivedTerminalChunk = true;
                         controller.enqueue(chunk);
                         stepFinishReason = "error";
                         break;
@@ -7541,6 +7580,19 @@ var DefaultStreamTextResult = class {
                   // invoke onFinish callback and resolve toolResults promise when the stream is about to close:
                   async flush(controller) {
                     var _a22, _b2, _c2, _d2, _e2, _f2, _g2;
+                    if (!hasReceivedTerminalChunk && !hasReceivedOutputChunk) {
+                      controller.enqueue({
+                        type: "error",
+                        error: new NoOutputGeneratedError({
+                          message: "No output generated. The model stream ended without a finish chunk."
+                        })
+                      });
+                      doStreamSpan.end();
+                      clearStepTimeout();
+                      clearChunkTimeout();
+                      self.closeStream();
+                      return;
+                    }
                     const stepToolCallsJson = stepToolCalls.length > 0 ? JSON.stringify(stepToolCalls) : void 0;
                     try {
                       doStreamSpan.setAttributes(
@@ -7818,14 +7870,29 @@ var DefaultStreamTextResult = class {
       )
     );
   }
+  rejectResultPromises(error) {
+    if (this._finishReason.isPending())
+      this._finishReason.reject(error);
+    if (this._rawFinishReason.isPending())
+      this._rawFinishReason.reject(error);
+    if (this._totalUsage.isPending())
+      this._totalUsage.reject(error);
+    if (this._steps.isPending())
+      this._steps.reject(error);
+  }
   async consumeStream(options) {
     var _a21;
     try {
       await consumeStream({
         stream: this.fullStream,
-        onError: options == null ? void 0 : options.onError
+        onError: (error) => {
+          var _a22;
+          this.rejectResultPromises(error);
+          (_a22 = options == null ? void 0 : options.onError) == null ? void 0 : _a22.call(options, error);
+        }
       });
     } catch (error) {
+      this.rejectResultPromises(error);
       (_a21 = options == null ? void 0 : options.onError) == null ? void 0 : _a21.call(options, error);
     }
   }
@@ -8518,7 +8585,7 @@ async function convertToModelMessages(messages, options) {
         if (message.parts != null) {
           let block = [];
           async function processBlock() {
-            var _a21, _b, _c, _d, _e, _f, _g;
+            var _a21, _b, _c, _d, _e, _f, _g, _h;
             if (block.length === 0) {
               return;
             }
@@ -8626,7 +8693,7 @@ async function convertToModelMessages(messages, options) {
                         toolName: getToolName(toolPart),
                         output: {
                           type: "error-text",
-                          value: (_f = toolPart.approval.reason) != null ? _f : "Tool execution denied."
+                          value: (_g = (_f = toolPart.approval) == null ? void 0 : _f.reason) != null ? _g : "Tool execution denied."
                         },
                         ...toolPart.callProviderMetadata != null ? { providerOptions: toolPart.callProviderMetadata } : {}
                       });
@@ -8643,7 +8710,7 @@ async function convertToModelMessages(messages, options) {
                           toolCallId: toolPart.toolCallId,
                           input: toolPart.input,
                           output: toolPart.state === "output-error" ? toolPart.errorText : toolPart.output,
-                          tool: (_g = options == null ? void 0 : options.tools) == null ? void 0 : _g[toolName],
+                          tool: (_h = options == null ? void 0 : options.tools) == null ? void 0 : _h[toolName],
                           errorMode: toolPart.state === "output-error" ? "text" : "none"
                         }),
                         ...toolPart.callProviderMetadata != null ? { providerOptions: toolPart.callProviderMetadata } : {}