@convex-dev/agent 0.2.2 → 0.2.3

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", () => {
@@ -417,4 +461,190 @@ describe("toUIMessages", () => {
     );
     expect(toolCallParts).toHaveLength(0);
   });
+  it("sets text field correctly when message has many parts with text part as final part", () => {
+    const messages = [
+      baseMessageDoc({
+        text: "what time is it in paris?",
+        message: {
+          role: "user",
+          content: "what time is it in paris?",
+        },
+      }),
+      baseMessageDoc({
+        tool: true,
+        finishReason: "tool-calls",
+        text: "",
+        stepOrder: 1,
+        message: {
+          role: "assistant",
+          content: [
+            {
+              type: "reasoning",
+              text: "**Finding the Time**\n\nI've pinpointed the core task: obtaining the current time in Paris. It involves using the `dateTime` tool. I've identified \"Europe/Paris\" as the necessary timezone identifier to provide to the tool. My next step is to test the tool.\n\n\n",
+            },
+            {
+              args: {
+                timezone: "Europe/Paris",
+              },
+              type: "tool-call",
+              toolName: "dateTime",
+              toolCallId: "tool_0_dateTime",
+            },
+          ],
+        },
+        reasoning:
+          "**Finding the Time**\n\nI've pinpointed the core task: obtaining the current time in Paris. It involves using the `dateTime` tool. I've identified \"Europe/Paris\" as the necessary timezone identifier to provide to the tool. My next step is to test the tool.\n\n\n",
+        reasoningDetails: [
+          {
+            text: "**Finding the Time**\n\nI've pinpointed the core task: obtaining the current time in Paris. It involves using the `dateTime` tool. I've identified \"Europe/Paris\" as the necessary timezone identifier to provide to the tool. My next step is to test the tool.\n\n\n",
+            type: "reasoning",
+          },
+        ],
+        warnings: [],
+      }),
+      baseMessageDoc({
+        tool: true,
+        stepOrder: 2,
+        message: {
+          role: "tool",
+          content: [
+            {
+              type: "tool-result",
+              toolCallId: "tool_0_dateTime",
+              toolName: "dateTime",
+              result: {
+                type: "json",
+                value: {
+                  day: "20",
+                  hours: 16,
+                  minutes: 3,
+                  month: "August",
+                  year: "2025",
+                },
+              },
+            },
+          ],
+        },
+        sources: [],
+      }),
+      baseMessageDoc({
+        finishReason: "stop",
+        text: "The time in Paris, France is 4:03 PM on August 20, 2025.",
+        stepOrder: 3,
+        message: {
+          role: "assistant",
+          content: [
+            {
+              type: "text",
+              text: "The time in Paris, France is 4:03 PM on August 20, 2025.",
+            },
+          ],
+        },
+        reasoningDetails: [],
+        sources: [],
+        warnings: [],
+      }),
+    ];
+
+    const uiMessages = toUIMessages(messages);
+
+    expect(uiMessages).toHaveLength(2);
+    expect(uiMessages[0].role).toBe("user");
+    expect(uiMessages[0].text).toBe("what time is it in paris?");
+    expect(uiMessages[1].role).toBe("assistant");
+    expect(uiMessages[1].text).toBe(
+      "The time in Paris, France is 4:03 PM on August 20, 2025.",
+    );
+  });
+
+  it("handles messages in reverse order (assistant response, then tool response, then tool call)", () => {
+    const messages = [
+      // Final assistant response comes first (stepOrder 3) - no tool flag since it's the final message
+      baseMessageDoc({
+        _id: "msg3",
+        order: 1,
+        stepOrder: 3,
+        finishReason: "stop",
+        text: "The result is 42.",
+        tool: false, // This is the final message without tool calls
+        message: {
+          role: "assistant",
+          content: [
+            {
+              type: "text",
+              text: "The result is 42.",
+            },
+          ],
+        },
+      }),
+      // Tool result comes second (stepOrder 2)
+      baseMessageDoc({
+        _id: "msg2",
+        order: 1,
+        stepOrder: 2,
+        tool: true,
+        message: {
+          role: "tool",
+          content: [
+            {
+              type: "tool-result",
+              toolCallId: "call1",
+              toolName: "calculator",
+              result: "42",
+            },
+          ],
+        },
+      }),
+      // Tool call comes last (stepOrder 1)
+      baseMessageDoc({
+        _id: "msg1",
+        order: 1,
+        stepOrder: 1,
+        tool: true,
+        text: "",
+        message: {
+          role: "assistant",
+          content: [
+            {
+              type: "tool-call",
+              toolName: "calculator",
+              toolCallId: "call1",
+              args: { operation: "add", a: 40, b: 2 },
+            },
+          ],
+        },
+      }),
+    ];
+
+    const uiMessages = toUIMessages(messages);
+
+    expect(uiMessages).toHaveLength(1);
+    expect(uiMessages[0].role).toBe("assistant");
+
+    // Should concatenate text from all messages (only the final response has text)
+    expect(uiMessages[0].text).toBe("The result is 42.");
+
+    // Should use first message's fields (msg1 with stepOrder 1)
+    expect(uiMessages[0].id).toBe("msg1");
+    expect(uiMessages[0].order).toBe(1);
+    expect(uiMessages[0].stepOrder).toBe(1);
+
+    // Should have both tool call and result parts
+    const toolParts = uiMessages[0].parts.filter(
+      (p) => p.type === "tool-calculator",
+    );
+    expect(toolParts).toHaveLength(1);
+    expect(toolParts[0]).toMatchObject({
+      type: "tool-calculator",
+      toolCallId: "call1",
+      state: "output-available",
+      input: { operation: "add", a: 40, b: 2 },
+      output: "42",
+    });
+
+    // Should also have text part
+    const textParts = uiMessages[0].parts.filter((p) => p.type === "text");
+    expect(textParts).toHaveLength(1);
+    expect(textParts[0].text).toBe("The result is 42.");
+  });
 });