ai 3.4.23 → 3.4.25

package/dist/index.mjs

@@ -3593,8 +3593,6 @@ function runToolsTransformation({
   telemetry,
   abortSignal
 }) {
-  let canClose = false;
-  const outstandingToolCalls = /* @__PURE__ */ new Set();
   let toolResultsStreamController = null;
   const toolResultsStream = new ReadableStream({
     start(controller) {
@@ -3602,6 +3600,17 @@ function runToolsTransformation({
     }
   });
   const activeToolCalls = {};
+  const outstandingToolResults = /* @__PURE__ */ new Set();
+  let canClose = false;
+  let finishChunk = void 0;
+  function attemptClose() {
+    if (canClose && outstandingToolResults.size === 0) {
+      if (finishChunk != null) {
+        toolResultsStreamController.enqueue(finishChunk);
+      }
+      toolResultsStreamController.close();
+    }
+  }
   const forwardStream = new TransformStream({
     transform(chunk, controller) {
       const chunkType = chunk.type;
@@ -3659,7 +3668,7 @@ function runToolsTransformation({
             controller.enqueue(toolCall);
             if (tool2.execute != null) {
               const toolExecutionId = generateId();
-              outstandingToolCalls.add(toolExecutionId);
+              outstandingToolResults.add(toolExecutionId);
               recordSpan({
                 name: "ai.toolCall",
                 attributes: selectTelemetryAttributes({
@@ -3684,10 +3693,8 @@ function runToolsTransformation({
                       type: "tool-result",
                       result
                     });
-                    outstandingToolCalls.delete(toolExecutionId);
-                    if (canClose && outstandingToolCalls.size === 0) {
-                      toolResultsStreamController.close();
-                    }
+                    outstandingToolResults.delete(toolExecutionId);
+                    attemptClose();
                     try {
                       span.setAttributes(
                         selectTelemetryAttributes({
@@ -3707,10 +3714,8 @@ function runToolsTransformation({
                       type: "error",
                       error
                     });
-                    outstandingToolCalls.delete(toolExecutionId);
-                    if (canClose && outstandingToolCalls.size === 0) {
-                      toolResultsStreamController.close();
-                    }
+                    outstandingToolResults.delete(toolExecutionId);
+                    attemptClose();
                   }
                 )
               });
@@ -3724,13 +3729,13 @@ function runToolsTransformation({
           break;
         }
         case "finish": {
-          controller.enqueue({
+          finishChunk = {
             type: "finish",
             finishReason: chunk.finishReason,
             logprobs: chunk.logprobs,
             usage: calculateLanguageModelUsage(chunk.usage),
             experimental_providerMetadata: chunk.providerMetadata
-          });
+          };
           break;
         }
         default: {
@@ -3741,9 +3746,7 @@ function runToolsTransformation({
     },
     flush() {
       canClose = true;
-      if (outstandingToolCalls.size === 0) {
-        toolResultsStreamController.close();
-      }
+      attemptClose();
     }
   });
   return new ReadableStream({