ai 4.1.31 → 4.1.32

package/dist/index.mjs

@@ -4714,13 +4714,23 @@ function createOutputTransformStream(output) {
   let text2 = "";
   let textChunk = "";
   let lastPublishedJson = "";
+  function publishTextChunk({
+    controller,
+    partialOutput = void 0
+  }) {
+    controller.enqueue({
+      part: { type: "text-delta", textDelta: textChunk },
+      partialOutput
+    });
+    textChunk = "";
+  }
   return new TransformStream({
     transform(chunk, controller) {
+      if (chunk.type === "step-finish") {
+        publishTextChunk({ controller });
+      }
       if (chunk.type !== "text-delta") {
-        controller.enqueue({
-          part: chunk,
-          partialOutput: void 0
-        });
+        controller.enqueue({ part: chunk, partialOutput: void 0 });
         return;
       }
       text2 += chunk.textDelta;
@@ -4729,27 +4739,14 @@ function createOutputTransformStream(output) {
       if (result != null) {
         const currentJson = JSON.stringify(result.partial);
         if (currentJson !== lastPublishedJson) {
-          controller.enqueue({
-            part: {
-              type: "text-delta",
-              textDelta: textChunk
-            },
-            partialOutput: result.partial
-          });
+          publishTextChunk({ controller, partialOutput: result.partial });
           lastPublishedJson = currentJson;
-          textChunk = "";
         }
       }
     },
     flush(controller) {
       if (textChunk.length > 0) {
-        controller.enqueue({
-          part: {
-            type: "text-delta",
-            textDelta: textChunk
-          },
-          partialOutput: void 0
-        });
+        publishTextChunk({ controller });
       }
     }
   });