@assistant-ui/core 0.3.18 → 0.3.19

package/src/react/runtimes/cloud/AssistantCloudThreadHistoryAdapter.test.tsx

@@ -178,6 +178,20 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
       threadId: "thread-1",
       messageId: "local-message-1",
     });
+    listeners.get("thread.toolApprovalAnswered")!({
+      threadId: "thread-1",
+      messageId: "local-message-1",
+      toolCallId: "tool-approved",
+      toolName: "send_email",
+      approved: true,
+    });
+    listeners.get("thread.toolApprovalAnswered")!({
+      threadId: "thread-1",
+      messageId: "local-message-1",
+      toolCallId: "tool-rejected",
+      toolName: "delete_account",
+      approved: false,
+    });
     listeners.get("composer.attachmentAdd")!({
       threadId: "thread-1",
       contentType: "image/png",
@@ -202,6 +216,11 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
       messageId: "local-message-1",
       reason: "error",
     });
+    listeners.get("message.error")!({
+      threadId: "thread-1",
+      messageId: "local-message-2",
+      reason: "error",
+    });
     threadState.isEmpty = true;
     threadState.suggestions = [{ prompt: "one" }, { prompt: "two" }];
     notify!();
@@ -232,6 +251,16 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
         expect.objectContaining({ kind: "message_regenerated" }),
         expect.objectContaining({ kind: "branch_switched" }),
         expect.objectContaining({ kind: "message_copied" }),
+        expect.objectContaining({
+          kind: "tool_approved",
+          message_id: "remote-message-1",
+          props: { toolCallId: "tool-approved", toolName: "send_email" },
+        }),
+        expect.objectContaining({
+          kind: "tool_rejected",
+          message_id: "remote-message-1",
+          props: { toolCallId: "tool-rejected", toolName: "delete_account" },
+        }),
         expect.objectContaining({
           kind: "attachment_added",
           props: { type: "image/png" },
@@ -250,6 +279,12 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
         expect.objectContaining({ kind: "suggestions_shown", value: 2 }),
       ]),
     );
+    expect(
+      vi
+        .mocked(cloud.events.track)
+        .mock.calls.map(([event]) => event)
+        .filter((event) => event.kind === "error_shown"),
+    ).toHaveLength(1);
   });
 
   it("refreshes formatted persistence when the Cloud client changes", async () => {
@@ -501,7 +536,7 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
         },
         timing: {
           streamStartTime: 100,
-          firstTokenTime: 155.6,
+          firstTokenTime: 55.6,
           totalChunks: 1,
           toolCallCount: 0,
         },
@@ -688,6 +723,7 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
                 tool_source: "mcp",
               }),
             ],
+            finish_reason: "tool-calls",
           },
           {
             tool_calls: [
@@ -696,6 +732,7 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
                 tool_source: "frontend",
               }),
             ],
+            finish_reason: "tool-calls",
           },
         ],
         trace_id: "00112233445566778899aabbccddeeff",
@@ -705,6 +742,189 @@ describe("useAssistantCloudThreadHistoryAdapter", () => {
     );
   });
 
+  it("reports a single ai-sdk/v6 step with its tool calls", () => {
+    mocks.aui = mocks.makeClient("thread-1");
+    const cloud = makeCloud();
+    const { result } = renderHook(() =>
+      useAssistantCloudThreadHistoryAdapter({ current: cloud }),
+    );
+    const formatted = result.current.withFormat({
+      format: "ai-sdk/v6",
+      encode: ({ message }) => message,
+      decode: ({ parent_id, content }) => ({
+        parentId: parent_id,
+        message: content as { id: string },
+      }),
+      getId: (message: { id: string }) => message.id,
+    });
+
+    formatted.reportTelemetry([
+      {
+        parentId: null,
+        message: {
+          id: "message-1",
+          role: "assistant",
+          parts: [
+            { type: "step-start" },
+            {
+              type: "tool-search",
+              toolCallId: "search-1",
+              input: { query: "test" },
+              output: { result: "ok" },
+            },
+          ],
+        },
+      },
+    ]);
+
+    expect(cloud.runs.report).toHaveBeenCalledWith(
+      expect.objectContaining({
+        total_steps: 1,
+        steps: [
+          {
+            tool_calls: [
+              expect.objectContaining({
+                tool_call_id: "search-1",
+                tool_source: "frontend",
+              }),
+            ],
+            finish_reason: "tool-calls",
+          },
+        ],
+      }),
+    );
+  });
+
+  it("reads the status of an ai-sdk/v6 run from its finish reason", () => {
+    mocks.aui = mocks.makeClient("thread-1");
+    const cloud = makeCloud();
+    const { result } = renderHook(() =>
+      useAssistantCloudThreadHistoryAdapter({ current: cloud }),
+    );
+    const formatted = result.current.withFormat({
+      format: "ai-sdk/v6",
+      encode: ({ message }) => message,
+      decode: ({ parent_id, content }) => ({
+        parentId: parent_id,
+        message: content as { id: string },
+      }),
+      getId: (message: { id: string }) => message.id,
+    });
+
+    formatted.reportTelemetry([
+      {
+        parentId: null,
+        message: {
+          id: "message-1",
+          role: "assistant",
+          parts: [{ type: "text", text: "cut" }],
+          metadata: { finishReason: "length" },
+        },
+      },
+    ]);
+
+    expect(cloud.runs.report).toHaveBeenCalledWith(
+      expect.objectContaining({ status: "incomplete", outcome_type: "length" }),
+    );
+  });
+
+  it("reads the error and timing of an ai-sdk/v6 run from the thread message", () => {
+    mocks.aui = mocks.makeClient("thread-1");
+    const cloud = makeCloud();
+    const { result } = renderHook(() =>
+      useAssistantCloudThreadHistoryAdapter({ current: cloud }),
+    );
+    const formatted = result.current.withFormat({
+      format: "ai-sdk/v6",
+      encode: ({ message }) => message,
+      decode: ({ parent_id, content }) => ({
+        parentId: parent_id,
+        message: content as { id: string },
+      }),
+      getId: (message: { id: string }) => message.id,
+    });
+    const base = makeAssistantMessage("assistant-1");
+    const message: ThreadAssistantMessage = {
+      ...base,
+      status: {
+        type: "incomplete",
+        reason: "error",
+        error: Object.assign(new Error("model unavailable"), {
+          code: "model_unavailable",
+        }),
+      },
+      metadata: {
+        ...base.metadata,
+        timing: {
+          streamStartTime: 100,
+          firstTokenTime: 240,
+          totalChunks: 3,
+          toolCallCount: 0,
+        },
+      },
+    };
+
+    formatted.reportTelemetry(
+      [
+        {
+          parentId: null,
+          message: {
+            id: "message-1",
+            role: "assistant",
+            parts: [{ type: "text", text: "partial" }],
+          },
+        },
+      ],
+      { message },
+    );
+
+    expect(cloud.runs.report).toHaveBeenCalledWith(
+      expect.objectContaining({
+        status: "error",
+        error: "model unavailable",
+        error_code: "model_unavailable",
+        first_token_ms: 240,
+      }),
+    );
+  });
+
+  it("reports a run that failed before any assistant message was stored", () => {
+    mocks.aui = mocks.makeClient("thread-1");
+    const cloud = makeCloud();
+    const { result } = renderHook(() =>
+      useAssistantCloudThreadHistoryAdapter({ current: cloud }),
+    );
+    const formatted = result.current.withFormat({
+      format: "ai-sdk/v6",
+      encode: ({ message }) => message,
+      decode: ({ parent_id, content }) => ({
+        parentId: parent_id,
+        message: content as { id: string },
+      }),
+      getId: (message: { id: string }) => message.id,
+    });
+    const message: ThreadAssistantMessage = {
+      ...makeAssistantMessage("assistant-1"),
+      content: [],
+      status: {
+        type: "incomplete",
+        reason: "error",
+        error: { code: "AI_APICallError", message: "upstream failed" },
+      },
+    };
+
+    formatted.reportTelemetry([], { message, durationMs: 120 });
+
+    expect(cloud.runs.report).toHaveBeenCalledWith(
+      expect.objectContaining({
+        status: "error",
+        error: "upstream failed",
+        error_code: "AI_APICallError",
+        duration_ms: 120,
+      }),
+    );
+  });
+
   it("reports the model ID carried by aui/v0 step metadata", () => {
     mocks.aui = mocks.makeClient("thread-1");
     const cloud = makeCloud();