assistant-ui 0.0.106 → 0.0.108

package/dist/lib/utils/registry.js.map

@@ -1 +1 @@
-{"version":3,"file":"registry.js","names":["path","fs"],"sources":["../../../src/lib/utils/registry.ts"],"sourcesContent":["import * as fs from \"node:fs\";\nimport * as path from \"node:path\";\n\nconst REGISTRY_BASE_URL = \"https://r.assistant-ui.com\";\n\nexport function getComponentsJsonStyle(cwd: string): string | undefined {\n  try {\n    const configPath = path.join(cwd, \"components.json\");\n    const config = JSON.parse(fs.readFileSync(configPath, \"utf8\")) as {\n      style?: unknown;\n    };\n\n    return typeof config.style === \"string\" ? config.style : undefined;\n  } catch {\n    return undefined;\n  }\n}\n\nexport function resolveQuickStartRegistryUrl(style?: string): string {\n  // The shadcn CLI appends a literal /json segment to fetched URLs containing\n  // /chat/b/ unless they already end with it, so this item must use the direct\n  // tree URLs rather than the /styles/ template.\n  if (style === undefined || style.startsWith(\"base-\")) {\n    return `${REGISTRY_BASE_URL}/base/chat/b/ai-sdk-quick-start/json`;\n  }\n\n  return `${REGISTRY_BASE_URL}/chat/b/ai-sdk-quick-start/json`;\n}\n\nexport function resolveRegistryItemUrl(\n  component: string,\n  style?: string,\n): string {\n  const registryUrl = style?.startsWith(\"base-\")\n    ? `${REGISTRY_BASE_URL}/styles/${encodeURIComponent(style)}`\n    : REGISTRY_BASE_URL;\n\n  return `${registryUrl}/${encodeURIComponent(component)}.json`;\n}\n"],"mappings":";;;AAGA,MAAM,oBAAoB;AAE1B,SAAgB,uBAAuB,KAAiC;CACtE,IAAI;EACF,MAAM,aAAaA,OAAK,KAAK,KAAK,iBAAiB;EACnD,MAAM,SAAS,KAAK,MAAMC,KAAG,aAAa,YAAY,MAAM,CAAC;EAI7D,OAAO,OAAO,OAAO,UAAU,WAAW,OAAO,QAAQ,KAAA;CAC3D,QAAQ;EACN;CACF;AACF;AAEA,SAAgB,6BAA6B,OAAwB;CAInE,IAAI,UAAU,KAAA,KAAa,MAAM,WAAW,OAAO,GACjD,OAAO,GAAG,kBAAkB;CAG9B,OAAO,GAAG,kBAAkB;AAC9B;AAEA,SAAgB,uBACd,WACA,OACQ;CAKR,OAAO,GAJa,OAAO,WAAW,OAAO,IACzC,GAAG,kBAAkB,UAAU,mBAAmB,KAAK,MACvD,kBAEkB,GAAG,mBAAmB,SAAS,EAAE;AACzD"}
+{"version":3,"file":"registry.js","names":["path","fs"],"sources":["../../../src/lib/utils/registry.ts"],"sourcesContent":["import * as fs from \"node:fs\";\nimport * as path from \"node:path\";\n\nconst REGISTRY_BASE_URL = \"https://r.assistant-ui.com\";\n\nexport function getComponentsJsonStyle(cwd: string): string | undefined {\n  try {\n    const configPath = path.join(cwd, \"components.json\");\n    const config = JSON.parse(fs.readFileSync(configPath, \"utf8\")) as {\n      style?: unknown;\n    };\n\n    return typeof config.style === \"string\" ? config.style : undefined;\n  } catch {\n    return undefined;\n  }\n}\n\nexport function resolveQuickStartRegistryUrl(style?: string): string {\n  // The shadcn CLI appends a literal /json segment to fetched URLs containing\n  // /chat/b/ unless they already end with it, so this item must use the direct\n  // tree URLs rather than the /styles/ template.\n  if (style === undefined || style.startsWith(\"base-\")) {\n    return `${REGISTRY_BASE_URL}/base/chat/b/ai-sdk-quick-start/json`;\n  }\n\n  return `${REGISTRY_BASE_URL}/chat/b/ai-sdk-quick-start/json`;\n}\n\nexport function resolveRegistryItemUrl(\n  component: string,\n  style?: string,\n): string {\n  if (style === undefined) {\n    return `${REGISTRY_BASE_URL}/base/${encodeURIComponent(component)}.json`;\n  }\n\n  const registryUrl = style.startsWith(\"base-\")\n    ? `${REGISTRY_BASE_URL}/styles/${encodeURIComponent(style)}`\n    : REGISTRY_BASE_URL;\n\n  return `${registryUrl}/${encodeURIComponent(component)}.json`;\n}\n"],"mappings":";;;AAGA,MAAM,oBAAoB;AAE1B,SAAgB,uBAAuB,KAAiC;CACtE,IAAI;EACF,MAAM,aAAaA,OAAK,KAAK,KAAK,iBAAiB;EACnD,MAAM,SAAS,KAAK,MAAMC,KAAG,aAAa,YAAY,MAAM,CAAC;EAI7D,OAAO,OAAO,OAAO,UAAU,WAAW,OAAO,QAAQ,KAAA;CAC3D,QAAQ;EACN;CACF;AACF;AAEA,SAAgB,6BAA6B,OAAwB;CAInE,IAAI,UAAU,KAAA,KAAa,MAAM,WAAW,OAAO,GACjD,OAAO,GAAG,kBAAkB;CAG9B,OAAO,GAAG,kBAAkB;AAC9B;AAEA,SAAgB,uBACd,WACA,OACQ;CACR,IAAI,UAAU,KAAA,GACZ,OAAO,GAAG,kBAAkB,QAAQ,mBAAmB,SAAS,EAAE;CAOpE,OAAO,GAJa,MAAM,WAAW,OAAO,IACxC,GAAG,kBAAkB,UAAU,mBAAmB,KAAK,MACvD,kBAEkB,GAAG,mBAAmB,SAAS,EAAE;AACzD"}

package/dist/lib/utils/workspace.d.ts

@@ -0,0 +1,6 @@
+//#region src/lib/utils/workspace.d.ts
+declare function resolveRealPath(inputPath: string): string;
+declare function findWorkspaceRoot(cwd: string): string | null;
+//#endregion
+export { findWorkspaceRoot, resolveRealPath };
+//# sourceMappingURL=workspace.d.ts.map

package/dist/lib/utils/workspace.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace.d.ts","names":[],"sources":["../../../src/lib/utils/workspace.ts"],"mappings":";iBAGgB,gBAAgB;iBAUhB,kBAAkB"}

package/dist/lib/utils/workspace.js

@@ -0,0 +1,28 @@
+import * as fs$1 from "node:fs";
+import * as path$1 from "node:path";
+//#region src/lib/utils/workspace.ts
+function resolveRealPath(inputPath) {
+	const resolved = path$1.resolve(inputPath);
+	try {
+		return fs$1.realpathSync(resolved);
+	} catch {
+		return resolved;
+	}
+}
+function findWorkspaceRoot(cwd) {
+	let dir = path$1.resolve(cwd);
+	const root = path$1.parse(dir).root;
+	while (true) {
+		if (fs$1.existsSync(path$1.join(dir, "pnpm-workspace.yaml"))) return dir;
+		const pkgPath = path$1.join(dir, "package.json");
+		if (fs$1.existsSync(pkgPath)) try {
+			if (JSON.parse(fs$1.readFileSync(pkgPath, "utf8")).workspaces) return dir;
+		} catch {}
+		if (dir === root) return null;
+		dir = path$1.dirname(dir);
+	}
+}
+//#endregion
+export { findWorkspaceRoot, resolveRealPath };
+
+//# sourceMappingURL=workspace.js.map

package/dist/lib/utils/workspace.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"workspace.js","names":["path","fs"],"sources":["../../../src/lib/utils/workspace.ts"],"sourcesContent":["import * as fs from \"node:fs\";\nimport * as path from \"node:path\";\n\nexport function resolveRealPath(inputPath: string): string {\n  const resolved = path.resolve(inputPath);\n\n  try {\n    return fs.realpathSync(resolved);\n  } catch {\n    return resolved;\n  }\n}\n\nexport function findWorkspaceRoot(cwd: string): string | null {\n  let dir = path.resolve(cwd);\n  const root = path.parse(dir).root;\n\n  while (true) {\n    if (fs.existsSync(path.join(dir, \"pnpm-workspace.yaml\"))) return dir;\n\n    const pkgPath = path.join(dir, \"package.json\");\n    if (fs.existsSync(pkgPath)) {\n      try {\n        const pkg = JSON.parse(fs.readFileSync(pkgPath, \"utf8\"));\n        if (pkg.workspaces) return dir;\n      } catch {\n        // Malformed package manifests are not workspace markers.\n      }\n    }\n\n    if (dir === root) return null;\n    dir = path.dirname(dir);\n  }\n}\n"],"mappings":";;;AAGA,SAAgB,gBAAgB,WAA2B;CACzD,MAAM,WAAWA,OAAK,QAAQ,SAAS;CAEvC,IAAI;EACF,OAAOC,KAAG,aAAa,QAAQ;CACjC,QAAQ;EACN,OAAO;CACT;AACF;AAEA,SAAgB,kBAAkB,KAA4B;CAC5D,IAAI,MAAMD,OAAK,QAAQ,GAAG;CAC1B,MAAM,OAAOA,OAAK,MAAM,GAAG,CAAC,CAAC;CAE7B,OAAO,MAAM;EACX,IAAIC,KAAG,WAAWD,OAAK,KAAK,KAAK,qBAAqB,CAAC,GAAG,OAAO;EAEjE,MAAM,UAAUA,OAAK,KAAK,KAAK,cAAc;EAC7C,IAAIC,KAAG,WAAW,OAAO,GACvB,IAAI;GAEF,IADY,KAAK,MAAMA,KAAG,aAAa,SAAS,MAAM,CAChD,CAAC,CAAC,YAAY,OAAO;EAC7B,QAAQ,CAER;EAGF,IAAI,QAAQ,MAAM,OAAO;EACzB,MAAMD,OAAK,QAAQ,GAAG;CACxB;AACF"}

package/dist/run.d.ts

@@ -0,0 +1,5 @@
+//#region src/run.d.ts
+declare function runCli(): Promise<void>;
+//#endregion
+export { runCli };
+//# sourceMappingURL=run.d.ts.map

package/dist/run.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"run.d.ts","names":[],"sources":["../src/run.ts"],"mappings":";iBAEsB,UAAU"}

package/dist/run.js

@@ -0,0 +1,9 @@
+import { buildProgram } from "./program.js";
+//#region src/run.ts
+async function runCli() {
+	await buildProgram().parseAsync();
+}
+//#endregion
+export { runCli };
+
+//# sourceMappingURL=run.js.map

package/dist/run.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run.js","names":[],"sources":["../src/run.ts"],"sourcesContent":["import { buildProgram } from \"./program\";\n\nexport async function runCli(): Promise<void> {\n  await buildProgram().parseAsync();\n}\n"],"mappings":";;AAEA,eAAsB,SAAwB;CAC5C,MAAM,aAAa,CAAC,CAAC,WAAW;AAClC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-ui",
-  "version": "0.0.106",
+  "version": "0.0.108",
   "description": "CLI for assistant-ui",
   "keywords": [
     "cli",
@@ -33,9 +33,9 @@
     "cross-spawn": "^7.0.6",
     "debug": "^4.4.3",
     "detect-package-manager": "^3.0.2",
-    "giget": "^3.3.0",
+    "giget": "^3.3.1",
     "glob": "^13.0.6",
-    "jscodeshift": "^17.3.0",
+    "jscodeshift": "^17.4.0",
     "jsonc-parser": "^3.3.1",
     "semver": "^7.8.5",
     "@assistant-ui/agent-launcher": "0.1.9"
@@ -49,7 +49,7 @@
     "@types/semver": "^7.7.1",
     "@vitest/coverage-v8": "^4.1.10",
     "vitest": "^4.1.10",
-    "@assistant-ui/x-buildutils": "0.0.18"
+    "@assistant-ui/x-buildutils": "0.0.20"
   },
   "publishConfig": {
     "access": "public",

package/plugin/skills/assistant-ui/SKILL.md

@@ -24,20 +24,39 @@ This initializes shadcn and installs the default assistant-ui chat components.
 Install components via the shadcn registry:
 
 ```bash
-npx shadcn@latest add "https://r.assistant-ui.com/chat/b/ai-sdk-quick-start/json"
+npx shadcn@latest add "https://r.assistant-ui.com/base/chat/b/ai-sdk-quick-start/json"
 ```
 
+For Radix-styled projects (`components.json` style not starting with `base-`), use `https://r.assistant-ui.com/chat/b/ai-sdk-quick-start/json` instead.
+
 Available component presets:
 
 | Preset | Registry URL |
 |--------|-------------|
-| AI SDK Quick Start | `https://r.assistant-ui.com/chat/b/ai-sdk-quick-start/json` |
+| AI SDK Quick Start | `https://r.assistant-ui.com/base/chat/b/ai-sdk-quick-start/json` |
+
+You can also add individual assistant-ui components with the assistant-ui CLI, which picks the right flavor from `components.json`:
+
+```bash
+npx assistant-ui add thread
+npx assistant-ui add markdown-text
+```
+
+To use the shadcn CLI directly instead, add the assistant-ui registry to `components.json`:
+
+```json
+{
+  "registries": {
+    "@assistant-ui": "https://r.assistant-ui.com/styles/{style}/{name}.json"
+  }
+}
+```
 
-You can also add individual assistant-ui shadcn components:
+Then add components with the namespaced form:
 
 ```bash
-npx shadcn@latest add assistant-ui/thread
-npx shadcn@latest add assistant-ui/markdown-text
+npx shadcn@latest add @assistant-ui/thread
+npx shadcn@latest add @assistant-ui/markdown-text
 ```
 
 ## Step 3: Runtime Setup

package/src/codemods/v0-15/__tests__/aui-accessor-calls-to-properties.test.ts

@@ -0,0 +1,109 @@
+import { describe, it, expect } from "vitest";
+import jscodeshift, { type API } from "jscodeshift";
+import transform from "../aui-accessor-calls-to-properties";
+
+const j = jscodeshift.withParser("tsx");
+
+function applyTransform(source: string): string | null {
+  const fileInfo = {
+    path: "test.tsx",
+    source,
+  };
+
+  const api: API = {
+    jscodeshift: j,
+    j,
+    stats: () => {},
+    report: () => {},
+  };
+
+  return transform(fileInfo, api, {});
+}
+
+describe("aui-accessor-calls-to-properties", () => {
+  it("rewrites nullary accessor calls on a useAui variable", () => {
+    const input = `
+const client = useAui();
+client.thread().cancelRun();
+const state = client.composer().getState();
+`;
+    const expected = `
+const client = useAui();
+client.thread.cancelRun();
+const state = client.composer.getState();
+`;
+    expect(applyTransform(input)?.trim()).toBe(expected.trim());
+  });
+
+  it("rewrites only the accessor call in chained expressions", () => {
+    const input = `
+const aui = useAui();
+const part = aui.message().part({ index: 0 });
+aui.message().composer().send();
+`;
+    const expected = `
+const aui = useAui();
+const part = aui.message.part({ index: 0 });
+aui.message.composer().send();
+`;
+    expect(applyTransform(input)?.trim()).toBe(expected.trim());
+  });
+
+  it("rewrites identifiers named aui without a declaration", () => {
+    const input = `
+const Derived = {
+  get: (aui) => aui.thread().message({ index: 0 }),
+};
+`;
+    const expected = `
+const Derived = {
+  get: (aui) => aui.thread.message({ index: 0 }),
+};
+`;
+    expect(applyTransform(input)?.trim()).toBe(expected.trim());
+  });
+
+  it("rewrites parameters typed AssistantClient", () => {
+    const input = `
+const getItem = (client: AssistantClient) => client.threadListItem().getState();
+`;
+    const expected = `
+const getItem = (client: AssistantClient) => client.threadListItem.getState();
+`;
+    expect(applyTransform(input)?.trim()).toBe(expected.trim());
+  });
+
+  it("leaves calls with arguments untouched", () => {
+    const input = `
+const aui = useAui();
+const t = aui.threads().thread({ id: "t1" });
+`;
+    const expected = `
+const aui = useAui();
+const t = aui.threads.thread({ id: "t1" });
+`;
+    expect(applyTransform(input)?.trim()).toBe(expected.trim());
+  });
+
+  it("does not rewrite unknown receivers", () => {
+    const input = `
+toolkit.tools();
+message.composer().send();
+ref.current.thread().getState();
+`;
+    expect(applyTransform(input)).toBeNull();
+  });
+
+  it("does not rewrite non-scope member calls on aui", () => {
+    const input = `
+const aui = useAui();
+aui.subscribe(() => {});
+aui.on("thread.updated", () => {});
+`;
+    expect(applyTransform(input)).toBeNull();
+  });
+
+  it("returns null when nothing changes", () => {
+    expect(applyTransform(`const x = 1;`)).toBeNull();
+  });
+});

package/src/codemods/v0-15/aui-accessor-calls-to-properties.ts

@@ -0,0 +1,83 @@
+import { createTransformer } from "../utils/createTransformer";
+
+// Nullary scope accessors that became properties in v0.15. Parameterized
+// lookups (e.g. `aui.thread.message({ id })`) stay as real calls.
+const NULLARY_SCOPES = new Set([
+  "threads",
+  "threadListItem",
+  "thread",
+  "message",
+  "part",
+  "composer",
+  "attachment",
+  "modelContext",
+  "suggestions",
+  "suggestion",
+  "chainOfThought",
+  "queueItem",
+  "tools",
+  "dataRenderers",
+  "interactables",
+  "unstable_interactables",
+  "mcp",
+  "mcpServer",
+  "span",
+]);
+
+const AUI_HOOKS = new Set(["useAui", "useAssistantApi"]);
+
+const auiAccessorCallsToProperties = createTransformer(
+  ({ j, root, markAsChanged }) => {
+    const auiNames = new Set(["aui"]);
+
+    root.find(j.VariableDeclarator).forEach((path: any) => {
+      const { id, init } = path.value;
+      if (
+        j.Identifier.check(id) &&
+        init &&
+        j.CallExpression.check(init) &&
+        j.Identifier.check(init.callee) &&
+        AUI_HOOKS.has(init.callee.name)
+      ) {
+        auiNames.add(id.name);
+      }
+    });
+
+    const collectParam = (param: any) => {
+      const annotation = param?.typeAnnotation?.typeAnnotation;
+      if (
+        j.Identifier.check(param) &&
+        annotation &&
+        j.TSTypeReference.check(annotation) &&
+        j.Identifier.check(annotation.typeName) &&
+        annotation.typeName.name === "AssistantClient"
+      ) {
+        auiNames.add(param.name);
+      }
+    };
+    for (const fnType of [
+      j.FunctionDeclaration,
+      j.FunctionExpression,
+      j.ArrowFunctionExpression,
+    ] as const) {
+      root.find(fnType as typeof j.FunctionDeclaration).forEach((path: any) => {
+        path.value.params.forEach(collectParam);
+      });
+    }
+
+    root.find(j.CallExpression).forEach((path: any) => {
+      const node = path.value;
+      if (node.arguments.length !== 0) return;
+      const callee = node.callee;
+      if (!j.MemberExpression.check(callee) || callee.computed) return;
+      if (!j.Identifier.check(callee.property)) return;
+      if (!NULLARY_SCOPES.has(callee.property.name)) return;
+      if (!j.Identifier.check(callee.object)) return;
+      if (!auiNames.has(callee.object.name)) return;
+      j(path).replaceWith(callee);
+      markAsChanged();
+    });
+  },
+);
+
+export default auiAccessorCallsToProperties;

package/src/commands/add.ts

@@ -67,7 +67,7 @@ export const add = new Command()
     // Check if project is initialized
     if (!hasConfig(opts.cwd)) {
       logger.warn(
-        "It looks like you haven't initialized your project yet. Run 'assistant-ui init' first.",
+        "It looks like you haven't initialized your project yet — defaulting to Base UI flavored components. Run 'assistant-ui init' first for a configured setup.",
       );
       logger.break();
     }

package/src/commands/create.ts

@@ -12,6 +12,7 @@ import {
   resolvePackageManagerForCwd,
   scaffoldProject,
   transformProject,
+  type TransformResult,
 } from "../lib/create-project";
 import { runSpawn, SpawnExitError } from "../lib/run-spawn";
 import {
@@ -165,7 +166,7 @@ export const PROJECT_METADATA: ProjectMetadata[] = [
     description: "Realtime voice with ElevenLabs",
     category: "example",
     path: "examples/with-elevenlabs-conversational",
-    hasLocalComponents: true,
+    hasLocalComponents: false,
   },
   {
     name: "with-elevenlabs-scribe",
@@ -181,7 +182,7 @@ export const PROJECT_METADATA: ProjectMetadata[] = [
     description: "Realtime voice with LiveKit",
     category: "example",
     path: "examples/with-livekit",
-    hasLocalComponents: true,
+    hasLocalComponents: false,
   },
   {
     name: "with-expo",
@@ -197,7 +198,7 @@ export const PROJECT_METADATA: ProjectMetadata[] = [
     description: "AI-driven interactive UI components",
     category: "example",
     path: "examples/with-interactables",
-    hasLocalComponents: true,
+    hasLocalComponents: false,
   },
   {
     name: "with-external-store",
@@ -609,6 +610,7 @@ export const create = new Command()
           ? `Copying project from local source: ${localSourceRoot}`
           : "Downloading project...",
       );
+      let transformResult: TransformResult;
       try {
         const source = localSourceRoot
           ? { kind: "local" as const, rootDir: localSourceRoot }
@@ -632,7 +634,7 @@ export const create = new Command()
         }
 
         // 5. Run transform pipeline
-        await transformProject(absoluteProjectDir, {
+        transformResult = await transformProject(absoluteProjectDir, {
           hasLocalComponents: project.hasLocalComponents,
           skipInstall: opts.skipInstall,
           packageManager: pm,
@@ -657,6 +659,16 @@ export const create = new Command()
         throw err;
       }
 
+      if (transformResult.registryInstallFailure) {
+        process.removeListener("exit", cleanupOnExit);
+        logger.break();
+        logger.error("Project created with missing components.");
+        logger.info("Retry the component install with:");
+        logger.info(`  cd ${resolvedProjectDirectory}`);
+        logger.info(`  ${transformResult.registryInstallFailure.retryCommand}`);
+        process.exit(1);
+      }
+
       // 6. Apply preset if provided
       if (scaffoldSelector.preset) {
         const presetUrl = resolvePresetUrl(scaffoldSelector.preset);

package/src/commands/doctor.ts

@@ -3,6 +3,7 @@ import * as fs from "node:fs";
 import * as path from "node:path";
 import chalk from "chalk";
 import { compare, valid } from "semver";
+import { findWorkspaceRoot, resolveRealPath } from "../lib/utils/workspace";
 
 const ASSISTANT_UI_PACKAGE_NAMES = new Set([
   "assistant-stream",
@@ -39,13 +40,7 @@ function processPackageDir(
   results: DiscoveredPackage[],
   visited: ProcessedDir,
 ): void {
-  const real = (() => {
-    try {
-      return fs.realpathSync(pkgDir);
-    } catch {
-      return pkgDir;
-    }
-  })();
+  const real = resolveRealPath(pkgDir);
   if (visited.set.has(real)) return;
   visited.set.add(real);
 
@@ -183,14 +178,23 @@ function walkPnpmStore(
 }
 
 // Discover every installation of an assistant-ui-family package reachable
-// from `cwd`. Recurses into nested node_modules so transitive copies
-// (the real source of duplicate-version bugs) are not missed, and scans the
-// pnpm `.pnpm` virtual store, which the hoisted walk cannot reach.
+// from `cwd`. Node resolves packages through ancestor node_modules directories,
+// so inspect each level to include workspace-hoisted installs. Nested installs
+// and pnpm virtual stores are scanned to catch duplicate transitive copies.
 export function discoverInstalledPackages(cwd: string): DiscoveredPackage[] {
   const results: DiscoveredPackage[] = [];
   const visited: ProcessedDir = { set: new Set() };
-  walkNodeModulesAt(cwd, results, visited);
-  walkPnpmStore(cwd, results, visited);
+  let dir = resolveRealPath(cwd);
+  const scanRoot = findWorkspaceRoot(dir) ?? dir;
+
+  while (true) {
+    walkNodeModulesAt(dir, results, visited);
+    walkPnpmStore(dir, results, visited);
+
+    if (dir === scanRoot) break;
+    dir = path.dirname(dir);
+  }
+
   return results;
 }
 
@@ -369,7 +373,7 @@ export const doctor = new Command()
   )
   .option("--no-network", "Skip the npm registry check for latest versions.")
   .action(async (opts: { cwd: string; network: boolean }) => {
-    const cwd = path.resolve(opts.cwd);
+    const cwd = resolveRealPath(opts.cwd);
     const packageJsonPath = path.join(cwd, "package.json");
 
     if (!fs.existsSync(packageJsonPath)) {

package/src/commands/info.ts

@@ -6,6 +6,9 @@ import { spawnSync } from "node:child_process";
 import chalk from "chalk";
 import { detect } from "detect-package-manager";
 import { satisfies } from "semver";
+import { findWorkspaceRoot, resolveRealPath } from "../lib/utils/workspace";
+
+export { findWorkspaceRoot };
 
 const ASSISTANT_UI_PACKAGES = [
   // Distribution
@@ -89,25 +92,6 @@ function getInstalledVersion(pkg: string, cwd: string): string | null {
   return null;
 }
 
-export function findWorkspaceRoot(cwd: string): string | null {
-  let dir = cwd;
-  const root = path.parse(dir).root;
-  while (true) {
-    if (fs.existsSync(path.join(dir, "pnpm-workspace.yaml"))) return dir;
-    const pkgPath = path.join(dir, "package.json");
-    if (fs.existsSync(pkgPath)) {
-      try {
-        const pkg = JSON.parse(fs.readFileSync(pkgPath, "utf8"));
-        if (pkg.workspaces) return dir;
-      } catch {
-        // ignore
-      }
-    }
-    if (dir === root) return null;
-    dir = path.dirname(dir);
-  }
-}
-
 function readProjectDeps(
   projectPkg: Record<string, unknown>,
 ): Record<string, string> {
@@ -197,6 +181,19 @@ function getOsInfo(): string {
   }
 }
 
+function getCliVersion(): string {
+  try {
+    const packageJson = JSON.parse(
+      fs.readFileSync(new URL("../../package.json", import.meta.url), "utf8"),
+    ) as { version?: unknown };
+    return typeof packageJson.version === "string"
+      ? packageJson.version
+      : "unknown";
+  } catch {
+    return "unknown";
+  }
+}
+
 async function getPackageManagerInfo(
   cwd: string,
 ): Promise<{ name: string; version: string }> {
@@ -230,6 +227,7 @@ interface PackageInfo {
 }
 
 interface InfoData {
+  cliVersion: string;
   os: string;
   node: string;
   pm: { name: string; version: string };
@@ -317,6 +315,7 @@ async function collectInfo(
   const warnings = collectWarnings(packages, cwd, projectPkg);
 
   return {
+    cliVersion: getCliVersion(),
     os: getOsInfo(),
     node: process.version,
     pm,
@@ -344,6 +343,7 @@ function renderPlain(data: InfoData): string[] {
   const lines: string[] = [];
 
   lines.push("Environment:");
+  lines.push(`  assistant-ui CLI: ${data.cliVersion}`);
   lines.push(`  OS:               ${data.os}`);
   lines.push(`  Node.js:          ${data.node}`);
   lines.push(`  Package Manager:  ${data.pm.name} ${data.pm.version}`);
@@ -370,6 +370,7 @@ function renderColored(data: InfoData): string[] {
   const lines: string[] = [];
 
   lines.push(chalk.bold("Environment:"));
+  lines.push(`  assistant-ui CLI: ${data.cliVersion}`);
   lines.push(`  OS:               ${data.os}`);
   lines.push(`  Node.js:          ${data.node}`);
   lines.push(`  Package Manager:  ${data.pm.name} ${data.pm.version}`);
@@ -415,7 +416,7 @@ export const info = new Command()
     process.cwd(),
   )
   .action(async (opts) => {
-    const cwd = path.resolve(opts.cwd);
+    const cwd = resolveRealPath(opts.cwd);
     const packageJsonPath = path.join(cwd, "package.json");
 
     if (!fs.existsSync(packageJsonPath)) {

package/src/commands/upgrade.ts

@@ -48,9 +48,9 @@ export const upgradeCommand = addTransformOptions(
   new Command()
     .command("upgrade")
     .description("Upgrade ai package dependencies and apply codemods"),
-).action((options: TransformOptions) => {
+).action(async (options: TransformOptions) => {
   try {
-    upgrade(options);
+    await upgrade(options);
   } catch (err) {
     const errorMessage = err instanceof Error ? err.message : String(err);
     const errorStack = err instanceof Error ? err.stack : undefined;

package/src/index.ts

@@ -1,12 +1,11 @@
 #!/usr/bin/env node
 
-import { buildProgram } from "./program";
+import { runCli } from "./run";
 
 process.on("SIGINT", () => process.exit(0));
 process.on("SIGTERM", () => process.exit(0));
 
-function main() {
-  buildProgram().parse();
-}
-
-main();
+void runCli().catch((error: unknown) => {
+  console.error(error);
+  process.exitCode = 1;
+});