assistant-ui 0.0.107 → 0.0.108

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

@@ -0,0 +1,5 @@
+//#region src/codemods/v0-15/aui-accessor-calls-to-properties.d.ts
+declare const auiAccessorCallsToProperties: (fileInfo: import("jscodeshift/src/core").FileInfo, api: import("jscodeshift/src/core").API, options: any) => string | null;
+//#endregion
+export { auiAccessorCallsToProperties as default };
+//# sourceMappingURL=aui-accessor-calls-to-properties.d.ts.map

package/dist/codemods/v0-15/aui-accessor-calls-to-properties.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"aui-accessor-calls-to-properties.d.ts","names":[],"sources":["../../../src/codemods/v0-15/aui-accessor-calls-to-properties.ts"],"mappings":";cA4BM,+BAA4B,yCAAA,UAAA,oCAAA,KAAA"}

package/dist/codemods/v0-15/aui-accessor-calls-to-properties.js

@@ -0,0 +1,58 @@
+import { createTransformer } from "../utils/createTransformer.js";
+//#region src/codemods/v0-15/aui-accessor-calls-to-properties.ts
+const NULLARY_SCOPES = /* @__PURE__ */ 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 = /* @__PURE__ */ new Set(["useAui", "useAssistantApi"]);
+const auiAccessorCallsToProperties = createTransformer(({ j, root, markAsChanged }) => {
+	const auiNames = /* @__PURE__ */ new Set(["aui"]);
+	root.find(j.VariableDeclarator).forEach((path) => {
+		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) => {
+		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
+	]) root.find(fnType).forEach((path) => {
+		path.value.params.forEach(collectParam);
+	});
+	root.find(j.CallExpression).forEach((path) => {
+		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();
+	});
+});
+//#endregion
+export { auiAccessorCallsToProperties as default };
+
+//# sourceMappingURL=aui-accessor-calls-to-properties.js.map

package/dist/codemods/v0-15/aui-accessor-calls-to-properties.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aui-accessor-calls-to-properties.js","names":[],"sources":["../../../src/codemods/v0-15/aui-accessor-calls-to-properties.ts"],"sourcesContent":["import { createTransformer } from \"../utils/createTransformer\";\n\n// Nullary scope accessors that became properties in v0.15. Parameterized\n// lookups (e.g. `aui.thread.message({ id })`) stay as real calls.\nconst NULLARY_SCOPES = new Set([\n  \"threads\",\n  \"threadListItem\",\n  \"thread\",\n  \"message\",\n  \"part\",\n  \"composer\",\n  \"attachment\",\n  \"modelContext\",\n  \"suggestions\",\n  \"suggestion\",\n  \"chainOfThought\",\n  \"queueItem\",\n  \"tools\",\n  \"dataRenderers\",\n  \"interactables\",\n  \"unstable_interactables\",\n  \"mcp\",\n  \"mcpServer\",\n  \"span\",\n]);\n\nconst AUI_HOOKS = new Set([\"useAui\", \"useAssistantApi\"]);\n\nconst auiAccessorCallsToProperties = createTransformer(\n  ({ j, root, markAsChanged }) => {\n    const auiNames = new Set([\"aui\"]);\n\n    root.find(j.VariableDeclarator).forEach((path: any) => {\n      const { id, init } = path.value;\n      if (\n        j.Identifier.check(id) &&\n        init &&\n        j.CallExpression.check(init) &&\n        j.Identifier.check(init.callee) &&\n        AUI_HOOKS.has(init.callee.name)\n      ) {\n        auiNames.add(id.name);\n      }\n    });\n\n    const collectParam = (param: any) => {\n      const annotation = param?.typeAnnotation?.typeAnnotation;\n      if (\n        j.Identifier.check(param) &&\n        annotation &&\n        j.TSTypeReference.check(annotation) &&\n        j.Identifier.check(annotation.typeName) &&\n        annotation.typeName.name === \"AssistantClient\"\n      ) {\n        auiNames.add(param.name);\n      }\n    };\n    for (const fnType of [\n      j.FunctionDeclaration,\n      j.FunctionExpression,\n      j.ArrowFunctionExpression,\n    ] as const) {\n      root.find(fnType as typeof j.FunctionDeclaration).forEach((path: any) => {\n        path.value.params.forEach(collectParam);\n      });\n    }\n\n    root.find(j.CallExpression).forEach((path: any) => {\n      const node = path.value;\n      if (node.arguments.length !== 0) return;\n      const callee = node.callee;\n      if (!j.MemberExpression.check(callee) || callee.computed) return;\n      if (!j.Identifier.check(callee.property)) return;\n      if (!NULLARY_SCOPES.has(callee.property.name)) return;\n      if (!j.Identifier.check(callee.object)) return;\n      if (!auiNames.has(callee.object.name)) return;\n      j(path).replaceWith(callee);\n      markAsChanged();\n    });\n  },\n);\n\nexport default auiAccessorCallsToProperties;\n"],"mappings":";;AAIA,MAAM,iCAAiB,IAAI,IAAI;CAC7B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;AACF,CAAC;AAED,MAAM,4BAAY,IAAI,IAAI,CAAC,UAAU,iBAAiB,CAAC;AAEvD,MAAM,+BAA+B,mBAClC,EAAE,GAAG,MAAM,oBAAoB;CAC9B,MAAM,2BAAW,IAAI,IAAI,CAAC,KAAK,CAAC;CAEhC,KAAK,KAAK,EAAE,kBAAkB,CAAC,CAAC,SAAS,SAAc;EACrD,MAAM,EAAE,IAAI,SAAS,KAAK;EAC1B,IACE,EAAE,WAAW,MAAM,EAAE,KACrB,QACA,EAAE,eAAe,MAAM,IAAI,KAC3B,EAAE,WAAW,MAAM,KAAK,MAAM,KAC9B,UAAU,IAAI,KAAK,OAAO,IAAI,GAE9B,SAAS,IAAI,GAAG,IAAI;CAExB,CAAC;CAED,MAAM,gBAAgB,UAAe;EACnC,MAAM,aAAa,OAAO,gBAAgB;EAC1C,IACE,EAAE,WAAW,MAAM,KAAK,KACxB,cACA,EAAE,gBAAgB,MAAM,UAAU,KAClC,EAAE,WAAW,MAAM,WAAW,QAAQ,KACtC,WAAW,SAAS,SAAS,mBAE7B,SAAS,IAAI,MAAM,IAAI;CAE3B;CACA,KAAK,MAAM,UAAU;EACnB,EAAE;EACF,EAAE;EACF,EAAE;CACJ,GACE,KAAK,KAAK,MAAsC,CAAC,CAAC,SAAS,SAAc;EACvE,KAAK,MAAM,OAAO,QAAQ,YAAY;CACxC,CAAC;CAGH,KAAK,KAAK,EAAE,cAAc,CAAC,CAAC,SAAS,SAAc;EACjD,MAAM,OAAO,KAAK;EAClB,IAAI,KAAK,UAAU,WAAW,GAAG;EACjC,MAAM,SAAS,KAAK;EACpB,IAAI,CAAC,EAAE,iBAAiB,MAAM,MAAM,KAAK,OAAO,UAAU;EAC1D,IAAI,CAAC,EAAE,WAAW,MAAM,OAAO,QAAQ,GAAG;EAC1C,IAAI,CAAC,eAAe,IAAI,OAAO,SAAS,IAAI,GAAG;EAC/C,IAAI,CAAC,EAAE,WAAW,MAAM,OAAO,MAAM,GAAG;EACxC,IAAI,CAAC,SAAS,IAAI,OAAO,OAAO,IAAI,GAAG;EACvC,EAAE,IAAI,CAAC,CAAC,YAAY,MAAM;EAC1B,cAAc;CAChB,CAAC;AACH,CACF"}

package/dist/commands/upgrade.js

@@ -19,9 +19,9 @@ const codemodCommand = addTransformOptions(new Command().name("codemod").descrip
 		process.exit(1);
 	}
 });
-const upgradeCommand = addTransformOptions(new Command().command("upgrade").description("Upgrade ai package dependencies and apply codemods")).action((options) => {
+const upgradeCommand = addTransformOptions(new Command().command("upgrade").description("Upgrade ai package dependencies and apply codemods")).action(async (options) => {
 	try {
-		upgrade(options);
+		await upgrade(options);
 	} catch (err) {
 		const errorMessage = err instanceof Error ? err.message : String(err);
 		const errorStack = err instanceof Error ? err.stack : void 0;

package/dist/commands/upgrade.js.map

@@ -1 +1 @@
-{"version":3,"file":"upgrade.js","names":[],"sources":["../../src/commands/upgrade.ts"],"sourcesContent":["import { Command } from \"commander\";\nimport { transform } from \"../lib/transform\";\nimport { upgrade } from \"../lib/upgrade\";\nimport debug from \"debug\";\n\nexport interface TransformOptions {\n  dry?: boolean;\n  print?: boolean;\n  verbose?: boolean;\n  jscodeshift?: string;\n}\n\nconst error = debug(\"codemod:error\");\ndebug.enable(\"codemod:*\");\n\nconst addTransformOptions = (command: Command): Command => {\n  return command\n    .option(\"-d, --dry\", \"Dry run (no changes are made to files)\")\n    .option(\"-p, --print\", \"Print transformed files to stdout\")\n    .option(\"--verbose\", \"Show more information about the transform process\")\n    .option(\n      \"-j, --jscodeshift <options>\",\n      \"Pass options directly to jscodeshift\",\n    );\n};\n\nexport const codemodCommand = addTransformOptions(\n  new Command()\n    .name(\"codemod\")\n    .description(\"CLI tool for running codemods\")\n    .argument(\"<codemod>\", \"Codemod to run (e.g., rewrite-framework-imports)\")\n    .argument(\"<source>\", \"Path to source files or directory to transform\"),\n).action((codemod, source, options: TransformOptions) => {\n  try {\n    transform(codemod, source, options);\n  } catch (err) {\n    const errorMessage = err instanceof Error ? err.message : String(err);\n    const errorStack = err instanceof Error ? err.stack : undefined;\n    error(`Error transforming: ${errorMessage}`);\n    if (errorStack) {\n      error(errorStack);\n    }\n    process.exit(1);\n  }\n});\n\nexport const upgradeCommand = addTransformOptions(\n  new Command()\n    .command(\"upgrade\")\n    .description(\"Upgrade ai package dependencies and apply codemods\"),\n).action((options: TransformOptions) => {\n  try {\n    upgrade(options);\n  } catch (err) {\n    const errorMessage = err instanceof Error ? err.message : String(err);\n    const errorStack = err instanceof Error ? err.stack : undefined;\n    error(`Error upgrading: ${errorMessage}`);\n    if (errorStack) {\n      error(errorStack);\n    }\n    process.exit(1);\n  }\n});\n"],"mappings":";;;;;AAYA,MAAM,QAAQ,MAAM,eAAe;AACnC,MAAM,OAAO,WAAW;AAExB,MAAM,uBAAuB,YAA8B;CACzD,OAAO,QACJ,OAAO,aAAa,wCAAwC,CAAC,CAC7D,OAAO,eAAe,mCAAmC,CAAC,CAC1D,OAAO,aAAa,mDAAmD,CAAC,CACxE,OACC,+BACA,sCACF;AACJ;AAEA,MAAa,iBAAiB,oBAC5B,IAAI,QAAQ,CAAC,CACV,KAAK,SAAS,CAAC,CACf,YAAY,+BAA+B,CAAC,CAC5C,SAAS,aAAa,kDAAkD,CAAC,CACzE,SAAS,YAAY,gDAAgD,CAC1E,CAAC,CAAC,QAAQ,SAAS,QAAQ,YAA8B;CACvD,IAAI;EACF,UAAU,SAAS,QAAQ,OAAO;CACpC,SAAS,KAAK;EACZ,MAAM,eAAe,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;EACpE,MAAM,aAAa,eAAe,QAAQ,IAAI,QAAQ,KAAA;EACtD,MAAM,uBAAuB,cAAc;EAC3C,IAAI,YACF,MAAM,UAAU;EAElB,QAAQ,KAAK,CAAC;CAChB;AACF,CAAC;AAED,MAAa,iBAAiB,oBAC5B,IAAI,QAAQ,CAAC,CACV,QAAQ,SAAS,CAAC,CAClB,YAAY,oDAAoD,CACrE,CAAC,CAAC,QAAQ,YAA8B;CACtC,IAAI;EACF,QAAQ,OAAO;CACjB,SAAS,KAAK;EACZ,MAAM,eAAe,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;EACpE,MAAM,aAAa,eAAe,QAAQ,IAAI,QAAQ,KAAA;EACtD,MAAM,oBAAoB,cAAc;EACxC,IAAI,YACF,MAAM,UAAU;EAElB,QAAQ,KAAK,CAAC;CAChB;AACF,CAAC"}
+{"version":3,"file":"upgrade.js","names":[],"sources":["../../src/commands/upgrade.ts"],"sourcesContent":["import { Command } from \"commander\";\nimport { transform } from \"../lib/transform\";\nimport { upgrade } from \"../lib/upgrade\";\nimport debug from \"debug\";\n\nexport interface TransformOptions {\n  dry?: boolean;\n  print?: boolean;\n  verbose?: boolean;\n  jscodeshift?: string;\n}\n\nconst error = debug(\"codemod:error\");\ndebug.enable(\"codemod:*\");\n\nconst addTransformOptions = (command: Command): Command => {\n  return command\n    .option(\"-d, --dry\", \"Dry run (no changes are made to files)\")\n    .option(\"-p, --print\", \"Print transformed files to stdout\")\n    .option(\"--verbose\", \"Show more information about the transform process\")\n    .option(\n      \"-j, --jscodeshift <options>\",\n      \"Pass options directly to jscodeshift\",\n    );\n};\n\nexport const codemodCommand = addTransformOptions(\n  new Command()\n    .name(\"codemod\")\n    .description(\"CLI tool for running codemods\")\n    .argument(\"<codemod>\", \"Codemod to run (e.g., rewrite-framework-imports)\")\n    .argument(\"<source>\", \"Path to source files or directory to transform\"),\n).action((codemod, source, options: TransformOptions) => {\n  try {\n    transform(codemod, source, options);\n  } catch (err) {\n    const errorMessage = err instanceof Error ? err.message : String(err);\n    const errorStack = err instanceof Error ? err.stack : undefined;\n    error(`Error transforming: ${errorMessage}`);\n    if (errorStack) {\n      error(errorStack);\n    }\n    process.exit(1);\n  }\n});\n\nexport const upgradeCommand = addTransformOptions(\n  new Command()\n    .command(\"upgrade\")\n    .description(\"Upgrade ai package dependencies and apply codemods\"),\n).action(async (options: TransformOptions) => {\n  try {\n    await upgrade(options);\n  } catch (err) {\n    const errorMessage = err instanceof Error ? err.message : String(err);\n    const errorStack = err instanceof Error ? err.stack : undefined;\n    error(`Error upgrading: ${errorMessage}`);\n    if (errorStack) {\n      error(errorStack);\n    }\n    process.exit(1);\n  }\n});\n"],"mappings":";;;;;AAYA,MAAM,QAAQ,MAAM,eAAe;AACnC,MAAM,OAAO,WAAW;AAExB,MAAM,uBAAuB,YAA8B;CACzD,OAAO,QACJ,OAAO,aAAa,wCAAwC,CAAC,CAC7D,OAAO,eAAe,mCAAmC,CAAC,CAC1D,OAAO,aAAa,mDAAmD,CAAC,CACxE,OACC,+BACA,sCACF;AACJ;AAEA,MAAa,iBAAiB,oBAC5B,IAAI,QAAQ,CAAC,CACV,KAAK,SAAS,CAAC,CACf,YAAY,+BAA+B,CAAC,CAC5C,SAAS,aAAa,kDAAkD,CAAC,CACzE,SAAS,YAAY,gDAAgD,CAC1E,CAAC,CAAC,QAAQ,SAAS,QAAQ,YAA8B;CACvD,IAAI;EACF,UAAU,SAAS,QAAQ,OAAO;CACpC,SAAS,KAAK;EACZ,MAAM,eAAe,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;EACpE,MAAM,aAAa,eAAe,QAAQ,IAAI,QAAQ,KAAA;EACtD,MAAM,uBAAuB,cAAc;EAC3C,IAAI,YACF,MAAM,UAAU;EAElB,QAAQ,KAAK,CAAC;CAChB;AACF,CAAC;AAED,MAAa,iBAAiB,oBAC5B,IAAI,QAAQ,CAAC,CACV,QAAQ,SAAS,CAAC,CAClB,YAAY,oDAAoD,CACrE,CAAC,CAAC,OAAO,OAAO,YAA8B;CAC5C,IAAI;EACF,MAAM,QAAQ,OAAO;CACvB,SAAS,KAAK;EACZ,MAAM,eAAe,eAAe,QAAQ,IAAI,UAAU,OAAO,GAAG;EACpE,MAAM,aAAa,eAAe,QAAQ,IAAI,QAAQ,KAAA;EACtD,MAAM,oBAAoB,cAAc;EACxC,IAAI,YACF,MAAM,UAAU;EAElB,QAAQ,KAAK,CAAC;CAChB;AACF,CAAC"}

package/dist/index.js

@@ -1,12 +1,12 @@
 #!/usr/bin/env node
-import { buildProgram } from "./program.js";
+import { runCli } from "./run.js";
 //#region src/index.ts
 process.on("SIGINT", () => process.exit(0));
 process.on("SIGTERM", () => process.exit(0));
-function main() {
-	buildProgram().parse();
-}
-main();
+runCli().catch((error) => {
+	console.error(error);
+	process.exitCode = 1;
+});
 //#endregion
 
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","names":[],"sources":["../src/index.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport { buildProgram } from \"./program\";\n\nprocess.on(\"SIGINT\", () => process.exit(0));\nprocess.on(\"SIGTERM\", () => process.exit(0));\n\nfunction main() {\n  buildProgram().parse();\n}\n\nmain();\n"],"mappings":";;;AAIA,QAAQ,GAAG,gBAAgB,QAAQ,KAAK,CAAC,CAAC;AAC1C,QAAQ,GAAG,iBAAiB,QAAQ,KAAK,CAAC,CAAC;AAE3C,SAAS,OAAO;CACd,aAAa,CAAC,CAAC,MAAM;AACvB;AAEA,KAAK"}
+{"version":3,"file":"index.js","names":[],"sources":["../src/index.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport { runCli } from \"./run\";\n\nprocess.on(\"SIGINT\", () => process.exit(0));\nprocess.on(\"SIGTERM\", () => process.exit(0));\n\nvoid runCli().catch((error: unknown) => {\n  console.error(error);\n  process.exitCode = 1;\n});\n"],"mappings":";;;AAIA,QAAQ,GAAG,gBAAgB,QAAQ,KAAK,CAAC,CAAC;AAC1C,QAAQ,GAAG,iBAAiB,QAAQ,KAAK,CAAC,CAAC;AAEtC,OAAO,CAAC,CAAC,OAAO,UAAmB;CACtC,QAAQ,MAAM,KAAK;CACnB,QAAQ,WAAW;AACrB,CAAC"}

package/dist/lib/upgrade.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"upgrade.d.ts","names":[],"sources":["../../src/lib/upgrade.ts"],"mappings":";;;;;;;;iBA2BsB,QAAQ,SAAS,mBAAgB"}
+{"version":3,"file":"upgrade.d.ts","names":[],"sources":["../../src/lib/upgrade.ts"],"mappings":";;;;;;;;iBA4BsB,QAAQ,SAAS,mBAAgB"}

package/dist/lib/upgrade.js

@@ -12,7 +12,8 @@ const bundle = [
 	"v0-11/content-part-to-message-part",
 	"v0-12/assistant-api-to-aui",
 	"v0-12/event-names-to-camelcase",
-	"v0-12/primitive-if-to-aui-if"
+	"v0-12/primitive-if-to-aui-if",
+	"v0-15/aui-accessor-calls-to-properties"
 ];
 const log = debug("codemod:upgrade");
 const error = debug("codemod:upgrade:error");

package/dist/lib/upgrade.js.map

@@ -1 +1 @@
-{"version":3,"file":"upgrade.js","names":[],"sources":["../../src/lib/upgrade.ts"],"sourcesContent":["import debug from \"debug\";\nimport { transform, type TransformErrors, getRelevantFiles } from \"./transform\";\nimport type { TransformOptions } from \"./transform-options\";\nimport { SingleBar, Presets } from \"cli-progress\";\nimport installReactUILib from \"./install-ui-lib\";\nimport installEdgeLib from \"./install-edge-lib\";\nimport installAiSdkLib from \"./install-ai-sdk-lib\";\nimport { logger } from \"./utils/logger\";\n\nconst bundle = [\n  \"v0-8/ui-package-split\",\n  \"v0-9/edge-package-split\",\n  \"v0-11/content-part-to-message-part\",\n  \"v0-12/assistant-api-to-aui\",\n  \"v0-12/event-names-to-camelcase\",\n  \"v0-12/primitive-if-to-aui-if\",\n];\n\nconst log = debug(\"codemod:upgrade\");\nconst error = debug(\"codemod:upgrade:error\");\n\n/**\n * Runs the upgrade cycle:\n *   - Runs each codemod in the bundle.\n *   - Displays progress using cli-progress.\n *   - After codemods run, checks if any file now imports from the new packages and prompts for install.\n */\nexport async function upgrade(options: TransformOptions) {\n  const cwd = process.cwd();\n  log(\"Starting upgrade...\");\n\n  // Find relevant files once to avoid duplicate work\n  logger.info(\"Analyzing codebase...\");\n  const relevantFiles = getRelevantFiles(cwd);\n  const fileCount = relevantFiles.length;\n  logger.info(`Found ${fileCount} files to process.`);\n\n  // Calculate total work units (files × codemods)\n  const totalWork = fileCount * bundle.length;\n  let completedWork = 0;\n\n  const bar = new SingleBar(\n    {\n      format: \"Progress |{bar}| {percentage}% | ETA: {eta}s || {status}\",\n      hideCursor: true,\n    },\n    Presets.shades_classic,\n  );\n\n  bar.start(totalWork, 0, { status: \"Starting...\" });\n  const allErrors: TransformErrors = [];\n\n  for (const codemod of bundle) {\n    bar.update(completedWork, { status: `Running ${codemod}...` });\n\n    // Use a custom progress callback to update the progress bar\n    const errors = transform(codemod, cwd, options, {\n      logStatus: false,\n      onProgress: (processedFiles: number) => {\n        completedWork = bundle.indexOf(codemod) * fileCount + processedFiles;\n        bar.update(Math.min(completedWork, totalWork), {\n          status: `Running ${codemod} (${processedFiles}/${fileCount} files)`,\n        });\n      },\n      relevantFiles, // Pass the pre-computed relevant files\n    });\n\n    allErrors.push(...errors);\n    completedWork = (bundle.indexOf(codemod) + 1) * fileCount;\n    bar.update(completedWork, { status: `Completed ${codemod}` });\n  }\n\n  bar.update(totalWork, { status: \"Checking dependencies...\" });\n  bar.stop();\n\n  if (allErrors.length > 0) {\n    log(\"Some codemods did not apply successfully to all files. Details:\");\n    allErrors.forEach(({ transform, filename, summary }) => {\n      error(`codemod=${transform}, path=${filename}, summary=${summary}`);\n    });\n  }\n\n  // After codemods run, check if files import from the new packages and prompt for install.\n  logger.info(\"Checking for package dependencies...\");\n  await installReactUILib();\n  await installEdgeLib();\n  await installAiSdkLib();\n\n  log(\"Upgrade complete.\");\n  logger.success(\"Upgrade complete!\");\n}\n"],"mappings":";;;;;;;;AASA,MAAM,SAAS;CACb;CACA;CACA;CACA;CACA;CACA;AACF;AAEA,MAAM,MAAM,MAAM,iBAAiB;AACnC,MAAM,QAAQ,MAAM,uBAAuB;;;;;;;AAQ3C,eAAsB,QAAQ,SAA2B;CACvD,MAAM,MAAM,QAAQ,IAAI;CACxB,IAAI,qBAAqB;CAGzB,OAAO,KAAK,uBAAuB;CACnC,MAAM,gBAAgB,iBAAiB,GAAG;CAC1C,MAAM,YAAY,cAAc;CAChC,OAAO,KAAK,SAAS,UAAU,mBAAmB;CAGlD,MAAM,YAAY,YAAY,OAAO;CACrC,IAAI,gBAAgB;CAEpB,MAAM,MAAM,IAAI,UACd;EACE,QAAQ;EACR,YAAY;CACd,GACA,QAAQ,cACV;CAEA,IAAI,MAAM,WAAW,GAAG,EAAE,QAAQ,cAAc,CAAC;CACjD,MAAM,YAA6B,CAAC;CAEpC,KAAK,MAAM,WAAW,QAAQ;EAC5B,IAAI,OAAO,eAAe,EAAE,QAAQ,WAAW,QAAQ,KAAK,CAAC;EAG7D,MAAM,SAAS,UAAU,SAAS,KAAK,SAAS;GAC9C,WAAW;GACX,aAAa,mBAA2B;IACtC,gBAAgB,OAAO,QAAQ,OAAO,IAAI,YAAY;IACtD,IAAI,OAAO,KAAK,IAAI,eAAe,SAAS,GAAG,EAC7C,QAAQ,WAAW,QAAQ,IAAI,eAAe,GAAG,UAAU,SAC7D,CAAC;GACH;GACA;EACF,CAAC;EAED,UAAU,KAAK,GAAG,MAAM;EACxB,iBAAiB,OAAO,QAAQ,OAAO,IAAI,KAAK;EAChD,IAAI,OAAO,eAAe,EAAE,QAAQ,aAAa,UAAU,CAAC;CAC9D;CAEA,IAAI,OAAO,WAAW,EAAE,QAAQ,2BAA2B,CAAC;CAC5D,IAAI,KAAK;CAET,IAAI,UAAU,SAAS,GAAG;EACxB,IAAI,iEAAiE;EACrE,UAAU,SAAS,EAAE,WAAW,UAAU,cAAc;GACtD,MAAM,WAAW,UAAU,SAAS,SAAS,YAAY,SAAS;EACpE,CAAC;CACH;CAGA,OAAO,KAAK,sCAAsC;CAClD,MAAM,kBAAkB;CACxB,MAAM,eAAe;CACrB,MAAM,gBAAgB;CAEtB,IAAI,mBAAmB;CACvB,OAAO,QAAQ,mBAAmB;AACpC"}
+{"version":3,"file":"upgrade.js","names":[],"sources":["../../src/lib/upgrade.ts"],"sourcesContent":["import debug from \"debug\";\nimport { transform, type TransformErrors, getRelevantFiles } from \"./transform\";\nimport type { TransformOptions } from \"./transform-options\";\nimport { SingleBar, Presets } from \"cli-progress\";\nimport installReactUILib from \"./install-ui-lib\";\nimport installEdgeLib from \"./install-edge-lib\";\nimport installAiSdkLib from \"./install-ai-sdk-lib\";\nimport { logger } from \"./utils/logger\";\n\nconst bundle = [\n  \"v0-8/ui-package-split\",\n  \"v0-9/edge-package-split\",\n  \"v0-11/content-part-to-message-part\",\n  \"v0-12/assistant-api-to-aui\",\n  \"v0-12/event-names-to-camelcase\",\n  \"v0-12/primitive-if-to-aui-if\",\n  \"v0-15/aui-accessor-calls-to-properties\",\n];\n\nconst log = debug(\"codemod:upgrade\");\nconst error = debug(\"codemod:upgrade:error\");\n\n/**\n * Runs the upgrade cycle:\n *   - Runs each codemod in the bundle.\n *   - Displays progress using cli-progress.\n *   - After codemods run, checks if any file now imports from the new packages and prompts for install.\n */\nexport async function upgrade(options: TransformOptions) {\n  const cwd = process.cwd();\n  log(\"Starting upgrade...\");\n\n  // Find relevant files once to avoid duplicate work\n  logger.info(\"Analyzing codebase...\");\n  const relevantFiles = getRelevantFiles(cwd);\n  const fileCount = relevantFiles.length;\n  logger.info(`Found ${fileCount} files to process.`);\n\n  // Calculate total work units (files × codemods)\n  const totalWork = fileCount * bundle.length;\n  let completedWork = 0;\n\n  const bar = new SingleBar(\n    {\n      format: \"Progress |{bar}| {percentage}% | ETA: {eta}s || {status}\",\n      hideCursor: true,\n    },\n    Presets.shades_classic,\n  );\n\n  bar.start(totalWork, 0, { status: \"Starting...\" });\n  const allErrors: TransformErrors = [];\n\n  for (const codemod of bundle) {\n    bar.update(completedWork, { status: `Running ${codemod}...` });\n\n    // Use a custom progress callback to update the progress bar\n    const errors = transform(codemod, cwd, options, {\n      logStatus: false,\n      onProgress: (processedFiles: number) => {\n        completedWork = bundle.indexOf(codemod) * fileCount + processedFiles;\n        bar.update(Math.min(completedWork, totalWork), {\n          status: `Running ${codemod} (${processedFiles}/${fileCount} files)`,\n        });\n      },\n      relevantFiles, // Pass the pre-computed relevant files\n    });\n\n    allErrors.push(...errors);\n    completedWork = (bundle.indexOf(codemod) + 1) * fileCount;\n    bar.update(completedWork, { status: `Completed ${codemod}` });\n  }\n\n  bar.update(totalWork, { status: \"Checking dependencies...\" });\n  bar.stop();\n\n  if (allErrors.length > 0) {\n    log(\"Some codemods did not apply successfully to all files. Details:\");\n    allErrors.forEach(({ transform, filename, summary }) => {\n      error(`codemod=${transform}, path=${filename}, summary=${summary}`);\n    });\n  }\n\n  // After codemods run, check if files import from the new packages and prompt for install.\n  logger.info(\"Checking for package dependencies...\");\n  await installReactUILib();\n  await installEdgeLib();\n  await installAiSdkLib();\n\n  log(\"Upgrade complete.\");\n  logger.success(\"Upgrade complete!\");\n}\n"],"mappings":";;;;;;;;AASA,MAAM,SAAS;CACb;CACA;CACA;CACA;CACA;CACA;CACA;AACF;AAEA,MAAM,MAAM,MAAM,iBAAiB;AACnC,MAAM,QAAQ,MAAM,uBAAuB;;;;;;;AAQ3C,eAAsB,QAAQ,SAA2B;CACvD,MAAM,MAAM,QAAQ,IAAI;CACxB,IAAI,qBAAqB;CAGzB,OAAO,KAAK,uBAAuB;CACnC,MAAM,gBAAgB,iBAAiB,GAAG;CAC1C,MAAM,YAAY,cAAc;CAChC,OAAO,KAAK,SAAS,UAAU,mBAAmB;CAGlD,MAAM,YAAY,YAAY,OAAO;CACrC,IAAI,gBAAgB;CAEpB,MAAM,MAAM,IAAI,UACd;EACE,QAAQ;EACR,YAAY;CACd,GACA,QAAQ,cACV;CAEA,IAAI,MAAM,WAAW,GAAG,EAAE,QAAQ,cAAc,CAAC;CACjD,MAAM,YAA6B,CAAC;CAEpC,KAAK,MAAM,WAAW,QAAQ;EAC5B,IAAI,OAAO,eAAe,EAAE,QAAQ,WAAW,QAAQ,KAAK,CAAC;EAG7D,MAAM,SAAS,UAAU,SAAS,KAAK,SAAS;GAC9C,WAAW;GACX,aAAa,mBAA2B;IACtC,gBAAgB,OAAO,QAAQ,OAAO,IAAI,YAAY;IACtD,IAAI,OAAO,KAAK,IAAI,eAAe,SAAS,GAAG,EAC7C,QAAQ,WAAW,QAAQ,IAAI,eAAe,GAAG,UAAU,SAC7D,CAAC;GACH;GACA;EACF,CAAC;EAED,UAAU,KAAK,GAAG,MAAM;EACxB,iBAAiB,OAAO,QAAQ,OAAO,IAAI,KAAK;EAChD,IAAI,OAAO,eAAe,EAAE,QAAQ,aAAa,UAAU,CAAC;CAC9D;CAEA,IAAI,OAAO,WAAW,EAAE,QAAQ,2BAA2B,CAAC;CAC5D,IAAI,KAAK;CAET,IAAI,UAAU,SAAS,GAAG;EACxB,IAAI,iEAAiE;EACrE,UAAU,SAAS,EAAE,WAAW,UAAU,cAAc;GACtD,MAAM,WAAW,UAAU,SAAS,SAAS,YAAY,SAAS;EACpE,CAAC;CACH;CAGA,OAAO,KAAK,sCAAsC;CAClD,MAAM,kBAAkB;CACxB,MAAM,eAAe;CACrB,MAAM,gBAAgB;CAEtB,IAAI,mBAAmB;CACvB,OAAO,QAAQ,mBAAmB;AACpC"}

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.107",
+  "version": "0.0.108",
   "description": "CLI for assistant-ui",
   "keywords": [
     "cli",
@@ -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.19"
+    "@assistant-ui/x-buildutils": "0.0.20"
   },
   "publishConfig": {
     "access": "public",

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/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;
+});

package/src/lib/upgrade.ts

@@ -14,6 +14,7 @@ const bundle = [
   "v0-12/assistant-api-to-aui",
   "v0-12/event-names-to-camelcase",
   "v0-12/primitive-if-to-aui-if",
+  "v0-15/aui-accessor-calls-to-properties",
 ];
 
 const log = debug("codemod:upgrade");

package/src/run.test.ts

@@ -0,0 +1,27 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+const mocks = vi.hoisted(() => ({
+  parseAsync: vi.fn(),
+}));
+
+vi.mock("./program", () => ({
+  buildProgram: () => ({
+    parseAsync: mocks.parseAsync,
+  }),
+}));
+
+import { runCli } from "./run";
+
+describe("runCli", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("awaits and propagates asynchronous command failures", async () => {
+    const error = new Error("command failed");
+    mocks.parseAsync.mockRejectedValue(error);
+
+    await expect(runCli()).rejects.toBe(error);
+    expect(mocks.parseAsync).toHaveBeenCalledOnce();
+  });
+});

package/src/run.ts

@@ -0,0 +1,5 @@
+import { buildProgram } from "./program";
+
+export async function runCli(): Promise<void> {
+  await buildProgram().parseAsync();
+}