assistant-ui 0.0.91 → 0.0.93

package/dist/codemods/v0-8/ui-package-split.js

@@ -1,170 +1,127 @@
 import { createTransformer } from "../utils/createTransformer.js";
+//#region src/codemods/v0-8/ui-package-split.ts
 const reactUIExports = [
-    "ThreadConfigProvider",
-    "useThreadConfig",
-    "ThreadConfig",
-    "ThreadWelcomeConfig",
-    "UserMessageConfig",
-    "AssistantMessageConfig",
-    "StringsConfig",
-    "SuggestionConfig",
-    "ThreadConfigProviderProps",
-    "AssistantActionBar",
-    "AssistantMessage",
-    "AssistantModal",
-    "BranchPicker",
-    "Composer",
-    "MessagePart",
-    "AttachmentUI",
-    "EditComposer",
-    "Thread",
-    "ThreadList",
-    "ThreadListItem",
-    "ThreadWelcome",
-    "UserMessage",
-    "makeMarkdownText",
-    "MakeMarkdownTextProps",
-    "CodeHeader",
+	"ThreadConfigProvider",
+	"useThreadConfig",
+	"ThreadConfig",
+	"ThreadWelcomeConfig",
+	"UserMessageConfig",
+	"AssistantMessageConfig",
+	"StringsConfig",
+	"SuggestionConfig",
+	"ThreadConfigProviderProps",
+	"AssistantActionBar",
+	"AssistantMessage",
+	"AssistantModal",
+	"BranchPicker",
+	"Composer",
+	"MessagePart",
+	"AttachmentUI",
+	"EditComposer",
+	"Thread",
+	"ThreadList",
+	"ThreadListItem",
+	"ThreadWelcome",
+	"UserMessage",
+	"makeMarkdownText",
+	"MakeMarkdownTextProps",
+	"CodeHeader"
 ];
 const migrateAssistantUI = createTransformer(({ j, root, markAsChanged }) => {
-    const sourcesToMigrate = [
-        "@assistant-ui/react",
-        "@assistant-ui/react-markdown",
-    ];
-    const movedSpecifiers = [];
-    let lastMigratedImportPath = null;
-    root
-        .find(j.ImportDeclaration)
-        .filter((path) => sourcesToMigrate.includes(path.value.source.value))
-        .forEach((path) => {
-        let hadMigratedSpecifiers = false;
-        const remainingSpecifiers = [];
-        path.value.specifiers.forEach((specifier) => {
-            if (j.ImportSpecifier.check(specifier) &&
-                reactUIExports.includes(specifier.imported.name)) {
-                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) => {
-                movedSpecifiers.forEach((specifier) => {
-                    if (!path.value.specifiers.some((s) => 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 = {
-        "@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) => {
-        const sourceValue = 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) => {
-        const arg = path.value.arguments[0];
-        return (arg &&
-            (arg.type === "Literal" || arg.type === "StringLiteral") &&
-            arg.value === "@assistant-ui/react-markdown/tailwindcss");
-    })
-        .forEach((path) => {
-        removedMarkdownPlugin = true;
-        const parent = path.parentPath;
-        if (parent?.value && parent.value.type === "VariableDeclarator") {
-            const varDecl = parent.parentPath;
-            if (varDecl?.value.declarations &&
-                varDecl.value.declarations.length === 1) {
-                j(varDecl).remove();
-            }
-            else {
-                varDecl.value.declarations = varDecl.value.declarations.filter((decl) => decl !== parent.value);
-            }
-            markAsChanged();
-        }
-        else {
-            j(path).remove();
-            markAsChanged();
-        }
-    });
-    root
-        .find(j.CallExpression, { callee: { name: "require" } })
-        .filter((path) => {
-        const arg = path.value.arguments[0];
-        return (arg &&
-            (arg.type === "Literal" || arg.type === "StringLiteral") &&
-            arg.value === "@assistant-ui/react/tailwindcss");
-    })
-        .forEach((path) => {
-        path.value.arguments[0].value = "@assistant-ui/react-ui/tailwindcss";
-        markAsChanged();
-        if (removedMarkdownPlugin) {
-            if (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) => {
-                        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) => el.type === "Literal" && el.value === "markdown");
-                        if (!hasMarkdown) {
-                            componentsArray.push(j.literal("markdown"));
-                            markAsChanged();
-                        }
-                    }
-                }
-            }
-        }
-    });
+	const sourcesToMigrate = ["@assistant-ui/react", "@assistant-ui/react-markdown"];
+	const movedSpecifiers = [];
+	let lastMigratedImportPath = null;
+	root.find(j.ImportDeclaration).filter((path) => sourcesToMigrate.includes(path.value.source.value)).forEach((path) => {
+		let hadMigratedSpecifiers = false;
+		const remainingSpecifiers = [];
+		path.value.specifiers.forEach((specifier) => {
+			if (j.ImportSpecifier.check(specifier) && reactUIExports.includes(specifier.imported.name)) {
+				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) => {
+			movedSpecifiers.forEach((specifier) => {
+				if (!path.value.specifiers.some((s) => 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 = {
+		"@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) => {
+		const sourceValue = 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) => {
+		const arg = path.value.arguments[0];
+		return arg && (arg.type === "Literal" || arg.type === "StringLiteral") && arg.value === "@assistant-ui/react-markdown/tailwindcss";
+	}).forEach((path) => {
+		removedMarkdownPlugin = true;
+		const parent = path.parentPath;
+		if (parent?.value && parent.value.type === "VariableDeclarator") {
+			const varDecl = parent.parentPath;
+			if (varDecl?.value.declarations && varDecl.value.declarations.length === 1) j(varDecl).remove();
+			else varDecl.value.declarations = varDecl.value.declarations.filter((decl) => decl !== parent.value);
+			markAsChanged();
+		} else {
+			j(path).remove();
+			markAsChanged();
+		}
+	});
+	root.find(j.CallExpression, { callee: { name: "require" } }).filter((path) => {
+		const arg = path.value.arguments[0];
+		return arg && (arg.type === "Literal" || arg.type === "StringLiteral") && arg.value === "@assistant-ui/react/tailwindcss";
+	}).forEach((path) => {
+		path.value.arguments[0].value = "@assistant-ui/react-ui/tailwindcss";
+		markAsChanged();
+		if (removedMarkdownPlugin) {
+			if (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) => {
+						return (prop.key.name === "components" || prop.key.value === "components") && prop.value.type === "ArrayExpression";
+					});
+					if (componentsProp) {
+						const componentsArray = componentsProp.value.elements;
+						if (!componentsArray.some((el) => el.type === "Literal" && el.value === "markdown")) {
+							componentsArray.push(j.literal("markdown"));
+							markAsChanged();
+						}
+					}
+				}
+			}
+		}
+	});
 });
-export default migrateAssistantUI;
+//#endregion
+export { migrateAssistantUI as default };
+
 //# sourceMappingURL=ui-package-split.js.map

package/dist/codemods/v0-8/ui-package-split.js.map

@@ -1 +1 @@
-{"version":3,"file":"ui-package-split.js","sourceRoot":"","sources":["../../../src/codemods/v0-8/ui-package-split.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,sCAAmC;AAE/D,MAAM,cAAc,GAAa;IAC/B,sBAAsB;IACtB,iBAAiB;IACjB,cAAc;IACd,qBAAqB;IACrB,mBAAmB;IACnB,wBAAwB;IACxB,eAAe;IACf,kBAAkB;IAClB,2BAA2B;IAC3B,oBAAoB;IACpB,kBAAkB;IAClB,gBAAgB;IAChB,cAAc;IACd,UAAU;IACV,aAAa;IACb,cAAc;IACd,cAAc;IACd,QAAQ;IACR,YAAY;IACZ,gBAAgB;IAChB,eAAe;IACf,aAAa;IACb,kBAAkB;IAClB,uBAAuB;IACvB,YAAY;CACb,CAAC;AAEF,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE;IAC1E,MAAM,gBAAgB,GAAa;QACjC,qBAAqB;QACrB,8BAA8B;KAC/B,CAAC;IACF,MAAM,eAAe,GAAU,EAAE,CAAC;IAClC,IAAI,sBAAsB,GAAQ,IAAI,CAAC;IAEvC,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC;SACzB,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE,CAAC,gBAAgB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;SACzE,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;QACrB,IAAI,qBAAqB,GAAG,KAAK,CAAC;QAClC,MAAM,mBAAmB,GAAU,EAAE,CAAC;QACtC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,SAAc,EAAE,EAAE;YAC/C,IACE,CAAC,CAAC,eAAe,CAAC,KAAK,CAAC,SAAS,CAAC;gBAClC,cAAc,CAAC,QAAQ,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAc,CAAC,EAC1D,CAAC;gBACD,eAAe,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAChC,qBAAqB,GAAG,IAAI,CAAC;YAC/B,CAAC;iBAAM,CAAC;gBACN,mBAAmB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,qBAAqB,EAAE,CAAC;YAC1B,sBAAsB,GAAG,IAAI,CAAC;QAChC,CAAC;QACD,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrC,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACjB,aAAa,EAAE,CAAC;QAClB,CAAC;aAAM,IAAI,mBAAmB,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC;YACvE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,mBAAmB,CAAC;YAC5C,aAAa,EAAE,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,EAAE;YAC3D,MAAM,EAAE,EAAE,KAAK,EAAE,wBAAwB,EAAE;SAC5C,CAAC,CAAC;QACH,IAAI,qBAAqB,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;YACrC,qBAAqB,CAAC,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;gBAC1C,eAAe,CAAC,OAAO,CAAC,CAAC,SAAc,EAAE,EAAE;oBACzC,IACE,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CACzB,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,CAAC,QAAQ,CAAC,IAAI,CACxD,EACD,CAAC;wBACD,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;oBACxC,CAAC;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,SAAS,GAAG,CAAC,CAAC,iBAAiB,CACnC,eAAe,EACf,CAAC,CAAC,OAAO,CAAC,wBAAwB,CAAC,CACpC,CAAC;YACF,IAAI,sBAAsB,EAAE,CAAC;gBAC3B,CAAC,CAAC,sBAAsB,CAAC,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC;YACnD,CAAC;iBAAM,CAAC;gBACN,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;gBACzD,IAAI,WAAW,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;oBAC3B,WAAW,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;gBACtC,CAAC;qBAAM,CAAC;oBACN,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAClD,CAAC;YACH,CAAC;QACH,CAAC;QACD,aAAa,EAAE,CAAC;IAClB,CAAC;IAED,MAAM,eAAe,GAA2B;QAC9C,sCAAsC,EACpC,yCAAyC;QAC3C,sCAAsC,EACpC,yCAAyC;QAC3C,kDAAkD,EAChD,4CAA4C;KAC/C,CAAC;IAEF,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;QACnD,MAAM,WAAW,GAAW,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;QACpD,IAAI,eAAe,CAAC,WAAW,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,WAAW,CAAC,CAAC,CAAC;YAC5D,aAAa,EAAE,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,IAAI,qBAAqB,GAAG,KAAK,CAAC;IAClC,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC;SACvD,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACpC,OAAO,CACL,GAAG;YACH,CAAC,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,CAAC;YACxD,GAAG,CAAC,KAAK,KAAK,0CAA0C,CACzD,CAAC;IACJ,CAAC,CAAC;SACD,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;QACrB,qBAAqB,GAAG,IAAI,CAAC;QAC7B,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC;QAC/B,IAAI,MAAM,EAAE,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;YAChE,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC;YAClC,IACE,OAAO,EAAE,KAAK,CAAC,YAAY;gBAC3B,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EACvC,CAAC;gBACD,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,CAAC;YACtB,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,KAAK,CAAC,YAAY,GAAG,OAAO,CAAC,KAAK,CAAC,YAAY,CAAC,MAAM,CAC5D,CAAC,IAAS,EAAE,EAAE,CAAC,IAAI,KAAK,MAAM,CAAC,KAAK,CACrC,CAAC;YACJ,CAAC;YACD,aAAa,EAAE,CAAC;QAClB,CAAC;aAAM,CAAC;YACN,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,CAAC;YACjB,aAAa,EAAE,CAAC;QAClB,CAAC;IACH,CAAC,CAAC,CAAC;IAEL,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,CAAC;SACvD,MAAM,CAAC,CAAC,IAAS,EAAE,EAAE;QACpB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QACpC,OAAO,CACL,GAAG;YACH,CAAC,GAAG,CAAC,IAAI,KAAK,SAAS,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,CAAC;YACxD,GAAG,CAAC,KAAK,KAAK,iCAAiC,CAChD,CAAC;IACJ,CAAC,CAAC;SACD,OAAO,CAAC,CAAC,IAAS,EAAE,EAAE;QACrB,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,oCAAoC,CAAC;QACrE,aAAa,EAAE,CAAC;QAChB,IAAI,qBAAqB,EAAE,CAAC;YAC1B,IACE,IAAI,CAAC,UAAU,EAAE,KAAK;gBACtB,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,KAAK,gBAAgB;gBAC/C,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAC1C,CAAC;gBACD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;gBACrD,IAAI,SAAS,IAAI,SAAS,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;oBACvD,MAAM,cAAc,GAAG,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAS,EAAE,EAAE;wBAC7D,OAAO,CACL,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,KAAK,YAAY;4BAC7B,IAAI,CAAC,GAAG,CAAC,KAAK,KAAK,YAAY,CAAC;4BAClC,IAAI,CAAC,KAAK,CAAC,IAAI,KAAK,iBAAiB,CACtC,CAAC;oBACJ,CAAC,CAAC,CAAC;oBACH,IAAI,cAAc,EAAE,CAAC;wBACnB,MAAM,eAAe,GAAG,cAAc,CAAC,KAAK,CAAC,QAAQ,CAAC;wBACtD,MAAM,WAAW,GAAG,eAAe,CAAC,IAAI,CACtC,CAAC,EAAO,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,KAAK,SAAS,IAAI,EAAE,CAAC,KAAK,KAAK,UAAU,CAC9D,CAAC;wBACF,IAAI,CAAC,WAAW,EAAE,CAAC;4BACjB,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;4BAC5C,aAAa,EAAE,CAAC;wBAClB,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC,CAAC,CAAC;AAEH,eAAe,kBAAkB,CAAC"}
+{"version":3,"file":"ui-package-split.js","names":[],"sources":["../../../src/codemods/v0-8/ui-package-split.ts"],"sourcesContent":["import { createTransformer } from \"../utils/createTransformer\";\n\nconst reactUIExports: string[] = [\n  \"ThreadConfigProvider\",\n  \"useThreadConfig\",\n  \"ThreadConfig\",\n  \"ThreadWelcomeConfig\",\n  \"UserMessageConfig\",\n  \"AssistantMessageConfig\",\n  \"StringsConfig\",\n  \"SuggestionConfig\",\n  \"ThreadConfigProviderProps\",\n  \"AssistantActionBar\",\n  \"AssistantMessage\",\n  \"AssistantModal\",\n  \"BranchPicker\",\n  \"Composer\",\n  \"MessagePart\",\n  \"AttachmentUI\",\n  \"EditComposer\",\n  \"Thread\",\n  \"ThreadList\",\n  \"ThreadListItem\",\n  \"ThreadWelcome\",\n  \"UserMessage\",\n  \"makeMarkdownText\",\n  \"MakeMarkdownTextProps\",\n  \"CodeHeader\",\n];\n\nconst migrateAssistantUI = createTransformer(({ j, root, markAsChanged }) => {\n  const sourcesToMigrate: string[] = [\n    \"@assistant-ui/react\",\n    \"@assistant-ui/react-markdown\",\n  ];\n  const movedSpecifiers: any[] = [];\n  let lastMigratedImportPath: any = null;\n\n  root\n    .find(j.ImportDeclaration)\n    .filter((path: any) => sourcesToMigrate.includes(path.value.source.value))\n    .forEach((path: any) => {\n      let hadMigratedSpecifiers = false;\n      const remainingSpecifiers: any[] = [];\n      path.value.specifiers.forEach((specifier: any) => {\n        if (\n          j.ImportSpecifier.check(specifier) &&\n          reactUIExports.includes(specifier.imported.name as string)\n        ) {\n          movedSpecifiers.push(specifier);\n          hadMigratedSpecifiers = true;\n        } else {\n          remainingSpecifiers.push(specifier);\n        }\n      });\n      if (hadMigratedSpecifiers) {\n        lastMigratedImportPath = path;\n      }\n      if (remainingSpecifiers.length === 0) {\n        j(path).remove();\n        markAsChanged();\n      } else if (remainingSpecifiers.length !== path.value.specifiers.length) {\n        path.value.specifiers = remainingSpecifiers;\n        markAsChanged();\n      }\n    });\n\n  if (movedSpecifiers.length > 0) {\n    const existingReactUIImport = root.find(j.ImportDeclaration, {\n      source: { value: \"@assistant-ui/react-ui\" },\n    });\n    if (existingReactUIImport.size() > 0) {\n      existingReactUIImport.forEach((path: any) => {\n        movedSpecifiers.forEach((specifier: any) => {\n          if (\n            !path.value.specifiers.some(\n              (s: any) => s.imported.name === specifier.imported.name,\n            )\n          ) {\n            path.value.specifiers.push(specifier);\n          }\n        });\n      });\n    } else {\n      const newImport = j.importDeclaration(\n        movedSpecifiers,\n        j.literal(\"@assistant-ui/react-ui\"),\n      );\n      if (lastMigratedImportPath) {\n        j(lastMigratedImportPath).insertAfter(newImport);\n      } else {\n        const firstImport = root.find(j.ImportDeclaration).at(0);\n        if (firstImport.size() > 0) {\n          firstImport.insertBefore(newImport);\n        } else {\n          root.get().node.program.body.unshift(newImport);\n        }\n      }\n    }\n    markAsChanged();\n  }\n\n  const cssReplacements: Record<string, string> = {\n    \"@assistant-ui/react/styles/index.css\":\n      \"@assistant-ui/react-ui/styles/index.css\",\n    \"@assistant-ui/react/styles/modal.css\":\n      \"@assistant-ui/react-ui/styles/modal.css\",\n    \"@assistant-ui/react-markdown/styles/markdown.css\":\n      \"@assistant-ui/react-ui/styles/markdown.css\",\n  };\n\n  root.find(j.ImportDeclaration).forEach((path: any) => {\n    const sourceValue: string = path.value.source.value;\n    if (cssReplacements[sourceValue]) {\n      path.value.source = j.literal(cssReplacements[sourceValue]);\n      markAsChanged();\n    }\n  });\n\n  let removedMarkdownPlugin = false;\n  root\n    .find(j.CallExpression, { callee: { name: \"require\" } })\n    .filter((path: any) => {\n      const arg = path.value.arguments[0];\n      return (\n        arg &&\n        (arg.type === \"Literal\" || arg.type === \"StringLiteral\") &&\n        arg.value === \"@assistant-ui/react-markdown/tailwindcss\"\n      );\n    })\n    .forEach((path: any) => {\n      removedMarkdownPlugin = true;\n      const parent = path.parentPath;\n      if (parent?.value && parent.value.type === \"VariableDeclarator\") {\n        const varDecl = parent.parentPath;\n        if (\n          varDecl?.value.declarations &&\n          varDecl.value.declarations.length === 1\n        ) {\n          j(varDecl).remove();\n        } else {\n          varDecl.value.declarations = varDecl.value.declarations.filter(\n            (decl: any) => decl !== parent.value,\n          );\n        }\n        markAsChanged();\n      } else {\n        j(path).remove();\n        markAsChanged();\n      }\n    });\n\n  root\n    .find(j.CallExpression, { callee: { name: \"require\" } })\n    .filter((path: any) => {\n      const arg = path.value.arguments[0];\n      return (\n        arg &&\n        (arg.type === \"Literal\" || arg.type === \"StringLiteral\") &&\n        arg.value === \"@assistant-ui/react/tailwindcss\"\n      );\n    })\n    .forEach((path: any) => {\n      path.value.arguments[0].value = \"@assistant-ui/react-ui/tailwindcss\";\n      markAsChanged();\n      if (removedMarkdownPlugin) {\n        if (\n          path.parentPath?.value &&\n          path.parentPath.value.type === \"CallExpression\" &&\n          path.parentPath.value.arguments.length > 0\n        ) {\n          const configObj = path.parentPath.value.arguments[0];\n          if (configObj && configObj.type === \"ObjectExpression\") {\n            const componentsProp = configObj.properties.find((prop: any) => {\n              return (\n                (prop.key.name === \"components\" ||\n                  prop.key.value === \"components\") &&\n                prop.value.type === \"ArrayExpression\"\n              );\n            });\n            if (componentsProp) {\n              const componentsArray = componentsProp.value.elements;\n              const hasMarkdown = componentsArray.some(\n                (el: any) => el.type === \"Literal\" && el.value === \"markdown\",\n              );\n              if (!hasMarkdown) {\n                componentsArray.push(j.literal(\"markdown\"));\n                markAsChanged();\n              }\n            }\n          }\n        }\n      }\n    });\n});\n\nexport default migrateAssistantUI;\n"],"mappings":";;AAEA,MAAM,iBAA2B;CAC/B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;AACF;AAEA,MAAM,qBAAqB,mBAAmB,EAAE,GAAG,MAAM,oBAAoB;CAC3E,MAAM,mBAA6B,CACjC,uBACA,8BACF;CACA,MAAM,kBAAyB,CAAC;CAChC,IAAI,yBAA8B;CAElC,KACG,KAAK,EAAE,iBAAiB,EACxB,QAAQ,SAAc,iBAAiB,SAAS,KAAK,MAAM,OAAO,KAAK,CAAC,EACxE,SAAS,SAAc;EACtB,IAAI,wBAAwB;EAC5B,MAAM,sBAA6B,CAAC;EACpC,KAAK,MAAM,WAAW,SAAS,cAAmB;GAChD,IACE,EAAE,gBAAgB,MAAM,SAAS,KACjC,eAAe,SAAS,UAAU,SAAS,IAAc,GACzD;IACA,gBAAgB,KAAK,SAAS;IAC9B,wBAAwB;GAC1B,OACE,oBAAoB,KAAK,SAAS;EAEtC,CAAC;EACD,IAAI,uBACF,yBAAyB;EAE3B,IAAI,oBAAoB,WAAW,GAAG;GACpC,EAAE,IAAI,EAAE,OAAO;GACf,cAAc;EAChB,OAAO,IAAI,oBAAoB,WAAW,KAAK,MAAM,WAAW,QAAQ;GACtE,KAAK,MAAM,aAAa;GACxB,cAAc;EAChB;CACF,CAAC;CAEH,IAAI,gBAAgB,SAAS,GAAG;EAC9B,MAAM,wBAAwB,KAAK,KAAK,EAAE,mBAAmB,EAC3D,QAAQ,EAAE,OAAO,yBAAyB,EAC5C,CAAC;EACD,IAAI,sBAAsB,KAAK,IAAI,GACjC,sBAAsB,SAAS,SAAc;GAC3C,gBAAgB,SAAS,cAAmB;IAC1C,IACE,CAAC,KAAK,MAAM,WAAW,MACpB,MAAW,EAAE,SAAS,SAAS,UAAU,SAAS,IACrD,GAEA,KAAK,MAAM,WAAW,KAAK,SAAS;GAExC,CAAC;EACH,CAAC;OACI;GACL,MAAM,YAAY,EAAE,kBAClB,iBACA,EAAE,QAAQ,wBAAwB,CACpC;GACA,IAAI,wBACF,EAAE,sBAAsB,EAAE,YAAY,SAAS;QAC1C;IACL,MAAM,cAAc,KAAK,KAAK,EAAE,iBAAiB,EAAE,GAAG,CAAC;IACvD,IAAI,YAAY,KAAK,IAAI,GACvB,YAAY,aAAa,SAAS;SAElC,KAAK,IAAI,EAAE,KAAK,QAAQ,KAAK,QAAQ,SAAS;GAElD;EACF;EACA,cAAc;CAChB;CAEA,MAAM,kBAA0C;EAC9C,wCACE;EACF,wCACE;EACF,oDACE;CACJ;CAEA,KAAK,KAAK,EAAE,iBAAiB,EAAE,SAAS,SAAc;EACpD,MAAM,cAAsB,KAAK,MAAM,OAAO;EAC9C,IAAI,gBAAgB,cAAc;GAChC,KAAK,MAAM,SAAS,EAAE,QAAQ,gBAAgB,YAAY;GAC1D,cAAc;EAChB;CACF,CAAC;CAED,IAAI,wBAAwB;CAC5B,KACG,KAAK,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,UAAU,EAAE,CAAC,EACtD,QAAQ,SAAc;EACrB,MAAM,MAAM,KAAK,MAAM,UAAU;EACjC,OACE,QACC,IAAI,SAAS,aAAa,IAAI,SAAS,oBACxC,IAAI,UAAU;CAElB,CAAC,EACA,SAAS,SAAc;EACtB,wBAAwB;EACxB,MAAM,SAAS,KAAK;EACpB,IAAI,QAAQ,SAAS,OAAO,MAAM,SAAS,sBAAsB;GAC/D,MAAM,UAAU,OAAO;GACvB,IACE,SAAS,MAAM,gBACf,QAAQ,MAAM,aAAa,WAAW,GAEtC,EAAE,OAAO,EAAE,OAAO;QAElB,QAAQ,MAAM,eAAe,QAAQ,MAAM,aAAa,QACrD,SAAc,SAAS,OAAO,KACjC;GAEF,cAAc;EAChB,OAAO;GACL,EAAE,IAAI,EAAE,OAAO;GACf,cAAc;EAChB;CACF,CAAC;CAEH,KACG,KAAK,EAAE,gBAAgB,EAAE,QAAQ,EAAE,MAAM,UAAU,EAAE,CAAC,EACtD,QAAQ,SAAc;EACrB,MAAM,MAAM,KAAK,MAAM,UAAU;EACjC,OACE,QACC,IAAI,SAAS,aAAa,IAAI,SAAS,oBACxC,IAAI,UAAU;CAElB,CAAC,EACA,SAAS,SAAc;EACtB,KAAK,MAAM,UAAU,GAAG,QAAQ;EAChC,cAAc;EACd,IAAI;OAEA,KAAK,YAAY,SACjB,KAAK,WAAW,MAAM,SAAS,oBAC/B,KAAK,WAAW,MAAM,UAAU,SAAS,GACzC;IACA,MAAM,YAAY,KAAK,WAAW,MAAM,UAAU;IAClD,IAAI,aAAa,UAAU,SAAS,oBAAoB;KACtD,MAAM,iBAAiB,UAAU,WAAW,MAAM,SAAc;MAC9D,QACG,KAAK,IAAI,SAAS,gBACjB,KAAK,IAAI,UAAU,iBACrB,KAAK,MAAM,SAAS;KAExB,CAAC;KACD,IAAI,gBAAgB;MAClB,MAAM,kBAAkB,eAAe,MAAM;MAI7C,IAAI,CAHgB,gBAAgB,MACjC,OAAY,GAAG,SAAS,aAAa,GAAG,UAAU,UAEtC,GAAG;OAChB,gBAAgB,KAAK,EAAE,QAAQ,UAAU,CAAC;OAC1C,cAAc;MAChB;KACF;IACF;GACF;;CAEJ,CAAC;AACL,CAAC"}

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

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

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

@@ -1 +1 @@
-{"version":3,"file":"edge-package-split.d.ts","sourceRoot":"","sources":["../../../src/codemods/v0-9/edge-package-split.ts"],"names":[],"mappings":"AA8BA,QAAA,MAAM,oBAAoB,6HAkJxB,CAAC;AAEH,eAAe,oBAAoB,CAAC"}
+{"version":3,"file":"edge-package-split.d.ts","names":[],"sources":["../../../src/codemods/v0-9/edge-package-split.ts"],"mappings":";cA8BM,oBAAA,GAAoB,QAAA,iCAAA,QAAA,EAAA,GAAA,iCAAA,GAAA,EAAA,OAAA"}

package/dist/codemods/v0-9/edge-package-split.js

@@ -1,146 +1,102 @@
 import { createTransformer } from "../utils/createTransformer.js";
-// Chat runtime and converter exports moved to react-ai-sdk
+//#region src/codemods/v0-9/edge-package-split.ts
 const reactAiSdkExports = [
-    // Chat Runtime (replaces edge runtime)
-    "useChatRuntime",
-    // Core Types
-    "CoreMessage",
-    "CoreUserMessage",
-    "CoreAssistantMessage",
-    "CoreSystemMessage",
-    "CoreUserMessagePart",
-    "CoreAssistantMessagePart",
-    "CoreToolCallMessagePart",
-    // Core message converters
-    "fromCoreMessages",
-    "fromCoreMessage",
-    "toCoreMessages",
-    "toCoreMessage",
-    // Language model converters
-    "toLanguageModelMessages",
-    "toLanguageModelTools",
-    "fromLanguageModelMessages",
-    "fromLanguageModelTools",
-    "useDangerousInBrowserRuntime",
+	"useChatRuntime",
+	"CoreMessage",
+	"CoreUserMessage",
+	"CoreAssistantMessage",
+	"CoreSystemMessage",
+	"CoreUserMessagePart",
+	"CoreAssistantMessagePart",
+	"CoreToolCallMessagePart",
+	"fromCoreMessages",
+	"fromCoreMessage",
+	"toCoreMessages",
+	"toCoreMessage",
+	"toLanguageModelMessages",
+	"toLanguageModelTools",
+	"fromLanguageModelMessages",
+	"fromLanguageModelTools",
+	"useDangerousInBrowserRuntime"
 ];
 const migrateToEdgePackage = createTransformer(({ j, root, markAsChanged }) => {
-    const sourcesToMigrate = ["@assistant-ui/react"];
-    const movedAiSdkSpecifiers = [];
-    let lastMigratedImportPath = null;
-    root
-        .find(j.ImportDeclaration)
-        .filter((path) => sourcesToMigrate.includes(path.value.source.value))
-        .forEach((path) => {
-        let hadMigratedSpecifiers = false;
-        const remainingSpecifiers = [];
-        path.value.specifiers.forEach((specifier) => {
-            if (j.ImportSpecifier.check(specifier) &&
-                reactAiSdkExports.includes(specifier.imported.name)) {
-                movedAiSdkSpecifiers.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();
-        }
-    });
-    // Add imports for react-ai-sdk (all edge-related imports now go here)
-    if (movedAiSdkSpecifiers.length > 0) {
-        const existingAiSdkImport = root.find(j.ImportDeclaration, {
-            source: { value: "@assistant-ui/react-ai-sdk" },
-        });
-        if (existingAiSdkImport.size() > 0) {
-            existingAiSdkImport.forEach((path) => {
-                movedAiSdkSpecifiers.forEach((specifier) => {
-                    if (!path.value.specifiers.some((s) => s.imported.name === specifier.imported.name)) {
-                        path.value.specifiers.push(specifier);
-                    }
-                });
-            });
-        }
-        else {
-            const newImport = j.importDeclaration(movedAiSdkSpecifiers, j.literal("@assistant-ui/react-ai-sdk"));
-            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();
-    }
-    // Migrate imports from edge/converters
-    root.find(j.ImportDeclaration).forEach((path) => {
-        const sourceValue = path.value.source.value;
-        if (sourceValue.startsWith("@assistant-ui/react/") &&
-            (sourceValue.includes("edge/") ||
-                sourceValue.includes("dangerous-in-browser/"))) {
-            path.value.source = j.literal(sourceValue.replace("@assistant-ui/react/", "@assistant-ui/react-ai-sdk/"));
-            markAsChanged();
-        }
-    });
-    // Migrate language model converter imports from react-edge to react-ai-sdk
-    root.find(j.ImportDeclaration).forEach((path) => {
-        const sourceValue = path.value.source.value;
-        if (sourceValue === "@assistant-ui/react-edge" ||
-            sourceValue === "@assistant-ui/react") {
-            let hasLanguageModelConverters = false;
-            const remainingSpecifiers = [];
-            const aiSdkSpecifiers = [];
-            path.value.specifiers.forEach((specifier) => {
-                if (j.ImportSpecifier.check(specifier) &&
-                    reactAiSdkExports.includes(specifier.imported.name)) {
-                    aiSdkSpecifiers.push(specifier);
-                    hasLanguageModelConverters = true;
-                }
-                else {
-                    remainingSpecifiers.push(specifier);
-                }
-            });
-            if (hasLanguageModelConverters) {
-                if (remainingSpecifiers.length === 0) {
-                    j(path).remove();
-                }
-                else {
-                    path.value.specifiers = remainingSpecifiers;
-                }
-                const existingAiSdkImport = root.find(j.ImportDeclaration, {
-                    source: { value: "@assistant-ui/react-ai-sdk" },
-                });
-                if (existingAiSdkImport.size() > 0) {
-                    existingAiSdkImport.forEach((importPath) => {
-                        aiSdkSpecifiers.forEach((specifier) => {
-                            if (!importPath.value.specifiers.some((s) => s.imported.name === specifier.imported.name)) {
-                                importPath.value.specifiers.push(specifier);
-                            }
-                        });
-                    });
-                }
-                else {
-                    const newImport = j.importDeclaration(aiSdkSpecifiers, j.literal("@assistant-ui/react-ai-sdk"));
-                    j(path).insertAfter(newImport);
-                }
-                markAsChanged();
-            }
-        }
-    });
+	const sourcesToMigrate = ["@assistant-ui/react"];
+	const movedAiSdkSpecifiers = [];
+	let lastMigratedImportPath = null;
+	root.find(j.ImportDeclaration).filter((path) => sourcesToMigrate.includes(path.value.source.value)).forEach((path) => {
+		let hadMigratedSpecifiers = false;
+		const remainingSpecifiers = [];
+		path.value.specifiers.forEach((specifier) => {
+			if (j.ImportSpecifier.check(specifier) && reactAiSdkExports.includes(specifier.imported.name)) {
+				movedAiSdkSpecifiers.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 (movedAiSdkSpecifiers.length > 0) {
+		const existingAiSdkImport = root.find(j.ImportDeclaration, { source: { value: "@assistant-ui/react-ai-sdk" } });
+		if (existingAiSdkImport.size() > 0) existingAiSdkImport.forEach((path) => {
+			movedAiSdkSpecifiers.forEach((specifier) => {
+				if (!path.value.specifiers.some((s) => s.imported.name === specifier.imported.name)) path.value.specifiers.push(specifier);
+			});
+		});
+		else {
+			const newImport = j.importDeclaration(movedAiSdkSpecifiers, j.literal("@assistant-ui/react-ai-sdk"));
+			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();
+	}
+	root.find(j.ImportDeclaration).forEach((path) => {
+		const sourceValue = path.value.source.value;
+		if (sourceValue.startsWith("@assistant-ui/react/") && (sourceValue.includes("edge/") || sourceValue.includes("dangerous-in-browser/"))) {
+			path.value.source = j.literal(sourceValue.replace("@assistant-ui/react/", "@assistant-ui/react-ai-sdk/"));
+			markAsChanged();
+		}
+	});
+	root.find(j.ImportDeclaration).forEach((path) => {
+		const sourceValue = path.value.source.value;
+		if (sourceValue === "@assistant-ui/react-edge" || sourceValue === "@assistant-ui/react") {
+			let hasLanguageModelConverters = false;
+			const remainingSpecifiers = [];
+			const aiSdkSpecifiers = [];
+			path.value.specifiers.forEach((specifier) => {
+				if (j.ImportSpecifier.check(specifier) && reactAiSdkExports.includes(specifier.imported.name)) {
+					aiSdkSpecifiers.push(specifier);
+					hasLanguageModelConverters = true;
+				} else remainingSpecifiers.push(specifier);
+			});
+			if (hasLanguageModelConverters) {
+				if (remainingSpecifiers.length === 0) j(path).remove();
+				else path.value.specifiers = remainingSpecifiers;
+				const existingAiSdkImport = root.find(j.ImportDeclaration, { source: { value: "@assistant-ui/react-ai-sdk" } });
+				if (existingAiSdkImport.size() > 0) existingAiSdkImport.forEach((importPath) => {
+					aiSdkSpecifiers.forEach((specifier) => {
+						if (!importPath.value.specifiers.some((s) => s.imported.name === specifier.imported.name)) importPath.value.specifiers.push(specifier);
+					});
+				});
+				else {
+					const newImport = j.importDeclaration(aiSdkSpecifiers, j.literal("@assistant-ui/react-ai-sdk"));
+					j(path).insertAfter(newImport);
+				}
+				markAsChanged();
+			}
+		}
+	});
 });
-export default migrateToEdgePackage;
+//#endregion
+export { migrateToEdgePackage as default };
+
 //# sourceMappingURL=edge-package-split.js.map