assistant-ui 0.0.123 → 0.0.125

package/src/lib/cloud-remote.ts

@@ -0,0 +1,150 @@
+import { validateCloudUrl } from "./cloud-url";
+
+export const CLOUD_CLI_RUN_PATH = "/api/cli/run";
+
+export type CloudCliAction = { type: string; [key: string]: unknown };
+
+export type CloudCliResponse = {
+  stdout?: string;
+  stderr?: string;
+  exitCode: number;
+  actions?: CloudCliAction[];
+};
+
+const SAFE = /^[A-Za-z0-9_@%+=:,./-]+$/;
+
+export const joinCloudArgs = (args: readonly string[]): string =>
+  args
+    .map((arg) => (SAFE.test(arg) ? arg : `'${arg.replaceAll("'", `'\\''`)}'`))
+    .join(" ");
+
+export const parseCloudCliResponse = (
+  value: unknown,
+): CloudCliResponse | null => {
+  if (!value || typeof value !== "object") return null;
+  const body = value as Record<string, unknown>;
+  const exitCode = body["exitCode"];
+  if (
+    typeof exitCode !== "number" ||
+    !Number.isInteger(exitCode) ||
+    exitCode < 0 ||
+    exitCode > 255
+  ) {
+    return null;
+  }
+  for (const key of ["stdout", "stderr"]) {
+    if (body[key] !== undefined && typeof body[key] !== "string") return null;
+  }
+  const actions = body["actions"];
+  if (
+    actions !== undefined &&
+    !(
+      Array.isArray(actions) &&
+      actions.every(
+        (action) =>
+          !!action &&
+          typeof action === "object" &&
+          typeof (action as CloudCliAction).type === "string",
+      )
+    )
+  ) {
+    return null;
+  }
+  return body as CloudCliResponse;
+};
+
+const cliVersion: string =
+  typeof __AUI_PACKAGE_VERSION__ === "string"
+    ? __AUI_PACKAGE_VERSION__
+    : "0.0.0";
+
+export const sendCloudCommand = async (
+  cloudUrl: string,
+  args: readonly string[],
+  accessToken: string | null,
+): Promise<CloudCliResponse> => {
+  const url = `${validateCloudUrl(cloudUrl)}${CLOUD_CLI_RUN_PATH}`;
+  let response: Response;
+  try {
+    response = await fetch(url, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        ...(accessToken ? { Authorization: `Bearer ${accessToken}` } : {}),
+      },
+      body: JSON.stringify({
+        command: joinCloudArgs(args),
+        client: {
+          name: "assistant-ui",
+          version: cliVersion,
+          platform: `${process.platform}-${process.arch}`,
+        },
+      }),
+      redirect: "error",
+      signal: AbortSignal.timeout(60_000),
+    });
+  } catch (error) {
+    const cause = (error as Error).cause as NodeJS.ErrnoException | undefined;
+    throw new Error(
+      `Could not reach ${url}: ${cause?.message || cause?.code || (error as Error).message}`,
+    );
+  }
+  const body = parseCloudCliResponse(await response.json().catch(() => null));
+  if (!body) {
+    throw new Error(
+      `Unexpected response from ${url} (${response.status} ${response.statusText}).`,
+    );
+  }
+  return body;
+};
+
+// A response naming an action this client lacks fails before printing, so the
+// server can roll a new action out behind the client version it is sent.
+const ACTION_HANDLERS: Record<
+  string,
+  (action: CloudCliAction) => Promise<number>
+> = {};
+
+export const runCloudCommand = async (
+  args: readonly string[],
+  deps: {
+    cloudUrl: string;
+    accessToken: () => Promise<string | null>;
+    stdout: (text: string) => void;
+    stderr: (text: string) => void;
+    send?: typeof sendCloudCommand;
+  },
+): Promise<number> => {
+  let response: CloudCliResponse;
+  try {
+    response = await (deps.send ?? sendCloudCommand)(
+      deps.cloudUrl,
+      args,
+      await deps.accessToken().catch((error: unknown) => {
+        deps.stderr(
+          `${error instanceof Error ? error.message : String(error)}\n`,
+        );
+        return null;
+      }),
+    );
+  } catch (error) {
+    deps.stderr(`${error instanceof Error ? error.message : String(error)}\n`);
+    return 1;
+  }
+  const unknown = response.actions?.find(
+    (action) => !Object.hasOwn(ACTION_HANDLERS, action.type),
+  );
+  if (unknown) {
+    deps.stderr(
+      `This command needs a newer assistant-ui (unsupported action "${unknown.type}"). Update the CLI and try again.\n`,
+    );
+    return 1;
+  }
+  if (response.stdout) deps.stdout(response.stdout);
+  if (response.stderr) deps.stderr(response.stderr);
+  for (const action of response.actions ?? []) {
+    const code = await ACTION_HANDLERS[action.type]!(action);
+    if (code !== 0) return code;
+  }
+  return response.exitCode;
+};

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": {