assistant-ui 0.0.49 → 0.0.50

package/dist/codemods/utils/createTransformer.d.ts

@@ -0,0 +1,15 @@
+import { FileInfo, API, JSCodeshift, Collection } from "jscodeshift";
+export type TransformerFunction = (context: TransformContext, fileInfo: FileInfo, api: API, options: any) => void;
+export interface TransformContext {
+    /**
+     * The jscodeshift API object.
+     */
+    j: JSCodeshift;
+    /**
+     * The root collection of the AST.
+     */
+    root: Collection<any>;
+    markAsChanged: () => void;
+}
+export declare function createTransformer(transformFn: TransformerFunction): (fileInfo: FileInfo, api: API, options: any) => string | null;
+//# sourceMappingURL=createTransformer.d.ts.map

package/dist/codemods/utils/createTransformer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"createTransformer.d.ts","sourceRoot":"","sources":["../../../src/codemods/utils/createTransformer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,GAAG,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErE,MAAM,MAAM,mBAAmB,GAAG,CAChC,OAAO,EAAE,gBAAgB,EACzB,QAAQ,EAAE,QAAQ,EAClB,GAAG,EAAE,GAAG,EACR,OAAO,EAAE,GAAG,KACT,IAAI,CAAC;AAEV,MAAM,WAAW,gBAAgB;IAC/B;;OAEG;IACH,CAAC,EAAE,WAAW,CAAC;IAEf;;OAEG;IACH,IAAI,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;IAEtB,aAAa,EAAE,MAAM,IAAI,CAAC;CAC3B;AAED,wBAAgB,iBAAiB,CAAC,WAAW,EAAE,mBAAmB,IACpC,UAAU,QAAQ,EAAE,KAAK,GAAG,EAAE,SAAS,GAAG,mBAgBvE"}

package/dist/codemods/v0-8/ui-package-split.d.ts

@@ -0,0 +1,3 @@
+declare const migrateAssistantUI: (fileInfo: import("jscodeshift/src/core").FileInfo, api: import("jscodeshift/src/core").API, options: any) => string | null;
+export default migrateAssistantUI;
+//# sourceMappingURL=ui-package-split.d.ts.map

package/dist/codemods/v0-8/ui-package-split.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ui-package-split.d.ts","sourceRoot":"","sources":["../../../src/codemods/v0-8/ui-package-split.ts"],"names":[],"mappings":"AA8BA,QAAA,MAAM,kBAAkB,6HA0KtB,CAAC;AAEH,eAAe,kBAAkB,CAAC"}

package/dist/codemods/v0-9/edge-package-split.d.ts

@@ -0,0 +1,3 @@
+declare const migrateToEdgePackage: (fileInfo: import("jscodeshift/src/core").FileInfo, api: import("jscodeshift/src/core").API, options: any) => string | null;
+export default migrateToEdgePackage;
+//# sourceMappingURL=edge-package-split.d.ts.map

package/dist/codemods/v0-9/edge-package-split.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"edge-package-split.d.ts","sourceRoot":"","sources":["../../../src/codemods/v0-9/edge-package-split.ts"],"names":[],"mappings":"AAqCA,QAAA,MAAM,oBAAoB,6HA0LxB,CAAC;AAEH,eAAe,oBAAoB,CAAC"}

package/dist/commands/add.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const add: Command;
+//# sourceMappingURL=add.d.ts.map

package/dist/commands/add.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"add.d.ts","sourceRoot":"","sources":["../../src/commands/add.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAKpC,eAAO,MAAM,GAAG,SAyCZ,CAAC"}

package/dist/commands/create.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const create: Command;
+//# sourceMappingURL=create.d.ts.map

package/dist/commands/create.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"create.d.ts","sourceRoot":"","sources":["../../src/commands/create.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAIpC,eAAO,MAAM,MAAM,SAuFf,CAAC"}

package/dist/commands/init.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const init: Command;
+//# sourceMappingURL=init.d.ts.map

package/dist/commands/init.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../src/commands/init.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAOpC,eAAO,MAAM,IAAI,SA0Cb,CAAC"}

package/dist/commands/update.d.ts

@@ -0,0 +1,3 @@
+import { Command } from "commander";
+export declare const update: Command;
+//# sourceMappingURL=update.d.ts.map

package/dist/commands/update.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"update.d.ts","sourceRoot":"","sources":["../../src/commands/update.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAOpC,eAAO,MAAM,MAAM,SAoDf,CAAC"}

package/dist/commands/upgrade.d.ts

@@ -0,0 +1,10 @@
+import { Command } from "commander";
+export interface TransformOptions {
+    dry?: boolean;
+    print?: boolean;
+    verbose?: boolean;
+    jscodeshift?: string;
+}
+export declare const codemodCommand: Command;
+export declare const upgradeCommand: Command;
+//# sourceMappingURL=upgrade.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"upgrade.d.ts","sourceRoot":"","sources":["../../src/commands/upgrade.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAKpC,MAAM,WAAW,gBAAgB;IAC/B,GAAG,CAAC,EAAE,OAAO,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAgBD,eAAO,MAAM,cAAc,SAczB,CAAC;AAEH,eAAO,MAAM,cAAc,SAYzB,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,3 @@
+#!/usr/bin/env node
+export {};
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}

package/dist/lib/install-ai-sdk-lib.d.ts

@@ -0,0 +1,2 @@
+export default function installAiSdkLib(): Promise<void>;
+//# sourceMappingURL=install-ai-sdk-lib.d.ts.map

package/dist/lib/install-ai-sdk-lib.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"install-ai-sdk-lib.d.ts","sourceRoot":"","sources":["../../src/lib/install-ai-sdk-lib.ts"],"names":[],"mappings":"AAuCA,wBAA8B,eAAe,IAAI,OAAO,CAAC,IAAI,CAAC,CAoD7D"}

package/dist/lib/install-edge-lib.d.ts

@@ -0,0 +1,2 @@
+export default function installEdgeLib(): Promise<void>;
+//# sourceMappingURL=install-edge-lib.d.ts.map

package/dist/lib/install-edge-lib.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"install-edge-lib.d.ts","sourceRoot":"","sources":["../../src/lib/install-edge-lib.ts"],"names":[],"mappings":"AAuCA,wBAA8B,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC,CAoD5D"}

package/dist/lib/install-ui-lib.d.ts

@@ -0,0 +1,2 @@
+export default function installReactUILib(): Promise<void>;
+//# sourceMappingURL=install-ui-lib.d.ts.map

package/dist/lib/install-ui-lib.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"install-ui-lib.d.ts","sourceRoot":"","sources":["../../src/lib/install-ui-lib.ts"],"names":[],"mappings":"AAuCA,wBAA8B,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC,CAoD/D"}

package/dist/lib/transform-options.d.ts

@@ -0,0 +1,7 @@
+export interface TransformOptions {
+    dry?: boolean;
+    print?: boolean;
+    verbose?: boolean;
+    jscodeshift?: string;
+}
+//# sourceMappingURL=transform-options.d.ts.map

package/dist/lib/transform-options.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"transform-options.d.ts","sourceRoot":"","sources":["../../src/lib/transform-options.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,gBAAgB;IAC/B,GAAG,CAAC,EAAE,OAAO,CAAC;IACd,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB"}

package/dist/lib/transform.d.ts

@@ -0,0 +1,21 @@
+import { TransformOptions } from "./transform-options";
+/**
+ * Gets the list of files that need to be processed in the codebase
+ * Only includes files that contain "assistant-ui" to optimize performance
+ */
+export declare function getRelevantFiles(cwd: string): string[];
+/**
+ * Counts the number of files that need to be processed
+ */
+export declare function countFilesToProcess(cwd: string): number;
+export type TransformErrors = {
+    transform: string;
+    filename: string;
+    summary: string;
+}[];
+export declare function transform(codemod: string, source: string, transformOptions: TransformOptions, options?: {
+    logStatus: boolean;
+    onProgress?: (processedFiles: number) => void;
+    relevantFiles?: string[];
+}): TransformErrors;
+//# sourceMappingURL=transform.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"transform.d.ts","sourceRoot":"","sources":["../../src/lib/transform.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAWvD;;;GAGG;AACH,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,EAAE,CAwBtD;AAED;;GAEG;AACH,wBAAgB,mBAAmB,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM,CAEvD;AAoCD,MAAM,MAAM,eAAe,GAAG;IAC5B,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB,EAAE,CAAC;AAoBJ,wBAAgB,SAAS,CACvB,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,MAAM,EACd,gBAAgB,EAAE,gBAAgB,EAClC,OAAO,GAAE;IACP,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,CAAC,EAAE,CAAC,cAAc,EAAE,MAAM,KAAK,IAAI,CAAC;IAC9C,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CACJ,GACtB,eAAe,CA0DjB"}

package/dist/lib/upgrade.d.ts

@@ -0,0 +1,9 @@
+import { TransformOptions } from "./transform-options";
+/**
+ * Runs the upgrade cycle:
+ *   - Runs each codemod in the bundle.
+ *   - Displays progress using cli-progress.
+ *   - After codemods run, checks if any file now imports from the new packages and prompts for install.
+ */
+export declare function upgrade(options: TransformOptions): Promise<void>;
+//# sourceMappingURL=upgrade.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"upgrade.d.ts","sourceRoot":"","sources":["../../src/lib/upgrade.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAWvD;;;;;GAKG;AACH,wBAAsB,OAAO,CAAC,OAAO,EAAE,gBAAgB,iBA+DtD"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "assistant-ui",
-  "version": "0.0.49",
+  "version": "0.0.50",
   "license": "MIT",
   "source": "./src/index.ts",
   "dependencies": {
@@ -29,6 +29,7 @@
   },
   "files": [
     "dist",
+    "src",
     "README.md"
   ],
   "bin": "./dist/index.js",
@@ -39,14 +40,11 @@
   "homepage": "https://www.assistant-ui.com/",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/assistant-ui/assistant-ui.git"
+    "url": "https://github.com/assistant-ui/assistant-ui/tree/main/packages/cli"
   },
   "bugs": {
     "url": "https://github.com/assistant-ui/assistant-ui/issues"
   },
-  "engines": {
-    "node": ">=20.0.0"
-  },
   "scripts": {
     "build": "tsx scripts/build.mts"
   }

package/src/codemods/utils/createTransformer.ts

@@ -0,0 +1,41 @@
+import { FileInfo, API, JSCodeshift, Collection } from "jscodeshift";
+
+export type TransformerFunction = (
+  context: TransformContext,
+  fileInfo: FileInfo,
+  api: API,
+  options: any,
+) => void;
+
+export interface TransformContext {
+  /**
+   * The jscodeshift API object.
+   */
+  j: JSCodeshift;
+
+  /**
+   * The root collection of the AST.
+   */
+  root: Collection<any>;
+
+  markAsChanged: () => void;
+}
+
+export function createTransformer(transformFn: TransformerFunction) {
+  return function transformer(fileInfo: FileInfo, api: API, options: any) {
+    const j = api.jscodeshift;
+    const root = j(fileInfo.source);
+
+    let isChanged = false;
+    const context: TransformContext = {
+      j,
+      root,
+      markAsChanged: () => {
+        isChanged = true;
+      },
+    };
+
+    transformFn(context, fileInfo, api, options);
+    return isChanged ? root.toSource() : null;
+  };
+}

package/src/codemods/v0-8/ui-package-split.ts

@@ -0,0 +1,203 @@
+import { createTransformer } from "../utils/createTransformer";
+
+const reactUIExports: string[] = [
+  "ThreadConfigProvider",
+  "useThreadConfig",
+  "ThreadConfig",
+  "ThreadWelcomeConfig",
+  "UserMessageConfig",
+  "AssistantMessageConfig",
+  "StringsConfig",
+  "SuggestionConfig",
+  "ThreadConfigProviderProps",
+  "AssistantActionBar",
+  "AssistantMessage",
+  "AssistantModal",
+  "BranchPicker",
+  "Composer",
+  "ContentPart",
+  "AttachmentUI",
+  "EditComposer",
+  "Thread",
+  "ThreadList",
+  "ThreadListItem",
+  "ThreadWelcome",
+  "UserMessage",
+  "makeMarkdownText",
+  "MakeMarkdownTextProps",
+  "CodeHeader",
+];
+
+const migrateAssistantUI = createTransformer(({ j, root, markAsChanged }) => {
+  const sourcesToMigrate: string[] = [
+    "@assistant-ui/react",
+    "@assistant-ui/react-markdown",
+  ];
+  const movedSpecifiers: 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) &&
+          reactUIExports.includes(specifier.imported.name as string)
+        ) {
+          movedSpecifiers.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();
+      }
+    });
+
+  if (movedSpecifiers.length > 0) {
+    const existingReactUIImport = root.find(j.ImportDeclaration, {
+      source: { value: "@assistant-ui/react-ui" },
+    });
+    if (existingReactUIImport.size() > 0) {
+      existingReactUIImport.forEach((path: any) => {
+        movedSpecifiers.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(
+        movedSpecifiers,
+        j.literal("@assistant-ui/react-ui"),
+      );
+      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();
+  }
+
+  const cssReplacements: Record<string, string> = {
+    "@assistant-ui/react/styles/index.css":
+      "@assistant-ui/react-ui/styles/index.css",
+    "@assistant-ui/react/styles/modal.css":
+      "@assistant-ui/react-ui/styles/modal.css",
+    "@assistant-ui/react-markdown/styles/markdown.css":
+      "@assistant-ui/react-ui/styles/markdown.css",
+  };
+
+  root.find(j.ImportDeclaration).forEach((path: any) => {
+    const sourceValue: string = path.value.source.value;
+    if (cssReplacements[sourceValue]) {
+      path.value.source = j.literal(cssReplacements[sourceValue]);
+      markAsChanged();
+    }
+  });
+
+  let removedMarkdownPlugin = false;
+  root
+    .find(j.CallExpression, { callee: { name: "require" } })
+    .filter((path: any) => {
+      const arg = path.value.arguments[0];
+      return (
+        arg &&
+        (arg.type === "Literal" || arg.type === "StringLiteral") &&
+        arg.value === "@assistant-ui/react-markdown/tailwindcss"
+      );
+    })
+    .forEach((path: any) => {
+      removedMarkdownPlugin = true;
+      const parent = path.parentPath;
+      if (
+        parent &&
+        parent.value &&
+        parent.value.type === "VariableDeclarator"
+      ) {
+        const varDecl = parent.parentPath;
+        if (
+          varDecl &&
+          varDecl.value.declarations &&
+          varDecl.value.declarations.length === 1
+        ) {
+          j(varDecl).remove();
+        } else {
+          varDecl.value.declarations = varDecl.value.declarations.filter(
+            (decl: any) => decl !== parent.value,
+          );
+        }
+        markAsChanged();
+      } else {
+        j(path).remove();
+        markAsChanged();
+      }
+    });
+
+  root
+    .find(j.CallExpression, { callee: { name: "require" } })
+    .filter((path: any) => {
+      const arg = path.value.arguments[0];
+      return (
+        arg &&
+        (arg.type === "Literal" || arg.type === "StringLiteral") &&
+        arg.value === "@assistant-ui/react/tailwindcss"
+      );
+    })
+    .forEach((path: any) => {
+      path.value.arguments[0].value = "@assistant-ui/react-ui/tailwindcss";
+      markAsChanged();
+      if (removedMarkdownPlugin) {
+        if (
+          path.parentPath &&
+          path.parentPath.value &&
+          path.parentPath.value.type === "CallExpression" &&
+          path.parentPath.value.arguments.length > 0
+        ) {
+          const configObj = path.parentPath.value.arguments[0];
+          if (configObj && configObj.type === "ObjectExpression") {
+            const componentsProp = configObj.properties.find((prop: any) => {
+              return (
+                (prop.key.name === "components" ||
+                  prop.key.value === "components") &&
+                prop.value.type === "ArrayExpression"
+              );
+            });
+            if (componentsProp) {
+              const componentsArray = componentsProp.value.elements;
+              const hasMarkdown = componentsArray.some(
+                (el: any) => el.type === "Literal" && el.value === "markdown",
+              );
+              if (!hasMarkdown) {
+                componentsArray.push(j.literal("markdown"));
+                markAsChanged();
+              }
+            }
+          }
+        }
+      }
+    });
+});
+
+export default migrateAssistantUI;