@convex-dev/agent 0.1.14-alpha.5 → 0.1.15-alpha.0

package/src/react/toUIMessages.test.ts

@@ -1,6 +1,7 @@
 import { describe, it, expect } from "vitest";
 import { toUIMessages } from "./toUIMessages.js";
 import type { MessageDoc } from "../client/index.js";
+import { assert } from "convex-helpers";
 
 // Helper to create a base message doc
 function baseMessageDoc(overrides: Partial<MessageDoc> = {}): MessageDoc {
@@ -170,6 +171,12 @@ describe("toUIMessages", () => {
     expect(
       uiMessages[0].parts.filter((p) => p.type === "reasoning")
     ).toHaveLength(1);
+    expect(uiMessages[0].parts[0].type).toBe("reasoning");
+    assert(uiMessages[0].parts[0].type === "reasoning");
+    expect(uiMessages[0].parts[0].reasoning).toBe(
+      "I'm thinking...I'm thinking..."
+    );
+
     expect(uiMessages[0].parts.filter((p) => p.type === "text")).toHaveLength(
       1
     );

package/src/react/toUIMessages.ts

@@ -80,13 +80,6 @@ export function toUIMessages(
       }
       // update it to the last message's id
       assistantMessage.id = message.id ?? message._id;
-      if (message.text) {
-        assistantMessage.parts.push({
-          type: "text",
-          text: message.text,
-        });
-        assistantMessage.content += message.text;
-      }
       if (message.reasoning) {
         assistantMessage.parts.push({
           type: "reasoning",
@@ -94,6 +87,13 @@ export function toUIMessages(
           details: message.reasoningDetails ?? [],
         });
       }
+      if (message.text) {
+        assistantMessage.parts.push({
+          type: "text",
+          text: message.text,
+        });
+        assistantMessage.content += message.text;
+      }
       for (const source of message.sources ?? []) {
         assistantMessage.parts.push({
           type: "source",

package/src/validators.ts

@@ -460,6 +460,7 @@ export const vStreamArgs = v.optional(
   v.union(
     v.object({
       kind: v.literal("list"),
+      startOrder: v.optional(v.number()),
     }),
     v.object({
       kind: v.literal("deltas"),
@@ -471,6 +472,11 @@ export type StreamArgs = Infer<typeof vStreamArgs>;
 
 export const vStreamMessage = v.object({
   streamId: v.string(),
+  status: v.union(
+    v.literal("streaming"),
+    v.literal("finished"),
+    v.literal("aborted")
+  ),
   order: v.number(),
   stepOrder: v.number(),
   // metadata