mastra 1.0.0-beta.9 → 1.0.0

package/dist/index.js

@@ -1,24 +1,26 @@
 #! /usr/bin/env node
-import { package_default, COMPONENTS, parseComponents, LLMProvider, parseLlmProvider, parseMcp, create, checkForPkgJson, getVersionTag, checkAndInstallCoreDeps, interactivePrompt, init, logger, FileService, createLogger, shouldSkipDotenvLoading, DepsService } from './chunk-SOGF2UZZ.js';
-export { create } from './chunk-SOGF2UZZ.js';
-import { PosthogAnalytics, setAnalytics } from './chunk-TL3K64EI.js';
-export { PosthogAnalytics } from './chunk-TL3K64EI.js';
+import { package_default, COMPONENTS, parseComponents, LLMProvider, parseLlmProvider, parseMcp, create, checkForPkgJson, getVersionTag, isGitInitialized, checkAndInstallCoreDeps, interactivePrompt, init, logger, FileService, createLogger, shouldSkipDotenvLoading, DepsService } from './chunk-CIB5EGSF.js';
+export { create } from './chunk-CIB5EGSF.js';
+import { PosthogAnalytics, setAnalytics } from './chunk-3MHPDBSK.js';
+export { PosthogAnalytics } from './chunk-3MHPDBSK.js';
 import { Command } from 'commander';
-import pc2 from 'picocolors';
+import pc3 from 'picocolors';
 import * as p2 from '@clack/prompts';
 import fs, { existsSync, readFileSync } from 'fs';
 import path, { dirname, join } from 'path';
-import { getServerOptions, normalizeStudioBase, FileService as FileService$2, getBundlerOptions, getWatcherInputOptions, createWatcher } from '@mastra/deployer/build';
-import { Bundler } from '@mastra/deployer/bundler';
+import { fileURLToPath } from 'url';
+import { getServerOptions, normalizeStudioBase, FileService as FileService$2, getWatcherInputOptions, createWatcher } from '@mastra/deployer/build';
+import { Bundler, IS_DEFAULT } from '@mastra/deployer/bundler';
+import * as fsExtra from 'fs-extra';
+import { copy } from 'fs-extra';
 import { getDeployer, FileService as FileService$1 } from '@mastra/deployer';
 import stripJsonComments from 'strip-json-comments';
-import process2 from 'process';
-import devcert from '@expo/devcert';
+import process3 from 'process';
 import { execa } from 'execa';
+import { writeFile, mkdir } from 'fs/promises';
+import devcert from '@expo/devcert';
 import getPort from 'get-port';
-import { writeFile } from 'fs/promises';
-import { fileURLToPath } from 'url';
-import * as fsExtra from 'fs-extra';
+import { getPackageInfo } from 'local-pkg';
 import { spawn } from 'child_process';
 import { config } from 'dotenv';
 import http from 'http';
@@ -406,7 +408,7 @@ async function initializeScorer(scorer, customPath) {
 }
 function showSuccessNote() {
   p2.note(`
-        ${pc2.green("To use: Add the Scorer to your workflow or agent!")}
+        ${pc3.green("To use: Add the Scorer to your workflow or agent!")}
         `);
 }
 async function showInteractivePrompt(providedCustomDir) {
@@ -417,7 +419,7 @@ async function showInteractivePrompt(providedCustomDir) {
   let customPath = providedCustomDir;
   if (!providedCustomDir) {
     const useCustomDir = await p2.confirm({
-      message: `Would you like to use a custom directory?${pc2.gray("(Default: src/mastra/scorers)")}`,
+      message: `Would you like to use a custom directory?${pc3.gray("(Default: src/mastra/scorers)")}`,
       initialValue: false
     });
     if (p2.isCancel(useCustomDir)) {
@@ -473,8 +475,21 @@ var addScorer = async (scorerName, args) => {
   });
 };
 var BuildBundler = class extends Bundler {
-  constructor() {
+  studio;
+  constructor({ studio: studio2 } = {}) {
     super("Build");
+    this.studio = studio2 ?? false;
+    this.platform = process.versions?.bun ? "neutral" : "node";
+  }
+  async getUserBundlerOptions(mastraEntryFile, outputDirectory) {
+    const bundlerOptions = await super.getUserBundlerOptions(mastraEntryFile, outputDirectory);
+    if (!bundlerOptions?.[IS_DEFAULT]) {
+      return bundlerOptions;
+    }
+    return {
+      ...bundlerOptions,
+      externals: true
+    };
   }
   getEnvFiles() {
     if (shouldSkipDotenvLoading()) {
@@ -491,6 +506,14 @@ var BuildBundler = class extends Bundler {
   }
   async prepare(outputDirectory) {
     await super.prepare(outputDirectory);
+    if (this.studio) {
+      const __filename2 = fileURLToPath(import.meta.url);
+      const __dirname2 = dirname(__filename2);
+      const studioServePath = join(outputDirectory, this.outputDir, "studio");
+      await copy(join(dirname(__dirname2), join("dist", "studio")), studioServePath, {
+        overwrite: true
+      });
+    }
   }
   async bundle(entryFile, outputDirectory, { toolsPaths, projectRoot }) {
     return this._bundle(this.getEntry(), entryFile, { outputDirectory, projectRoot }, toolsPaths);
@@ -503,7 +526,7 @@ var BuildBundler = class extends Bundler {
     import { createNodeServer, getToolExports } from '#server';
     import { tools } from '#tools';
     // @ts-ignore
-    await createNodeServer(mastra, { tools: getToolExports(tools) });
+    await createNodeServer(mastra, { tools: getToolExports(tools), studio: ${this.studio} });
 
     if (mastra.getStorage()) {
       // start storage init in the background
@@ -520,6 +543,7 @@ async function build({
   dir: dir2,
   tools,
   root,
+  studio: studio2,
   debug
 }) {
   const rootDir = root || process.cwd();
@@ -531,7 +555,7 @@ async function build({
     const mastraEntryFile = fs3.getFirstExistingFile([join(mastraDir, "index.ts"), join(mastraDir, "index.js")]);
     const platformDeployer = await getDeployer(mastraEntryFile, outputDirectory);
     if (!platformDeployer) {
-      const deployer = new BuildBundler();
+      const deployer = new BuildBundler({ studio: studio2 });
       deployer.__setLogger(logger2);
       const discoveredTools2 = deployer.getAllToolPaths(mastraDir, tools);
       await deployer.prepare(outputDirectory);
@@ -540,7 +564,13 @@ async function build({
         projectRoot: rootDir
       });
       logger2.info(`Build successful, you can now deploy the .mastra/output directory to your target platform.`);
-      logger2.info(`To start the server, run: node .mastra/output/index.mjs`);
+      if (studio2) {
+        logger2.info(
+          `To start the server with studio, run: MASTRA_STUDIO_PATH=.mastra/output/studio node .mastra/output/index.mjs`
+        );
+      } else {
+        logger2.info(`To start the server, run: node .mastra/output/index.mjs`);
+      }
       return;
     }
     logger2.info("Deployer found, preparing deployer build...");
@@ -580,6 +610,7 @@ var buildProject = async (args) => {
         dir: args?.dir,
         root: args?.root,
         tools: args?.tools ? args.tools.split(",") : [],
+        studio: args?.studio,
         debug: args.debug
       });
     },
@@ -632,9 +663,10 @@ var initProject = async (args) => {
     execution: async () => {
       await checkForPkgJson();
       const versionTag = await getVersionTag();
+      const skipGitInit = await isGitInitialized({ cwd: process.cwd() });
       await checkAndInstallCoreDeps(Boolean(args?.example || args?.default), versionTag);
       if (!Object.keys(args).length) {
-        const result = await interactivePrompt();
+        const result = await interactivePrompt({ skip: { gitInit: skipGitInit } });
         await init({
           ...result,
           llmApiKey: result?.llmApiKey,
@@ -863,15 +895,15 @@ function formatTable(scorers) {
   const idWidth = Math.max(2, Math.max(...scorers.map((s) => s.id.length)));
   const typeWidth = Math.max(4, Math.max(...scorers.map((s) => s.type.length)));
   const descWidth = Math.max(11, Math.max(...scorers.map((s) => s.description.length)));
-  const header = `${pc2.bold("Name".padEnd(nameWidth))} \u2502 ${pc2.bold("ID".padEnd(idWidth))} \u2502 ${pc2.bold("Type".padEnd(typeWidth))} \u2502 ${pc2.bold("Description".padEnd(descWidth))}`;
+  const header = `${pc3.bold("Name".padEnd(nameWidth))} \u2502 ${pc3.bold("ID".padEnd(idWidth))} \u2502 ${pc3.bold("Type".padEnd(typeWidth))} \u2502 ${pc3.bold("Description".padEnd(descWidth))}`;
   const separator = "\u2500".repeat(nameWidth) + "\u2500\u253C\u2500" + "\u2500".repeat(idWidth) + "\u2500\u253C\u2500" + "\u2500".repeat(typeWidth) + "\u2500\u253C\u2500" + "\u2500".repeat(descWidth) + "\u2500";
   const rows = scorers.map(
-    (scorer) => `${(scorer?.name ?? scorer?.id).padEnd(nameWidth)} \u2502 ${pc2.dim(scorer.id.padEnd(idWidth))} \u2502 ${scorer.type.padEnd(typeWidth)} \u2502 ${pc2.dim(scorer.description.padEnd(descWidth))}`
+    (scorer) => `${(scorer?.name ?? scorer?.id).padEnd(nameWidth)} \u2502 ${pc3.dim(scorer.id.padEnd(idWidth))} \u2502 ${scorer.type.padEnd(typeWidth)} \u2502 ${pc3.dim(scorer.description.padEnd(descWidth))}`
   );
   return [header, separator, ...rows].join("\n");
 }
 function listAllScorers() {
-  p2.intro(pc2.inverse(" Available Scorers "));
+  p2.intro(pc3.inverse(" Available Scorers "));
   const groupedScorers = AVAILABLE_SCORERS.reduce(
     (acc, scorer) => {
       if (!acc[scorer.category]) {
@@ -883,7 +915,7 @@ function listAllScorers() {
     {}
   );
   for (const [category, scorers] of Object.entries(groupedScorers)) {
-    p2.log.info(`${pc2.bold(pc2.cyan(formatCategoryName(category)))} Scorers:`);
+    p2.log.info(`${pc3.bold(pc3.cyan(formatCategoryName(category)))} Scorers:`);
     p2.log.message(formatTable(scorers));
   }
 }
@@ -900,6 +932,241 @@ var listScorers = async (args) => {
     origin: origin5
   });
 };
+var MigrateBundler = class extends BuildBundler {
+  customEnvFile;
+  constructor(customEnvFile) {
+    super({ studio: false });
+    this.customEnvFile = customEnvFile;
+  }
+  getEnvFiles() {
+    if (shouldSkipDotenvLoading()) {
+      return Promise.resolve([]);
+    }
+    const possibleFiles = [".env.development", ".env.local", ".env"];
+    if (this.customEnvFile) {
+      possibleFiles.unshift(this.customEnvFile);
+    }
+    try {
+      const fileService = new FileService$2();
+      const envFile = fileService.getFirstExistingFile(possibleFiles);
+      return Promise.resolve([envFile]);
+    } catch {
+    }
+    return Promise.resolve([]);
+  }
+  async bundle(entryFile, outputDirectory, { toolsPaths, projectRoot }) {
+    return this._bundle(this.getEntry(), entryFile, { outputDirectory, projectRoot }, toolsPaths);
+  }
+  getEntry() {
+    return `
+    import { mastra } from '#mastra';
+
+    async function runMigration() {
+      const storage = mastra.getStorage();
+
+      if (!storage) {
+        console.log(JSON.stringify({
+          success: false,
+          alreadyMigrated: false,
+          duplicatesRemoved: 0,
+          message: 'Storage not configured. Please configure storage in your Mastra instance.',
+        }));
+        process.exit(1);
+      }
+
+      // Access the observability store directly from storage.stores
+      const observabilityStore = storage.stores?.observability;
+
+      if (!observabilityStore) {
+        console.log(JSON.stringify({
+          success: false,
+          alreadyMigrated: false,
+          duplicatesRemoved: 0,
+          message: 'Observability storage not configured. Migration not required.',
+        }));
+        process.exit(0);
+      }
+
+      // Check if the store has a migrateSpans method
+      if (typeof observabilityStore.migrateSpans !== 'function') {
+        console.log(JSON.stringify({
+          success: false,
+          alreadyMigrated: false,
+          duplicatesRemoved: 0,
+          message: 'Migration not supported for this storage backend.',
+        }));
+        process.exit(1);
+      }
+
+      try {
+        // Run the migration - migrateSpans handles everything internally
+        const result = await observabilityStore.migrateSpans();
+
+        console.log(JSON.stringify({
+          success: result.success,
+          alreadyMigrated: result.alreadyMigrated,
+          duplicatesRemoved: result.duplicatesRemoved,
+          message: result.message,
+        }));
+
+        process.exit(result.success ? 0 : 1);
+      } catch (error) {
+        console.log(JSON.stringify({
+          success: false,
+          alreadyMigrated: false,
+          duplicatesRemoved: 0,
+          message: error instanceof Error ? error.message : 'Unknown error during migration',
+        }));
+        process.exit(1);
+      }
+    }
+
+    runMigration();
+    `;
+  }
+};
+
+// src/commands/migrate/migrate.ts
+async function migrate({
+  dir: dir2,
+  root,
+  env,
+  debug,
+  yes
+}) {
+  const logger2 = createLogger(debug);
+  const rootDir = root || process3.cwd();
+  const mastraDir = dir2 ? dir2.startsWith("/") ? dir2 : join(process3.cwd(), dir2) : join(process3.cwd(), "src", "mastra");
+  const dotMastraPath = join(rootDir, ".mastra");
+  p2.intro(pc3.cyan("Mastra Storage Migration"));
+  if (!yes) {
+    p2.log.warn(pc3.yellow("Warning: This migration will modify your database."));
+    p2.log.message("Before proceeding, please ensure you have:");
+    p2.log.message("  \u2022 Created a backup of your database");
+    p2.log.message("  \u2022 Tested this migration in a non-production environment");
+    const confirmed = await p2.confirm({
+      message: "Have you backed up your database and are ready to proceed?",
+      initialValue: false
+    });
+    if (p2.isCancel(confirmed) || !confirmed) {
+      p2.log.info("Migration cancelled. Please back up your database before running this command.");
+      p2.log.message(pc3.dim("Tip: Use --yes or -y to skip this prompt in CI/automation."));
+      process3.exit(0);
+    }
+  }
+  try {
+    const fileService = new FileService$1();
+    const entryFile = fileService.getFirstExistingFile([join(mastraDir, "index.ts"), join(mastraDir, "index.js")]);
+    const bundler = new MigrateBundler(env);
+    bundler.__setLogger(logger2);
+    logger2.info("Building project for migration...");
+    await bundler.prepare(dotMastraPath);
+    const discoveredTools = bundler.getAllToolPaths(mastraDir, []);
+    await bundler.bundle(entryFile, dotMastraPath, {
+      toolsPaths: discoveredTools,
+      projectRoot: rootDir
+    });
+    logger2.info("Running migration...");
+    logger2.info(pc3.dim("This may take a while for large tables."));
+    logger2.info("");
+    const loadedEnv = await bundler.loadEnvVars();
+    const migrationProcess = execa(process3.execPath, [join(dotMastraPath, "output", "index.mjs")], {
+      cwd: rootDir,
+      env: {
+        NODE_ENV: "production",
+        MASTRA_DISABLE_STORAGE_INIT: "true",
+        // Prevent MIGRATION_REQUIRED error during import
+        ...Object.fromEntries(loadedEnv)
+      },
+      stdio: ["inherit", "pipe", "pipe"],
+      reject: false
+    });
+    let stdoutData = "";
+    let stderrData = "";
+    migrationProcess.stdout?.on("data", (data) => {
+      stdoutData += data.toString();
+    });
+    migrationProcess.stderr?.on("data", (data) => {
+      stderrData += data.toString();
+      if (debug) {
+        process3.stderr.write(data);
+      }
+    });
+    const processResult = await migrationProcess;
+    let result;
+    try {
+      const jsonMatch = stdoutData.match(/\{[^{}]*"success"[^{}]*\}/g);
+      if (jsonMatch && jsonMatch.length > 0) {
+        result = JSON.parse(jsonMatch[jsonMatch.length - 1]);
+      }
+    } catch {
+    }
+    if (result) {
+      if (result.success) {
+        if (result.alreadyMigrated) {
+          logger2.info(pc3.green("\u2713 Migration already complete."));
+        } else {
+          logger2.info(pc3.green("\u2713 Migration completed successfully!"));
+          if (result.duplicatesRemoved > 0) {
+            logger2.info(`  Removed ${result.duplicatesRemoved} duplicate entries.`);
+          }
+        }
+        logger2.info(`  ${result.message}`);
+      } else {
+        logger2.error(pc3.red("\u2717 Migration failed."));
+        logger2.error(`  ${result.message}`);
+        process3.exit(1);
+      }
+    } else {
+      if (processResult.exitCode !== 0) {
+        logger2.error(pc3.red("\u2717 Migration failed."));
+        if (stderrData) {
+          logger2.error(stderrData);
+        }
+        if (stdoutData && !stdoutData.includes('"success"')) {
+          logger2.error(stdoutData);
+        }
+        process3.exit(1);
+      } else {
+        logger2.info(pc3.green("\u2713 Migration completed."));
+      }
+    }
+  } catch (error) {
+    if (error.code === "ERR_MODULE_NOT_FOUND" || error.message?.includes("Cannot find module")) {
+      logger2.error(pc3.red("Error: Could not find Mastra entry file."));
+      logger2.info("");
+      logger2.info("Make sure you have a mastra directory with an index.ts or index.js file.");
+      logger2.info(`Expected location: ${mastraDir}`);
+      logger2.info("");
+      logger2.info("You can specify a custom directory:");
+      logger2.info(pc3.cyan("  npx mastra migrate --dir path/to/mastra"));
+    } else {
+      logger2.error(pc3.red(`Error: ${error.message}`));
+      if (debug) {
+        logger2.error(error);
+      }
+    }
+    process3.exit(1);
+  }
+}
+
+// src/commands/actions/migrate.ts
+var migrate2 = async (args) => {
+  await analytics.trackCommandExecution({
+    command: "mastra migrate",
+    args: { ...args },
+    execution: async () => {
+      await migrate({
+        dir: args?.dir,
+        root: args?.root,
+        env: args?.env,
+        debug: args?.debug ?? false,
+        yes: args?.yes ?? false
+      });
+    },
+    origin: origin2
+  });
+};
 var DevLogger = class {
   options;
   constructor(options = {}) {
@@ -911,32 +1178,32 @@ var DevLogger = class {
   }
   formatTime() {
     if (!this.options.timestamp) return "";
-    return pc2.dim((/* @__PURE__ */ new Date()).toLocaleTimeString());
+    return pc3.dim((/* @__PURE__ */ new Date()).toLocaleTimeString());
   }
   formatPrefix(text2, color2) {
     const time = this.formatTime();
-    const prefix = pc2.bold(color2(text2));
+    const prefix = pc3.bold(color2(text2));
     return time ? `${time} ${prefix}` : prefix;
   }
   info(message) {
-    const prefix = this.formatPrefix("\u25D0", pc2.cyan);
+    const prefix = this.formatPrefix("\u25D0", pc3.cyan);
     console.info(`${prefix} ${message}`);
   }
   success(message) {
-    const prefix = this.formatPrefix("\u2713", pc2.green);
-    console.info(`${prefix} ${pc2.green(message)}`);
+    const prefix = this.formatPrefix("\u2713", pc3.green);
+    console.info(`${prefix} ${pc3.green(message)}`);
   }
   warn(message) {
-    const prefix = this.formatPrefix("\u26A0", pc2.yellow);
-    console.info(`${prefix} ${pc2.yellow(message)}`);
+    const prefix = this.formatPrefix("\u26A0", pc3.yellow);
+    console.info(`${prefix} ${pc3.yellow(message)}`);
   }
   error(message) {
-    const prefix = this.formatPrefix("\u2717", pc2.red);
-    console.info(`${prefix} ${pc2.red(message)}`);
+    const prefix = this.formatPrefix("\u2717", pc3.red);
+    console.info(`${prefix} ${pc3.red(message)}`);
   }
   starting() {
-    const prefix = this.formatPrefix("\u25C7", pc2.blue);
-    console.info(`${prefix} ${pc2.blue("Starting Mastra dev server...")}`);
+    const prefix = this.formatPrefix("\u25C7", pc3.blue);
+    console.info(`${prefix} ${pc3.blue("Starting Mastra dev server...")}`);
   }
   ready(host, port, studioBasePath, startTime, https) {
     let protocol = "http";
@@ -945,62 +1212,62 @@ var DevLogger = class {
     }
     console.info("");
     const timing = startTime ? `${Date.now() - startTime} ms` : "XXX ms";
-    console.info(pc2.inverse(pc2.green(" mastra ")) + ` ${pc2.green(version)} ${pc2.gray("ready in")} ${timing}`);
+    console.info(pc3.inverse(pc3.green(" mastra ")) + ` ${pc3.green(version)} ${pc3.gray("ready in")} ${timing}`);
     console.info("");
-    console.info(`${pc2.dim("\u2502")} ${pc2.bold("Studio:")} ${pc2.cyan(`${protocol}://${host}:${port}${studioBasePath}`)}`);
-    console.info(`${pc2.dim("\u2502")} ${pc2.bold("API:")}    ${`${protocol}://${host}:${port}/api`}`);
+    console.info(`${pc3.dim("\u2502")} ${pc3.bold("Studio:")} ${pc3.cyan(`${protocol}://${host}:${port}${studioBasePath}`)}`);
+    console.info(`${pc3.dim("\u2502")} ${pc3.bold("API:")}    ${`${protocol}://${host}:${port}/api`}`);
     console.info("");
   }
   bundling() {
-    const prefix = this.formatPrefix("\u25D0", pc2.magenta);
-    console.info(`${prefix} ${pc2.magenta("Bundling...")}`);
+    const prefix = this.formatPrefix("\u25D0", pc3.magenta);
+    console.info(`${prefix} ${pc3.magenta("Bundling...")}`);
   }
   bundleComplete() {
-    const prefix = this.formatPrefix("\u2713", pc2.green);
-    console.info(`${prefix} ${pc2.green("Bundle complete")}`);
+    const prefix = this.formatPrefix("\u2713", pc3.green);
+    console.info(`${prefix} ${pc3.green("Bundle complete")}`);
   }
   watching() {
     const time = this.formatTime();
-    const icon = pc2.dim("\u25EF");
-    const message = pc2.dim("watching for file changes...");
+    const icon = pc3.dim("\u25EF");
+    const message = pc3.dim("watching for file changes...");
     const fullMessage = `${icon} ${message}`;
     console.info(time ? `${time} ${fullMessage}` : fullMessage);
   }
   restarting() {
-    const prefix = this.formatPrefix("\u21BB", pc2.blue);
-    console.info(`${prefix} ${pc2.blue("Restarting server...")}`);
+    const prefix = this.formatPrefix("\u21BB", pc3.blue);
+    console.info(`${prefix} ${pc3.blue("Restarting server...")}`);
   }
   fileChange(file) {
-    const prefix = this.formatPrefix("\u26A1", pc2.cyan);
+    const prefix = this.formatPrefix("\u26A1", pc3.cyan);
     const fileName = path.basename(file);
-    console.info(`${prefix} ${pc2.cyan("File changed:")} ${pc2.dim(fileName)}`);
+    console.info(`${prefix} ${pc3.cyan("File changed:")} ${pc3.dim(fileName)}`);
   }
   // Enhanced error reporting
   serverError(error) {
     console.info("");
-    console.info(pc2.red("  \u2717 ") + pc2.bold(pc2.red("Server Error")));
+    console.info(pc3.red("  \u2717 ") + pc3.bold(pc3.red("Server Error")));
     console.info("");
-    console.info(`  ${pc2.red("\u2502")} ${error}`);
+    console.info(`  ${pc3.red("\u2502")} ${error}`);
     console.info("");
   }
   shutdown() {
     console.info("");
-    const prefix = this.formatPrefix("\u2713", pc2.green);
-    console.info(`${prefix} ${pc2.green("Dev server stopped")}`);
+    const prefix = this.formatPrefix("\u2713", pc3.green);
+    console.info(`${prefix} ${pc3.green("Dev server stopped")}`);
   }
   envInfo(info) {
     console.info("");
-    console.info(`  ${pc2.dim("\u2502")} ${pc2.bold("Environment:")} ${pc2.cyan(info.env || "development")}`);
-    console.info(`  ${pc2.dim("\u2502")} ${pc2.bold("Root:")} ${pc2.dim(info.root)}`);
-    console.info(`  ${pc2.dim("\u2502")} ${pc2.bold("Port:")} ${pc2.cyan(info.port.toString())}`);
+    console.info(`  ${pc3.dim("\u2502")} ${pc3.bold("Environment:")} ${pc3.cyan(info.env || "development")}`);
+    console.info(`  ${pc3.dim("\u2502")} ${pc3.bold("Root:")} ${pc3.dim(info.root)}`);
+    console.info(`  ${pc3.dim("\u2502")} ${pc3.bold("Port:")} ${pc3.cyan(info.port.toString())}`);
   }
   raw(message) {
     console.info(message);
   }
   debug(message) {
     if (process.env.DEBUG || process.env.MASTRA_DEBUG) {
-      const prefix = this.formatPrefix("\u25E6", pc2.gray);
-      console.info(`${prefix} ${pc2.gray(message)}`);
+      const prefix = this.formatPrefix("\u25E6", pc3.gray);
+      console.info(`${prefix} ${pc3.gray(message)}`);
     }
   }
   spinnerChars = ["\u25D0", "\u25D3", "\u25D1", "\u25D2"];
@@ -1015,16 +1282,48 @@ var DevLogger = class {
   }
   update(message) {
     this.clearLine();
-    const prefix = this.formatPrefix(this.getSpinnerChar(), pc2.cyan);
+    const prefix = this.formatPrefix(this.getSpinnerChar(), pc3.cyan);
     process.stdout.write(`${prefix} ${message}`);
   }
 };
 var devLogger = new DevLogger();
+async function getResolvedVersion(packageName, specifiedVersion) {
+  try {
+    const packageInfo = await getPackageInfo(packageName);
+    return packageInfo?.version ?? specifiedVersion;
+  } catch {
+    return specifiedVersion;
+  }
+}
+async function getMastraPackages2(rootDir) {
+  try {
+    const packageJsonPath = join(rootDir, "package.json");
+    const packageJsonContent = readFileSync(packageJsonPath, "utf-8");
+    const packageJson = JSON.parse(packageJsonContent);
+    const allDependencies = {
+      ...packageJson.dependencies ?? {},
+      ...packageJson.devDependencies ?? {}
+    };
+    const mastraDeps = Object.entries(allDependencies).filter(
+      ([name]) => name.startsWith("@mastra/") || name === "mastra"
+    );
+    const packages = await Promise.all(
+      mastraDeps.map(async ([name, specifiedVersion]) => ({
+        name,
+        version: await getResolvedVersion(name, specifiedVersion)
+      }))
+    );
+    return packages;
+  } catch {
+    return [];
+  }
+}
 var DevBundler = class extends Bundler {
   customEnvFile;
   constructor(customEnvFile) {
     super("Dev");
     this.customEnvFile = customEnvFile;
+    this.platform = process.versions?.bun ? "neutral" : "node";
   }
   getEnvFiles() {
     if (shouldSkipDotenvLoading()) {
@@ -1046,8 +1345,8 @@ var DevBundler = class extends Bundler {
     await super.prepare(outputDirectory);
     const __filename2 = fileURLToPath(import.meta.url);
     const __dirname2 = dirname(__filename2);
-    const playgroundServePath = join(outputDirectory, this.outputDir, "playground");
-    await fsExtra.copy(join(dirname(__dirname2), "dist/playground"), playgroundServePath, {
+    const studioServePath = join(outputDirectory, this.outputDir, "studio");
+    await fsExtra.copy(join(dirname(__dirname2), join("dist", "studio")), studioServePath, {
       overwrite: true
     });
   }
@@ -1055,16 +1354,11 @@ var DevBundler = class extends Bundler {
     const __filename2 = fileURLToPath(import.meta.url);
     const __dirname2 = dirname(__filename2);
     const envFiles = await this.getEnvFiles();
-    let sourcemapEnabled = false;
-    try {
-      const bundlerOptions = await getBundlerOptions(entryFile, outputDirectory);
-      sourcemapEnabled = !!bundlerOptions?.sourcemap;
-    } catch (error) {
-      this.logger.debug("Failed to get bundler options, sourcemap will be disabled", { error });
-    }
+    const bundlerOptions = await this.getUserBundlerOptions(entryFile, outputDirectory);
+    const sourcemapEnabled = !!bundlerOptions?.sourcemap;
     const inputOptions = await getWatcherInputOptions(
       entryFile,
-      "node",
+      this.platform,
       {
         "process.env.NODE_ENV": JSON.stringify(process.env.NODE_ENV || "development")
       },
@@ -1073,7 +1367,6 @@ var DevBundler = class extends Bundler {
     const toolsInputOptions = await this.listToolsInputOptions(toolsPaths);
     const outputDir = join(outputDirectory, this.outputDir);
     await this.writePackageJson(outputDir, /* @__PURE__ */ new Map(), {});
-    const copyPublic = this.copyPublic.bind(this);
     const watcher = await createWatcher(
       {
         ...inputOptions,
@@ -1097,15 +1390,6 @@ var DevBundler = class extends Bundler {
               }
             }
           },
-          {
-            name: "public-dir-watcher",
-            buildStart() {
-              this.addWatchFile(join(dirname(entryFile), "public"));
-            },
-            buildEnd() {
-              return copyPublic(dirname(entryFile), outputDirectory);
-            }
-          },
           {
             name: "tools-watcher",
             async buildEnd() {
@@ -1178,11 +1462,7 @@ var restartAllActiveWorkflowRuns = async ({ host, port }) => {
     }
   }
 };
-var startServer = async (dotMastraPath, {
-  port,
-  host,
-  studioBasePath
-}, env, startOptions = {}, errorRestartCount = 0) => {
+var startServer = async (dotMastraPath, { port, host, studioBasePath, publicDir }, env, startOptions = {}, errorRestartCount = 0) => {
   let serverIsReady = false;
   try {
     serverStartTime = Date.now();
@@ -1201,15 +1481,21 @@ var startServer = async (dotMastraPath, {
     if (startOptions.customArgs) {
       commands.push(...startOptions.customArgs);
     }
-    commands.push("index.mjs");
-    currentServerProcess = execa(process2.execPath, commands, {
-      cwd: dotMastraPath,
+    commands.push(join(dotMastraPath, "index.mjs"));
+    const packagesFilePath = join(dotMastraPath, "..", "mastra-packages.json");
+    await mkdir(dotMastraPath, { recursive: true });
+    if (startOptions.mastraPackages) {
+      await writeFile(packagesFilePath, JSON.stringify(startOptions.mastraPackages), "utf-8");
+    }
+    await mkdir(publicDir, { recursive: true });
+    currentServerProcess = execa(process3.execPath, commands, {
+      cwd: publicDir,
       env: {
         NODE_ENV: "production",
         ...Object.fromEntries(env),
         MASTRA_DEV: "true",
         PORT: port.toString(),
-        MASTRA_DEFAULT_STORAGE_URL: `file:${join(dotMastraPath, "..", "mastra.db")}`,
+        MASTRA_PACKAGES_FILE: packagesFilePath,
         ...startOptions?.https ? {
           MASTRA_HTTPS_KEY: startOptions.https.key.toString("base64"),
           MASTRA_HTTPS_CERT: startOptions.https.cert.toString("base64")
@@ -1229,16 +1515,16 @@ var startServer = async (dotMastraPath, {
     if (currentServerProcess.stdout) {
       currentServerProcess.stdout.on("data", (data) => {
         const output = data.toString();
-        if (!output.includes("Studio available") && !output.includes("\u{1F468}\u200D\u{1F4BB}") && !output.includes("Mastra API running on port")) {
-          process2.stdout.write(output);
+        if (!output.includes("Studio available") && !output.includes("\u{1F468}\u200D\u{1F4BB}") && !output.includes("Mastra API running on ")) {
+          process3.stdout.write(output);
         }
       });
     }
     if (currentServerProcess.stderr) {
       currentServerProcess.stderr.on("data", (data) => {
         const output = data.toString();
-        if (!output.includes("Studio available") && !output.includes("\u{1F468}\u200D\u{1F4BB}") && !output.includes("Mastra API running on port")) {
-          process2.stderr.write(output);
+        if (!output.includes("Studio available") && !output.includes("\u{1F468}\u200D\u{1F4BB}") && !output.includes("Mastra API running on ")) {
+          process3.stderr.write(output);
         }
       });
     }
@@ -1289,7 +1575,7 @@ var startServer = async (dotMastraPath, {
         errorRestartCount++;
         if (errorRestartCount > ON_ERROR_MAX_RESTARTS) {
           devLogger.error(`Server failed to start after ${ON_ERROR_MAX_RESTARTS} error attempts. Giving up.`);
-          process2.exit(1);
+          process3.exit(1);
         }
         devLogger.warn(
           `Attempting to restart server after error... (Attempt ${errorRestartCount}/${ON_ERROR_MAX_RESTARTS})`
@@ -1299,7 +1585,8 @@ var startServer = async (dotMastraPath, {
           {
             port,
             host,
-            studioBasePath
+            studioBasePath,
+            publicDir
           },
           env,
           startOptions,
@@ -1309,11 +1596,7 @@ var startServer = async (dotMastraPath, {
     }, 1e3);
   }
 };
-async function checkAndRestart(dotMastraPath, {
-  port,
-  host,
-  studioBasePath
-}, bundler, startOptions = {}) {
+async function checkAndRestart(dotMastraPath, { port, host, studioBasePath, publicDir }, bundler, startOptions = {}) {
   if (isRestarting) {
     return;
   }
@@ -1330,13 +1613,9 @@ async function checkAndRestart(dotMastraPath, {
     devLogger.debug(`[Mastra Dev] - Could not check hot reload status: ${error}`);
   }
   devLogger.info("[Mastra Dev] - \u2705 Restarting server...");
-  await rebundleAndRestart(dotMastraPath, { port, host, studioBasePath }, bundler, startOptions);
+  await rebundleAndRestart(dotMastraPath, { port, host, studioBasePath, publicDir }, bundler, startOptions);
 }
-async function rebundleAndRestart(dotMastraPath, {
-  port,
-  host,
-  studioBasePath
-}, bundler, startOptions = {}) {
+async function rebundleAndRestart(dotMastraPath, { port, host, studioBasePath, publicDir }, bundler, startOptions = {}) {
   if (isRestarting) {
     return;
   }
@@ -1349,14 +1628,15 @@ async function rebundleAndRestart(dotMastraPath, {
     }
     const env = await bundler.loadEnvVars();
     for (const [key, value] of env.entries()) {
-      process2.env[key] = value;
+      process3.env[key] = value;
     }
     await startServer(
       join(dotMastraPath, "output"),
       {
         port,
         host,
-        studioBasePath
+        studioBasePath,
+        publicDir
       },
       env,
       startOptions
@@ -1376,8 +1656,8 @@ async function dev({
   https,
   debug
 }) {
-  const rootDir = root || process2.cwd();
-  const mastraDir = dir2 ? dir2.startsWith("/") ? dir2 : join(process2.cwd(), dir2) : join(process2.cwd(), "src", "mastra");
+  const rootDir = root || process3.cwd();
+  const mastraDir = dir2 ? dir2.startsWith("/") ? dir2 : join(process3.cwd(), dir2) : join(process3.cwd(), "src", "mastra");
   const dotMastraPath = join(rootDir, ".mastra");
   const fileService = new FileService$1();
   const entryFile = fileService.getFirstExistingFile([join(mastraDir, "index.ts"), join(mastraDir, "index.js")]);
@@ -1386,11 +1666,11 @@ async function dev({
   const discoveredTools = bundler.getAllToolPaths(mastraDir, tools ?? []);
   const loadedEnv = await bundler.loadEnvVars();
   for (const [key, value] of loadedEnv.entries()) {
-    process2.env[key] = value;
+    process3.env[key] = value;
   }
   const serverOptions = await getServerOptions(entryFile, join(dotMastraPath, "output"));
-  let portToUse = serverOptions?.port ?? process2.env.PORT;
-  let hostToUse = serverOptions?.host ?? process2.env.HOST ?? "localhost";
+  let portToUse = serverOptions?.port ?? process3.env.PORT;
+  let hostToUse = serverOptions?.host ?? process3.env.HOST ?? "localhost";
   const studioBasePathToUse = normalizeStudioBase(serverOptions?.studioBase ?? "/");
   if (!portToUse || isNaN(Number(portToUse))) {
     const portList = Array.from({ length: 21 }, (_, i) => 4111 + i);
@@ -1410,7 +1690,8 @@ async function dev({
     const { key, cert } = await devcert.certificateFor(serverOptions?.host ?? "localhost");
     httpsOptions = { key, cert };
   }
-  const startOptions = { inspect, inspectBrk, customArgs, https: httpsOptions };
+  const mastraPackages = await getMastraPackages2(rootDir);
+  const startOptions = { inspect, inspectBrk, customArgs, https: httpsOptions, mastraPackages };
   await bundler.prepare(dotMastraPath);
   const watcher = await bundler.watch(entryFile, dotMastraPath, discoveredTools);
   await startServer(
@@ -1418,7 +1699,8 @@ async function dev({
     {
       port: Number(portToUse),
       host: hostToUse,
-      studioBasePath: studioBasePathToUse
+      studioBasePath: studioBasePathToUse,
+      publicDir: join(mastraDir, "public")
     },
     loadedEnv,
     startOptions
@@ -1435,20 +1717,21 @@ async function dev({
         {
           port: Number(portToUse),
           host: hostToUse,
-          studioBasePath: studioBasePathToUse
+          studioBasePath: studioBasePathToUse,
+          publicDir: join(mastraDir, "public")
         },
         bundler,
         startOptions
       );
     }
   });
-  process2.on("SIGINT", () => {
+  process3.on("SIGINT", () => {
     devLogger.shutdown();
     if (currentServerProcess) {
       currentServerProcess.kill();
     }
     watcher.close().catch(() => {
-    }).finally(() => process2.exit(0));
+    }).finally(() => process3.exit(0));
   });
 }
 
@@ -1484,7 +1767,7 @@ async function start(options = {}) {
     }
     const commands = [];
     commands.push("index.mjs");
-    const server = spawn("node", commands, {
+    const server = spawn(process.execPath, commands, {
       cwd: outputPath,
       stdio: ["inherit", "inherit", "pipe"],
       env: {
@@ -1560,16 +1843,20 @@ var startProject = async (args) => {
 };
 var __filename = fileURLToPath(import.meta.url);
 var __dirname = dirname(__filename);
-async function studio(options = {}) {
+async function studio(options = {
+  serverHost: "localhost",
+  serverPort: 4111,
+  serverProtocol: "http"
+}) {
   config({ path: [options.env || ".env.production", ".env"] });
   try {
-    const distPath = join(__dirname, "playground");
+    const distPath = join(__dirname, "studio");
     if (!existsSync(distPath)) {
       logger.error(`Studio distribution not found at ${distPath}.`);
       process.exit(1);
     }
     const port = options.port || 3e3;
-    const server = createServer(distPath);
+    const server = createServer(distPath, options);
     server.listen(port, () => {
       logger.info(`Mastra Studio running on http://localhost:${port}`);
     });
@@ -1588,16 +1875,19 @@ async function studio(options = {}) {
     process.exit(1);
   }
 }
-var createServer = (builtStudioPath) => {
-  const server = http.createServer((request, response) => {
-    return handler(request, response, {
-      public: builtStudioPath,
-      rewrites: [
-        {
-          source: "**",
-          destination: "/index.html"
-        }
-      ]
+var createServer = (builtStudioPath, options) => {
+  const indexHtmlPath = join(builtStudioPath, "index.html");
+  const basePath = "";
+  let html = readFileSync(indexHtmlPath, "utf8").replaceAll("%%MASTRA_STUDIO_BASE_PATH%%", basePath).replace("%%MASTRA_SERVER_HOST%%", options.serverHost || "localhost").replace("%%MASTRA_SERVER_PORT%%", String(options.serverPort || 4111)).replace("%%MASTRA_SERVER_PROTOCOL%%", options.serverProtocol || "http");
+  const server = http.createServer((req, res) => {
+    const url = req.url || basePath;
+    const isStaticAsset = url.includes("/assets/") || url.includes("/dist/assets/") || url.includes("/mastra.svg") || url.includes("/favicon.ico");
+    if (!isStaticAsset) {
+      res.writeHead(200, { "Content-Type": "text/html" });
+      return res.end(html);
+    }
+    return handler(req, res, {
+      public: builtStudioPath
     });
   });
   return server;
@@ -1617,7 +1907,7 @@ var origin2 = process.env.MASTRA_ANALYTICS_ORIGIN;
 program.name("mastra").version(`${version}`, "-v, --version").addHelpText(
   "before",
   `
-${pc2.bold(pc2.cyan("Mastra"))} is a typescript framework for building AI applications, agents, and workflows.
+${pc3.bold(pc3.cyan("Mastra"))} is a typescript framework for building AI applications, agents, and workflows.
 `
 ).action(() => {
   program.help();
@@ -1629,7 +1919,11 @@ program.command("create [project-name]").description("Create a new Mastra projec
 ).option("-l, --llm <model-provider>", `Default model provider (${LLMProvider.join(", ")})`, parseLlmProvider).option("-k, --llm-api-key <api-key>", "API key for the model provider").option("-e, --example", "Include example code").option("-n, --no-example", "Do not include example code").option("-t, --timeout [timeout]", "Configurable timeout for package installation, defaults to 60000 ms").option("-d, --dir <directory>", "Target directory for Mastra source code (default: src/)").option(
   "-p, --project-name <string>",
   "Project name that will be used in package.json and as the project directory name."
-).option("-m, --mcp <editor>", "MCP Server for code editor (cursor, cursor-global, windsurf, vscode)", parseMcp).option(
+).option(
+  "-m, --mcp <editor>",
+  "MCP Server for code editor (cursor, cursor-global, windsurf, vscode, antigravity)",
+  parseMcp
+).option(
   "--template [template-name]",
   "Create project from a template (use template name, public GitHub URL, or leave blank to select from list)"
 ).action(createProject);
@@ -1637,7 +1931,11 @@ program.command("init").description("Initialize Mastra in your project").option(
   "-c, --components <components>",
   `Comma-separated list of components (${COMPONENTS.join(", ")})`,
   parseComponents
-).option("-l, --llm <model-provider>", `Default model provider (${LLMProvider.join(", ")})`, parseLlmProvider).option("-k, --llm-api-key <api-key>", "API key for the model provider").option("-e, --example", "Include example code").option("-n, --no-example", "Do not include example code").option("-m, --mcp <editor>", "MCP Server for code editor (cursor, cursor-global, windsurf, vscode)", parseMcp).action(initProject);
+).option("-l, --llm <model-provider>", `Default model provider (${LLMProvider.join(", ")})`, parseLlmProvider).option("-k, --llm-api-key <api-key>", "API key for the model provider").option("-e, --example", "Include example code").option("-n, --no-example", "Do not include example code").option(
+  "-m, --mcp <editor>",
+  "MCP Server for code editor (cursor, cursor-global, windsurf, vscode, antigravity)",
+  parseMcp
+).action(initProject);
 program.command("lint").description("Lint your Mastra project").option("-d, --dir <path>", "Path to your Mastra folder").option("-r, --root <path>", "Path to your root folder").option("-t, --tools <toolsDirs>", "Comma-separated list of paths to tool files to include").action(lintProject);
 program.command("dev").description("Start mastra server").option("-d, --dir <dir>", "Path to your mastra folder").option("-r, --root <root>", "Path to your root folder").option("-t, --tools <toolsDirs>", "Comma-separated list of paths to tool files to include").option("-e, --env <env>", "Custom env file to include in the dev server").option(
   "-i, --inspect [host:port]",
@@ -1649,9 +1947,10 @@ program.command("dev").description("Start mastra server").option("-d, --dir <dir
   "-c, --custom-args <args>",
   "Comma-separated list of custom arguments to pass to the dev server. IE: --experimental-transform-types"
 ).option("-s, --https", "Enable local HTTPS").option("--debug", "Enable debug logs", false).action(startDevServer);
-program.command("build").description("Build your Mastra project").option("-d, --dir <path>", "Path to your Mastra Folder").option("-r, --root <path>", "Path to your root folder").option("-t, --tools <toolsDirs>", "Comma-separated list of paths to tool files to include").option("--debug", "Enable debug logs", false).action(buildProject);
+program.command("build").description("Build your Mastra project").option("-d, --dir <path>", "Path to your Mastra Folder").option("-r, --root <path>", "Path to your root folder").option("-t, --tools <toolsDirs>", "Comma-separated list of paths to tool files to include").option("-s, --studio", "Bundle the studio UI with the build").option("--debug", "Enable debug logs", false).action(buildProject);
 program.command("start").description("Start your built Mastra application").option("-d, --dir <path>", "Path to your built Mastra output directory (default: .mastra/output)").option("-e, --env <env>", "Custom env file to include in the start").action(startProject);
-program.command("studio").description("Start the Mastra studio").option("-p, --port <port>", "Port to run the studio on (default: 3000)").option("-e, --env <env>", "Custom env file to include in the studio").action(studio);
+program.command("studio").description("Start the Mastra studio").option("-p, --port <port>", "Port to run the studio on (default: 3000)").option("-e, --env <env>", "Custom env file to include in the studio").option("-h, --server-host <serverHost>", "Host of the Mastra API server (default: localhost)").option("-s, --server-port <serverPort>", "Port of the Mastra API server (default: 4111)").option("-x, --server-protocol <serverProtocol>", "Protocol of the Mastra API server (default: http)").action(studio);
+program.command("migrate").description("Run database migrations to update storage schema").option("-d, --dir <path>", "Path to your Mastra folder").option("-r, --root <path>", "Path to your root folder").option("-e, --env <env>", "Custom env file to include").option("--debug", "Enable debug logs", false).option("-y, --yes", "Skip confirmation prompt (for CI/automation)").action(migrate2);
 var scorersCommand = program.command("scorers").description("Manage scorers for evaluating AI outputs");
 scorersCommand.command("add [scorer-name]").description("Add a new scorer to your project").option("-d, --dir <path>", "Path to your Mastra directory (default: auto-detect)").action(addScorer);
 scorersCommand.command("list").description("List available scorer templates").action(listScorers);