refresh views after assistant changes

apps/nextjs/src/components/assistant/assistant-mutation-refresh.spec.ts

@@ -0,0 +1,94 @@
+import { describe, expect, test } from "vitest";
+import type { ThreadMessage } from "@assistant-ui/react";
+
+import {
+  getSuccessfulApprovedAssistantMutationIds,
+  updateAssistantMutationRefreshState,
+} from "./assistant-mutation-refresh";
+
+const assistantMessage = (...content: ThreadMessage["content"]): ThreadMessage =>
+  ({
+    id: "assistant-message",
+    role: "assistant",
+    content,
+    status: { type: "complete", reason: "stop" },
+    createdAt: new Date(),
+    metadata: { custom: {} },
+  }) as ThreadMessage;
+
+describe("assistant mutation refresh", () => {
+  test("returns only completed, approved mutation calls", () => {
+    expect(
+      getSuccessfulApprovedAssistantMutationIds([
+        assistantMessage(
+          {
+            type: "tool-call",
+            toolCallId: "approved-1",
+            toolName: "app_create",
+            args: { name: "Wikipedia" },
+            argsText: "{}",
+            approval: { id: "approval-1", approved: true },
+            result: { id: "app-1" },
+          },
+          {
+            type: "tool-call",
+            toolCallId: "read-1",
+            toolName: "app_getAll",
+            args: {},
+            argsText: "{}",
+            result: [],
+          },
+          {
+            type: "tool-call",
+            toolCallId: "pending-1",
+            toolName: "board_savePartialBoardSettings",
+            args: { id: "board-1" },
+            argsText: "{}",
+            approval: { id: "approval-2", approved: true },
+          },
+        ),
+      ]),
+    ).toEqual(["approved-1"]);
+  });
+
+  test("ignores denied and failed mutations", () => {
+    expect(
+      getSuccessfulApprovedAssistantMutationIds([
+        assistantMessage(
+          {
+            type: "tool-call",
+            toolCallId: "denied-1",
+            toolName: "app_create",
+            args: {},
+            argsText: "{}",
+            approval: { id: "approval-1", approved: false },
+            result: null,
+          },
+          {
+            type: "tool-call",
+            toolCallId: "failed-1",
+            toolName: "app_create",
+            args: {},
+            argsText: "{}",
+            approval: { id: "approval-2", approved: true },
+            result: { error: "Creation failed" },
+          },
+        ),
+      ]),
+    ).toEqual([]);
+  });
+
+  test("baselines loaded history, refreshes once for new calls, and baselines a switched conversation", () => {
+    const initial = updateAssistantMutationRefreshState({ conversationId: null, toolCallIds: new Set() }, "thread-1", [
+      "historical-1",
+    ]);
+    expect(initial.shouldRefresh).toBe(false);
+
+    const mutation = updateAssistantMutationRefreshState(initial.state, "thread-1", ["historical-1", "new-1", "new-2"]);
+    expect(mutation.shouldRefresh).toBe(true);
+    expect(updateAssistantMutationRefreshState(mutation.state, "thread-1", ["new-1", "new-2"]).shouldRefresh).toBe(
+      false,
+    );
+    expect(updateAssistantMutationRefreshState(mutation.state, "thread-2", ["historical-2"]).shouldRefresh).toBe(false);
+  });
+});