mastra 0.1.57-unstable.57 → 0.1.57-unstable.88

package/dist/commands/create/create.d.ts

@@ -0,0 +1,8 @@
+import { LLMProvider } from '../init/utils.js';
+export declare const create: (args: {
+    components?: string[];
+    llmProvider?: LLMProvider;
+    addExample?: boolean;
+    llmApiKey?: string;
+}) => Promise<void>;
+//# sourceMappingURL=create.d.ts.map

package/dist/commands/create/create.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"create.d.ts","sourceRoot":"","sources":["../../../src/commands/create/create.ts"],"names":[],"mappings":"AAIA,OAAO,EAAqB,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAIlE,eAAO,MAAM,MAAM,SAAgB;IACjC,UAAU,CAAC,EAAE,MAAM,EAAE,CAAC;IACtB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,kBAyBA,CAAC"}

package/dist/commands/create/create.js

@@ -0,0 +1,35 @@
+import * as p from '@clack/prompts';
+import color from 'picocolors';
+import { init } from '../init/init.js';
+import { interactivePrompt } from '../init/utils.js';
+import { createMastraProject } from './utils.js';
+export const create = async (args) => {
+    const { projectName } = await createMastraProject();
+    const directory = '/src';
+    if (!args.components || !args.llmProvider || !args.addExample) {
+        const result = await interactivePrompt();
+        await init({
+            ...result,
+            llmApiKey: result?.llmApiKey,
+        });
+        postCreate({ projectName });
+        return;
+    }
+    const { components = [], llmProvider = 'openai', addExample = false, llmApiKey } = args;
+    await init({
+        directory,
+        components,
+        llmProvider,
+        addExample,
+        llmApiKey,
+    });
+    postCreate({ projectName });
+};
+const postCreate = ({ projectName }) => {
+    p.outro(`
+   ${color.green('To start your project:')}
+
+    ${color.cyan('cd')} ${projectName}
+    ${color.cyan('npm run dev')}
+  `);
+};

package/dist/commands/create/utils.d.ts

@@ -0,0 +1,4 @@
+export declare const createMastraProject: () => Promise<{
+    projectName: string;
+}>;
+//# sourceMappingURL=utils.d.ts.map

package/dist/commands/create/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/commands/create/utils.ts"],"names":[],"mappings":"AAYA,eAAO,MAAM,mBAAmB;;EA6D/B,CAAC"}

package/dist/commands/create/utils.js

@@ -0,0 +1,58 @@
+import * as p from '@clack/prompts';
+import child_process from 'node:child_process';
+import util from 'node:util';
+import color from 'picocolors';
+import fs from 'fs/promises';
+import { DepsService } from '../../services/service.deps.js';
+import { logger } from '../../utils/logger.js';
+const exec = util.promisify(child_process.exec);
+export const createMastraProject = async () => {
+    p.intro(color.inverse('Mastra Create'));
+    const projectName = await p.text({
+        message: 'What do you want to name your project?',
+        placeholder: 'my-mastra-app',
+        defaultValue: 'my-mastra-app',
+    });
+    if (p.isCancel(projectName)) {
+        p.cancel('Operation cancelled');
+        process.exit(0);
+    }
+    const s = p.spinner();
+    s.start('Creating project');
+    try {
+        await fs.mkdir(projectName);
+    }
+    catch (error) {
+        if (error instanceof Error && 'code' in error && error.code === 'EEXIST') {
+            s.stop(`A directory named "${projectName}" already exists. Please choose a different name or delete the existing directory.`);
+            process.exit(1);
+        }
+        throw error;
+    }
+    process.chdir(projectName);
+    s.message('Creating project');
+    await exec(`npm init -y`);
+    const depsService = new DepsService();
+    await depsService.addScriptsToPackageJson({
+        dev: 'mastra dev',
+    });
+    s.stop('Project created');
+    s.start('Installing npm dependencies');
+    await exec(`npm i zod`);
+    await exec(`npm i typescript tsx @types/node --save-dev`);
+    s.stop('NPM dependencies installed');
+    s.start('Installing mastra');
+    await exec(`npm i -D mastra`);
+    s.stop('mastra installed');
+    s.start('Installing @mastra/core');
+    await exec(`npm i @mastra/core@alpha`);
+    s.stop('@mastra/core installed');
+    s.start('Adding .gitignore');
+    await exec(`echo output.txt >> .gitignore`);
+    await exec(`echo node_modules >> .gitignore`);
+    await exec(`echo dist >> .gitignore`);
+    s.stop('.gitignore added');
+    p.outro('Project created successfully');
+    logger.break();
+    return { projectName };
+};

package/dist/commands/deploy/cloudflare/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/commands/deploy/cloudflare/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAG1C,qBAAa,kBAAmB,SAAQ,QAAQ;IAC9C,IAAI,SAAgB;IAEd,UAAU;IAMV,YAAY;IAyBlB,UAAU,IAAI,IAAI;IA+BZ,aAAa,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAajE"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/commands/deploy/cloudflare/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAG1C,qBAAa,kBAAmB,SAAQ,QAAQ;IAC9C,IAAI,SAAgB;IAEd,UAAU;IAMV,YAAY;IA2BlB,UAAU,IAAI,IAAI;IA+BZ,aAAa,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CAajE"}

package/dist/commands/deploy/cloudflare/index.js

@@ -28,6 +28,8 @@ export class CloudflareDeployer extends Deployer {
             dependencies: {
                 '@mastra/core': '0.1.27-alpha.18',
                 'itty-router': '5.0.18',
+                superjson: '^2.2.2',
+                'zod-to-json-schema': '^3.24.1',
             },
         }, null, 2));
     }

package/dist/commands/deploy/netlify/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/commands/deploy/netlify/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAG1C,qBAAa,eAAgB,SAAQ,QAAQ;IAC3C,IAAI,SAAa;IAEjB,mBAAmB;IAMb,UAAU;IAMV,YAAY;IA4BlB,UAAU,IAAI,IAAI;IAoBZ,aAAa,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CASzF"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/commands/deploy/netlify/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAG1C,qBAAa,eAAgB,SAAQ,QAAQ;IAC3C,IAAI,SAAa;IAEjB,mBAAmB;IAMb,UAAU;IAMV,YAAY;IA8BlB,UAAU,IAAI,IAAI;IAoBZ,aAAa,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;CASzF"}

package/dist/commands/deploy/netlify/index.js

@@ -35,6 +35,8 @@ export class NetlifyDeployer extends Deployer {
                 '@mastra/core': '0.1.27-alpha.18',
                 'serverless-http': 'latest',
                 '@netlify/functions': 'latest',
+                superjson: '^2.2.2',
+                'zod-to-json-schema': '^3.24.1',
                 express: '^4.21.1',
             },
         }, null, 2));

package/dist/commands/deploy/vercel/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/commands/deploy/vercel/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAe1C,qBAAa,cAAe,SAAQ,QAAQ;IAC1C,IAAI,SAAY;IACV,UAAU;IAMV,YAAY;IA4BlB,UAAU;IA4BV,OAAO,CAAC,YAAY;IAUd,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE;IAqDpC,aAAa,CAAC,EAClB,KAAK,EACL,WAA8B,GAC/B,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,IAAI,CAAC;CA6ClB"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../../src/commands/deploy/vercel/index.ts"],"names":[],"mappings":"AAKA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAe1C,qBAAa,cAAe,SAAQ,QAAQ;IAC1C,IAAI,SAAY;IACV,UAAU;IAMV,YAAY;IA8BlB,UAAU;IA4BV,OAAO,CAAC,YAAY;IAUd,OAAO,CAAC,EAAE,KAAK,EAAE,EAAE;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE;IAqDpC,aAAa,CAAC,EAClB,KAAK,EACL,WAA8B,GAC/B,EAAE;QACD,KAAK,EAAE,MAAM,CAAC;QACd,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,GAAG,OAAO,CAAC,IAAI,CAAC;CA6ClB"}

package/dist/commands/deploy/vercel/index.js

@@ -30,6 +30,8 @@ export class VercelDeployer extends Deployer {
                 '@mastra/core': '0.1.27-alpha.35',
                 'express-jsdoc-swagger': '^1.8.0',
                 'serverless-http': '^3.2.0',
+                superjson: '^2.2.2',
+                'zod-to-json-schema': '^3.24.1',
                 zod: '3.24.0',
             },
         }, null, 2));

package/dist/commands/dev.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"dev.d.ts","sourceRoot":"","sources":["../../src/commands/dev.ts"],"names":[],"mappings":"AAkBA,wBAAsB,GAAG,CAAC,EACxB,IAAI,EACJ,GAAG,EACH,GAAG,EACH,SAAS,GACV,EAAE;IACD,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,iBA4GA"}
+{"version":3,"file":"dev.d.ts","sourceRoot":"","sources":["../../src/commands/dev.ts"],"names":[],"mappings":"AAqGA,wBAAsB,GAAG,CAAC,EACxB,IAAI,EACJ,GAAG,EACH,GAAG,EACH,SAAS,GACV,EAAE;IACD,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,iBA4HA"}

package/dist/commands/dev.js

@@ -1,9 +1,10 @@
+import { watch } from 'chokidar';
 import { config } from 'dotenv';
 import { execa } from 'execa';
 import { writeFileSync } from 'fs';
 import { join } from 'path';
 import path from 'path';
-import { fileURLToPath, pathToFileURL } from 'url';
+import { fileURLToPath } from 'url';
 import fsExtra from 'fs-extra/esm';
 import fs from 'fs/promises';
 import { FileService } from '../services/service.file.js';
@@ -11,6 +12,78 @@ import { bundle, bundleServer } from '../utils/bundle.js';
 import { EXPRESS_SERVER } from './deploy/server.js';
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = path.dirname(__filename);
+let currentServerProcess;
+async function rebundleAndRestart(dirPath, dotMastraPath, port, toolsDirs) {
+    if (currentServerProcess) {
+        console.log('Stopping current server...');
+        currentServerProcess.kill();
+        await new Promise(resolve => setTimeout(resolve, 1000));
+    }
+    /*
+      Bundle mastra
+    */
+    await bundle(dirPath, { buildName: 'Mastra' });
+    /*
+      Bundle tools
+    */
+    const defaultToolsPath = path.join(dirPath, 'tools');
+    const toolsPaths = [...(toolsDirs?.split(',').map(tool => path.join(process.cwd(), tool)) || []), defaultToolsPath];
+    const toolPathsWithFileNames = (await Promise.all(toolsPaths.map(async (toolPath) => {
+        try {
+            const files = await fs.readdir(toolPath);
+            return files.map(file => {
+                const fullPath = path.join(toolPath, file);
+                const fileName = path.parse(file).name;
+                const name = fileName === 'index' ? path.basename(path.dirname(fullPath)) : fileName;
+                return {
+                    path: toolPath,
+                    name,
+                    fileName,
+                };
+            });
+        }
+        catch (err) {
+            if (toolPath === defaultToolsPath) {
+                return [];
+            }
+            console.warn(`Error reading tools directory ${toolPath}:`, err);
+            return [];
+        }
+    }))).flat();
+    for (const { path, name, fileName } of toolPathsWithFileNames) {
+        await bundle(path, {
+            outfile: join(dotMastraPath, 'tools', `${name}.mjs`),
+            entryFile: fileName,
+            buildName: `${name}`,
+        });
+    }
+    const MASTRA_TOOLS_PATH = toolPathsWithFileNames?.length
+        ? toolPathsWithFileNames.map(tool => path.join(dotMastraPath, 'tools', `${tool.name}.mjs`)).join(',')
+        : undefined;
+    try {
+        console.log('Restarting server...');
+        currentServerProcess = execa('node', ['server.mjs'], {
+            cwd: dotMastraPath,
+            env: {
+                PORT: port.toString() || '4111',
+                MASTRA_TOOLS_PATH,
+            },
+            stdio: 'inherit',
+            reject: false,
+        });
+        if (currentServerProcess.failed) {
+            console.error('Server failed to restart with error:', currentServerProcess.stderr);
+            return;
+        }
+    }
+    catch (err) {
+        const execaError = err;
+        if (execaError.stderr)
+            console.error('Server error output:', execaError.stderr);
+        if (execaError.stdout)
+            console.error('Server output:', execaError.stdout);
+    }
+}
 export async function dev({ port, env, dir, toolsDirs, }) {
     const dotMastraPath = join(process.cwd(), '.mastra');
     const playgroundServePath = join(dotMastraPath, 'playground');
@@ -39,7 +112,7 @@ export async function dev({ port, env, dir, toolsDirs, }) {
       Bundle mastra
     */
     const dirPath = dir || path.join(process.cwd(), 'src/mastra');
-    await bundle(dirPath);
+    await bundle(dirPath, { buildName: 'Mastra' });
     /*
       Bundle tools
     */
@@ -71,20 +144,20 @@ export async function dev({ port, env, dir, toolsDirs, }) {
         await bundle(path, {
             outfile: join(dotMastraPath, 'tools', `${name}.mjs`),
             entryFile: fileName,
+            buildName: `${name}`,
         });
     }
     /*
       Bundle server
     */
-    const serverPath = join(dotMastraPath, 'index.mjs');
-    writeFileSync(serverPath, EXPRESS_SERVER);
-    await bundleServer(serverPath);
+    writeFileSync(join(dotMastraPath, 'index.mjs'), EXPRESS_SERVER);
+    await bundleServer(join(dotMastraPath, 'index.mjs'));
     const MASTRA_TOOLS_PATH = toolPathsWithFileNames?.length
-        ? toolPathsWithFileNames.map(tool => pathToFileURL(join(dotMastraPath, 'tools', `${tool.name}.mjs`)).href).join(',')
+        ? toolPathsWithFileNames.map(tool => path.join(dotMastraPath, 'tools', `${tool.name}.mjs`)).join(',')
         : undefined;
     try {
         console.log('Starting server...');
-        const serverProcess = await execa('node', ['server.mjs'], {
+        currentServerProcess = execa('node', ['server.mjs'], {
             cwd: dotMastraPath,
             env: {
                 PORT: port.toString() || '4111',
@@ -93,11 +166,25 @@ export async function dev({ port, env, dir, toolsDirs, }) {
             stdio: 'inherit',
             reject: false,
         });
-        if (serverProcess.failed) {
-            console.error('Server failed to start with error:', serverProcess.stderr);
+        if (currentServerProcess.failed) {
+            console.error('Server failed to start with error:', currentServerProcess.stderr);
             process.exit(1);
         }
-        console.log('Server started successfully');
+        const watcher = watch(dirPath, {
+            persistent: true,
+        });
+        watcher.on('change', async () => {
+            console.log(`Changes detected`);
+            await rebundleAndRestart(dirPath, dotMastraPath, port, toolsDirs);
+        });
+        process.on('SIGINT', () => {
+            console.log('Stopping server...');
+            if (currentServerProcess) {
+                currentServerProcess.kill();
+            }
+            watcher.close();
+            process.exit(0);
+        });
     }
     catch (err) {
         const execaError = err;

package/dist/commands/engine/down.d.ts

@@ -1,2 +1,2 @@
-export declare function down(): Promise<void>;
+export declare function down(composePath?: string): Promise<void>;
 //# sourceMappingURL=down.d.ts.map

package/dist/commands/engine/down.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"down.d.ts","sourceRoot":"","sources":["../../../src/commands/engine/down.ts"],"names":[],"mappings":"AAIA,wBAAsB,IAAI,kBAWzB"}
+{"version":3,"file":"down.d.ts","sourceRoot":"","sources":["../../../src/commands/engine/down.ts"],"names":[],"mappings":"AAIA,wBAAsB,IAAI,CAAC,WAAW,CAAC,EAAE,MAAM,iBAY9C"}

package/dist/commands/engine/down.js

@@ -1,11 +1,12 @@
 import yoctoSpinner from 'yocto-spinner';
 import { DockerService } from '../../services/service.docker.js';
-export async function down() {
+export async function down(composePath) {
     const spinner = yoctoSpinner({ text: 'Shutting down docker container\n' });
     spinner.start();
     try {
         const dockerService = new DockerService();
-        await dockerService.stopDockerContainer('mastra-pg.docker-compose.yaml');
+        const composeFile = dockerService.getComposeFile(composePath);
+        await dockerService.stopDockerContainer(composeFile);
         spinner.success('Docker container shut down successfully\n');
     }
     catch (error) {

package/dist/commands/engine/up.d.ts

@@ -1,2 +1,2 @@
-export declare function up(): Promise<void>;
+export declare function up(composePath?: string): Promise<void>;
 //# sourceMappingURL=up.d.ts.map

package/dist/commands/engine/up.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"up.d.ts","sourceRoot":"","sources":["../../../src/commands/engine/up.ts"],"names":[],"mappings":"AAIA,wBAAsB,EAAE,kBAWvB"}
+{"version":3,"file":"up.d.ts","sourceRoot":"","sources":["../../../src/commands/engine/up.ts"],"names":[],"mappings":"AAIA,wBAAsB,EAAE,CAAC,WAAW,CAAC,EAAE,MAAM,iBAY5C"}

package/dist/commands/engine/up.js

@@ -1,11 +1,12 @@
 import yoctoSpinner from 'yocto-spinner';
 import { DockerService } from '../../services/service.docker.js';
-export async function up() {
+export async function up(composePath) {
     const spinner = yoctoSpinner({ text: 'Starting docker container\n' });
     spinner.start();
     try {
         const dockerService = new DockerService();
-        await dockerService.startDockerContainer('mastra-pg.docker-compose.yaml');
+        const composeFile = dockerService.getComposeFile(composePath);
+        await dockerService.startDockerContainer(composeFile);
         spinner.success('Docker containers started successfully\n');
     }
     catch (error) {

package/dist/commands/init/init.d.ts

@@ -1,10 +1,10 @@
 import { LLMProvider } from './utils.js';
-export declare const init: ({ directory, addExample, components, llmProvider, showSpinner, }: {
+export declare const init: ({ directory, addExample, components, llmProvider, llmApiKey, }: {
     directory: string;
     components: string[];
     llmProvider: LLMProvider;
     addExample: boolean;
-    showSpinner?: boolean;
+    llmApiKey?: string;
 }) => Promise<{
     success: boolean;
 }>;

package/dist/commands/init/init.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../../src/commands/init/init.ts"],"names":[],"mappings":"AAEA,OAAO,EAIL,WAAW,EAIZ,MAAM,YAAY,CAAC;AAEpB,eAAO,MAAM,IAAI,qEAMd;IACD,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,WAAW,EAAE,WAAW,CAAC;IACzB,UAAU,EAAE,OAAO,CAAC;IACpB,WAAW,CAAC,EAAE,OAAO,CAAC;CACvB;;EA2BA,CAAC"}
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../../src/commands/init/init.ts"],"names":[],"mappings":"AAGA,OAAO,EAKL,WAAW,EAIZ,MAAM,YAAY,CAAC;AAIpB,eAAO,MAAM,IAAI,mEAMd;IACD,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,WAAW,EAAE,WAAW,CAAC;IACzB,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;;EAqDA,CAAC"}

package/dist/commands/init/init.js

@@ -1,26 +1,50 @@
-import { logger } from '../../utils/logger.js';
-import { createComponentsDir, createMastraDir, writeAPIKey, writeCodeSample, writeIndexFile, } from './utils.js';
-export const init = async ({ directory, addExample = false, components, llmProvider = 'openai', showSpinner = false, }) => {
+import * as p from '@clack/prompts';
+import color from 'picocolors';
+import { createComponentsDir, createMastraDir, getAPIKey, writeAPIKey, writeCodeSample, writeIndexFile, } from './utils.js';
+const s = p.spinner();
+export const init = async ({ directory, addExample = false, components, llmProvider = 'openai', llmApiKey, }) => {
+    s.start('Initializing Mastra');
     try {
         const result = await createMastraDir(directory);
         if (!result.ok) {
-            logger.info('Mastra already initialized');
+            s.stop(color.inverse(' Mastra already initialized '));
             return { success: false };
         }
         const dirPath = result.dirPath;
         await Promise.all([
-            writeIndexFile(dirPath, addExample),
+            writeIndexFile({
+                dirPath,
+                addExample,
+                addWorkflow: components.includes('workflows'),
+                addAgent: components.includes('agents'),
+            }),
             ...components.map(component => createComponentsDir(dirPath, component)),
-            writeAPIKey(llmProvider),
+            writeAPIKey({ provider: llmProvider, apiKey: llmApiKey }),
         ]);
         if (addExample) {
-            await Promise.all([components.map(component => writeCodeSample(dirPath, component, llmProvider))]);
+            await Promise.all([
+                components.map(component => writeCodeSample(dirPath, component, llmProvider, components)),
+            ]);
+        }
+        const key = await getAPIKey(llmProvider || 'openai');
+        s.stop();
+        if (!llmApiKey) {
+            p.note(`
+      ${color.green('Mastra initialized successfully!')}
+
+      Add your ${color.cyan(key)} as an environment variable
+      in your ${color.cyan('.env.development')} file
+      `);
+        }
+        else {
+            p.note(`
+      ${color.green('Mastra initialized successfully!')}
+      `);
         }
-        showSpinner && logger.success('Mastra initialized successfully');
         return { success: true };
     }
     catch (err) {
-        showSpinner && logger.error('Could not initialize mastra');
+        s.stop(color.inverse('An error occurred while initializing Mastra'));
         console.error(err);
         return { success: false };
     }

package/dist/commands/init/utils.d.ts

@@ -2,26 +2,38 @@ import { ModelConfig } from '@mastra/core';
 export type LLMProvider = 'openai' | 'anthropic' | 'groq';
 export type Components = 'agents' | 'workflows' | 'tools';
 export declare const modelToConfigMap: Record<LLMProvider, ModelConfig>;
-export declare function writeAgentSample(llmProvider: LLMProvider, destPath: string): Promise<void>;
+export declare function writeAgentSample(llmProvider: LLMProvider, destPath: string, addExampleTool: boolean): Promise<void>;
 export declare function writeWorkflowSample(destPath: string): Promise<void>;
 export declare function writeToolSample(destPath: string): Promise<void>;
-export declare function writeCodeSampleForComponents(llmprovider: LLMProvider, component: Components, destPath: string): Promise<void | "">;
+export declare function writeCodeSampleForComponents(llmprovider: LLMProvider, component: Components, destPath: string, components: Components[]): Promise<void | "">;
 export declare const createComponentsDir: (dirPath: string, component: string) => Promise<void>;
-export declare const writeIndexFile: (dirPath: string, addExample: boolean) => Promise<void>;
+export declare const writeIndexFile: ({ dirPath, addAgent, addExample, addWorkflow, }: {
+    dirPath: string;
+    addExample: boolean;
+    addWorkflow: boolean;
+    addAgent: boolean;
+}) => Promise<void>;
 export declare const checkInitialization: (dirPath: string) => Promise<boolean>;
 export declare const checkAndInstallCoreDeps: () => Promise<void>;
 export declare function installCoreDeps(): Promise<void>;
-export declare const writeAPIKey: (provider: LLMProvider) => Promise<void>;
+export declare const getAPIKey: (provider: LLMProvider) => Promise<string>;
+export declare const writeAPIKey: ({ provider, apiKey, }: {
+    provider: LLMProvider;
+    apiKey?: string;
+}) => Promise<void>;
 export declare const createMastraDir: (directory: string) => Promise<{
     ok: true;
     dirPath: string;
 } | {
     ok: false;
 }>;
-export declare const writeCodeSample: (dirPath: string, component: Components, llmProvider: LLMProvider) => Promise<void>;
-export declare const interactivePrompt: () => Promise<void>;
-export declare const initializeMinimal: () => Promise<void>;
-export declare const checkPkgJsonAndCreateStarter: () => Promise<{
-    isStarter: boolean;
+export declare const writeCodeSample: (dirPath: string, component: Components, llmProvider: LLMProvider, components: Components[]) => Promise<void>;
+export declare const interactivePrompt: () => Promise<{
+    components: string[];
+    directory: string;
+    llmProvider: "openai" | "anthropic" | "groq";
+    llmApiKey: unknown;
+    addExample: boolean;
 }>;
+export declare const checkPkgJson: () => Promise<void>;
 //# sourceMappingURL=utils.d.ts.map

package/dist/commands/init/utils.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/commands/init/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAmB3C,MAAM,MAAM,WAAW,GAAG,QAAQ,GAAG,WAAW,GAAG,MAAM,CAAC;AAC1D,MAAM,MAAM,UAAU,GAAG,QAAQ,GAAG,WAAW,GAAG,OAAO,CAAC;AAE1D,eAAO,MAAM,gBAAgB,EAAE,MAAM,CAAC,WAAW,EAAE,WAAW,CAI7D,CAAC;AAEF,wBAAsB,gBAAgB,CAAC,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,iBAkBhF;AAED,wBAAsB,mBAAmB,CAAC,QAAQ,EAAE,MAAM,iBAGzD;AAED,wBAAsB,eAAe,CAAC,QAAQ,EAAE,MAAM,iBAGrD;AAED,wBAAsB,4BAA4B,CAAC,WAAW,EAAE,WAAW,EAAE,SAAS,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,sBAWnH;AAED,eAAO,MAAM,mBAAmB,YAAmB,MAAM,aAAa,MAAM,kBAI3E,CAAC;AAEF,eAAO,MAAM,cAAc,YAAmB,MAAM,cAAc,OAAO,kBAoCxE,CAAC;AAEF,eAAO,MAAM,mBAAmB,YAAmB,MAAM,qBAOxD,CAAC;AAEF,eAAO,MAAM,uBAAuB,qBAOnC,CAAC;AAGF,wBAAsB,eAAe,kBA0BpC;AAED,eAAO,MAAM,WAAW,aAAoB,WAAW,kBAatD,CAAC;AACF,eAAO,MAAM,eAAe,cAAqB,MAAM,KAAG,OAAO,CAAC;IAAE,EAAE,EAAE,IAAI,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,EAAE,EAAE,KAAK,CAAA;CAAE,CAe9G,CAAC;AAEF,eAAO,MAAM,eAAe,YAAmB,MAAM,aAAa,UAAU,eAAe,WAAW,kBAQrG,CAAC;AAEF,eAAO,MAAM,iBAAiB,qBAyE7B,CAAC;AAEF,eAAO,MAAM,iBAAiB,qBA8B7B,CAAC;AAEF,eAAO,MAAM,4BAA4B;;EAwBxC,CAAC"}
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/commands/init/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAiB3C,MAAM,MAAM,WAAW,GAAG,QAAQ,GAAG,WAAW,GAAG,MAAM,CAAC;AAC1D,MAAM,MAAM,UAAU,GAAG,QAAQ,GAAG,WAAW,GAAG,OAAO,CAAC;AAE1D,eAAO,MAAM,gBAAgB,EAAE,MAAM,CAAC,WAAW,EAAE,WAAW,CAI7D,CAAC;AAEF,wBAAsB,gBAAgB,CAAC,WAAW,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAE,cAAc,EAAE,OAAO,iBA8BzG;AAED,wBAAsB,mBAAmB,CAAC,QAAQ,EAAE,MAAM,iBAGzD;AAED,wBAAsB,eAAe,CAAC,QAAQ,EAAE,MAAM,iBAGrD;AAED,wBAAsB,4BAA4B,CAChD,WAAW,EAAE,WAAW,EACxB,SAAS,EAAE,UAAU,EACrB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,UAAU,EAAE,sBAYzB;AAED,eAAO,MAAM,mBAAmB,YAAmB,MAAM,aAAa,MAAM,kBAI3E,CAAC;AAEF,eAAO,MAAM,cAAc,oDAKxB;IACD,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,EAAE,OAAO,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;IACrB,QAAQ,EAAE,OAAO,CAAC;CACnB,kBAwCA,CAAC;AAEF,eAAO,MAAM,mBAAmB,YAAmB,MAAM,qBAOxD,CAAC;AAEF,eAAO,MAAM,uBAAuB,qBAOnC,CAAC;AAGF,wBAAsB,eAAe,kBA0BpC;AAED,eAAO,MAAM,SAAS,aAAoB,WAAW,oBAYpD,CAAC;AAEF,eAAO,MAAM,WAAW,0BAGrB;IACD,QAAQ,EAAE,WAAW,CAAC;IACtB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB,kBAGA,CAAC;AACF,eAAO,MAAM,eAAe,cAAqB,MAAM,KAAG,OAAO,CAAC;IAAE,EAAE,EAAE,IAAI,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,GAAG;IAAE,EAAE,EAAE,KAAK,CAAA;CAAE,CAe9G,CAAC;AAEF,eAAO,MAAM,eAAe,YACjB,MAAM,aACJ,UAAU,eACR,WAAW,cACZ,UAAU,EAAE,kBASzB,CAAC;AAEF,eAAO,MAAM,iBAAiB;;;;;;EAuE7B,CAAC;AAEF,eAAO,MAAM,YAAY,qBAmBxB,CAAC"}