@convex-dev/agent 0.2.2 → 0.2.3-alpha.0

package/src/react/toUIMessages.test.ts

@@ -167,14 +167,14 @@ describe("toUIMessages", () => {
     expect(uiMessages[0].text).toBe("Here's one idea. Here's another idea.");
     expect(uiMessages[0].parts.filter((p) => p.type === "reasoning")).toEqual([
       {
-        providerMetadata: undefined,
-        state: undefined,
+        providerOptions: undefined,
+        state: "done",
         text: "I'm thinking...",
         type: "reasoning",
       },
       {
-        providerMetadata: undefined,
-        state: undefined,
+        providerOptions: undefined,
+        state: "done",
         text: "I'm thinking...",
         type: "reasoning",
       },
@@ -356,6 +356,50 @@ describe("toUIMessages", () => {
     expect(textParts[0].text).toBe("Hello!");
   });
 
+  it("sets text field correctly when message has many parts with text part as final part", () => {
+    const messages = [
+      baseMessageDoc({
+        message: {
+          role: "assistant",
+          content: [
+            {
+              type: "reasoning",
+              text: "Let me think about this...",
+            },
+            {
+              type: "tool-call",
+              toolName: "calculator",
+              toolCallId: "call1",
+              args: { operation: "add", a: 2, b: 3 },
+            },
+            {
+              type: "file",
+              mimeType: "application/json",
+              data: "some-file-data",
+            },
+            {
+              type: "text",
+              text: "Here's my final answer.",
+            },
+          ],
+        },
+        text: "Here's my final answer.",
+        reasoning: "Let me think about this...",
+      }),
+    ];
+
+    const uiMessages = toUIMessages(messages);
+
+    expect(uiMessages).toHaveLength(1);
+    expect(uiMessages[0].role).toBe("assistant");
+    expect(uiMessages[0].text).toBe("Here's my final answer.");
+
+    // Verify that the text part exists in parts
+    const textParts = uiMessages[0].parts.filter((p) => p.type === "text");
+    expect(textParts).toHaveLength(1);
+    expect(textParts[0].text).toBe("Here's my final answer.");
+  });
+
   // Add more tests for array content, tool calls, etc. as needed
 
   it("should update tool call state from input-available to output-available", () => {

package/src/validators.ts

@@ -63,7 +63,6 @@ export const vReasoningPart = v.object({
   text: v.string(),
   signature: v.optional(v.string()),
   providerOptions,
-  state: v.optional(v.union(v.literal("streaming"), v.literal("done"))),
 });
 
 export const vRedactedReasoningPart = v.object({
@@ -89,23 +88,10 @@ export const vToolCallPart = v.object({
   toolCallId: v.string(),
   toolName: v.string(),
   args: v.any(),
-  providerOptions,
   providerExecuted: v.optional(v.boolean()),
+  providerOptions,
 });
 
-export const vAssistantContent = v.union(
-  v.string(),
-  v.array(
-    v.union(
-      vTextPart,
-      vFilePart,
-      vReasoningPart,
-      vRedactedReasoningPart,
-      vToolCallPart,
-    ),
-  ),
-);
-
 const vToolResultContent = v.array(
   v.union(
     v.object({ type: v.literal("text"), text: v.string() }),
@@ -134,6 +120,20 @@ export const vToolResultPart = v.object({
 });
 export const vToolContent = v.array(vToolResultPart);
 
+export const vAssistantContent = v.union(
+  v.string(),
+  v.array(
+    v.union(
+      vTextPart,
+      vFilePart,
+      vReasoningPart,
+      vRedactedReasoningPart,
+      vToolCallPart,
+      vToolResultPart,
+    ),
+  ),
+);
+
 export const vContent = v.union(vUserContent, vAssistantContent, vToolContent);
 export type Content = Infer<typeof vContent>;