fix: restore assistant approval flows

apps/nextjs/src/components/assistant/assistant-runtime-provider.spec.ts

@@ -0,0 +1,80 @@
+// @vitest-environment jsdom
+
+import { act, createElement } from "react";
+import { createRoot } from "react-dom/client";
+import type { ChatModelAdapter, ThreadMessageLike } from "@assistant-ui/react";
+import { defineToolkit, MessagePrimitive, ThreadPrimitive, useLocalRuntime } from "@assistant-ui/react";
+import { afterEach, describe, expect, test } from "vitest";
+import { z } from "zod/v4";
+
+import { AssistantRuntimeProviderWithTools } from "./assistant-runtime-provider";
+
+(globalThis as typeof globalThis & { IS_REACT_ACT_ENVIRONMENT: boolean }).IS_REACT_ACT_ENVIRONMENT = true;
+
+const noOpAdapter: ChatModelAdapter = { async *run() {} };
+const toolkit = defineToolkit({
+  ask_user: {
+    type: "human",
+    display: "standalone",
+    parameters: z.object({ question: z.string() }),
+    render: ({ args }) => createElement("div", { "data-testid": "ask-user-ui" }, `Question: ${args.question}`),
+  },
+});
+const messages: ThreadMessageLike[] = [
+  {
+    role: "assistant",
+    content: [
+      {
+        type: "tool-call",
+        toolCallId: "ask-1",
+        toolName: "ask_user",
+        args: { question: "Which board?" },
+        argsText: JSON.stringify({ question: "Which board?" }),
+      },
+    ],
+    status: { type: "requires-action", reason: "tool-calls" },
+  },
+];
+
+const Message = () =>
+  createElement(MessagePrimitive.Parts, {
+    components: {
+      tools: {
+        Fallback: () => createElement("div", { "data-testid": "fallback-tool-ui" }, "Fallback"),
+      },
+    },
+  });
+
+const TestAssistant = () => {
+  const runtime = useLocalRuntime(noOpAdapter, { initialMessages: messages });
+  return createElement(
+    AssistantRuntimeProviderWithTools,
+    { runtime, toolkit },
+    createElement(ThreadPrimitive.Messages, { components: { Message } }),
+  );
+};
+
+const containers: HTMLDivElement[] = [];
+
+afterEach(() => {
+  for (const container of containers.splice(0)) container.remove();
+});
+
+describe("AssistantRuntimeProviderWithTools", () => {
+  test("makes the registered human-tool renderer available to message parts", async () => {
+    const container = document.createElement("div");
+    containers.push(container);
+    document.body.append(container);
+    const root = createRoot(container);
+
+    await act(async () => {
+      root.render(createElement(TestAssistant));
+      await new Promise((resolve) => setTimeout(resolve, 0));
+    });
+
+    expect(container.querySelector('[data-testid="ask-user-ui"]')?.textContent).toBe("Question: Which board?");
+    expect(container.querySelector('[data-testid="fallback-tool-ui"]')).toBeNull();
+
+    await act(async () => root.unmount());
+  });
+});