ai 3.1.24 → 3.1.26

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "3.1.24",
+  "version": "3.1.26",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",

package/rsc/dist/rsc-server.mjs

@@ -1640,7 +1640,7 @@ async function streamUI({
   const ui = createStreamableUI(initial);
   const textRender = text || defaultTextRenderer;
   let finished;
-  async function handleRender(args, renderer, res) {
+  async function handleRender(args, renderer, res, lastCall = false) {
     if (!renderer)
       return;
     const resolvable = createResolvablePromise();
@@ -1652,13 +1652,21 @@ async function streamUI({
     const value = renderer(...args);
     if (value instanceof Promise || value && typeof value === "object" && "then" in value && typeof value.then === "function") {
       const node = await value;
-      res.update(node);
+      if (lastCall) {
+        res.done(node);
+      } else {
+        res.update(node);
+      }
       resolvable.resolve(void 0);
     } else if (value && typeof value === "object" && Symbol.asyncIterator in value) {
       const it = value;
       while (true) {
         const { done, value: value2 } = await it.next();
-        res.update(value2);
+        if (lastCall && done) {
+          res.done(value2);
+        } else {
+          res.update(value2);
+        }
         if (done)
           break;
       }
@@ -1667,13 +1675,21 @@ async function streamUI({
       const it = value;
       while (true) {
         const { done, value: value2 } = it.next();
-        res.update(value2);
+        if (lastCall && done) {
+          res.done(value2);
+        } else {
+          res.update(value2);
+        }
         if (done)
           break;
       }
       resolvable.resolve(void 0);
     } else {
-      res.update(value);
+      if (lastCall) {
+        res.done(value);
+      } else {
+        res.update(value);
+      }
       resolvable.resolve(void 0);
     }
   }
@@ -1727,6 +1743,7 @@ async function streamUI({
                 availableTools: Object.keys(tools)
               });
             }
+            hasToolCall = true;
             const parseResult = safeParseJSON({
               text: value.args,
               schema: tool.parameters
@@ -1747,7 +1764,8 @@ async function streamUI({
                 }
               ],
               tool.generate,
-              ui
+              ui,
+              true
             );
             break;
           }
@@ -1760,11 +1778,9 @@ async function streamUI({
       }
       if (hasToolCall) {
         await finished;
-        ui.done();
       } else {
-        handleRender([{ content, done: true }], textRender, ui);
+        handleRender([{ content, done: true }], textRender, ui, true);
         await finished;
-        ui.done();
       }
     } catch (error) {
       ui.error(error);