@mastra/codemod 0.1.0-beta.0 → 0.1.0-beta.1

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/codemod
 
+## 0.1.0-beta.1
+
+### Patch Changes
+
+- - Improve existing codemods ([#9959](https://github.com/mastra-ai/mastra/pull/9959))
+  - Make package ESM-only
+  - Add new codemods
+
 ## 0.1.0-beta.0
 
 ### Minor Changes

package/README.md

@@ -9,7 +9,7 @@ Codemods are transformations that run on your codebase programmatically, allowin
 ### Run Version-Specific Codemods
 
 ```sh
-npx @mastra/codemod v1
+npx @mastra/codemod@beta v1
 ```
 
 ### Run Individual Codemods
@@ -17,49 +17,66 @@ npx @mastra/codemod v1
 To run a specific codemod:
 
 ```sh
-npx @mastra/codemod <codemod-name> <path>
+npx @mastra/codemod@beta <codemod-name> <path>
 ```
 
 Examples:
 
 ```sh
 # Transform a specific file
-npx @mastra/codemod v1/mastra-core-imports src/mastra.ts
+npx @mastra/codemod@beta v1/mastra-core-imports src/mastra.ts
 
 # Transform a directory
-npx @mastra/codemod v1/mastra-core-imports src/lib/
+npx @mastra/codemod@beta v1/mastra-core-imports src/lib/
 
 # Transform entire project
-npx @mastra/codemod v1/mastra-core-imports .
+npx @mastra/codemod@beta v1/mastra-core-imports .
 ```
 
 ## Available Codemods
 
 ### v1 Codemods (v0 → v1 Migration)
 
-| Codemod                           | Description                                                                                                                                                                              |
-| --------------------------------- | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
-| `v1/mastra-core-imports`          | Updates all imports from `@mastra/core` to use the new subpath imports. For v1, all exports except `Mastra` and `Config` have moved to subpaths.                                         |
-| `v1/runtime-context`              | Renames `RuntimeContext` to `RequestContext` and updates all parameter names from `runtimeContext` to `requestContext` across all APIs, including string literals in middleware.         |
-| `v1/agent-generate-stream-v-next` | Transforms Agent VNext methods: `agent.generateVNext()` → `agent.generate()` and `agent.streamVNext()` → `agent.stream()`                                                                |
-| `v1/agent-get-agents`             | Transforms Mastra method: `mastra.getAgents()` → `mastra.listAgents()`                                                                                                                   |
-| `v1/agent-processor-methods`      | Transforms Agent processor methods: `agent.getInputProcessors()` → `agent.listInputProcessors()` and `agent.getOutputProcessors()` → `agent.listOutputProcessors()`                      |
-| `v1/agent-property-access`        | Transforms Agent property access to method calls: `agent.llm` → `agent.getLLM()`, `agent.tools` → `agent.getTools()`, `agent.instructions` → `agent.getInstructions()`                   |
-| `v1/agent-voice`                  | Transforms Agent voice methods to use namespace: `agent.speak()` → `agent.voice.speak()`, `agent.listen()` → `agent.voice.listen()`, `agent.getSpeakers()` → `agent.voice.getSpeakers()` |
-| `v1/evals-get-scorers`            | Transforms Mastra method: `mastra.getScorers()` → `mastra.listScorers()`                                                                                                                 |
-| `v1/experimental-auth`            | Renames `experimental_auth` to `auth` in Mastra configuration                                                                                                                            |
-| `v1/mcp-get-mcp-servers`          | Transforms Mastra method: `mastra.getMCPServers()` → `mastra.listMCPServers()`                                                                                                           |
-| `v1/mcp-get-tools`                | Transforms MCPServer method: `mcp.getTools()` → `mcp.listTools()`                                                                                                                        |
-| `v1/mcp-get-toolsets`             | Transforms MCPServer method: `mcp.getToolsets()` → `mcp.listToolsets()`                                                                                                                  |
-| `v1/voice-property-names`         | Transforms voice property names in Agent configuration: `speakProvider` → `output`, `listenProvider` → `input`, `realtimeProvider` → `realtime`                                          |
-| `v1/workflows-get-workflows`      | Transforms Mastra method: `mastra.getWorkflows()` → `mastra.listWorkflows()`                                                                                                             |
+| Codemod                              | Description                                                                                                        |
+| ------------------------------------ | ------------------------------------------------------------------------------------------------------------------ |
+| `v1/agent-abort-signal`              | Moves `abortSignal` from `modelSettings` to top-level options in agent method calls                                |
+| `v1/agent-generate-stream-v-next`    | Renames `agent.generateVNext()` → `agent.generate()` and `agent.streamVNext()` → `agent.stream()`                  |
+| `v1/agent-processor-methods`         | Renames `agent.getInputProcessors()` → `agent.listInputProcessors()` and similar output processor methods          |
+| `v1/agent-property-access`           | Transforms agent property access to method calls: `agent.llm` → `agent.getLLM()`                                   |
+| `v1/agent-voice`                     | Moves agent voice methods to namespace: `agent.speak()` → `agent.voice.speak()`                                    |
+| `v1/client-get-memory-thread`        | Updates `client.getMemoryThread(threadId, agentId)` to use object parameter                                        |
+| `v1/client-offset-limit`             | Renames pagination properties from `offset`/`limit` to `page`/`perPage`                                            |
+| `v1/client-sdk-types`                | Renames Client SDK types from Get\* to List\* pattern                                                              |
+| `v1/client-to-ai-sdk-format`         | Renames `toAISdkFormat` to `toAISdkStream`                                                                         |
+| `v1/evals-prebuilt-imports`          | Updates prebuilt scorer imports from `scorers/llm` and `scorers/code` to `scorers/prebuilt`                        |
+| `v1/evals-run-experiment`            | Renames `runExperiment()` → `runEvals()` in imports and usages                                                     |
+| `v1/evals-scorer-by-name`            | Renames `mastra.getScorerByName()` → `mastra.getScorerById()`                                                      |
+| `v1/experimental-auth`               | Renames `experimental_auth` to `auth` in Mastra configuration                                                      |
+| `v1/mastra-core-imports`             | Updates imports from `@mastra/core` to use new subpath imports                                                     |
+| `v1/mastra-plural-apis`              | Renames Mastra plural API methods from get\* to list\*                                                             |
+| `v1/mcp-get-tools`                   | Renames `mcp.getTools()` → `mcp.listTools()`                                                                       |
+| `v1/mcp-get-toolsets`                | Renames `mcp.getToolsets()` → `mcp.listToolsets()`                                                                 |
+| `v1/memory-message-v2-type`          | Renames `MastraMessageV2` type → `MastraDBMessage` in imports and usages                                           |
+| `v1/memory-query-to-recall`          | Renames `memory.query()` → `memory.recall()`                                                                       |
+| `v1/memory-vector-search-param`      | Renames `vectorMessageSearch` parameter → `vectorSearchString` in `memory.recall()` calls                          |
+| `v1/runtime-context`                 | Renames `RuntimeContext` to `RequestContext` and updates parameter names from `runtimeContext` to `requestContext` |
+| `v1/storage-get-messages-paginated`  | Renames `storage.getMessagesPaginated()` → `storage.listMessages()` and `offset`/`limit` → `page`/`perPage`        |
+| `v1/storage-get-threads-by-resource` | Renames `storage.getThreadsByResourceId()` → `storage.listThreadsByResourceId()`                                   |
+| `v1/storage-list-messages-by-id`     | Renames `storage.getMessagesById()` → `storage.listMessagesById()`                                                 |
+| `v1/storage-list-workflow-runs`      | Renames `storage.getWorkflowRuns()` → `storage.listWorkflowRuns()`                                                 |
+| `v1/storage-postgres-schema-name`    | Renames `schema` property → `schemaName` in PostgresStore constructor                                              |
+| `v1/vector-pg-constructor`           | Converts `new PgVector(connectionString)` to `new PgVector({ connectionString })`                                  |
+| `v1/voice-property-names`            | Renames voice property names in Agent configuration: `speakProvider` → `output`                                    |
+| `v1/workflow-create-run-async`       | Renames `workflow.createRunAsync()` → `workflow.createRun()`                                                       |
+| `v1/workflow-list-runs`              | Renames `workflow.getWorkflowRuns()` → `workflow.listWorkflowRuns()`                                               |
+| `v1/workflow-run-count`              | Renames `context.runCount` → `context.retryCount` in step execution functions                                      |
 
 ## CLI Options
 
 ### Commands
 
 ```sh
-npx @mastra/codemod <command> [options]
+npx @mastra/codemod@beta <command> [options]
 ```
 
 **Available Commands:**
@@ -76,10 +93,10 @@ npx @mastra/codemod <command> [options]
 
 ```sh
 # Show verbose output for specific codemod
-npx @mastra/codemod --verbose v1/mastra-core-imports src/
+npx @mastra/codemod@beta --verbose v1/mastra-core-imports src/
 
 # Print transformed code for specific codemod
-npx @mastra/codemod --print v1/mastra-core-imports src/mastra.ts
+npx @mastra/codemod@beta --print v1/mastra-core-imports src/mastra.ts
 ```
 
 ## Contributing

package/dist/codemods/chunk-JNFQ6J6B.js

@@ -0,0 +1,21 @@
+// src/codemods/lib/create-transformer.ts
+function createTransformer(transformFn) {
+  return function transformer(fileInfo, api, options) {
+    const j = api.jscodeshift;
+    const root = j(fileInfo.source);
+    const context = {
+      j,
+      root,
+      hasChanges: false,
+      messages: []
+    };
+    transformFn(fileInfo, api, options, context);
+    context.messages.forEach((message) => api.report(message));
+    return context.hasChanges ? root.toSource({ quote: "single" }) : null;
+  };
+}
+
+export {
+  createTransformer
+};
+//# sourceMappingURL=chunk-JNFQ6J6B.js.map

package/dist/codemods/chunk-JNFQ6J6B.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/codemods/lib/create-transformer.ts"],"sourcesContent":["// Copied from https://github.com/vercel/ai/blob/main/packages/codemod/src/codemods/lib/create-transformer.ts\n// License: Apache-2.0\n\nimport type { FileInfo, API, JSCodeshift, Collection } from 'jscodeshift';\n\ntype TransformerFunction = (\n  fileInfo: FileInfo,\n  api: API,\n  options: Record<string, unknown>,\n  context: TransformContext,\n) => void;\n\nexport interface TransformContext {\n  /**\n   * The jscodeshift API object.\n   */\n  j: JSCodeshift;\n\n  /**\n   * The root collection of the AST.\n   */\n  root: Collection<any>;\n\n  /**\n   * Codemods should set this to true if they make any changes to the AST.\n   */\n  hasChanges: boolean;\n\n  /**\n   * Codemods can append messages to this array to report information to the user.\n   */\n  messages: string[];\n}\n\nexport function createTransformer(transformFn: TransformerFunction) {\n  // Note the return type of this function is explicitly designed to conform to\n  // the signature expected by jscodeshift. For more see\n  // https://github.com/facebook/jscodeshift\n  return function transformer(fileInfo: FileInfo, api: API, options: Record<string, unknown>) {\n    const j = api.jscodeshift;\n    const root = j(fileInfo.source);\n    const context: TransformContext = {\n      j,\n      root,\n      hasChanges: false,\n      messages: [],\n    };\n\n    // Execute the transformation\n    transformFn(fileInfo, api, options, context);\n\n    // Report any messages\n    context.messages.forEach(message => api.report(message));\n\n    // Return the transformed source code if changes were made\n    return context.hasChanges ? root.toSource({ quote: 'single' }) : null;\n  };\n}\n"],"mappings":";AAkCO,SAAS,kBAAkB,aAAkC;AAIlE,SAAO,SAAS,YAAY,UAAoB,KAAU,SAAkC;AAC1F,UAAM,IAAI,IAAI;AACd,UAAM,OAAO,EAAE,SAAS,MAAM;AAC9B,UAAM,UAA4B;AAAA,MAChC;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,UAAU,CAAC;AAAA,IACb;AAGA,gBAAY,UAAU,KAAK,SAAS,OAAO;AAG3C,YAAQ,SAAS,QAAQ,aAAW,IAAI,OAAO,OAAO,CAAC;AAGvD,WAAO,QAAQ,aAAa,KAAK,SAAS,EAAE,OAAO,SAAS,CAAC,IAAI;AAAA,EACnE;AACF;","names":[]}

package/dist/codemods/chunk-LGW3NUAJ.js

@@ -0,0 +1,32 @@
+// src/codemods/lib/add-comment.ts
+var ERROR_PREFIX = "FIXME(mastra): ";
+function existsComment(comments, comment) {
+  let hasComment = false;
+  if (comments) {
+    comments.forEach((commentNode) => {
+      const currentComment = commentNode.value.trim();
+      if (currentComment === comment) {
+        hasComment = true;
+      }
+    });
+    if (hasComment) {
+      return true;
+    }
+  }
+  return false;
+}
+function insertCommentOnce(node, j, comment) {
+  const hasCommentInInlineComments = existsComment(node.comments, comment);
+  const hasCommentInLeadingComments = existsComment(node.leadingComments, comment);
+  if (!hasCommentInInlineComments && !hasCommentInLeadingComments) {
+    node.comments = [...node.comments || [], j.commentBlock(` ${comment} `)];
+    return true;
+  }
+  return false;
+}
+
+export {
+  ERROR_PREFIX,
+  insertCommentOnce
+};
+//# sourceMappingURL=chunk-LGW3NUAJ.js.map

package/dist/codemods/chunk-LGW3NUAJ.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/codemods/lib/add-comment.ts"],"sourcesContent":["import type { API, ASTPath } from 'jscodeshift';\n\nexport const ERROR_PREFIX = 'FIXME(mastra): ';\n\nfunction existsComment(comments: ASTPath<any>['node']['comments'], comment: string): boolean {\n  let hasComment = false;\n\n  if (comments) {\n    comments.forEach((commentNode: { value: string; type: string }) => {\n      const currentComment = commentNode.value.trim();\n      if (currentComment === comment) {\n        hasComment = true;\n      }\n    });\n\n    if (hasComment) {\n      return true;\n    }\n  }\n  return false;\n}\n\nexport function insertCommentOnce(node: ASTPath<any>['node'], j: API['j'], comment: string): boolean {\n  const hasCommentInInlineComments = existsComment(node.comments, comment);\n  const hasCommentInLeadingComments = existsComment(node.leadingComments, comment);\n\n  if (!hasCommentInInlineComments && !hasCommentInLeadingComments) {\n    node.comments = [...(node.comments || []), j.commentBlock(` ${comment} `)];\n    return true;\n  }\n\n  return false;\n}\n"],"mappings":";AAEO,IAAM,eAAe;AAE5B,SAAS,cAAc,UAA4C,SAA0B;AAC3F,MAAI,aAAa;AAEjB,MAAI,UAAU;AACZ,aAAS,QAAQ,CAAC,gBAAiD;AACjE,YAAM,iBAAiB,YAAY,MAAM,KAAK;AAC9C,UAAI,mBAAmB,SAAS;AAC9B,qBAAa;AAAA,MACf;AAAA,IACF,CAAC;AAED,QAAI,YAAY;AACd,aAAO;AAAA,IACT;AAAA,EACF;AACA,SAAO;AACT;AAEO,SAAS,kBAAkB,MAA4B,GAAa,SAA0B;AACnG,QAAM,6BAA6B,cAAc,KAAK,UAAU,OAAO;AACvE,QAAM,8BAA8B,cAAc,KAAK,iBAAiB,OAAO;AAE/E,MAAI,CAAC,8BAA8B,CAAC,6BAA6B;AAC/D,SAAK,WAAW,CAAC,GAAI,KAAK,YAAY,CAAC,GAAI,EAAE,aAAa,IAAI,OAAO,GAAG,CAAC;AACzE,WAAO;AAAA,EACT;AAEA,SAAO;AACT;","names":[]}

package/dist/codemods/chunk-PKCFDS7J.js

@@ -0,0 +1,177 @@
+// src/codemods/lib/utils.ts
+function trackClassInstances(j, root, className) {
+  const instances = /* @__PURE__ */ new Set();
+  root.find(j.NewExpression, {
+    callee: {
+      type: "Identifier",
+      name: className
+    }
+  }).forEach((path) => {
+    const parent = path.parent.value;
+    if (parent.type === "VariableDeclarator" && parent.id.type === "Identifier") {
+      instances.add(parent.id.name);
+    }
+  });
+  return instances;
+}
+function trackMultipleClassInstances(j, root, classNames) {
+  const instances = /* @__PURE__ */ new Set();
+  const classNameSet = new Set(classNames);
+  root.find(j.NewExpression).forEach((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "Identifier") return;
+    if (!classNameSet.has(callee.name)) return;
+    const parent = path.parent.value;
+    if (parent.type === "VariableDeclarator" && parent.id.type === "Identifier") {
+      instances.add(parent.id.name);
+    }
+  });
+  return instances;
+}
+function transformMethodCalls(j, root, instances, methodName, transform) {
+  if (instances.size === 0) return 0;
+  let count = 0;
+  root.find(j.CallExpression).forEach((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "MemberExpression") return;
+    if (callee.object.type !== "Identifier") return;
+    if (callee.property.type !== "Identifier") return;
+    if (!instances.has(callee.object.name)) return;
+    if (methodName && callee.property.name !== methodName) return;
+    transform(path);
+    count++;
+  });
+  return count;
+}
+function renameMethod(j, root, instances, oldMethodName, newMethodName) {
+  if (instances.size === 0) return 0;
+  let count = 0;
+  root.find(j.CallExpression).forEach((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "MemberExpression") return;
+    if (callee.object.type !== "Identifier") return;
+    if (callee.property.type !== "Identifier") return;
+    if (!instances.has(callee.object.name)) return;
+    if (callee.property.name !== oldMethodName) return;
+    callee.property.name = newMethodName;
+    count++;
+  });
+  return count;
+}
+function renameMethods(j, root, instances, methodRenames) {
+  if (instances.size === 0) return 0;
+  let count = 0;
+  root.find(j.CallExpression).forEach((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "MemberExpression") return;
+    if (callee.object.type !== "Identifier") return;
+    if (callee.property.type !== "Identifier") return;
+    if (!instances.has(callee.object.name)) return;
+    const oldName = callee.property.name;
+    const newName = methodRenames[oldName];
+    if (newName) {
+      callee.property.name = newName;
+      count++;
+    }
+  });
+  return count;
+}
+function transformObjectProperties(obj, propertyRenames) {
+  let count = 0;
+  const recurse = (o) => {
+    if (!o.properties) return;
+    o.properties.forEach((prop) => {
+      if ((prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier") {
+        const oldName = prop.key.name;
+        const newName = propertyRenames[oldName];
+        if (newName) {
+          prop.key.name = newName;
+          count++;
+        }
+        if (prop.value?.type === "ObjectExpression") {
+          recurse(prop.value);
+        }
+      }
+    });
+  };
+  recurse(obj);
+  return count;
+}
+function isMemberExpressionOnInstance(node, instances, propertyName) {
+  if (node.type !== "MemberExpression") return false;
+  if (node.object.type !== "Identifier") return false;
+  if (!instances.has(node.object.name)) return false;
+  if (propertyName && node.property.type === "Identifier" && node.property.name !== propertyName) {
+    return false;
+  }
+  return true;
+}
+function renameImportAndUsages(j, root, packageName, oldName, newName) {
+  let changes = 0;
+  const localNamesToReplace = /* @__PURE__ */ new Set();
+  root.find(j.ImportDeclaration).filter((path) => {
+    const source = path.value.source.value;
+    return typeof source === "string" && source === packageName;
+  }).forEach((path) => {
+    if (!path.value.specifiers) return;
+    path.value.specifiers.forEach((specifier) => {
+      if (specifier.type === "ImportSpecifier" && specifier.imported.type === "Identifier" && specifier.imported.name === oldName) {
+        const isAliased = specifier.local && specifier.local.name !== oldName;
+        specifier.imported.name = newName;
+        changes++;
+        if (!isAliased) {
+          if (specifier.local) {
+            specifier.local.name = newName;
+          }
+          localNamesToReplace.add(oldName);
+        }
+      }
+    });
+  });
+  localNamesToReplace.forEach((localName) => {
+    root.find(j.Identifier, { name: localName }).forEach((path) => {
+      const parent = path.parent;
+      if (parent && parent.value.type === "ImportSpecifier") {
+        return;
+      }
+      path.value.name = newName;
+      changes++;
+    });
+  });
+  return changes;
+}
+function transformConstructorProperties(j, root, className, propertyRenames) {
+  let count = 0;
+  root.find(j.NewExpression, {
+    callee: { type: "Identifier", name: className }
+  }).forEach((path) => {
+    const args = path.value.arguments;
+    if (args.length === 0) return;
+    const firstArg = args[0];
+    if (!firstArg || firstArg.type !== "ObjectExpression" || !firstArg.properties) return;
+    firstArg.properties.forEach((prop) => {
+      if ((prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier") {
+        const oldName = prop.key.name;
+        const newName = propertyRenames[oldName];
+        if (newName) {
+          prop.key.name = newName;
+          count++;
+        }
+      }
+    });
+  });
+  return count;
+}
+
+export {
+  trackClassInstances,
+  trackMultipleClassInstances,
+  transformMethodCalls,
+  renameMethod,
+  renameMethods,
+  transformObjectProperties,
+  isMemberExpressionOnInstance,
+  renameImportAndUsages,
+  transformConstructorProperties
+};
+//# sourceMappingURL=chunk-PKCFDS7J.js.map

package/dist/codemods/chunk-PKCFDS7J.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/codemods/lib/utils.ts"],"sourcesContent":["// Shared utility functions for codemods\n\nimport type { Collection, JSCodeshift } from 'jscodeshift';\n\n/**\n * Efficiently tracks instances of a specific class by finding all `new ClassName()` expressions\n * and extracting the variable names they're assigned to.\n *\n * @param j - JSCodeshift API\n * @param root - Root collection\n * @param className - Name of the class to track\n * @returns Set of variable names that are instances of the class\n */\nexport function trackClassInstances(j: JSCodeshift, root: Collection<any>, className: string): Set<string> {\n  const instances = new Set<string>();\n\n  root\n    .find(j.NewExpression, {\n      callee: {\n        type: 'Identifier',\n        name: className,\n      },\n    })\n    .forEach(path => {\n      const parent = path.parent.value;\n      if (parent.type === 'VariableDeclarator' && parent.id.type === 'Identifier') {\n        instances.add(parent.id.name);\n      }\n    });\n\n  return instances;\n}\n\n/**\n * Efficiently tracks instances of multiple classes in a single pass.\n * This is optimized for codemods that need to track several store types or class variants.\n *\n * @param j - JSCodeshift API\n * @param root - Root collection\n * @param classNames - Array of class names to track\n * @returns Set of variable names that are instances of any of the classes\n */\nexport function trackMultipleClassInstances(j: JSCodeshift, root: Collection<any>, classNames: string[]): Set<string> {\n  const instances = new Set<string>();\n  const classNameSet = new Set(classNames);\n\n  root.find(j.NewExpression).forEach(path => {\n    const { callee } = path.value;\n    if (callee.type !== 'Identifier') return;\n    if (!classNameSet.has(callee.name)) return;\n\n    const parent = path.parent.value;\n    if (parent.type === 'VariableDeclarator' && parent.id.type === 'Identifier') {\n      instances.add(parent.id.name);\n    }\n  });\n\n  return instances;\n}\n\n/**\n * Efficiently finds and transforms method calls on tracked instances.\n * This combines finding, filtering, and transforming in a single pass.\n *\n * @param j - JSCodeshift API\n * @param root - Root collection\n * @param instances - Set of instance variable names to track\n * @param methodName - Name of the method to find (or undefined to match any method)\n * @param transform - Callback to transform matching call expressions\n * @returns Number of transformations made\n */\nexport function transformMethodCalls(\n  j: JSCodeshift,\n  root: Collection<any>,\n  instances: Set<string>,\n  methodName: string | undefined,\n  transform: (path: any) => void,\n): number {\n  if (instances.size === 0) return 0;\n\n  let count = 0;\n\n  root.find(j.CallExpression).forEach(path => {\n    const { callee } = path.value;\n    if (callee.type !== 'MemberExpression') return;\n    if (callee.object.type !== 'Identifier') return;\n    if (callee.property.type !== 'Identifier') return;\n\n    // Only process if called on a tracked instance\n    if (!instances.has(callee.object.name)) return;\n\n    // Only process if it's the method we want (or any method if undefined)\n    if (methodName && callee.property.name !== methodName) return;\n\n    transform(path);\n    count++;\n  });\n\n  return count;\n}\n\n/**\n * Renames a method on tracked instances efficiently in a single pass.\n *\n * @param j - JSCodeshift API\n * @param root - Root collection\n * @param instances - Set of instance variable names to track\n * @param oldMethodName - Current method name\n * @param newMethodName - New method name\n * @returns Number of renames performed\n */\nexport function renameMethod(\n  j: JSCodeshift,\n  root: Collection<any>,\n  instances: Set<string>,\n  oldMethodName: string,\n  newMethodName: string,\n): number {\n  if (instances.size === 0) return 0;\n\n  let count = 0;\n\n  root.find(j.CallExpression).forEach(path => {\n    const { callee } = path.value;\n    if (callee.type !== 'MemberExpression') return;\n    if (callee.object.type !== 'Identifier') return;\n    if (callee.property.type !== 'Identifier') return;\n\n    // Only process if called on tracked instance\n    if (!instances.has(callee.object.name)) return;\n\n    // Only process if it's the method we want to rename\n    if (callee.property.name !== oldMethodName) return;\n\n    callee.property.name = newMethodName;\n    count++;\n  });\n\n  return count;\n}\n\n/**\n * Renames multiple methods on tracked instances in a single pass.\n *\n * @param j - JSCodeshift API\n * @param root - Root collection\n * @param instances - Set of instance variable names to track\n * @param methodRenames - Map of old method names to new method names\n * @returns Number of renames performed\n */\nexport function renameMethods(\n  j: JSCodeshift,\n  root: Collection<any>,\n  instances: Set<string>,\n  methodRenames: Record<string, string>,\n): number {\n  if (instances.size === 0) return 0;\n\n  let count = 0;\n\n  root.find(j.CallExpression).forEach(path => {\n    const { callee } = path.value;\n    if (callee.type !== 'MemberExpression') return;\n    if (callee.object.type !== 'Identifier') return;\n    if (callee.property.type !== 'Identifier') return;\n\n    // Only process if called on tracked instance\n    if (!instances.has(callee.object.name)) return;\n\n    // Check if this is one of the methods we want to rename\n    const oldName = callee.property.name;\n    const newName = methodRenames[oldName];\n\n    if (newName) {\n      callee.property.name = newName;\n      count++;\n    }\n  });\n\n  return count;\n}\n\n/**\n * Transforms object properties in method call arguments.\n * This is a helper for codemods that need to rename properties in object arguments.\n *\n * @param obj - Object expression to transform\n * @param propertyRenames - Map of old property names to new property names\n * @returns Number of properties renamed\n */\nexport function transformObjectProperties(obj: any, propertyRenames: Record<string, string>): number {\n  let count = 0;\n\n  const recurse = (o: any) => {\n    if (!o.properties) return;\n\n    o.properties.forEach((prop: any) => {\n      if ((prop.type === 'Property' || prop.type === 'ObjectProperty') && prop.key?.type === 'Identifier') {\n        const oldName = prop.key.name;\n        const newName = propertyRenames[oldName];\n\n        if (newName) {\n          prop.key.name = newName;\n          count++;\n        }\n\n        // Recursively transform nested objects\n        if (prop.value?.type === 'ObjectExpression') {\n          recurse(prop.value);\n        }\n      }\n    });\n  };\n\n  recurse(obj);\n  return count;\n}\n\n/**\n * Checks if a node is a member expression accessing a specific property on tracked instances.\n *\n * @param node - AST node to check\n * @param instances - Set of instance variable names to track\n * @param propertyName - Property name to match (or undefined to match any property)\n * @returns true if the node matches\n */\nexport function isMemberExpressionOnInstance(node: any, instances: Set<string>, propertyName?: string): boolean {\n  if (node.type !== 'MemberExpression') return false;\n  if (node.object.type !== 'Identifier') return false;\n  if (!instances.has(node.object.name)) return false;\n\n  if (propertyName && node.property.type === 'Identifier' && node.property.name !== propertyName) {\n    return false;\n  }\n\n  return true;\n}\n\n/**\n * Renames an import and all its usages in a single optimized pass.\n * Handles aliased imports correctly - only transforms usages for non-aliased imports.\n * Handles multiple imports of the same name (with different aliases) correctly.\n *\n * For non-aliased imports: Renames both import and all usages\n *   import { oldName } → import { newName }\n *   oldName() → newName()\n *\n * For aliased imports: Only renames the import, keeps alias in usages\n *   import { oldName as alias } → import { newName as alias }\n *   alias() → alias() (unchanged)\n *\n * @param j - JSCodeshift API\n * @param root - Root collection\n * @param packageName - Package to import from (e.g., '@mastra/core/evals')\n * @param oldName - Current import name\n * @param newName - New import name\n * @returns Number of changes made\n */\nexport function renameImportAndUsages(\n  j: JSCodeshift,\n  root: Collection<any>,\n  packageName: string,\n  oldName: string,\n  newName: string,\n): number {\n  let changes = 0;\n  const localNamesToReplace = new Set<string>();\n\n  // First: Transform import specifiers from the specific package and collect local names to replace\n  root\n    .find(j.ImportDeclaration)\n    .filter(path => {\n      const source = path.value.source.value;\n      return typeof source === 'string' && source === packageName;\n    })\n    .forEach(path => {\n      if (!path.value.specifiers) return;\n\n      path.value.specifiers.forEach((specifier: any) => {\n        if (\n          specifier.type === 'ImportSpecifier' &&\n          specifier.imported.type === 'Identifier' &&\n          specifier.imported.name === oldName\n        ) {\n          const isAliased = specifier.local && specifier.local.name !== oldName;\n\n          // Always rename the imported name\n          specifier.imported.name = newName;\n          changes++;\n\n          // Only rename the local name and track for usage replacement if NOT aliased\n          if (!isAliased) {\n            if (specifier.local) {\n              specifier.local.name = newName;\n            }\n            // Track for usage replacement (only non-aliased imports)\n            localNamesToReplace.add(oldName);\n          }\n        }\n      });\n    });\n\n  // Second: Transform usages only for non-aliased imports\n  localNamesToReplace.forEach(localName => {\n    root.find(j.Identifier, { name: localName }).forEach(path => {\n      // Skip identifiers that are part of import declarations\n      const parent = path.parent;\n      if (parent && parent.value.type === 'ImportSpecifier') {\n        return;\n      }\n\n      path.value.name = newName;\n      changes++;\n    });\n  });\n\n  return changes;\n}\n\n/**\n * Transforms properties in constructor call arguments.\n *\n * @param j - JSCodeshift API\n * @param root - Root collection\n * @param className - Name of the class whose constructor to transform\n * @param propertyRenames - Map of old property names to new property names\n * @returns Number of properties renamed\n */\nexport function transformConstructorProperties(\n  j: JSCodeshift,\n  root: Collection<any>,\n  className: string,\n  propertyRenames: Record<string, string>,\n): number {\n  let count = 0;\n\n  root\n    .find(j.NewExpression, {\n      callee: { type: 'Identifier', name: className },\n    })\n    .forEach(path => {\n      const args = path.value.arguments;\n      if (args.length === 0) return;\n\n      const firstArg = args[0];\n      if (!firstArg || firstArg.type !== 'ObjectExpression' || !firstArg.properties) return;\n\n      firstArg.properties.forEach((prop: any) => {\n        if ((prop.type === 'Property' || prop.type === 'ObjectProperty') && prop.key?.type === 'Identifier') {\n          const oldName = prop.key.name;\n          const newName = propertyRenames[oldName];\n\n          if (newName) {\n            prop.key.name = newName;\n            count++;\n          }\n        }\n      });\n    });\n\n  return count;\n}\n"],"mappings":";AAaO,SAAS,oBAAoB,GAAgB,MAAuB,WAAgC;AACzG,QAAM,YAAY,oBAAI,IAAY;AAElC,OACG,KAAK,EAAE,eAAe;AAAA,IACrB,QAAQ;AAAA,MACN,MAAM;AAAA,MACN,MAAM;AAAA,IACR;AAAA,EACF,CAAC,EACA,QAAQ,UAAQ;AACf,UAAM,SAAS,KAAK,OAAO;AAC3B,QAAI,OAAO,SAAS,wBAAwB,OAAO,GAAG,SAAS,cAAc;AAC3E,gBAAU,IAAI,OAAO,GAAG,IAAI;AAAA,IAC9B;AAAA,EACF,CAAC;AAEH,SAAO;AACT;AAWO,SAAS,4BAA4B,GAAgB,MAAuB,YAAmC;AACpH,QAAM,YAAY,oBAAI,IAAY;AAClC,QAAM,eAAe,IAAI,IAAI,UAAU;AAEvC,OAAK,KAAK,EAAE,aAAa,EAAE,QAAQ,UAAQ;AACzC,UAAM,EAAE,OAAO,IAAI,KAAK;AACxB,QAAI,OAAO,SAAS,aAAc;AAClC,QAAI,CAAC,aAAa,IAAI,OAAO,IAAI,EAAG;AAEpC,UAAM,SAAS,KAAK,OAAO;AAC3B,QAAI,OAAO,SAAS,wBAAwB,OAAO,GAAG,SAAS,cAAc;AAC3E,gBAAU,IAAI,OAAO,GAAG,IAAI;AAAA,IAC9B;AAAA,EACF,CAAC;AAED,SAAO;AACT;AAaO,SAAS,qBACd,GACA,MACA,WACA,YACA,WACQ;AACR,MAAI,UAAU,SAAS,EAAG,QAAO;AAEjC,MAAI,QAAQ;AAEZ,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,UAAM,EAAE,OAAO,IAAI,KAAK;AACxB,QAAI,OAAO,SAAS,mBAAoB;AACxC,QAAI,OAAO,OAAO,SAAS,aAAc;AACzC,QAAI,OAAO,SAAS,SAAS,aAAc;AAG3C,QAAI,CAAC,UAAU,IAAI,OAAO,OAAO,IAAI,EAAG;AAGxC,QAAI,cAAc,OAAO,SAAS,SAAS,WAAY;AAEvD,cAAU,IAAI;AACd;AAAA,EACF,CAAC;AAED,SAAO;AACT;AAYO,SAAS,aACd,GACA,MACA,WACA,eACA,eACQ;AACR,MAAI,UAAU,SAAS,EAAG,QAAO;AAEjC,MAAI,QAAQ;AAEZ,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,UAAM,EAAE,OAAO,IAAI,KAAK;AACxB,QAAI,OAAO,SAAS,mBAAoB;AACxC,QAAI,OAAO,OAAO,SAAS,aAAc;AACzC,QAAI,OAAO,SAAS,SAAS,aAAc;AAG3C,QAAI,CAAC,UAAU,IAAI,OAAO,OAAO,IAAI,EAAG;AAGxC,QAAI,OAAO,SAAS,SAAS,cAAe;AAE5C,WAAO,SAAS,OAAO;AACvB;AAAA,EACF,CAAC;AAED,SAAO;AACT;AAWO,SAAS,cACd,GACA,MACA,WACA,eACQ;AACR,MAAI,UAAU,SAAS,EAAG,QAAO;AAEjC,MAAI,QAAQ;AAEZ,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,UAAM,EAAE,OAAO,IAAI,KAAK;AACxB,QAAI,OAAO,SAAS,mBAAoB;AACxC,QAAI,OAAO,OAAO,SAAS,aAAc;AACzC,QAAI,OAAO,SAAS,SAAS,aAAc;AAG3C,QAAI,CAAC,UAAU,IAAI,OAAO,OAAO,IAAI,EAAG;AAGxC,UAAM,UAAU,OAAO,SAAS;AAChC,UAAM,UAAU,cAAc,OAAO;AAErC,QAAI,SAAS;AACX,aAAO,SAAS,OAAO;AACvB;AAAA,IACF;AAAA,EACF,CAAC;AAED,SAAO;AACT;AAUO,SAAS,0BAA0B,KAAU,iBAAiD;AACnG,MAAI,QAAQ;AAEZ,QAAM,UAAU,CAAC,MAAW;AAC1B,QAAI,CAAC,EAAE,WAAY;AAEnB,MAAE,WAAW,QAAQ,CAAC,SAAc;AAClC,WAAK,KAAK,SAAS,cAAc,KAAK,SAAS,qBAAqB,KAAK,KAAK,SAAS,cAAc;AACnG,cAAM,UAAU,KAAK,IAAI;AACzB,cAAM,UAAU,gBAAgB,OAAO;AAEvC,YAAI,SAAS;AACX,eAAK,IAAI,OAAO;AAChB;AAAA,QACF;AAGA,YAAI,KAAK,OAAO,SAAS,oBAAoB;AAC3C,kBAAQ,KAAK,KAAK;AAAA,QACpB;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAEA,UAAQ,GAAG;AACX,SAAO;AACT;AAUO,SAAS,6BAA6B,MAAW,WAAwB,cAAgC;AAC9G,MAAI,KAAK,SAAS,mBAAoB,QAAO;AAC7C,MAAI,KAAK,OAAO,SAAS,aAAc,QAAO;AAC9C,MAAI,CAAC,UAAU,IAAI,KAAK,OAAO,IAAI,EAAG,QAAO;AAE7C,MAAI,gBAAgB,KAAK,SAAS,SAAS,gBAAgB,KAAK,SAAS,SAAS,cAAc;AAC9F,WAAO;AAAA,EACT;AAEA,SAAO;AACT;AAsBO,SAAS,sBACd,GACA,MACA,aACA,SACA,SACQ;AACR,MAAI,UAAU;AACd,QAAM,sBAAsB,oBAAI,IAAY;AAG5C,OACG,KAAK,EAAE,iBAAiB,EACxB,OAAO,UAAQ;AACd,UAAM,SAAS,KAAK,MAAM,OAAO;AACjC,WAAO,OAAO,WAAW,YAAY,WAAW;AAAA,EAClD,CAAC,EACA,QAAQ,UAAQ;AACf,QAAI,CAAC,KAAK,MAAM,WAAY;AAE5B,SAAK,MAAM,WAAW,QAAQ,CAAC,cAAmB;AAChD,UACE,UAAU,SAAS,qBACnB,UAAU,SAAS,SAAS,gBAC5B,UAAU,SAAS,SAAS,SAC5B;AACA,cAAM,YAAY,UAAU,SAAS,UAAU,MAAM,SAAS;AAG9D,kBAAU,SAAS,OAAO;AAC1B;AAGA,YAAI,CAAC,WAAW;AACd,cAAI,UAAU,OAAO;AACnB,sBAAU,MAAM,OAAO;AAAA,UACzB;AAEA,8BAAoB,IAAI,OAAO;AAAA,QACjC;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAGH,sBAAoB,QAAQ,eAAa;AACvC,SAAK,KAAK,EAAE,YAAY,EAAE,MAAM,UAAU,CAAC,EAAE,QAAQ,UAAQ;AAE3D,YAAM,SAAS,KAAK;AACpB,UAAI,UAAU,OAAO,MAAM,SAAS,mBAAmB;AACrD;AAAA,MACF;AAEA,WAAK,MAAM,OAAO;AAClB;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAED,SAAO;AACT;AAWO,SAAS,+BACd,GACA,MACA,WACA,iBACQ;AACR,MAAI,QAAQ;AAEZ,OACG,KAAK,EAAE,eAAe;AAAA,IACrB,QAAQ,EAAE,MAAM,cAAc,MAAM,UAAU;AAAA,EAChD,CAAC,EACA,QAAQ,UAAQ;AACf,UAAM,OAAO,KAAK,MAAM;AACxB,QAAI,KAAK,WAAW,EAAG;AAEvB,UAAM,WAAW,KAAK,CAAC;AACvB,QAAI,CAAC,YAAY,SAAS,SAAS,sBAAsB,CAAC,SAAS,WAAY;AAE/E,aAAS,WAAW,QAAQ,CAAC,SAAc;AACzC,WAAK,KAAK,SAAS,cAAc,KAAK,SAAS,qBAAqB,KAAK,KAAK,SAAS,cAAc;AACnG,cAAM,UAAU,KAAK,IAAI;AACzB,cAAM,UAAU,gBAAgB,OAAO;AAEvC,YAAI,SAAS;AACX,eAAK,IAAI,OAAO;AAChB;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAEH,SAAO;AACT;","names":[]}

package/dist/codemods/lib/add-comment.js

@@ -1,57 +1,9 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/codemods/lib/add-comment.ts
-var add_comment_exports = {};
-__export(add_comment_exports, {
-  ERROR_PREFIX: () => ERROR_PREFIX,
-  insertCommentOnce: () => insertCommentOnce
-});
-module.exports = __toCommonJS(add_comment_exports);
-var ERROR_PREFIX = "FIXME(mastra): ";
-function existsComment(comments, comment) {
-  let hasComment = false;
-  if (comments) {
-    comments.forEach((commentNode) => {
-      const currentComment = commentNode.value.trim();
-      if (currentComment === comment) {
-        hasComment = true;
-      }
-    });
-    if (hasComment) {
-      return true;
-    }
-  }
-  return false;
-}
-function insertCommentOnce(node, j, comment) {
-  const hasCommentInInlineComments = existsComment(node.comments, comment);
-  const hasCommentInLeadingComments = existsComment(node.leadingComments, comment);
-  if (!hasCommentInInlineComments && !hasCommentInLeadingComments) {
-    node.comments = [...node.comments || [], j.commentBlock(` ${comment} `)];
-    return true;
-  }
-  return false;
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
+import {
   ERROR_PREFIX,
   insertCommentOnce
-});
+} from "../chunk-LGW3NUAJ.js";
+export {
+  ERROR_PREFIX,
+  insertCommentOnce
+};
 //# sourceMappingURL=add-comment.js.map

package/dist/codemods/lib/add-comment.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/codemods/lib/add-comment.ts"],"sourcesContent":["import type { API, ASTPath } from 'jscodeshift';\n\nexport const ERROR_PREFIX = 'FIXME(mastra): ';\n\nfunction existsComment(comments: ASTPath<any>['node']['comments'], comment: string): boolean {\n  let hasComment = false;\n\n  if (comments) {\n    comments.forEach((commentNode: any) => {\n      const currentComment = commentNode.value.trim();\n      if (currentComment === comment) {\n        hasComment = true;\n      }\n    });\n\n    if (hasComment) {\n      return true;\n    }\n  }\n  return false;\n}\n\nexport function insertCommentOnce(node: ASTPath<any>['node'], j: API['j'], comment: string): boolean {\n  const hasCommentInInlineComments = existsComment(node.comments, comment);\n  const hasCommentInLeadingComments = existsComment(node.leadingComments, comment);\n\n  if (!hasCommentInInlineComments && !hasCommentInLeadingComments) {\n    node.comments = [...(node.comments || []), j.commentBlock(` ${comment} `)];\n    return true;\n  }\n\n  return false;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEO,IAAM,eAAe;AAE5B,SAAS,cAAc,UAA4C,SAA0B;AAC3F,MAAI,aAAa;AAEjB,MAAI,UAAU;AACZ,aAAS,QAAQ,CAAC,gBAAqB;AACrC,YAAM,iBAAiB,YAAY,MAAM,KAAK;AAC9C,UAAI,mBAAmB,SAAS;AAC9B,qBAAa;AAAA,MACf;AAAA,IACF,CAAC;AAED,QAAI,YAAY;AACd,aAAO;AAAA,IACT;AAAA,EACF;AACA,SAAO;AACT;AAEO,SAAS,kBAAkB,MAA4B,GAAa,SAA0B;AACnG,QAAM,6BAA6B,cAAc,KAAK,UAAU,OAAO;AACvE,QAAM,8BAA8B,cAAc,KAAK,iBAAiB,OAAO;AAE/E,MAAI,CAAC,8BAA8B,CAAC,6BAA6B;AAC/D,SAAK,WAAW,CAAC,GAAI,KAAK,YAAY,CAAC,GAAI,EAAE,aAAa,IAAI,OAAO,GAAG,CAAC;AACzE,WAAO;AAAA,EACT;AAEA,SAAO;AACT;","names":[]}
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/codemods/lib/create-transformer.js

@@ -1,45 +1,7 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/codemods/lib/create-transformer.ts
-var create_transformer_exports = {};
-__export(create_transformer_exports, {
-  createTransformer: () => createTransformer
-});
-module.exports = __toCommonJS(create_transformer_exports);
-function createTransformer(transformFn) {
-  return function transformer(fileInfo, api, options) {
-    const j = api.jscodeshift;
-    const root = j(fileInfo.source);
-    const context = {
-      j,
-      root,
-      hasChanges: false,
-      messages: []
-    };
-    transformFn(fileInfo, api, options, context);
-    context.messages.forEach((message) => api.report(message));
-    return context.hasChanges ? root.toSource({ quote: "single" }) : null;
-  };
-}
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
+import {
   createTransformer
-});
+} from "../chunk-JNFQ6J6B.js";
+export {
+  createTransformer
+};
 //# sourceMappingURL=create-transformer.js.map

package/dist/codemods/lib/create-transformer.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/codemods/lib/create-transformer.ts"],"sourcesContent":["// Copied from https://github.com/vercel/ai/blob/main/packages/codemod/src/codemods/lib/create-transformer.ts\n// License: Apache-2.0\n\nimport type { FileInfo, API, JSCodeshift, Collection } from 'jscodeshift';\n\ntype TransformerFunction = (fileInfo: FileInfo, api: API, options: any, context: TransformContext) => void;\n\nexport interface TransformContext {\n  /**\n   * The jscodeshift API object.\n   */\n  j: JSCodeshift;\n\n  /**\n   * The root collection of the AST.\n   */\n  root: Collection<any>;\n\n  /**\n   * Codemods should set this to true if they make any changes to the AST.\n   */\n  hasChanges: boolean;\n\n  /**\n   * Codemods can append messages to this array to report information to the user.\n   */\n  messages: string[];\n}\n\nexport function createTransformer(transformFn: TransformerFunction) {\n  // Note the return type of this function is explicitly designed to conform to\n  // the signature expected by jscodeshift. For more see\n  // https://github.com/facebook/jscodeshift\n  return function transformer(fileInfo: FileInfo, api: API, options: any) {\n    const j = api.jscodeshift;\n    const root = j(fileInfo.source);\n    const context: TransformContext = {\n      j,\n      root,\n      hasChanges: false,\n      messages: [],\n    };\n\n    // Execute the transformation\n    transformFn(fileInfo, api, options, context);\n\n    // Report any messages\n    context.messages.forEach(message => api.report(message));\n\n    // Return the transformed source code if changes were made\n    return context.hasChanges ? root.toSource({ quote: 'single' }) : null;\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AA6BO,SAAS,kBAAkB,aAAkC;AAIlE,SAAO,SAAS,YAAY,UAAoB,KAAU,SAAc;AACtE,UAAM,IAAI,IAAI;AACd,UAAM,OAAO,EAAE,SAAS,MAAM;AAC9B,UAAM,UAA4B;AAAA,MAChC;AAAA,MACA;AAAA,MACA,YAAY;AAAA,MACZ,UAAU,CAAC;AAAA,IACb;AAGA,gBAAY,UAAU,KAAK,SAAS,OAAO;AAG3C,YAAQ,SAAS,QAAQ,aAAW,IAAI,OAAO,OAAO,CAAC;AAGvD,WAAO,QAAQ,aAAa,KAAK,SAAS,EAAE,OAAO,SAAS,CAAC,IAAI;AAAA,EACnE;AACF;","names":[]}
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/codemods/lib/utils.js

@@ -0,0 +1,23 @@
+import {
+  isMemberExpressionOnInstance,
+  renameImportAndUsages,
+  renameMethod,
+  renameMethods,
+  trackClassInstances,
+  trackMultipleClassInstances,
+  transformConstructorProperties,
+  transformMethodCalls,
+  transformObjectProperties
+} from "../chunk-PKCFDS7J.js";
+export {
+  isMemberExpressionOnInstance,
+  renameImportAndUsages,
+  renameMethod,
+  renameMethods,
+  trackClassInstances,
+  trackMultipleClassInstances,
+  transformConstructorProperties,
+  transformMethodCalls,
+  transformObjectProperties
+};
+//# sourceMappingURL=utils.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/codemods/v1/agent-abort-signal.js

@@ -0,0 +1,61 @@
+import {
+  createTransformer
+} from "../chunk-JNFQ6J6B.js";
+import {
+  trackClassInstances
+} from "../chunk-PKCFDS7J.js";
+
+// src/codemods/v1/agent-abort-signal.ts
+var agent_abort_signal_default = createTransformer((_fileInfo, _api, _options, context) => {
+  const { j, root } = context;
+  const agentInstances = trackClassInstances(j, root, "Agent");
+  if (agentInstances.size === 0) return;
+  root.find(j.CallExpression).forEach((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "MemberExpression") return;
+    if (callee.object.type !== "Identifier") return;
+    if (!agentInstances.has(callee.object.name)) return;
+    const args = path.value.arguments;
+    if (args.length < 2) return;
+    const optionsArg = args[1];
+    if (!optionsArg || optionsArg.type !== "ObjectExpression") return;
+    if (!optionsArg.properties) return;
+    let modelSettingsIndex = -1;
+    const modelSettingsProp = optionsArg.properties.find((prop, index) => {
+      if ((prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier" && prop.key.name === "modelSettings" && prop.value?.type === "ObjectExpression") {
+        modelSettingsIndex = index;
+        return true;
+      }
+      return false;
+    });
+    if (!modelSettingsProp || modelSettingsProp.value?.type !== "ObjectExpression") return;
+    if (modelSettingsIndex === -1) return;
+    const modelSettingsValue = modelSettingsProp.value;
+    let abortSignalProp;
+    const filteredProperties = modelSettingsValue.properties?.filter((prop) => {
+      if ((prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier" && prop.key.name === "abortSignal") {
+        abortSignalProp = prop;
+        return false;
+      }
+      return true;
+    });
+    if (!abortSignalProp) return;
+    modelSettingsValue.properties = filteredProperties;
+    const newProperties = [];
+    optionsArg.properties.forEach((prop, index) => {
+      newProperties.push(prop);
+      if (index === modelSettingsIndex) {
+        newProperties.push(abortSignalProp);
+      }
+    });
+    optionsArg.properties = newProperties;
+    context.hasChanges = true;
+  });
+  if (context.hasChanges) {
+    context.messages.push("Moved abortSignal from modelSettings to top-level options in agent method calls");
+  }
+});
+export {
+  agent_abort_signal_default as default
+};
+//# sourceMappingURL=agent-abort-signal.js.map