ai 4.1.12 → 4.1.14

package/CHANGELOG.md

@@ -1,5 +1,17 @@
 # ai
 
+## 4.1.14
+
+### Patch Changes
+
+- ca89615: fix (ai/core): only append assistant response at the end when there is a final user message
+
+## 4.1.13
+
+### Patch Changes
+
+- 999085e: feat (ai/core): add write function to DataStreamWriter
+
 ## 4.1.12
 
 ### Patch Changes

package/dist/index.d.mts

@@ -10,6 +10,10 @@ import { z } from 'zod';
 import { ServerResponse as ServerResponse$1 } from 'http';
 
 interface DataStreamWriter {
+    /**
+     * Appends a data part to the stream.
+     */
+    write(data: DataStreamString): void;
     /**
      * Appends a data part to the stream.
      */

package/dist/index.d.ts

@@ -10,6 +10,10 @@ import { z } from 'zod';
 import { ServerResponse as ServerResponse$1 } from 'http';
 
 interface DataStreamWriter {
+    /**
+     * Appends a data part to the stream.
+     */
+    write(data: DataStreamString): void;
     /**
      * Appends a data part to the stream.
      */

package/dist/index.js

@@ -115,6 +115,9 @@ function createDataStream({
   }
   try {
     const result = execute({
+      write(data) {
+        safeEnqueue(data);
+      },
       writeData(data) {
         safeEnqueue((0, import_ui_utils.formatDataStreamPart)("data", [data]));
       },
@@ -1593,7 +1596,9 @@ function convertToCoreMessages(messages, options) {
   var _a15;
   const tools = (_a15 = options == null ? void 0 : options.tools) != null ? _a15 : {};
   const coreMessages = [];
-  for (const message of messages) {
+  for (let i = 0; i < messages.length; i++) {
+    const message = messages[i];
+    const isLastMessage = i === messages.length - 1;
     const { role, content, toolInvocations, experimental_attachments } = message;
     switch (role) {
       case "system": {
@@ -1622,11 +1627,11 @@ function convertToCoreMessages(messages, options) {
           var _a16;
           return Math.max(max, (_a16 = toolInvocation.step) != null ? _a16 : 0);
         }, 0);
-        for (let i = 0; i <= maxStep; i++) {
+        for (let i2 = 0; i2 <= maxStep; i2++) {
           const stepInvocations = toolInvocations.filter(
             (toolInvocation) => {
               var _a16;
-              return ((_a16 = toolInvocation.step) != null ? _a16 : 0) === i;
+              return ((_a16 = toolInvocation.step) != null ? _a16 : 0) === i2;
             }
           );
           if (stepInvocations.length === 0) {
@@ -1635,6 +1640,7 @@ function convertToCoreMessages(messages, options) {
           coreMessages.push({
             role: "assistant",
             content: [
+              ...isLastMessage && content && i2 === 0 ? [{ type: "text", text: content }] : [],
               ...stepInvocations.map(
                 ({ toolCallId, toolName, args }) => ({
                   type: "tool-call",
@@ -1671,7 +1677,7 @@ function convertToCoreMessages(messages, options) {
             })
           });
         }
-        if (content) {
+        if (content && !isLastMessage) {
           coreMessages.push({ role: "assistant", content });
         }
         break;