@mastra/codemod 0.0.0-main-test-05-11-2025-2-20251106053353 → 0.0.0-main-test-2-20251127203846

package/dist/codemods/v1/mcp-get-toolsets.js

@@ -1,74 +1,24 @@
-"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/v1/mcp-get-toolsets.ts
-var mcp_get_toolsets_exports = {};
-__export(mcp_get_toolsets_exports, {
-  default: () => mcp_get_toolsets_default
-});
-module.exports = __toCommonJS(mcp_get_toolsets_exports);
-
-// 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;
-  };
-}
+import {
+  createTransformer
+} from "../chunk-JNFQ6J6B.js";
+import {
+  renameMethod,
+  trackClassInstances
+} from "../chunk-PKCFDS7J.js";
 
 // src/codemods/v1/mcp-get-toolsets.ts
-var mcp_get_toolsets_default = createTransformer((fileInfo, api, options, context) => {
+var mcp_get_toolsets_default = createTransformer((_fileInfo, _api, _options, context) => {
   const { j, root } = context;
-  const mcpVariables = /* @__PURE__ */ new Set();
-  root.find(j.VariableDeclarator).forEach((path) => {
-    const node = path.node;
-    if (node.init && node.init.type === "NewExpression" && node.init.callee.type === "Identifier" && node.init.callee.name === "MCPServer" && node.id.type === "Identifier") {
-      mcpVariables.add(node.id.name);
-    }
-  });
-  if (mcpVariables.size === 0) return;
-  root.find(j.CallExpression).forEach((path) => {
-    const node = path.node;
-    if (node.callee.type !== "MemberExpression") {
-      return;
-    }
-    const callee = node.callee;
-    if (callee.object.type !== "Identifier" || !mcpVariables.has(callee.object.name)) {
-      return;
-    }
-    if (callee.property.type !== "Identifier" || callee.property.name !== "getToolsets") {
-      return;
-    }
-    callee.property.name = "listToolsets";
+  const mcpServerInstances = trackClassInstances(j, root, "MCPServer");
+  const mcpClientInstances = trackClassInstances(j, root, "MCPClient");
+  const mcpInstances = /* @__PURE__ */ new Set([...mcpServerInstances, ...mcpClientInstances]);
+  const count = renameMethod(j, root, mcpInstances, "getToolsets", "listToolsets");
+  if (count > 0) {
     context.hasChanges = true;
-  });
-  if (context.hasChanges) {
     context.messages.push(`Transformed MCPServer method: getToolsets \u2192 listToolsets`);
   }
 });
+export {
+  mcp_get_toolsets_default as default
+};
 //# sourceMappingURL=mcp-get-toolsets.js.map

package/dist/codemods/v1/mcp-get-toolsets.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/codemods/v1/mcp-get-toolsets.ts","../../../src/codemods/lib/create-transformer.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\n\n/**\n * Transforms MCPServer getToolsets method to listToolsets:\n * - mcp.getToolsets() → mcp.listToolsets()\n *\n * Only transforms methods on variables that were instantiated with `new MCPServer(...)`\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  // Track variable names that are MCPServer instances\n  const mcpVariables = new Set<string>();\n\n  // Find all variable declarations with new MCPServer() assignments\n  root.find(j.VariableDeclarator).forEach(path => {\n    const node = path.node;\n\n    // Check if the init is a new MCPServer() expression\n    if (\n      node.init &&\n      node.init.type === 'NewExpression' &&\n      node.init.callee.type === 'Identifier' &&\n      node.init.callee.name === 'MCPServer' &&\n      node.id.type === 'Identifier'\n    ) {\n      mcpVariables.add(node.id.name);\n    }\n  });\n\n  // Early return if no MCPServer instances found\n  if (mcpVariables.size === 0) return;\n\n  // Find all call expressions where the callee is mcp.getToolsets\n  root.find(j.CallExpression).forEach(path => {\n    const node = path.node;\n\n    // Check if callee is a member expression (e.g., mcp.getToolsets)\n    if (node.callee.type !== 'MemberExpression') {\n      return;\n    }\n\n    const callee = node.callee;\n\n    // Check if the object is a MCPServer variable\n    if (callee.object.type !== 'Identifier' || !mcpVariables.has(callee.object.name)) {\n      return;\n    }\n\n    // Check if the property is 'getToolsets'\n    if (callee.property.type !== 'Identifier' || callee.property.name !== 'getToolsets') {\n      return;\n    }\n\n    // Rename the method to 'listToolsets'\n    callee.property.name = 'listToolsets';\n    context.hasChanges = true;\n  });\n\n  if (context.hasChanges) {\n    context.messages.push(`Transformed MCPServer method: getToolsets → listToolsets`);\n  }\n});\n","// 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;;;AC6BO,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;;;AD5CA,IAAO,2BAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,eAAe,oBAAI,IAAY;AAGrC,OAAK,KAAK,EAAE,kBAAkB,EAAE,QAAQ,UAAQ;AAC9C,UAAM,OAAO,KAAK;AAGlB,QACE,KAAK,QACL,KAAK,KAAK,SAAS,mBACnB,KAAK,KAAK,OAAO,SAAS,gBAC1B,KAAK,KAAK,OAAO,SAAS,eAC1B,KAAK,GAAG,SAAS,cACjB;AACA,mBAAa,IAAI,KAAK,GAAG,IAAI;AAAA,IAC/B;AAAA,EACF,CAAC;AAGD,MAAI,aAAa,SAAS,EAAG;AAG7B,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,UAAM,OAAO,KAAK;AAGlB,QAAI,KAAK,OAAO,SAAS,oBAAoB;AAC3C;AAAA,IACF;AAEA,UAAM,SAAS,KAAK;AAGpB,QAAI,OAAO,OAAO,SAAS,gBAAgB,CAAC,aAAa,IAAI,OAAO,OAAO,IAAI,GAAG;AAChF;AAAA,IACF;AAGA,QAAI,OAAO,SAAS,SAAS,gBAAgB,OAAO,SAAS,SAAS,eAAe;AACnF;AAAA,IACF;AAGA,WAAO,SAAS,OAAO;AACvB,YAAQ,aAAa;AAAA,EACvB,CAAC;AAED,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,+DAA0D;AAAA,EAClF;AACF,CAAC;","names":[]}
+{"version":3,"sources":["../../../src/codemods/v1/mcp-get-toolsets.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\nimport { trackClassInstances, renameMethod } from '../lib/utils';\n\n/**\n * Transforms MCPServer and MCPClient getToolsets method to listToolsets:\n * - mcp.getToolsets() → mcp.listToolsets()\n *\n * Only transforms methods on variables that were instantiated with `new MCPServer(...)` or `new MCPClient(...)`\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  // Track MCPServer and MCPClient instances and rename method in a single optimized pass\n  const mcpServerInstances = trackClassInstances(j, root, 'MCPServer');\n  const mcpClientInstances = trackClassInstances(j, root, 'MCPClient');\n  const mcpInstances = new Set([...mcpServerInstances, ...mcpClientInstances]);\n  const count = renameMethod(j, root, mcpInstances, 'getToolsets', 'listToolsets');\n\n  if (count > 0) {\n    context.hasChanges = true;\n    context.messages.push(`Transformed MCPServer method: getToolsets → listToolsets`);\n  }\n});\n"],"mappings":";;;;;;;;;AASA,IAAO,2BAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,qBAAqB,oBAAoB,GAAG,MAAM,WAAW;AACnE,QAAM,qBAAqB,oBAAoB,GAAG,MAAM,WAAW;AACnE,QAAM,eAAe,oBAAI,IAAI,CAAC,GAAG,oBAAoB,GAAG,kBAAkB,CAAC;AAC3E,QAAM,QAAQ,aAAa,GAAG,MAAM,cAAc,eAAe,cAAc;AAE/E,MAAI,QAAQ,GAAG;AACb,YAAQ,aAAa;AACrB,YAAQ,SAAS,KAAK,+DAA0D;AAAA,EAClF;AACF,CAAC;","names":[]}

package/dist/codemods/v1/memory-message-v2-type.js

@@ -0,0 +1,50 @@
+import {
+  createTransformer
+} from "../chunk-JNFQ6J6B.js";
+
+// src/codemods/v1/memory-message-v2-type.ts
+var memory_message_v2_type_default = createTransformer((_fileInfo, _api, _options, context) => {
+  const { j, root } = context;
+  const oldTypeName = "MastraMessageV2";
+  const newTypeName = "MastraDBMessage";
+  const localNamesToRewrite = /* @__PURE__ */ new Set();
+  root.find(j.ImportDeclaration).filter((path) => {
+    const source = path.value.source.value;
+    return typeof source === "string" && source === "@mastra/core";
+  }).forEach((path) => {
+    if (!path.value.specifiers) return;
+    path.value.specifiers.forEach((specifier) => {
+      if (specifier.type === "ImportSpecifier" && specifier.imported.type === "Identifier" && specifier.imported.name === oldTypeName) {
+        const localName = specifier.local?.name || oldTypeName;
+        const isAliased = localName !== oldTypeName;
+        specifier.imported.name = newTypeName;
+        if (!isAliased) {
+          if (specifier.local) {
+            specifier.local.name = newTypeName;
+          }
+          localNamesToRewrite.add(oldTypeName);
+        }
+        context.hasChanges = true;
+      }
+    });
+  });
+  if (localNamesToRewrite.size > 0) {
+    localNamesToRewrite.forEach((oldLocalName) => {
+      root.find(j.Identifier, { name: oldLocalName }).forEach((path) => {
+        const parent = path.parent;
+        if (parent && parent.value.type === "ImportSpecifier") {
+          return;
+        }
+        path.value.name = newTypeName;
+        context.hasChanges = true;
+      });
+    });
+  }
+  if (context.hasChanges) {
+    context.messages.push("Renamed MastraMessageV2 type to MastraDBMessage");
+  }
+});
+export {
+  memory_message_v2_type_default as default
+};
+//# sourceMappingURL=memory-message-v2-type.js.map

package/dist/codemods/v1/memory-message-v2-type.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/v1/memory-message-v2-type.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\n\n/**\n * Renames MastraMessageV2 type to MastraDBMessage.\n * This better describes the purpose as the database message format.\n *\n * Before:\n * import { MastraMessageV2 } from '@mastra/core';\n * function processMessage(message: MastraMessageV2) {}\n *\n * After:\n * import { MastraDBMessage } from '@mastra/core';\n * function processMessage(message: MastraDBMessage) {}\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  const oldTypeName = 'MastraMessageV2';\n  const newTypeName = 'MastraDBMessage';\n\n  // Track which local names should be rewritten (only non-aliased imports)\n  const localNamesToRewrite = new Set<string>();\n\n  // Transform import specifiers from @mastra/core\n  root\n    .find(j.ImportDeclaration)\n    .filter(path => {\n      const source = path.value.source.value;\n      return typeof source === 'string' && source === '@mastra/core';\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 === oldTypeName\n        ) {\n          const localName = specifier.local?.name || oldTypeName;\n          const isAliased = localName !== oldTypeName;\n\n          // Rename the imported name\n          specifier.imported.name = newTypeName;\n\n          // Only update local name and track for rewriting if not aliased\n          if (!isAliased) {\n            if (specifier.local) {\n              specifier.local.name = newTypeName;\n            }\n            // Track the old local name (oldTypeName) to rewrite all its usages\n            localNamesToRewrite.add(oldTypeName);\n          }\n          // If aliased, leave the local name intact and don't track for rewriting\n\n          context.hasChanges = true;\n        }\n      });\n    });\n\n  // Only transform usages for non-aliased imports\n  if (localNamesToRewrite.size > 0) {\n    // Transform all references that need to be rewritten\n    localNamesToRewrite.forEach(oldLocalName => {\n      root.find(j.Identifier, { name: oldLocalName }).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 = newTypeName;\n        context.hasChanges = true;\n      });\n    });\n  }\n\n  if (context.hasChanges) {\n    context.messages.push('Renamed MastraMessageV2 type to MastraDBMessage');\n  }\n});\n"],"mappings":";;;;;AAcA,IAAO,iCAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,cAAc;AACpB,QAAM,cAAc;AAGpB,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,aAC5B;AACA,cAAM,YAAY,UAAU,OAAO,QAAQ;AAC3C,cAAM,YAAY,cAAc;AAGhC,kBAAU,SAAS,OAAO;AAG1B,YAAI,CAAC,WAAW;AACd,cAAI,UAAU,OAAO;AACnB,sBAAU,MAAM,OAAO;AAAA,UACzB;AAEA,8BAAoB,IAAI,WAAW;AAAA,QACrC;AAGA,gBAAQ,aAAa;AAAA,MACvB;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAGH,MAAI,oBAAoB,OAAO,GAAG;AAEhC,wBAAoB,QAAQ,kBAAgB;AAC1C,WAAK,KAAK,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC,EAAE,QAAQ,UAAQ;AAE9D,cAAM,SAAS,KAAK;AACpB,YAAI,UAAU,OAAO,MAAM,SAAS,mBAAmB;AACrD;AAAA,QACF;AAEA,aAAK,MAAM,OAAO;AAClB,gBAAQ,aAAa;AAAA,MACvB,CAAC;AAAA,IACH,CAAC;AAAA,EACH;AAEA,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,iDAAiD;AAAA,EACzE;AACF,CAAC;","names":[]}

package/dist/codemods/v1/memory-query-to-recall.js

@@ -0,0 +1,22 @@
+import {
+  createTransformer
+} from "../chunk-JNFQ6J6B.js";
+import {
+  renameMethod,
+  trackClassInstances
+} from "../chunk-PKCFDS7J.js";
+
+// src/codemods/v1/memory-query-to-recall.ts
+var memory_query_to_recall_default = createTransformer((_fileInfo, _api, _options, context) => {
+  const { j, root } = context;
+  const memoryInstances = trackClassInstances(j, root, "Memory");
+  const count = renameMethod(j, root, memoryInstances, "query", "recall");
+  if (count > 0) {
+    context.hasChanges = true;
+    context.messages.push("Renamed query to recall on Memory instances");
+  }
+});
+export {
+  memory_query_to_recall_default as default
+};
+//# sourceMappingURL=memory-query-to-recall.js.map

package/dist/codemods/v1/memory-query-to-recall.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/v1/memory-query-to-recall.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\nimport { trackClassInstances, renameMethod } from '../lib/utils';\n\n/**\n * Renames memory.query() to memory.recall().\n * This better describes the action of retrieving messages from memory.\n *\n * Before:\n * const result = await memory.query({ threadId: 'thread-123' });\n *\n * After:\n * const result = await memory.recall({ threadId: 'thread-123' });\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  // Track Memory instances and rename query to recall in a single optimized pass\n  const memoryInstances = trackClassInstances(j, root, 'Memory');\n  const count = renameMethod(j, root, memoryInstances, 'query', 'recall');\n\n  if (count > 0) {\n    context.hasChanges = true;\n    context.messages.push('Renamed query to recall on Memory instances');\n  }\n});\n"],"mappings":";;;;;;;;;AAaA,IAAO,iCAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAGpB,QAAM,kBAAkB,oBAAoB,GAAG,MAAM,QAAQ;AAC7D,QAAM,QAAQ,aAAa,GAAG,MAAM,iBAAiB,SAAS,QAAQ;AAEtE,MAAI,QAAQ,GAAG;AACb,YAAQ,aAAa;AACrB,YAAQ,SAAS,KAAK,6CAA6C;AAAA,EACrE;AACF,CAAC;","names":[]}

package/dist/codemods/v1/memory-vector-search-param.js

@@ -0,0 +1,31 @@
+import {
+  createTransformer
+} from "../chunk-JNFQ6J6B.js";
+import {
+  trackClassInstances,
+  transformMethodCalls
+} from "../chunk-PKCFDS7J.js";
+
+// src/codemods/v1/memory-vector-search-param.ts
+var memory_vector_search_param_default = createTransformer((_fileInfo, _api, _options, context) => {
+  const { j, root } = context;
+  const memoryInstances = trackClassInstances(j, root, "Memory");
+  transformMethodCalls(j, root, memoryInstances, "recall", (path) => {
+    const args = path.value.arguments;
+    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 && prop.key.type === "Identifier" && prop.key.name === "vectorMessageSearch") {
+        prop.key.name = "vectorSearchString";
+        context.hasChanges = true;
+      }
+    });
+  });
+  if (context.hasChanges) {
+    context.messages.push("Renamed vectorMessageSearch to vectorSearchString in memory.recall() calls");
+  }
+});
+export {
+  memory_vector_search_param_default as default
+};
+//# sourceMappingURL=memory-vector-search-param.js.map

package/dist/codemods/v1/memory-vector-search-param.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/codemods/v1/memory-vector-search-param.ts"],"sourcesContent":["import { createTransformer } from '../lib/create-transformer';\nimport { trackClassInstances, transformMethodCalls } from '../lib/utils';\n\n/**\n * Renames vectorMessageSearch parameter to vectorSearchString in memory.recall() calls.\n * This provides more consistent naming.\n *\n * Before:\n * memory.recall({\n *   threadId: 'thread-123',\n *   vectorMessageSearch: 'What did we discuss?',\n * });\n *\n * After:\n * memory.recall({\n *   threadId: 'thread-123',\n *   vectorSearchString: 'What did we discuss?',\n * });\n */\nexport default createTransformer((_fileInfo, _api, _options, context) => {\n  const { j, root } = context;\n\n  const memoryInstances = trackClassInstances(j, root, 'Memory');\n\n  transformMethodCalls(j, root, memoryInstances, 'recall', path => {\n    const args = path.value.arguments;\n    const firstArg = args[0];\n    if (!firstArg || firstArg.type !== 'ObjectExpression' || !firstArg.properties) return;\n\n    firstArg.properties.forEach((prop: any) => {\n      if (\n        (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n        prop.key &&\n        prop.key.type === 'Identifier' &&\n        prop.key.name === 'vectorMessageSearch'\n      ) {\n        prop.key.name = 'vectorSearchString';\n        context.hasChanges = true;\n      }\n    });\n  });\n\n  if (context.hasChanges) {\n    context.messages.push('Renamed vectorMessageSearch to vectorSearchString in memory.recall() calls');\n  }\n});\n"],"mappings":";;;;;;;;;AAmBA,IAAO,qCAAQ,kBAAkB,CAAC,WAAW,MAAM,UAAU,YAAY;AACvE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,kBAAkB,oBAAoB,GAAG,MAAM,QAAQ;AAE7D,uBAAqB,GAAG,MAAM,iBAAiB,UAAU,UAAQ;AAC/D,UAAM,OAAO,KAAK,MAAM;AACxB,UAAM,WAAW,KAAK,CAAC;AACvB,QAAI,CAAC,YAAY,SAAS,SAAS,sBAAsB,CAAC,SAAS,WAAY;AAE/E,aAAS,WAAW,QAAQ,CAAC,SAAc;AACzC,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,OACL,KAAK,IAAI,SAAS,gBAClB,KAAK,IAAI,SAAS,uBAClB;AACA,aAAK,IAAI,OAAO;AAChB,gBAAQ,aAAa;AAAA,MACvB;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAED,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,4EAA4E;AAAA,EACpG;AACF,CAAC;","names":[]}

package/dist/codemods/v1/not-implemented/agent-format-parameter.js

@@ -0,0 +1,53 @@
+import {
+  insertCommentOnce
+} from "../../chunk-LGW3NUAJ.js";
+import {
+  createTransformer
+} from "../../chunk-JNFQ6J6B.js";
+
+// src/codemods/v1/not-implemented/agent-format-parameter.ts
+var agent_format_parameter_default = createTransformer((fileInfo, api, options, context) => {
+  const { j, root } = context;
+  const COMMENT_MESSAGE = "FIXME(mastra): The format parameter has been removed. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/agent#format-parameter-from-stream-and-generate";
+  const agentInstances = /* @__PURE__ */ new Set();
+  root.find(j.NewExpression, {
+    callee: {
+      type: "Identifier",
+      name: "Agent"
+    }
+  }).forEach((path) => {
+    const parent = path.parent.value;
+    if (parent.type === "VariableDeclarator" && parent.id.type === "Identifier") {
+      agentInstances.add(parent.id.name);
+    }
+  });
+  root.find(j.CallExpression).filter((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "MemberExpression") return false;
+    if (callee.object.type !== "Identifier") return false;
+    if (callee.property.type !== "Identifier") return false;
+    if (!agentInstances.has(callee.object.name)) return false;
+    return callee.property.name === "generate" || callee.property.name === "stream";
+  }).forEach((path) => {
+    const args = path.value.arguments;
+    if (args.length < 2) return;
+    const optionsArg = args[1];
+    if (!optionsArg || optionsArg.type !== "ObjectExpression") return;
+    if (!optionsArg.properties) return;
+    optionsArg.properties.forEach((prop) => {
+      if ((prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier" && prop.key.name === "format") {
+        const added = insertCommentOnce(prop, j, COMMENT_MESSAGE);
+        if (added) {
+          context.hasChanges = true;
+        }
+      }
+    });
+  });
+  if (context.hasChanges) {
+    context.messages.push(`Not Implemented ${fileInfo.path}: The format 'aisdk' parameter has been removed.`);
+  }
+});
+export {
+  agent_format_parameter_default as default
+};
+//# sourceMappingURL=agent-format-parameter.js.map

package/dist/codemods/v1/not-implemented/agent-format-parameter.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/codemods/v1/not-implemented/agent-format-parameter.ts"],"sourcesContent":["/* eslint-disable no-warning-comments */\nimport { insertCommentOnce } from '../../lib/add-comment';\nimport { createTransformer } from '../../lib/create-transformer';\n\n/**\n * Adds a FIXME comment above the format parameter in agent.generate() and agent.stream() calls.\n * The format parameter has been removed in v1 and requires manual migration.\n *\n * Before:\n * agent.generate('prompt', { format: 'aisdk' })\n *\n * After:\n * agent.generate('prompt', {\n *   /* FIXME(mastra): The format parameter has been removed. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/agent#format-parameter-from-stream-and-generate *\\/\n *   format: 'aisdk'\n * })\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  const COMMENT_MESSAGE =\n    'FIXME(mastra): The format parameter has been removed. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/agent#format-parameter-from-stream-and-generate';\n\n  // Track Agent instances\n  const agentInstances = new Set<string>();\n\n  root\n    .find(j.NewExpression, {\n      callee: {\n        type: 'Identifier',\n        name: 'Agent',\n      },\n    })\n    .forEach(path => {\n      const parent = path.parent.value;\n      if (parent.type === 'VariableDeclarator' && parent.id.type === 'Identifier') {\n        agentInstances.add(parent.id.name);\n      }\n    });\n\n  // Find agent.generate() and agent.stream() calls\n  root\n    .find(j.CallExpression)\n    .filter(path => {\n      const { callee } = path.value;\n      if (callee.type !== 'MemberExpression') return false;\n      if (callee.object.type !== 'Identifier') return false;\n      if (callee.property.type !== 'Identifier') return false;\n\n      // Only process if called on an Agent instance\n      if (!agentInstances.has(callee.object.name)) return false;\n\n      // Only process generate() and stream() methods\n      return callee.property.name === 'generate' || callee.property.name === 'stream';\n    })\n    .forEach(path => {\n      const args = path.value.arguments;\n\n      // We're looking for calls with an options object that has format parameter\n      if (args.length < 2) return;\n\n      const optionsArg = args[1];\n      if (!optionsArg || optionsArg.type !== 'ObjectExpression') return;\n      if (!optionsArg.properties) return;\n\n      // Find the format property\n      optionsArg.properties.forEach(prop => {\n        if (\n          (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n          prop.key?.type === 'Identifier' &&\n          prop.key.name === 'format'\n        ) {\n          // Add FIXME comment to the format property\n          const added = insertCommentOnce(prop, j, COMMENT_MESSAGE);\n          if (added) {\n            context.hasChanges = true;\n          }\n        }\n      });\n    });\n\n  if (context.hasChanges) {\n    context.messages.push(`Not Implemented ${fileInfo.path}: The format 'aisdk' parameter has been removed.`);\n  }\n});\n"],"mappings":";;;;;;;;AAiBA,IAAO,iCAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,kBACJ;AAGF,QAAM,iBAAiB,oBAAI,IAAY;AAEvC,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,qBAAe,IAAI,OAAO,GAAG,IAAI;AAAA,IACnC;AAAA,EACF,CAAC;AAGH,OACG,KAAK,EAAE,cAAc,EACrB,OAAO,UAAQ;AACd,UAAM,EAAE,OAAO,IAAI,KAAK;AACxB,QAAI,OAAO,SAAS,mBAAoB,QAAO;AAC/C,QAAI,OAAO,OAAO,SAAS,aAAc,QAAO;AAChD,QAAI,OAAO,SAAS,SAAS,aAAc,QAAO;AAGlD,QAAI,CAAC,eAAe,IAAI,OAAO,OAAO,IAAI,EAAG,QAAO;AAGpD,WAAO,OAAO,SAAS,SAAS,cAAc,OAAO,SAAS,SAAS;AAAA,EACzE,CAAC,EACA,QAAQ,UAAQ;AACf,UAAM,OAAO,KAAK,MAAM;AAGxB,QAAI,KAAK,SAAS,EAAG;AAErB,UAAM,aAAa,KAAK,CAAC;AACzB,QAAI,CAAC,cAAc,WAAW,SAAS,mBAAoB;AAC3D,QAAI,CAAC,WAAW,WAAY;AAG5B,eAAW,WAAW,QAAQ,UAAQ;AACpC,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS,UAClB;AAEA,cAAM,QAAQ,kBAAkB,MAAM,GAAG,eAAe;AACxD,YAAI,OAAO;AACT,kBAAQ,aAAa;AAAA,QACvB;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH,CAAC;AAEH,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,mBAAmB,SAAS,IAAI,kDAAkD;AAAA,EAC1G;AACF,CAAC;","names":[]}

package/dist/codemods/v1/not-implemented/agent-to-step.js

@@ -0,0 +1,54 @@
+import {
+  insertCommentOnce
+} from "../../chunk-LGW3NUAJ.js";
+import {
+  createTransformer
+} from "../../chunk-JNFQ6J6B.js";
+
+// src/codemods/v1/not-implemented/agent-to-step.ts
+var agent_to_step_default = createTransformer((fileInfo, api, options, context) => {
+  const { j, root } = context;
+  const COMMENT_MESSAGE = "FIXME(mastra): The toStep() method has been removed. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/agent#agenttostep-method";
+  const agentInstances = /* @__PURE__ */ new Set();
+  root.find(j.NewExpression, {
+    callee: {
+      type: "Identifier",
+      name: "Agent"
+    }
+  }).forEach((path) => {
+    const parent = path.parent.value;
+    if (parent.type === "VariableDeclarator" && parent.id.type === "Identifier") {
+      agentInstances.add(parent.id.name);
+    }
+  });
+  root.find(j.CallExpression).filter((path) => {
+    const { callee } = path.value;
+    if (callee.type !== "MemberExpression") return false;
+    if (callee.object.type !== "Identifier") return false;
+    if (callee.property.type !== "Identifier") return false;
+    if (!agentInstances.has(callee.object.name)) return false;
+    return callee.property.name === "toStep";
+  }).forEach((path) => {
+    let parent = path.parent;
+    while (parent && parent.value.type !== "VariableDeclaration" && parent.value.type !== "ExpressionStatement") {
+      parent = parent.parent;
+    }
+    if (parent && parent.value) {
+      let targetNode = parent.value;
+      if (parent.parent && parent.parent.value.type === "ExportNamedDeclaration") {
+        targetNode = parent.parent.value;
+      }
+      const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);
+      if (added) {
+        context.hasChanges = true;
+      }
+    }
+  });
+  if (context.hasChanges) {
+    context.messages.push(`Not Implemented ${fileInfo.path}: The toStep() method has been removed.`);
+  }
+});
+export {
+  agent_to_step_default as default
+};
+//# sourceMappingURL=agent-to-step.js.map

package/dist/codemods/v1/not-implemented/agent-to-step.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/codemods/v1/not-implemented/agent-to-step.ts"],"sourcesContent":["/* eslint-disable no-warning-comments */\nimport { insertCommentOnce } from '../../lib/add-comment';\nimport { createTransformer } from '../../lib/create-transformer';\n\n/**\n * Adds a FIXME comment above agent.toStep() method calls.\n * The toStep() method has been removed in v1 and requires manual migration.\n *\n * Before:\n * const step = agent.toStep();\n *\n * After:\n * /* FIXME(mastra): The toStep() method has been removed. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/agent#agenttostep-method *\\/\n * const step = agent.toStep();\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  const COMMENT_MESSAGE =\n    'FIXME(mastra): The toStep() method has been removed. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/agent#agenttostep-method';\n\n  // Track Agent instances\n  const agentInstances = new Set<string>();\n\n  root\n    .find(j.NewExpression, {\n      callee: {\n        type: 'Identifier',\n        name: 'Agent',\n      },\n    })\n    .forEach(path => {\n      const parent = path.parent.value;\n      if (parent.type === 'VariableDeclarator' && parent.id.type === 'Identifier') {\n        agentInstances.add(parent.id.name);\n      }\n    });\n\n  // Find agent.toStep() calls\n  root\n    .find(j.CallExpression)\n    .filter(path => {\n      const { callee } = path.value;\n      if (callee.type !== 'MemberExpression') return false;\n      if (callee.object.type !== 'Identifier') return false;\n      if (callee.property.type !== 'Identifier') return false;\n\n      // Only process if called on an Agent instance\n      if (!agentInstances.has(callee.object.name)) return false;\n\n      // Only process toStep() method\n      return callee.property.name === 'toStep';\n    })\n    .forEach(path => {\n      // Find the parent statement to add the comment to\n      let parent = path.parent;\n      while (parent && parent.value.type !== 'VariableDeclaration' && parent.value.type !== 'ExpressionStatement') {\n        parent = parent.parent;\n      }\n\n      if (parent && parent.value) {\n        // Check if this statement is wrapped in an export declaration\n        let targetNode = parent.value;\n        if (parent.parent && parent.parent.value.type === 'ExportNamedDeclaration') {\n          targetNode = parent.parent.value;\n        }\n\n        // Add FIXME comment to the statement (or export if it's exported)\n        const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n        if (added) {\n          context.hasChanges = true;\n        }\n      }\n    });\n\n  if (context.hasChanges) {\n    context.messages.push(`Not Implemented ${fileInfo.path}: The toStep() method has been removed.`);\n  }\n});\n"],"mappings":";;;;;;;;AAeA,IAAO,wBAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,kBACJ;AAGF,QAAM,iBAAiB,oBAAI,IAAY;AAEvC,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,qBAAe,IAAI,OAAO,GAAG,IAAI;AAAA,IACnC;AAAA,EACF,CAAC;AAGH,OACG,KAAK,EAAE,cAAc,EACrB,OAAO,UAAQ;AACd,UAAM,EAAE,OAAO,IAAI,KAAK;AACxB,QAAI,OAAO,SAAS,mBAAoB,QAAO;AAC/C,QAAI,OAAO,OAAO,SAAS,aAAc,QAAO;AAChD,QAAI,OAAO,SAAS,SAAS,aAAc,QAAO;AAGlD,QAAI,CAAC,eAAe,IAAI,OAAO,OAAO,IAAI,EAAG,QAAO;AAGpD,WAAO,OAAO,SAAS,SAAS;AAAA,EAClC,CAAC,EACA,QAAQ,UAAQ;AAEf,QAAI,SAAS,KAAK;AAClB,WAAO,UAAU,OAAO,MAAM,SAAS,yBAAyB,OAAO,MAAM,SAAS,uBAAuB;AAC3G,eAAS,OAAO;AAAA,IAClB;AAEA,QAAI,UAAU,OAAO,OAAO;AAE1B,UAAI,aAAa,OAAO;AACxB,UAAI,OAAO,UAAU,OAAO,OAAO,MAAM,SAAS,0BAA0B;AAC1E,qBAAa,OAAO,OAAO;AAAA,MAC7B;AAGA,YAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,UAAI,OAAO;AACT,gBAAQ,aAAa;AAAA,MACvB;AAAA,IACF;AAAA,EACF,CAAC;AAEH,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,mBAAmB,SAAS,IAAI,yCAAyC;AAAA,EACjG;AACF,CAAC;","names":[]}

package/dist/codemods/v1/not-implemented/mastra-memory.js

@@ -0,0 +1,50 @@
+import {
+  insertCommentOnce
+} from "../../chunk-LGW3NUAJ.js";
+import {
+  createTransformer
+} from "../../chunk-JNFQ6J6B.js";
+
+// src/codemods/v1/not-implemented/mastra-memory.ts
+var mastra_memory_default = createTransformer((fileInfo, api, options, context) => {
+  const { j, root } = context;
+  const COMMENT_MESSAGE = "FIXME(mastra): `memory` property has been removed. Memory is configured at the agent level. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/mastra#memory-property-from-mastra-class";
+  root.find(j.NewExpression, {
+    callee: {
+      type: "Identifier",
+      name: "Mastra"
+    }
+  }).forEach((path) => {
+    const hasMemory = path.value.arguments.some((arg) => {
+      if (arg.type === "ObjectExpression") {
+        return arg.properties?.some(
+          (prop) => (prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier" && prop.key.name === "memory"
+        );
+      }
+      return false;
+    });
+    if (hasMemory) {
+      let parent = path.parent;
+      while (parent && parent.value.type !== "VariableDeclaration" && parent.value.type !== "ExpressionStatement") {
+        parent = parent.parent;
+      }
+      if (parent && parent.value) {
+        let targetNode = parent.value;
+        if (parent.parent && parent.parent.value.type === "ExportNamedDeclaration") {
+          targetNode = parent.parent.value;
+        }
+        const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);
+        if (added) {
+          context.hasChanges = true;
+        }
+      }
+    }
+  });
+  if (context.hasChanges) {
+    context.messages.push(`Not Implemented ${fileInfo.path}: The memory property has been removed.`);
+  }
+});
+export {
+  mastra_memory_default as default
+};
+//# sourceMappingURL=mastra-memory.js.map

package/dist/codemods/v1/not-implemented/mastra-memory.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/codemods/v1/not-implemented/mastra-memory.ts"],"sourcesContent":["/* eslint-disable no-warning-comments */\nimport { insertCommentOnce } from '../../lib/add-comment';\nimport { createTransformer } from '../../lib/create-transformer';\n\n/**\n * Adds FIXME comment when Mastra config includes a memory property.\n * The memory property has been removed from Mastra class and should be configured at the agent level.\n *\n * Before:\n * const mastra = new Mastra({ memory: new Memory() });\n *\n * After:\n * /* FIXME(mastra): `memory` property has been removed. Memory is configured at the agent level. See: ... *\\/\n * const mastra = new Mastra({ memory: new Memory() });\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  const COMMENT_MESSAGE =\n    'FIXME(mastra): `memory` property has been removed. Memory is configured at the agent level. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/mastra#memory-property-from-mastra-class';\n\n  // Find new Mastra() expressions\n  root\n    .find(j.NewExpression, {\n      callee: {\n        type: 'Identifier',\n        name: 'Mastra',\n      },\n    })\n    .forEach(path => {\n      // Check if config has a memory property\n      const hasMemory = path.value.arguments.some(arg => {\n        if (arg.type === 'ObjectExpression') {\n          return arg.properties?.some(\n            prop =>\n              (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n              prop.key?.type === 'Identifier' &&\n              prop.key.name === 'memory',\n          );\n        }\n        return false;\n      });\n\n      if (hasMemory) {\n        // Find the parent statement to add comment\n        let parent = path.parent;\n        while (parent && parent.value.type !== 'VariableDeclaration' && parent.value.type !== 'ExpressionStatement') {\n          parent = parent.parent;\n        }\n\n        if (parent && parent.value) {\n          // Check if this statement is wrapped in an export declaration\n          let targetNode = parent.value;\n          if (parent.parent && parent.parent.value.type === 'ExportNamedDeclaration') {\n            targetNode = parent.parent.value;\n          }\n\n          const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n          if (added) {\n            context.hasChanges = true;\n          }\n        }\n      }\n    });\n\n  if (context.hasChanges) {\n    context.messages.push(`Not Implemented ${fileInfo.path}: The memory property has been removed.`);\n  }\n});\n"],"mappings":";;;;;;;;AAeA,IAAO,wBAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,kBACJ;AAGF,OACG,KAAK,EAAE,eAAe;AAAA,IACrB,QAAQ;AAAA,MACN,MAAM;AAAA,MACN,MAAM;AAAA,IACR;AAAA,EACF,CAAC,EACA,QAAQ,UAAQ;AAEf,UAAM,YAAY,KAAK,MAAM,UAAU,KAAK,SAAO;AACjD,UAAI,IAAI,SAAS,oBAAoB;AACnC,eAAO,IAAI,YAAY;AAAA,UACrB,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS;AAAA,QACtB;AAAA,MACF;AACA,aAAO;AAAA,IACT,CAAC;AAED,QAAI,WAAW;AAEb,UAAI,SAAS,KAAK;AAClB,aAAO,UAAU,OAAO,MAAM,SAAS,yBAAyB,OAAO,MAAM,SAAS,uBAAuB;AAC3G,iBAAS,OAAO;AAAA,MAClB;AAEA,UAAI,UAAU,OAAO,OAAO;AAE1B,YAAI,aAAa,OAAO;AACxB,YAAI,OAAO,UAAU,OAAO,OAAO,MAAM,SAAS,0BAA0B;AAC1E,uBAAa,OAAO,OAAO;AAAA,QAC7B;AAEA,cAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,YAAI,OAAO;AACT,kBAAQ,aAAa;AAAA,QACvB;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAEH,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,mBAAmB,SAAS,IAAI,yCAAyC;AAAA,EACjG;AACF,CAAC;","names":[]}

package/dist/codemods/v1/not-implemented/mastra-required-id.js

@@ -0,0 +1,113 @@
+import {
+  insertCommentOnce
+} from "../../chunk-LGW3NUAJ.js";
+import {
+  createTransformer
+} from "../../chunk-JNFQ6J6B.js";
+
+// src/codemods/v1/not-implemented/mastra-required-id.ts
+var mastra_required_id_default = createTransformer((fileInfo, api, options, context) => {
+  const { j, root } = context;
+  const COMMENT_MESSAGE = "FIXME(mastra): Add a unique `id` parameter. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/mastra#required-id-parameter-for-all-mastra-primitives";
+  const storageClasses = [
+    "LibSQLStore",
+    "PostgresStore",
+    "D1Store",
+    "MongoDBStore",
+    "DynamoDBStore",
+    "LibSQLVector",
+    "PgVector",
+    "ChromaVector",
+    "PineconeVector",
+    "QdrantVector",
+    "LanceVector",
+    "Agent",
+    "MCPServer"
+  ];
+  const createFunctions = ["createWorkflow", "createTool", "createScorer"];
+  root.find(j.NewExpression).forEach((path) => {
+    if (path.value.callee.type === "Identifier") {
+      const className = path.value.callee.name;
+      if (storageClasses.includes(className)) {
+        const hasId = path.value.arguments.some((arg) => {
+          if (arg.type === "ObjectExpression") {
+            return arg.properties?.some(
+              (prop) => (prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier" && prop.key.name === "id"
+            );
+          }
+          return false;
+        });
+        if (!hasId) {
+          let parent = path.parent;
+          const statementTypes = /* @__PURE__ */ new Set([
+            "VariableDeclaration",
+            "ExpressionStatement",
+            "ReturnStatement",
+            "ExportDefaultDeclaration",
+            "ExportNamedDeclaration",
+            "Program"
+          ]);
+          while (parent && !statementTypes.has(parent.value.type)) {
+            parent = parent.parent;
+          }
+          if (parent && parent.value) {
+            let targetNode = parent.value;
+            if (targetNode.type !== "ExportDefaultDeclaration" && targetNode.type !== "ExportNamedDeclaration" && parent.parent && (parent.parent.value.type === "ExportDefaultDeclaration" || parent.parent.value.type === "ExportNamedDeclaration")) {
+              targetNode = parent.parent.value;
+            }
+            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);
+            if (added) {
+              context.hasChanges = true;
+            }
+          }
+        }
+      }
+    }
+  });
+  root.find(j.CallExpression).forEach((path) => {
+    if (path.value.callee.type === "Identifier") {
+      const functionName = path.value.callee.name;
+      if (createFunctions.includes(functionName)) {
+        const hasId = path.value.arguments.some((arg) => {
+          if (arg.type === "ObjectExpression") {
+            return arg.properties?.some(
+              (prop) => (prop.type === "Property" || prop.type === "ObjectProperty") && prop.key?.type === "Identifier" && prop.key.name === "id"
+            );
+          }
+          return false;
+        });
+        if (!hasId) {
+          let parent = path.parent;
+          const statementTypes = /* @__PURE__ */ new Set([
+            "VariableDeclaration",
+            "ExpressionStatement",
+            "ReturnStatement",
+            "ExportDefaultDeclaration",
+            "ExportNamedDeclaration",
+            "Program"
+          ]);
+          while (parent && !statementTypes.has(parent.value.type)) {
+            parent = parent.parent;
+          }
+          if (parent && parent.value) {
+            let targetNode = parent.value;
+            if (targetNode.type !== "ExportDefaultDeclaration" && targetNode.type !== "ExportNamedDeclaration" && parent.parent && (parent.parent.value.type === "ExportDefaultDeclaration" || parent.parent.value.type === "ExportNamedDeclaration")) {
+              targetNode = parent.parent.value;
+            }
+            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);
+            if (added) {
+              context.hasChanges = true;
+            }
+          }
+        }
+      }
+    }
+  });
+  if (context.hasChanges) {
+    context.messages.push(`Not Implemented ${fileInfo.path}: Mastra primitives now require a unique id parameter.`);
+  }
+});
+export {
+  mastra_required_id_default as default
+};
+//# sourceMappingURL=mastra-required-id.js.map

package/dist/codemods/v1/not-implemented/mastra-required-id.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../../src/codemods/v1/not-implemented/mastra-required-id.ts"],"sourcesContent":["/* eslint-disable no-warning-comments */\nimport { insertCommentOnce } from '../../lib/add-comment';\nimport { createTransformer } from '../../lib/create-transformer';\n\n/**\n * Adds FIXME comments to Mastra primitives that now require an `id` parameter.\n * This includes storages, vector stores, agents, workflows, tools, scorers, and MCP servers.\n *\n * Before:\n * const agent = new Agent({ name: 'Support Agent' });\n * const tool = createTool({ description: 'Get weather' });\n *\n * After:\n * /* FIXME(mastra): Add a unique `id` parameter. See: ... *\\/\n * const agent = new Agent({ name: 'Support Agent' });\n * /* FIXME(mastra): Add a unique `id` parameter. See: ... *\\/\n * const tool = createTool({ description: 'Get weather' });\n */\nexport default createTransformer((fileInfo, api, options, context) => {\n  const { j, root } = context;\n\n  const COMMENT_MESSAGE =\n    'FIXME(mastra): Add a unique `id` parameter. See: https://mastra.ai/guides/v1/migrations/upgrade-to-v1/mastra#required-id-parameter-for-all-mastra-primitives';\n\n  // List of class names that require id\n  const storageClasses = [\n    'LibSQLStore',\n    'PostgresStore',\n    'D1Store',\n    'MongoDBStore',\n    'DynamoDBStore',\n    'LibSQLVector',\n    'PgVector',\n    'ChromaVector',\n    'PineconeVector',\n    'QdrantVector',\n    'LanceVector',\n    'Agent',\n    'MCPServer',\n  ];\n\n  // List of function names that require id\n  const createFunctions = ['createWorkflow', 'createTool', 'createScorer'];\n\n  // Find NewExpression for classes\n  root.find(j.NewExpression).forEach(path => {\n    if (path.value.callee.type === 'Identifier') {\n      const className = path.value.callee.name;\n\n      if (storageClasses.includes(className)) {\n        // Check if the config already has an id property\n        const hasId = path.value.arguments.some(arg => {\n          if (arg.type === 'ObjectExpression') {\n            return arg.properties?.some(\n              prop =>\n                (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n                prop.key?.type === 'Identifier' &&\n                prop.key.name === 'id',\n            );\n          }\n          return false;\n        });\n\n        if (!hasId) {\n          // Find the parent statement to add comment\n          let parent = path.parent;\n          const statementTypes = new Set([\n            'VariableDeclaration',\n            'ExpressionStatement',\n            'ReturnStatement',\n            'ExportDefaultDeclaration',\n            'ExportNamedDeclaration',\n            'Program',\n          ]);\n\n          while (parent && !statementTypes.has(parent.value.type)) {\n            parent = parent.parent;\n          }\n\n          if (parent && parent.value) {\n            // For export declarations, add comment to the export itself\n            let targetNode = parent.value;\n            if (\n              targetNode.type !== 'ExportDefaultDeclaration' &&\n              targetNode.type !== 'ExportNamedDeclaration' &&\n              parent.parent &&\n              (parent.parent.value.type === 'ExportDefaultDeclaration' ||\n                parent.parent.value.type === 'ExportNamedDeclaration')\n            ) {\n              targetNode = parent.parent.value;\n            }\n\n            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n            if (added) {\n              context.hasChanges = true;\n            }\n          }\n        }\n      }\n    }\n  });\n\n  // Find CallExpression for create functions\n  root.find(j.CallExpression).forEach(path => {\n    if (path.value.callee.type === 'Identifier') {\n      const functionName = path.value.callee.name;\n\n      if (createFunctions.includes(functionName)) {\n        // Check if the config already has an id property\n        const hasId = path.value.arguments.some(arg => {\n          if (arg.type === 'ObjectExpression') {\n            return arg.properties?.some(\n              prop =>\n                (prop.type === 'Property' || prop.type === 'ObjectProperty') &&\n                prop.key?.type === 'Identifier' &&\n                prop.key.name === 'id',\n            );\n          }\n          return false;\n        });\n\n        if (!hasId) {\n          // Find the parent statement to add comment\n          let parent = path.parent;\n          const statementTypes = new Set([\n            'VariableDeclaration',\n            'ExpressionStatement',\n            'ReturnStatement',\n            'ExportDefaultDeclaration',\n            'ExportNamedDeclaration',\n            'Program',\n          ]);\n\n          while (parent && !statementTypes.has(parent.value.type)) {\n            parent = parent.parent;\n          }\n\n          if (parent && parent.value) {\n            // For export declarations, add comment to the export itself\n            let targetNode = parent.value;\n            if (\n              targetNode.type !== 'ExportDefaultDeclaration' &&\n              targetNode.type !== 'ExportNamedDeclaration' &&\n              parent.parent &&\n              (parent.parent.value.type === 'ExportDefaultDeclaration' ||\n                parent.parent.value.type === 'ExportNamedDeclaration')\n            ) {\n              targetNode = parent.parent.value;\n            }\n\n            const added = insertCommentOnce(targetNode, j, COMMENT_MESSAGE);\n            if (added) {\n              context.hasChanges = true;\n            }\n          }\n        }\n      }\n    }\n  });\n\n  if (context.hasChanges) {\n    context.messages.push(`Not Implemented ${fileInfo.path}: Mastra primitives now require a unique id parameter.`);\n  }\n});\n"],"mappings":";;;;;;;;AAkBA,IAAO,6BAAQ,kBAAkB,CAAC,UAAU,KAAK,SAAS,YAAY;AACpE,QAAM,EAAE,GAAG,KAAK,IAAI;AAEpB,QAAM,kBACJ;AAGF,QAAM,iBAAiB;AAAA,IACrB;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAGA,QAAM,kBAAkB,CAAC,kBAAkB,cAAc,cAAc;AAGvE,OAAK,KAAK,EAAE,aAAa,EAAE,QAAQ,UAAQ;AACzC,QAAI,KAAK,MAAM,OAAO,SAAS,cAAc;AAC3C,YAAM,YAAY,KAAK,MAAM,OAAO;AAEpC,UAAI,eAAe,SAAS,SAAS,GAAG;AAEtC,cAAM,QAAQ,KAAK,MAAM,UAAU,KAAK,SAAO;AAC7C,cAAI,IAAI,SAAS,oBAAoB;AACnC,mBAAO,IAAI,YAAY;AAAA,cACrB,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS;AAAA,YACtB;AAAA,UACF;AACA,iBAAO;AAAA,QACT,CAAC;AAED,YAAI,CAAC,OAAO;AAEV,cAAI,SAAS,KAAK;AAClB,gBAAM,iBAAiB,oBAAI,IAAI;AAAA,YAC7B;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAED,iBAAO,UAAU,CAAC,eAAe,IAAI,OAAO,MAAM,IAAI,GAAG;AACvD,qBAAS,OAAO;AAAA,UAClB;AAEA,cAAI,UAAU,OAAO,OAAO;AAE1B,gBAAI,aAAa,OAAO;AACxB,gBACE,WAAW,SAAS,8BACpB,WAAW,SAAS,4BACpB,OAAO,WACN,OAAO,OAAO,MAAM,SAAS,8BAC5B,OAAO,OAAO,MAAM,SAAS,2BAC/B;AACA,2BAAa,OAAO,OAAO;AAAA,YAC7B;AAEA,kBAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,gBAAI,OAAO;AACT,sBAAQ,aAAa;AAAA,YACvB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAGD,OAAK,KAAK,EAAE,cAAc,EAAE,QAAQ,UAAQ;AAC1C,QAAI,KAAK,MAAM,OAAO,SAAS,cAAc;AAC3C,YAAM,eAAe,KAAK,MAAM,OAAO;AAEvC,UAAI,gBAAgB,SAAS,YAAY,GAAG;AAE1C,cAAM,QAAQ,KAAK,MAAM,UAAU,KAAK,SAAO;AAC7C,cAAI,IAAI,SAAS,oBAAoB;AACnC,mBAAO,IAAI,YAAY;AAAA,cACrB,WACG,KAAK,SAAS,cAAc,KAAK,SAAS,qBAC3C,KAAK,KAAK,SAAS,gBACnB,KAAK,IAAI,SAAS;AAAA,YACtB;AAAA,UACF;AACA,iBAAO;AAAA,QACT,CAAC;AAED,YAAI,CAAC,OAAO;AAEV,cAAI,SAAS,KAAK;AAClB,gBAAM,iBAAiB,oBAAI,IAAI;AAAA,YAC7B;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,YACA;AAAA,UACF,CAAC;AAED,iBAAO,UAAU,CAAC,eAAe,IAAI,OAAO,MAAM,IAAI,GAAG;AACvD,qBAAS,OAAO;AAAA,UAClB;AAEA,cAAI,UAAU,OAAO,OAAO;AAE1B,gBAAI,aAAa,OAAO;AACxB,gBACE,WAAW,SAAS,8BACpB,WAAW,SAAS,4BACpB,OAAO,WACN,OAAO,OAAO,MAAM,SAAS,8BAC5B,OAAO,OAAO,MAAM,SAAS,2BAC/B;AACA,2BAAa,OAAO,OAAO;AAAA,YAC7B;AAEA,kBAAM,QAAQ,kBAAkB,YAAY,GAAG,eAAe;AAC9D,gBAAI,OAAO;AACT,sBAAQ,aAAa;AAAA,YACvB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF,CAAC;AAED,MAAI,QAAQ,YAAY;AACtB,YAAQ,SAAS,KAAK,mBAAmB,SAAS,IAAI,wDAAwD;AAAA,EAChH;AACF,CAAC;","names":[]}

package/dist/codemods/v1/runtime-context.js

@@ -1,45 +1,6 @@
-"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/v1/runtime-context.ts
-var runtime_context_exports = {};
-__export(runtime_context_exports, {
-  default: () => runtime_context_default
-});
-module.exports = __toCommonJS(runtime_context_exports);
-
-// 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;
-  };
-}
+import {
+  createTransformer
+} from "../chunk-JNFQ6J6B.js";
 
 // src/codemods/v1/runtime-context.ts
 var runtime_context_default = createTransformer((fileInfo, api, options, context) => {
@@ -163,4 +124,7 @@ function shouldProcessKey(key, node) {
   if (key === "loc" || key === "comments") return false;
   return true;
 }
+export {
+  runtime_context_default as default
+};
 //# sourceMappingURL=runtime-context.js.map