assistant-ui 0.0.123 → 0.0.124

package/src/codemods/v0-9/edge-package-split.ts

@@ -1,11 +1,8 @@
 import { createTransformer } from "../utils/createTransformer";
 
-// Chat runtime and converter exports moved to react-ai-sdk
 const reactAiSdkExports: string[] = [
-  // Chat Runtime (replaces edge runtime)
   "useChatRuntime",
 
-  // Core Types
   "CoreMessage",
   "CoreUserMessage",
   "CoreAssistantMessage",
@@ -14,13 +11,11 @@ const reactAiSdkExports: string[] = [
   "CoreAssistantMessagePart",
   "CoreToolCallMessagePart",
 
-  // Core message converters
   "fromCoreMessages",
   "fromCoreMessage",
   "toCoreMessages",
   "toCoreMessage",
 
-  // Language model converters
   "toLanguageModelMessages",
   "toLanguageModelTools",
   "fromLanguageModelMessages",
@@ -29,150 +24,46 @@ const reactAiSdkExports: string[] = [
 ];
 
 const migrateToEdgePackage = createTransformer(({ j, root, markAsChanged }) => {
-  const sourcesToMigrate: string[] = ["@assistant-ui/react"];
-  const movedAiSdkSpecifiers: any[] = [];
-  let lastMigratedImportPath: any = null;
-
-  root
-    .find(j.ImportDeclaration)
-    .filter((path: any) => sourcesToMigrate.includes(path.value.source.value))
-    .forEach((path: any) => {
-      let hadMigratedSpecifiers = false;
-      const remainingSpecifiers: any[] = [];
-      path.value.specifiers.forEach((specifier: any) => {
-        if (
-          j.ImportSpecifier.check(specifier) &&
-          reactAiSdkExports.includes(specifier.imported.name as string)
-        ) {
-          movedAiSdkSpecifiers.push(specifier);
-          hadMigratedSpecifiers = true;
-        } else {
-          remainingSpecifiers.push(specifier);
-        }
-      });
-      if (hadMigratedSpecifiers) {
-        lastMigratedImportPath = path;
-      }
-      if (remainingSpecifiers.length === 0) {
-        j(path).remove();
-        markAsChanged();
-      } else if (remainingSpecifiers.length !== path.value.specifiers.length) {
-        path.value.specifiers = remainingSpecifiers;
-        markAsChanged();
-      }
-    });
-
-  // Add imports for react-ai-sdk (all edge-related imports now go here)
-  if (movedAiSdkSpecifiers.length > 0) {
-    const existingAiSdkImport = root.find(j.ImportDeclaration, {
-      source: { value: "@assistant-ui/react-ai-sdk" },
-    });
-    if (existingAiSdkImport.size() > 0) {
-      existingAiSdkImport.forEach((path: any) => {
-        movedAiSdkSpecifiers.forEach((specifier: any) => {
-          if (
-            !path.value.specifiers.some(
-              (s: any) => s.imported.name === specifier.imported.name,
-            )
-          ) {
-            path.value.specifiers.push(specifier);
-          }
-        });
-      });
-    } else {
-      const newImport = j.importDeclaration(
-        movedAiSdkSpecifiers,
-        j.literal("@assistant-ui/react-ai-sdk"),
-      );
-      if (lastMigratedImportPath) {
-        j(lastMigratedImportPath).insertAfter(newImport);
-      } else {
-        const firstImport = root.find(j.ImportDeclaration).at(0);
-        if (firstImport.size() > 0) {
-          firstImport.insertBefore(newImport);
-        } else {
-          root.get().node.program.body.unshift(newImport);
-        }
-      }
-    }
-    markAsChanged();
-  }
-
-  // Migrate imports from edge/converters
-  root.find(j.ImportDeclaration).forEach((path: any) => {
-    const sourceValue: string = path.value.source.value;
+  root.find(j.ImportDeclaration).forEach((path) => {
+    const source = path.node.source.value;
     if (
-      sourceValue.startsWith("@assistant-ui/react/") &&
-      (sourceValue.includes("edge/") ||
-        sourceValue.includes("dangerous-in-browser/"))
+      typeof source === "string" &&
+      source.startsWith("@assistant-ui/react/") &&
+      (source.includes("edge/") || source.includes("dangerous-in-browser/"))
     ) {
-      path.value.source = j.literal(
-        sourceValue.replace(
-          "@assistant-ui/react/",
-          "@assistant-ui/react-ai-sdk/",
-        ),
+      path.node.source = j.literal(
+        source.replace("@assistant-ui/react/", "@assistant-ui/react-ai-sdk/"),
       );
       markAsChanged();
+      return;
     }
-  });
 
-  // Migrate language model converter imports from react-edge to react-ai-sdk
-  root.find(j.ImportDeclaration).forEach((path: any) => {
-    const sourceValue: string = path.value.source.value;
     if (
-      sourceValue === "@assistant-ui/react-edge" ||
-      sourceValue === "@assistant-ui/react"
-    ) {
-      let hasLanguageModelConverters = false;
-      const remainingSpecifiers: any[] = [];
-      const aiSdkSpecifiers: any[] = [];
-
-      path.value.specifiers.forEach((specifier: any) => {
-        if (
-          j.ImportSpecifier.check(specifier) &&
-          reactAiSdkExports.includes(specifier.imported.name as string)
-        ) {
-          aiSdkSpecifiers.push(specifier);
-          hasLanguageModelConverters = true;
-        } else {
-          remainingSpecifiers.push(specifier);
-        }
-      });
-
-      if (hasLanguageModelConverters) {
-        if (remainingSpecifiers.length === 0) {
-          j(path).remove();
-        } else {
-          path.value.specifiers = remainingSpecifiers;
-        }
-
-        const existingAiSdkImport = root.find(j.ImportDeclaration, {
-          source: { value: "@assistant-ui/react-ai-sdk" },
-        });
+      source !== "@assistant-ui/react" &&
+      source !== "@assistant-ui/react-edge"
+    )
+      return;
 
-        if (existingAiSdkImport.size() > 0) {
-          existingAiSdkImport.forEach((importPath: any) => {
-            aiSdkSpecifiers.forEach((specifier: any) => {
-              if (
-                !importPath.value.specifiers.some(
-                  (s: any) => s.imported.name === specifier.imported.name,
-                )
-              ) {
-                importPath.value.specifiers.push(specifier);
-              }
-            });
-          });
-        } else {
-          const newImport = j.importDeclaration(
-            aiSdkSpecifiers,
-            j.literal("@assistant-ui/react-ai-sdk"),
-          );
-          j(path).insertAfter(newImport);
-        }
+    const specifiers = path.node.specifiers ?? [];
+    const moved = specifiers.filter(
+      (specifier) =>
+        j.ImportSpecifier.check(specifier) &&
+        reactAiSdkExports.includes(specifier.imported.name as string),
+    );
+    if (moved.length === 0) return;
 
-        markAsChanged();
-      }
+    const destination = j.literal("@assistant-ui/react-ai-sdk");
+    if (moved.length === specifiers.length) {
+      path.node.source = destination;
+    } else {
+      path.node.specifiers = specifiers.filter(
+        (specifier) => !moved.includes(specifier),
+      );
+      const declaration = j.importDeclaration(moved, destination);
+      declaration.importKind = path.node.importKind ?? "value";
+      j(path).insertAfter(declaration);
     }
+    markAsChanged();
   });
 });
 

package/src/lib/utils/package-installer.test.ts

@@ -0,0 +1,39 @@
+import { describe, expect, it, vi } from "vitest";
+import { installPackageIfNeeded } from "./package-installer";
+
+const mocks = vi.hoisted(() => ({
+  askQuestion: vi.fn(),
+  installPackage: vi.fn(),
+}));
+
+vi.mock("./file-scanner", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("./file-scanner")>()),
+  scanForImport: () => true,
+}));
+
+vi.mock("./package-manager", async (importOriginal) => ({
+  ...(await importOriginal<typeof import("./package-manager")>()),
+  isPackageInstalled: () => false,
+  askQuestion: mocks.askQuestion,
+  installPackage: mocks.installPackage,
+}));
+
+describe("installPackageIfNeeded", () => {
+  it.each(["y", ""])(
+    "rejects a failed required install after accepting with %j",
+    async (answer) => {
+      mocks.askQuestion.mockResolvedValue(answer);
+      mocks.installPackage.mockResolvedValue(false);
+
+      await expect(
+        installPackageIfNeeded({
+          packageName: "@assistant-ui/ai-sdk",
+          importPatterns: ["@assistant-ui/ai-sdk"],
+          promptMessage: "Install?",
+          skipMessage: "Already installed",
+          notFoundMessage: "Not found",
+        }),
+      ).rejects.toThrow("Failed to install @assistant-ui/ai-sdk");
+    },
+  );
+});

package/src/lib/utils/package-installer.ts

@@ -31,7 +31,9 @@ export async function installPackageIfNeeded(
 
   const answer = await askQuestion(config.promptMessage);
   if (answer === "" || answer.toLowerCase().startsWith("y")) {
-    await installPackage(config.packageName);
+    if (!(await installPackage(config.packageName))) {
+      throw new Error(`Failed to install ${config.packageName}`);
+    }
   } else {
     logger.info("Skipping installation.");
   }

package/templates/cloud-harness/components/assistant-ui/elements/tool-fallback.aui.tsx

@@ -141,11 +141,13 @@ function ToolFallbackDuration({
 function ToolFallbackTrigger({
   toolName,
   status,
+  responseDeliveryStatus,
   className,
   ...props
 }: React.ComponentProps<typeof CollapsibleTrigger> & {
   toolName: string;
   status?: ToolCallMessagePartStatus;
+  responseDeliveryStatus?: ToolCallMessagePart["responseDeliveryStatus"];
 }) {
   const statusType = status?.type ?? "complete";
   const isRunning = statusType === "running";
@@ -154,13 +156,15 @@ function ToolFallbackTrigger({
 
   const Icon = statusIconMap[statusType];
   const label =
-    statusType === "running"
-      ? "Running tool"
-      : statusType === "requires-action"
-        ? "Waiting on tool"
-        : statusType === "incomplete"
-          ? `${isCancelled ? "Cancelled" : "Failed"} tool`
-          : "Used tool";
+    responseDeliveryStatus === "queued"
+      ? "Reply queued, not sent yet"
+      : statusType === "running"
+        ? "Running tool"
+        : statusType === "requires-action"
+          ? "Waiting on tool"
+          : statusType === "incomplete"
+            ? `${isCancelled ? "Cancelled" : "Failed"} tool`
+            : "Used tool";
 
   return (
     <CollapsibleTrigger
@@ -1195,6 +1199,7 @@ const ToolFallbackImpl: ToolCallMessagePartComponent = ({
   argsText,
   result,
   status,
+  responseDeliveryStatus,
   addResult,
   resume,
   interrupt,
@@ -1243,7 +1248,11 @@ const ToolFallbackImpl: ToolCallMessagePartComponent = ({
 
   return (
     <ToolFallbackRoot open={open} onOpenChange={setOpen}>
-      <ToolFallbackTrigger toolName={toolName} status={status} />
+      <ToolFallbackTrigger
+        toolName={toolName}
+        status={status}
+        responseDeliveryStatus={responseDeliveryStatus}
+      />
       <ToolFallbackContent>
         <ToolFallbackError status={status} />
         <ToolFallbackArgs

package/templates/cloud-harness/package.json

@@ -28,13 +28,13 @@
     "cn": "^0.4.0",
     "harness-sdk": "file:vendor/harness-sdk-0.3.1.tgz",
     "lucide-react": "^1.52.0",
-    "next": "^16.4.0",
+    "next": "^16.4.1",
     "react": "^19.3.0",
     "react-dom": "^19.3.0",
     "remark-gfm": "^4.0.1",
     "statewire": "file:vendor/statewire-0.19.3.tgz",
     "tw-animate-css": "^1.4.0",
-    "tw-shimmer": "^0.4.13",
+    "tw-shimmer": "^0.4.14",
     "zustand": "^5.0.15"
   },
   "devDependencies": {