feat (packages/codemod): Add codemod to rm old provider registry expo…

…rts. (#3649)

.changeset/slimy-zoos-double.md

@@ -0,0 +1,5 @@
+---
+'@ai-sdk/codemod': patch
+---
+
+feat (providers/codemod): Add codemod to remove deprecated prov reg exports.

packages/codemod/src/codemods/remove-deprecated-provider-registry-exports.ts

@@ -0,0 +1,80 @@
+import { API, FileInfo } from 'jscodeshift';
+
+export default function transformer(file: FileInfo, api: API) {
+  const j = api.jscodeshift;
+  const root = j(file.source);
+
+  // Replace imports
+  root
+    .find(j.ImportDeclaration)
+    .filter(path => path.node.source.value === 'ai')
+    .forEach(path => {
+      const newSpecifiers = path.node.specifiers
+        ?.map(spec => {
+          if (spec.type !== 'ImportSpecifier') return spec;
+
+          const oldName = spec.imported.name;
+          if (oldName === 'experimental_createModelRegistry') {
+            return j.importSpecifier(
+              j.identifier('experimental_createProviderRegistry'),
+            );
+          }
+
+          if (
+            [
+              'experimental_Provider',
+              'experimental_ProviderRegistry',
+              'experimental_ModelRegistry',
+            ].includes(oldName)
+          ) {
+            return j.importSpecifier(j.identifier('Provider'));
+          }
+
+          return spec;
+        })
+        .filter((spec, index, arr) => {
+          // Deduplicate specifiers
+          if (!spec) return false;
+          return (
+            arr.findIndex(
+              s =>
+                s?.type === 'ImportSpecifier' &&
+                spec.type === 'ImportSpecifier' &&
+                s.imported.name === spec.imported.name,
+            ) === index
+          );
+        });
+
+      path.node.specifiers = newSpecifiers;
+    });
+
+  // Replace type references
+  root
+    .find(j.TSTypeReference)
+    .filter(
+      path =>
+        path.node.typeName.type === 'Identifier' &&
+        [
+          'experimental_Provider',
+          'experimental_ProviderRegistry',
+          'experimental_ModelRegistry',
+        ].includes(path.node.typeName.name),
+    )
+    .forEach(path => {
+      path.node.typeName = j.identifier('Provider');
+    });
+
+  // Replace function calls
+  root
+    .find(j.CallExpression, {
+      callee: {
+        type: 'Identifier',
+        name: 'experimental_createModelRegistry',
+      },
+    })
+    .forEach(path => {
+      path.node.callee = j.identifier('experimental_createProviderRegistry');
+    });
+
+  return root.toSource();
+}

packages/codemod/src/lib/upgrade.ts

@@ -16,6 +16,7 @@ const bundle = [
   'replace-nanoid',
   'replace-token-usage-types',
   'rewrite-framework-imports',
+  'remove-deprecated-provider-registry-exports',
   'remove-experimental-message-types',
   'replace-roundtrips-with-maxsteps',
 ];

packages/codemod/src/test/__testfixtures__/remove-deprecated-provider-registry-exports.input.ts

@@ -0,0 +1,17 @@
+// @ts-nocheck
+import { experimental_Provider, experimental_ProviderRegistry, experimental_ModelRegistry, experimental_createModelRegistry } from 'ai';
+
+function createProvider(): experimental_Provider {
+  return {
+    languageModel: () => null,
+    textEmbeddingModel: () => null
+  };
+}
+
+function createRegistry(): experimental_ProviderRegistry {
+  return experimental_createModelRegistry({
+    test: createProvider()
+  });
+}
+
+const registry: experimental_ModelRegistry = createRegistry();

packages/codemod/src/test/__testfixtures__/remove-deprecated-provider-registry-exports.output.ts

@@ -0,0 +1,17 @@
+// @ts-nocheck
+import { Provider, experimental_createProviderRegistry } from 'ai';
+
+function createProvider(): Provider {
+  return {
+    languageModel: () => null,
+    textEmbeddingModel: () => null
+  };
+}
+
+function createRegistry(): Provider {
+  return experimental_createProviderRegistry({
+    test: createProvider()
+  });
+}
+
+const registry: Provider = createRegistry();

packages/codemod/src/test/remove-deprecated-provider-registry-exports.test.ts

@@ -0,0 +1,9 @@
+import { describe, it } from 'vitest';
+import transformer from '../codemods/remove-deprecated-provider-registry-exports';
+import { testTransform } from './test-utils';
+
+describe('remove-deprecated-provider-registry-exports', () => {
+  it('transforms correctly', () => {
+    testTransform(transformer, 'remove-deprecated-provider-registry-exports');
+  });
+});