assistant-ui 0.0.120 → 0.0.122

package/src/lib/cloud-url.ts

@@ -0,0 +1,19 @@
+export const validateCloudUrl = (value: string): string => {
+  const url = new URL(value);
+  if (
+    url.username ||
+    url.password ||
+    url.search ||
+    url.hash ||
+    (url.protocol !== "https:" &&
+      !(
+        url.protocol === "http:" &&
+        ["localhost", "127.0.0.1", "[::1]"].includes(url.hostname)
+      ))
+  ) {
+    throw new Error(
+      "Cloud, accounts and setup URLs must use HTTPS or HTTP localhost for development.",
+    );
+  }
+  return value.replace(/\/+$/, "");
+};

package/src/lib/create-project.test.ts

@@ -24,23 +24,6 @@ describe("reconcileAssistantUIImportLayout", () => {
   const read = (file: string) =>
     fs.readFileSync(path.join(projectDir, file), "utf8");
 
-  it("rewrites a legacy import when only the elements layout exists", async () => {
-    write(
-      "app/page.tsx",
-      'import { Thread } from "@/components/assistant-ui/thread";\n' +
-        'import { ThreadList } from "@/components/assistant-ui/thread-list";\n',
-    );
-    write("components/assistant-ui/elements/thread.aui.tsx", "export {};");
-    write("components/assistant-ui/elements/thread-list.aui.tsx", "export {};");
-
-    await reconcileAssistantUIImportLayout(projectDir);
-
-    expect(read("app/page.tsx")).toBe(
-      'import { Thread } from "@/components/assistant-ui/elements/thread.aui";\n' +
-        'import { ThreadList } from "@/components/assistant-ui/elements/thread-list.aui";\n',
-    );
-  });
-
   it("rewrites module declarations without changing import-like source text", async () => {
     write(
       "app/page.tsx",

package/src/lib/create-project.ts

@@ -1,7 +1,6 @@
 import * as fs from "node:fs";
 import * as os from "node:os";
 import * as path from "node:path";
-import { downloadTemplate } from "giget";
 import {
   parse as parseJsonc,
   printParseErrorCode,
@@ -9,26 +8,14 @@ import {
 } from "jsonc-parser";
 import { logger } from "./utils/logger";
 import { runSpawn, SpawnExitError, SpawnSignalError } from "./run-spawn";
-import { type PackageManagerName } from "./utils/package-manager";
+import { dlxCommand, type PackageManagerName } from "./utils/package-manager";
+import { resolveGitHubAuthToken, withStagedDownload } from "./utils/download";
 import { readProjectFiles } from "./utils/file-scanner";
 import {
   detectRegistryPlatform,
   resolveRegistryItemUrl,
 } from "./utils/registry";
 
-export function dlxCommand(pm: PackageManagerName): [string, string[]] {
-  switch (pm) {
-    case "pnpm":
-      return ["pnpm", ["dlx"]];
-    case "yarn":
-      return ["yarn", ["dlx"]];
-    case "bun":
-      return ["bunx", []];
-    case "npm":
-      return ["npx", ["--yes"]];
-  }
-}
-
 export interface TransformOptions {
   hasLocalComponents: boolean;
   skipInstall?: boolean;
@@ -56,26 +43,6 @@ const LOCAL_PROJECT_ARTIFACT_GLOB_IGNORES = LOCAL_PROJECT_ARTIFACT_DIRS.map(
   (dir) => `**/${dir}/**`,
 );
 
-export function resolvePackageManager(opts: {
-  useNpm?: boolean;
-  usePnpm?: boolean;
-  useYarn?: boolean;
-  useBun?: boolean;
-}): PackageManagerName | undefined {
-  if (opts.useNpm) return "npm";
-  if (opts.usePnpm) return "pnpm";
-  if (opts.useYarn) return "yarn";
-  if (opts.useBun) return "bun";
-  return undefined;
-}
-
-export function resolveGitHubAuthToken(): string | undefined {
-  const token =
-    process.env.GIGET_AUTH ?? process.env.GITHUB_TOKEN ?? process.env.GH_TOKEN;
-  const trimmed = token?.trim();
-  return trimmed || undefined;
-}
-
 function toBearerAuthHeader(token: string): string {
   return token.toLowerCase().startsWith("bearer ") ? token : `Bearer ${token}`;
 }
@@ -97,7 +64,6 @@ export async function resolveLatestReleaseRef(): Promise<string | undefined> {
   }
 }
 
-export const DOWNLOAD_TIMEOUT_MS = 30_000;
 const pendingDownloadCleanups = new Set<() => void>();
 
 export function cleanupPendingProjectDownloads(): void {
@@ -117,16 +83,9 @@ export async function downloadProject(
     ? `gh:assistant-ui/assistant-ui/${repoPath}#${ref}`
     : `gh:assistant-ui/assistant-ui/${repoPath}`;
 
-  // Suppress giget's debug output. The `debug` package (used by the upgrade
-  // command) sets process.env.DEBUG at module-load time, and giget logs to
-  // console.debug whenever that env var is truthy — even for unrelated
-  // namespaces. Temporarily unsetting it targets the root cause.
-  const origDebug = process.env.DEBUG;
-  delete process.env.DEBUG;
   let destinationCreated = false;
   let stagingDir: string | undefined;
-  let downloadPromise: Promise<unknown> | undefined;
-  let downloadFinished = false;
+  let downloadStarted = false;
   let downloadCommitted = false;
   let cleanupRequested = false;
   const removeCleanupListeners = () => {
@@ -183,62 +142,38 @@ export async function downloadProject(
     process.once("exit", cleanupOnExit);
     pendingDownloadCleanups.add(cleanupOnExit);
 
-    const authToken = resolveGitHubAuthToken();
-    downloadPromise = downloadTemplate(source, {
-      dir: stagingDir,
-      force: true,
-      silent: true,
-      ...(authToken ? { auth: authToken } : {}),
-    }).finally(() => {
-      downloadFinished = true;
-    });
-
-    let timer: ReturnType<typeof setTimeout>;
-    const timeoutPromise = new Promise<never>((_, reject) => {
-      timer = setTimeout(
-        () =>
-          reject(
-            new Error(
-              "Download timed out. This may be due to GitHub rate limiting or a network issue. Try again in a few minutes.",
-            ),
-          ),
-        DOWNLOAD_TIMEOUT_MS,
-      );
-    });
-
-    try {
-      await Promise.race([downloadPromise, timeoutPromise]);
-      if (cleanupRequested) throw new Error("Download was interrupted.");
-      destinationCreated = !fs.existsSync(destDir);
-      await fs.promises.mkdir(destDir, { recursive: true });
-      for (const entry of await fs.promises.readdir(stagingDir)) {
-        const target = path.join(destDir, entry);
-        const sourceEntry = path.join(stagingDir, entry);
-        await fs.promises.rm(target, { recursive: true, force: true });
-        try {
-          await fs.promises.rename(sourceEntry, target);
-        } catch (error) {
-          if ((error as NodeJS.ErrnoException).code !== "EXDEV") throw error;
-          await fs.promises.cp(sourceEntry, target, {
-            recursive: true,
-            force: true,
-          });
-          await fs.promises.rm(sourceEntry, { recursive: true, force: true });
+    const currentStagingDir = stagingDir;
+    downloadStarted = true;
+    await withStagedDownload(
+      source,
+      { dir: currentStagingDir, force: true, silent: true },
+      "Download timed out. This may be due to GitHub rate limiting or a network issue. Try again in a few minutes.",
+      removeStagingDir,
+      async (download) => {
+        await download;
+        if (cleanupRequested) throw new Error("Download was interrupted.");
+        destinationCreated = !fs.existsSync(destDir);
+        await fs.promises.mkdir(destDir, { recursive: true });
+        for (const entry of await fs.promises.readdir(currentStagingDir)) {
+          const target = path.join(destDir, entry);
+          const sourceEntry = path.join(currentStagingDir, entry);
+          await fs.promises.rm(target, { recursive: true, force: true });
+          try {
+            await fs.promises.rename(sourceEntry, target);
+          } catch (error) {
+            if ((error as NodeJS.ErrnoException).code !== "EXDEV") throw error;
+            await fs.promises.cp(sourceEntry, target, {
+              recursive: true,
+              force: true,
+            });
+            await fs.promises.rm(sourceEntry, { recursive: true, force: true });
+          }
         }
-      }
-      downloadCommitted = true;
-    } finally {
-      clearTimeout(timer!);
-    }
+        downloadCommitted = true;
+      },
+    );
   } finally {
-    if (!downloadPromise || downloadFinished) {
-      await removeStagingDir();
-    } else {
-      void downloadPromise.then(removeStagingDir, removeStagingDir);
-    }
-    if (origDebug !== undefined) {
-      process.env.DEBUG = origDebug;
-    }
+    if (!downloadStarted) await removeStagingDir();
   }
 }
 

package/src/lib/transform-files.test.ts

@@ -2,7 +2,7 @@ import { describe, expect, it, onTestFinished } from "vitest";
 import { mkdirSync, mkdtempSync, writeFileSync, rmSync } from "node:fs";
 import { tmpdir } from "node:os";
 import { dirname, join, relative } from "node:path";
-import { getRelevantFiles, countFilesToProcess } from "./transform";
+import { getRelevantFiles } from "./transform";
 
 const fixture = () => {
   const directory = mkdtempSync(join(tmpdir(), "aui-transform-files-"));
@@ -26,7 +26,6 @@ describe("codemod file discovery", () => {
       const { write } = fixture();
       const file = write(`my app.${extension}`);
       expect(getRelevantFiles(file)).toEqual([file]);
-      expect(countFilesToProcess(file)).toBe(1);
       expect(getRelevantFiles(relative(process.cwd(), file))).toEqual([file]);
     },
   );
@@ -54,7 +53,6 @@ describe("codemod file discovery", () => {
     const { write } = fixture();
     const file = write("app.tsx", source);
     expect(getRelevantFiles(file)).toEqual([file]);
-    expect(countFilesToProcess(file)).toBe(1);
   });
 
   it("skips unsupported explicit files", () => {

package/src/lib/transform.ts

@@ -3,8 +3,8 @@ import path from "node:path";
 import type { TransformOptions } from "./transform-options";
 import { fileURLToPath } from "node:url";
 import * as fs from "node:fs";
-import { sync as globSync } from "glob";
 import { runSpawnCapture, SpawnExitError, SpawnSignalError } from "./run-spawn";
+import { readProjectFiles } from "./utils/file-scanner";
 
 const log = debug("codemod:transform");
 
@@ -22,7 +22,8 @@ export function getRelevantFiles(source: string): string[] {
   }
 
   const pattern = "**/*.{js,jsx,ts,tsx}";
-  const files = globSync(pattern, {
+  const relevantFiles: string[] = [];
+  for (const { fullPath, content } of readProjectFiles(pattern, {
     cwd: target,
     ignore: [
       "**/node_modules/**",
@@ -31,27 +32,12 @@ export function getRelevantFiles(source: string): string[] {
       "**/*.min.js",
       "**/*.bundle.js",
     ],
-  }).map((file) => path.join(target, file));
-
-  const relevantFiles = files.filter((file) => {
-    try {
-      const content = fs.readFileSync(file, "utf8");
-      return content.includes("assistant-ui");
-    } catch {
-      return false;
-    }
-  });
-
+  })) {
+    if (content.includes("assistant-ui")) relevantFiles.push(fullPath);
+  }
   return relevantFiles;
 }
 
-/**
- * Counts the number of files that need to be processed
- */
-export function countFilesToProcess(source: string): number {
-  return getRelevantFiles(source).length;
-}
-
 function buildCommand(
   codemodPath: string,
   targetFiles: string[],

package/src/lib/utils/download.ts

@@ -0,0 +1,57 @@
+import { downloadTemplate } from "giget";
+
+export const DOWNLOAD_TIMEOUT_MS = 30_000;
+
+export function resolveGitHubAuthToken(): string | undefined {
+  const token =
+    process.env.GIGET_AUTH ?? process.env.GITHUB_TOKEN ?? process.env.GH_TOKEN;
+  const trimmed = token?.trim();
+  return trimmed || undefined;
+}
+
+export async function withStagedDownload<T>(
+  source: string,
+  options: NonNullable<Parameters<typeof downloadTemplate>[1]>,
+  timeoutMessage: string,
+  cleanup: () => void | Promise<void>,
+  consumeDownload: (download: Promise<unknown>) => Promise<T>,
+): Promise<T> {
+  // giget logs to console.debug whenever DEBUG is set, which the `debug`
+  // package does at module load for an unrelated namespace.
+  const origDebug = process.env.DEBUG;
+  delete process.env.DEBUG;
+  let download: Promise<unknown> | undefined;
+  let settled = false;
+  try {
+    const authToken = resolveGitHubAuthToken();
+    download = downloadTemplate(source, {
+      ...options,
+      ...(authToken ? { auth: authToken } : {}),
+    }).finally(() => {
+      settled = true;
+    });
+
+    let timer: ReturnType<typeof setTimeout>;
+    const timeout = new Promise<never>((_, reject) => {
+      timer = setTimeout(
+        () => reject(new Error(timeoutMessage)),
+        DOWNLOAD_TIMEOUT_MS,
+      );
+    });
+    try {
+      return await consumeDownload(Promise.race([download, timeout]));
+    } finally {
+      clearTimeout(timer!);
+    }
+  } finally {
+    try {
+      if (!download || settled) {
+        await cleanup();
+      } else {
+        void download.then(cleanup, cleanup);
+      }
+    } finally {
+      if (origDebug !== undefined) process.env.DEBUG = origDebug;
+    }
+  }
+}

package/src/lib/utils/file-scanner.ts

@@ -49,29 +49,3 @@ export function scanForImport(
 
   return false;
 }
-
-export function getFilesContaining(
-  searchString: string,
-  options: ScanOptions = {},
-): string[] {
-  const cwd = options.cwd || process.cwd();
-  const pattern = options.pattern || "**/*.{js,jsx,ts,tsx}";
-  const ignore = options.ignore || [
-    "**/node_modules/**",
-    "**/dist/**",
-    "**/build/**",
-  ];
-
-  const result: string[] = [];
-
-  for (const { fullPath, content } of readProjectFiles(pattern, {
-    cwd,
-    ignore,
-  })) {
-    if (content.includes(searchString)) {
-      result.push(fullPath);
-    }
-  }
-
-  return result;
-}

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

@@ -7,6 +7,32 @@ import { runSpawn, SpawnSignalError } from "../run-spawn";
 
 export type PackageManagerName = "npm" | "pnpm" | "yarn" | "bun";
 
+export function dlxCommand(pm: PackageManagerName): [string, string[]] {
+  switch (pm) {
+    case "pnpm":
+      return ["pnpm", ["dlx"]];
+    case "yarn":
+      return ["yarn", ["dlx"]];
+    case "bun":
+      return ["bunx", []];
+    case "npm":
+      return ["npx", ["--yes"]];
+  }
+}
+
+export function resolvePackageManager(opts: {
+  useNpm?: boolean;
+  usePnpm?: boolean;
+  useYarn?: boolean;
+  useBun?: boolean;
+}): PackageManagerName | undefined {
+  if (opts.useNpm) return "npm";
+  if (opts.usePnpm) return "pnpm";
+  if (opts.useYarn) return "yarn";
+  if (opts.useBun) return "bun";
+  return undefined;
+}
+
 export function askQuestion(query: string): Promise<string> {
   return new Promise((resolve) => {
     // A stream only reaches EOF once, so a run that already consumed stdin gets

package/src/lib/utils/registry.test.ts

@@ -144,12 +144,6 @@ describe("resolveRegistryItemUrl", () => {
     );
   });
 
-  it("uses the base URL for an explicit undefined style", () => {
-    expect(resolveRegistryItemUrl("thread", undefined)).toBe(
-      "https://r.assistant-ui.com/base/thread.json",
-    );
-  });
-
   it("keeps the style inside a single path segment", () => {
     expect(resolveRegistryItemUrl("thread", "base-nova?preview=1")).toBe(
       "https://r.assistant-ui.com/styles/base-nova%3Fpreview%3D1/thread.json",

package/src/program.ts

@@ -8,6 +8,7 @@ import { mcp } from "./commands/mcp";
 import { agent } from "./commands/agent";
 import { info } from "./commands/info";
 import { doctor } from "./commands/doctor";
+import { cloud } from "./commands/cloud";
 
 export function buildProgram() {
   return new Command()
@@ -21,6 +22,7 @@ export function buildProgram() {
     .addCommand(upgradeCommand)
     .addCommand(update)
     .addCommand(agent)
+    .addCommand(cloud)
     .addCommand(info)
     .addCommand(doctor);
 }

package/templates/cloud-harness/.assistant-ui/template.json

@@ -0,0 +1,3 @@
+{
+  "template": "cloud-harness"
+}

package/templates/cloud-harness/.env.example

@@ -0,0 +1 @@
+OPENAI_API_KEY=

package/templates/cloud-harness/.gitignore

@@ -0,0 +1,42 @@
+# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
+
+# dependencies
+/node_modules
+/.pnp
+.pnp.*
+.yarn/*
+!.yarn/patches
+!.yarn/plugins
+!.yarn/releases
+!.yarn/versions
+
+# testing
+/coverage
+
+# next.js
+/.next/
+/out/
+
+# production
+/build
+
+# misc
+.DS_Store
+*.pem
+
+# debug
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+.pnpm-debug.log*
+
+# env files (can opt-in for committing if needed)
+.env*
+!.env.example
+
+# vercel
+.vercel
+
+# typescript
+*.tsbuildinfo
+next-env.d.ts

package/templates/cloud-harness/README.md

@@ -0,0 +1,28 @@
+# Shared cloud chat
+
+This Next.js starter connects assistant-ui to a hosted harness. Browser clients
+on the same URL share conversation state, replies, and cancellation through
+`HarnessCloudThreadList`.
+
+Run `npx assistant-ui@latest cloud setup .` from this app to sign in and configure a project and harness.
+Add your `OPENAI_API_KEY` to `.env.local`, install dependencies, and run `npm run dev`.
+Open the app in two browser windows, or use **Share this chat** to copy its link.
+Send a message from either window; both show the same reply as it streams.
+Reload either window to restore the conversation.
+
+The URL fragment selects the thread. A fresh app uses `main` in the
+configured shared workspace. The server mints short-lived browser credentials for the
+shared `hackathon` user. `ASSISTANT_API_KEY` and `OPENAI_API_KEY` stay in the
+server environment. This starter admits visitors as the same shared user.
+Both `/api/credential` and `/api/chat` admit public visitors in this hackathon
+starter. Add your application authentication to both routes before deploying a
+private workspace.
+
+During local development, the existing harness runtime serves model requests
+through the browser's localhost tunnel. Keep a browser connected while a run
+is active. For deployment, add the application's HTTPS `/api/chat` endpoint to
+the harness's allowed endpoints.
+
+The harness runtimes are bundled as local package tarballs while their public
+packages are in preview. See [vendor/README.md](vendor/README.md) for their source
+revision, license, and reproduction command.

package/templates/cloud-harness/app/api/chat/route.ts

@@ -0,0 +1,14 @@
+import { openai } from "@ai-sdk/openai";
+import { convertToModelMessages, streamText, type UIMessage } from "ai";
+
+export const maxDuration = 30;
+
+export const POST = async (request: Request) => {
+  const { messages } = (await request.json()) as { messages: UIMessage[] };
+  const result = streamText({
+    abortSignal: request.signal,
+    model: openai.responses("gpt-6-luna"),
+    messages: await convertToModelMessages(messages),
+  });
+  return result.toUIMessageStreamResponse();
+};

package/templates/cloud-harness/app/api/credential/route.ts

@@ -0,0 +1,14 @@
+import { AssistantCloud } from "assistant-cloud";
+
+export const GET = async () => {
+  const cloud = new AssistantCloud({
+    baseUrl: process.env.NEXT_PUBLIC_ASSISTANT_BASE_URL!,
+    apiKey: process.env.ASSISTANT_API_KEY!,
+    workspaceId: process.env.NEXT_PUBLIC_ASSISTANT_WORKSPACE_ID!,
+    userId: "hackathon",
+  });
+  const credential = await cloud.auth.tokens.create();
+  return Response.json(credential, {
+    headers: { "Cache-Control": "no-store" },
+  });
+};

package/templates/cloud-harness/app/assistant.tsx

@@ -0,0 +1,83 @@
+"use client";
+
+import { AuiConfig, AuiProvider } from "@assistant-ui/react";
+import { HarnessCloudThreadList } from "@assistant-ui/react-harness-sdk";
+import { useSyncExternalStore } from "react";
+import { Thread } from "@/components/assistant-ui/elements/thread.aui";
+import { Button } from "@/components/ui/button";
+import { useCopyToClipboard } from "@/hooks/use-copy-to-clipboard";
+
+const credential = async () => {
+  const response = await fetch("/api/credential", { cache: "no-store" });
+  if (!response.ok) throw new Error(await response.text());
+  const { token } = (await response.json()) as { token: string };
+  return token;
+};
+
+const ShareThread = () => {
+  const { isCopied, copyToClipboard } = useCopyToClipboard();
+  return (
+    <Button
+      variant="outline"
+      size="sm"
+      onClick={() => copyToClipboard(window.location.href)}
+    >
+      {isCopied ? "Link copied" : "Share this chat"}
+    </Button>
+  );
+};
+
+const ConnectedAssistant = ({ threadId }: { threadId: string }) => {
+  const config = AuiConfig({
+    threads: HarnessCloudThreadList({
+      url: "/api/chat",
+      origin: process.env.NEXT_PUBLIC_ASSISTANT_HARNESS_URL!,
+      workspaceId: process.env.NEXT_PUBLIC_ASSISTANT_WORKSPACE_ID!,
+      credential,
+      threadId,
+      onThreadIdChange: (threadId) => {
+        window.location.hash = threadId ?? crypto.randomUUID();
+      },
+    }),
+  });
+  return (
+    <AuiProvider config={config}>
+      <div className="flex h-dvh flex-col">
+        <header className="flex items-center justify-between border-b px-6 py-3">
+          <div>
+            <h1 className="font-semibold">Shared chat</h1>
+            <p className="text-muted-foreground text-sm">
+              Open the same link in another browser to chat together.
+            </p>
+          </div>
+          <ShareThread />
+        </header>
+        <div className="min-h-0 flex-1">
+          <Thread />
+        </div>
+      </div>
+    </AuiProvider>
+  );
+};
+
+export const Assistant = () => {
+  const threadId = useSyncExternalStore(
+    subscribeToHash,
+    currentThreadId,
+    () => null,
+  );
+  if (threadId === null)
+    return (
+      <p className="p-6" role="status">
+        Connecting to your chat…
+      </p>
+    );
+  return <ConnectedAssistant threadId={threadId} />;
+};
+
+const currentThreadId = () => window.location.hash.slice(1) || "main";
+
+const subscribeToHash = (onChange: () => void) => {
+  window.addEventListener("hashchange", onChange);
+  return () => window.removeEventListener("hashchange", onChange);
+};