mastra 0.1.57-unstable.57 → 0.1.57-unstable.89

package/dist/commands/init/utils.js

@@ -10,22 +10,33 @@ import fs from 'fs/promises';
 import { DepsService } from '../../services/service.deps.js';
 import { FileService } from '../../services/service.file.js';
 import { logger } from '../../utils/logger.js';
-import { init } from './init.js';
 const exec = util.promisify(child_process.exec);
 export const modelToConfigMap = {
     openai: { provider: 'OPEN_AI', name: 'gpt-4o', toolChoice: 'auto' },
     anthropic: { provider: 'ANTHROPIC', name: 'claude-3-5-sonnet-20241022', toolChoice: 'auto' },
     groq: { provider: 'GROQ', name: 'llama3-groq-70b-8192-tool-use-preview', toolChoice: 'auto' },
 };
-export async function writeAgentSample(llmProvider, destPath) {
+export async function writeAgentSample(llmProvider, destPath, addExampleTool) {
     const model = modelToConfigMap[llmProvider];
+    const instructions = `
+      You are a helpful weather assistant that provides accurate weather information.
+
+      Your primary function is to help users get weather details for specific locations. When responding:
+      - Always ask for a location if none is provided
+      - Include relevant details like humidity, wind conditions, and precipitation
+      - Keep responses concise but informative
+
+      ${addExampleTool ? 'Use the weatherTool to fetch current weather data.' : ''}
+`;
     const content = `
 import { Agent } from '@mastra/core';
+${addExampleTool ? `import { weatherTool } from '../tools';` : ''}
 
-export const catOne = new Agent({
-  name: 'cat-one',
-  instructions: 'You are a feline expert with comprehensive knowledge of all cat species, from domestic breeds to wild big cats. As a lifelong cat specialist, you understand their behavior, biology, social structures, and evolutionary history in great depth.',
+export const weatherAgent = new Agent({
+  name: 'Weather Agent',
+  instructions: \`${instructions}\`,
   model: ${JSON.stringify(model, null, 2)},
+  ${addExampleTool ? 'tools: { weatherTool },' : ''}
 });
     `;
     const formattedContent = await prettier.format(content, {
@@ -43,13 +54,13 @@ export async function writeToolSample(destPath) {
     const fileService = new FileService();
     await fileService.copyStarterFile('tools.ts', destPath);
 }
-export async function writeCodeSampleForComponents(llmprovider, component, destPath) {
-    switch (true) {
-        case component === 'agents':
-            return writeAgentSample(llmprovider, destPath);
-        case component === 'tools':
+export async function writeCodeSampleForComponents(llmprovider, component, destPath, components) {
+    switch (component) {
+        case 'agents':
+            return writeAgentSample(llmprovider, destPath, components.includes('tools'));
+        case 'tools':
             return writeToolSample(destPath);
-        case component === 'workflows':
+        case 'workflows':
             return writeWorkflowSample(destPath);
         default:
             return '';
@@ -59,26 +70,30 @@ export const createComponentsDir = async (dirPath, component) => {
     const componentPath = dirPath + `/${component}`;
     await fsExtra.ensureDir(componentPath);
 };
-export const writeIndexFile = async (dirPath, addExample) => {
+export const writeIndexFile = async ({ dirPath, addAgent, addExample, addWorkflow, }) => {
     const indexPath = dirPath + '/index.ts';
     const destPath = path.join(indexPath);
     try {
         await fs.writeFile(destPath, '');
+        const filteredExports = [
+            addWorkflow ? `workflows: { weatherWorkflow },` : '',
+            addAgent ? `agents: { weatherAgent },` : '',
+        ].filter(Boolean);
         if (!addExample) {
             await fs.writeFile(destPath, `
 import { Mastra } from '@mastra/core';
 
-export const mastra = new Mastra({})
+export const mastra = new Mastra()
         `);
             return;
         }
         await fs.writeFile(destPath, `
 import { Mastra, createLogger } from '@mastra/core';
-
-import { catOne } from './agents/index';
+${addWorkflow ? `import { weatherWorkflow } from './workflows';` : ''}
+${addAgent ? `import { weatherAgent } from './agents';` : ''}
 
 export const mastra = new Mastra({
-  agents: { catOne },
+  ${filteredExports.join('\n  ')}
   logger: createLogger({
     type: 'CONSOLE',
     level: 'INFO',
@@ -123,26 +138,29 @@ export async function installCoreDeps() {
         }
         spinner.start();
         const depsService = new DepsService();
-        await depsService.installPackages(['@mastra/core']);
+        await depsService.installPackages(['@mastra/core@alpha']);
         spinner.success('@mastra/core installed successfully');
     }
     catch (err) {
         console.error(err);
     }
 }
-export const writeAPIKey = async (provider) => {
+export const getAPIKey = async (provider) => {
     let key = 'OPENAI_API_KEY';
     switch (provider) {
         case 'anthropic':
             key = 'ANTHROPIC_API_KEY';
-            return;
+            return key;
         case 'groq':
             key = 'GROQ_API_KEY';
-            return;
+            return key;
         default:
-            key = 'OPENAI_API_KEY';
+            return key;
     }
-    await exec(`echo ${key}= >> .env.development`);
+};
+export const writeAPIKey = async ({ provider, apiKey = 'your-api-key', }) => {
+    const key = await getAPIKey(provider);
+    await exec(`echo ${key}=${apiKey} >> .env.development`);
 };
 export const createMastraDir = async (directory) => {
     let dir = directory
@@ -159,17 +177,17 @@ export const createMastraDir = async (directory) => {
         return { ok: true, dirPath };
     }
 };
-export const writeCodeSample = async (dirPath, component, llmProvider) => {
+export const writeCodeSample = async (dirPath, component, llmProvider, components) => {
     const destPath = dirPath + `/${component}/index.ts`;
     try {
-        await writeCodeSampleForComponents(llmProvider, component, destPath);
+        await writeCodeSampleForComponents(llmProvider, component, destPath, components);
     }
     catch (err) {
         throw err;
     }
 };
 export const interactivePrompt = async () => {
-    p.intro(color.inverse(' Initializing Mastra '));
+    p.intro(color.inverse('Mastra Init'));
     const mastraProject = await p.group({
         directory: () => p.text({
             message: 'Where should we create the Mastra files? (default: src/)',
@@ -198,6 +216,23 @@ export const interactivePrompt = async () => {
                 { value: 'groq', label: 'Groq' },
             ],
         }),
+        llmApiKey: async ({ results: { llmProvider } }) => {
+            const keyChoice = await p.select({
+                message: `Enter your ${llmProvider} API key?`,
+                options: [
+                    { value: 'skip', label: 'Skip for now', hint: 'default' },
+                    { value: 'enter', label: 'Enter API key' },
+                ],
+                initialValue: 'skip',
+            });
+            if (keyChoice === 'enter') {
+                return p.text({
+                    message: 'Enter your API key:',
+                    placeholder: 'sk-...',
+                });
+            }
+            return undefined;
+        },
         addExample: () => p.confirm({
             message: 'Add example',
             initialValue: false,
@@ -208,50 +243,11 @@ export const interactivePrompt = async () => {
             process.exit(0);
         },
     });
-    const s = p.spinner();
-    s.start('Initializing Mastra');
     const { shouldAddTools, components, ...rest } = mastraProject;
     const mastraComponents = shouldAddTools ? [...components, 'tools'] : components;
-    try {
-        const result = { ...rest, components: mastraComponents };
-        const { success } = await init(result);
-        if (!success) {
-            return;
-        }
-        s.stop('Mastra initialized successfully');
-        p.note('You are all set!');
-        p.outro(`Problems? ${color.underline(color.cyan('https://github.com/mastra-ai/mastra'))}`);
-    }
-    catch (err) {
-        s.stop('Could not initialize Mastra');
-        logger.error(err);
-    }
-};
-export const initializeMinimal = async () => {
-    const confirm = await p.confirm({
-        message: 'No package.json detected. Create a new project?',
-        initialValue: true,
-    });
-    if (p.isCancel(confirm)) {
-        p.cancel('Operation cancelled');
-        process.exit(0);
-    }
-    if (!confirm) {
-        p.cancel('Operation cancelled');
-        process.exit(0);
-    }
-    const s = p.spinner();
-    s.start('Creating project');
-    await exec(`npm init -y`);
-    await exec(`npm i zod typescript tsx @types/node --save-dev`);
-    s.message('Installing dependencies');
-    await exec(`echo output.txt >> .gitignore`);
-    await exec(`echo node_modules >> .gitignore`);
-    await exec(`npm i @mastra/core@alpha`);
-    s.stop('Project creation successful');
-    logger.break();
+    return { ...rest, components: mastraComponents };
 };
-export const checkPkgJsonAndCreateStarter = async () => {
+export const checkPkgJson = async () => {
     const cwd = process.cwd();
     const pkgJsonPath = path.join(cwd, 'package.json');
     let isPkgJsonPresent = false;
@@ -265,12 +261,6 @@ export const checkPkgJsonAndCreateStarter = async () => {
     if (isPkgJsonPresent) {
         return;
     }
-    try {
-        await initializeMinimal();
-        return { isStarter: true };
-    }
-    catch (err) {
-        logger.error(`Could not create project: ${err}`);
-        process.exit(0);
-    }
+    logger.info('no package.json found, create one or run "mastra create" to create a new project');
+    process.exit(0);
 };

package/dist/index.d.ts

@@ -1,3 +1,4 @@
 #! /usr/bin/env node
-export {};
+export { create } from './commands/create/create.js';
+export { PosthogAnalytics } from './analytics/index.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAsQA,OAAO,EAAE,MAAM,EAAE,MAAM,6BAA6B,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC"}

package/dist/index.js

@@ -1,11 +1,8 @@
 #! /usr/bin/env node
-import * as p from '@clack/prompts';
 import { Command } from 'commander';
 import color from 'picocolors';
 import { PosthogAnalytics } from './analytics/index.js';
-import { createNewAgent } from './commands/agents/create-new-agent.js';
-import { listAgents } from './commands/agents/list-agent.js';
-import { updateAgentIndexFile } from './commands/agents/update-agent-file.js';
+import { create } from './commands/create/create.js';
 import { cloudflareDeploy, netlifyDeploy, vercelDeploy } from './commands/deploy/index.js';
 import { dev } from './commands/dev.js';
 import { add } from './commands/engine/add.js';
@@ -14,7 +11,7 @@ import { generate } from './commands/engine/generate.js';
 import { migrate } from './commands/engine/migrate.js';
 import { up } from './commands/engine/up.js';
 import { init } from './commands/init/init.js';
-import { checkAndInstallCoreDeps, checkPkgJsonAndCreateStarter, interactivePrompt } from './commands/init/utils.js';
+import { checkAndInstallCoreDeps, checkPkgJson, interactivePrompt } from './commands/init/utils.js';
 import { DepsService } from './services/service.deps.js';
 import { findApiKeys } from './utils/find-api-keys.js';
 import { getEnv } from './utils/get-env.js';
@@ -40,31 +37,65 @@ program
     }
     catch (e) { }
 });
+program
+    .command('create')
+    .description('Create a new Mastra project')
+    .option('--default', 'Quick start with defaults(src, OpenAI, no examples)')
+    .option('-c, --components <components>', 'Comma-separated list of components (agents, tools, workflows)')
+    .option('-l, --llm <model-provider>', 'Default model provider (openai, anthropic, or groq))')
+    .option('-k, --llm-api-key <api-key>', 'API key for the model provider')
+    .option('-e, --example', 'Include example code')
+    .action(async (args) => {
+    await analytics.trackCommandExecution({
+        command: 'create',
+        args,
+        execution: async () => {
+            if (args.default) {
+                await create({
+                    components: ['agents', 'tools', 'workflows'],
+                    llmProvider: 'openai',
+                    addExample: false,
+                });
+            }
+            await create({
+                components: args.components,
+                llmProvider: args.llm,
+                addExample: args.example,
+                llmApiKey: args['llm-api-key'],
+            });
+        },
+    });
+});
 program
     .command('init')
-    .description('Initialize a new Mastra project')
+    .description('Initialize Mastra in your project')
     .option('--default', 'Quick start with defaults(src, OpenAI, no examples)')
     .option('-d, --dir <directory>', 'Directory for Mastra files to (defaults to src/)')
     .option('-c, --components <components>', 'Comma-separated list of components (agents, tools, workflows)')
     .option('-l, --llm <model-provider>', 'Default model provider (openai, anthropic, or groq))')
+    .option('-k, --llm-api-key <api-key>', 'API key for the model provider')
     .option('-e, --example', 'Include example code')
-    .option('-ne, --no-example', 'Skip example code')
     .action(async (args) => {
     await analytics.trackCommandExecution({
         command: 'init',
         args,
         execution: async () => {
-            await checkPkgJsonAndCreateStarter();
+            await checkPkgJson();
             await checkAndInstallCoreDeps();
-            if (!Object.keys(args).length)
-                return interactivePrompt();
+            if (!Object.keys(args).length) {
+                const result = await interactivePrompt();
+                await init({
+                    ...result,
+                    llmApiKey: result?.llmApiKey,
+                });
+                return;
+            }
             if (args?.default) {
                 init({
                     directory: 'src/',
                     components: ['agents', 'tools', 'workflows'],
                     llmProvider: 'openai',
                     addExample: false,
-                    showSpinner: true,
                 });
                 return;
             }
@@ -74,7 +105,7 @@ program
                 components: componentsArr,
                 llmProvider: args.llm,
                 addExample: args.example,
-                showSpinner: true,
+                llmApiKey: args['llm-api-key'],
             });
             return;
         },
@@ -86,12 +117,13 @@ program
     .option('-d, --dir <dir>', 'Path to your mastra folder')
     .option('-e, --env <env>', 'Environment File to use (defaults to .env.development)')
     .option('-t, --tools <toolsDirs>', 'Comma-separated list of paths to tool files to include')
+    .option('-p, --port <port>', 'Port number for the development server (defaults to 4111)')
     .action(args => {
     analytics.trackCommand({
         command: 'dev',
     });
     const apiKeys = findApiKeys();
-    dev({ port: 4111, env: apiKeys, dir: args?.dir, toolsDirs: args?.tools });
+    dev({ port: args?.port ? parseInt(args.port) : 4111, env: apiKeys, dir: args?.dir, toolsDirs: args?.tools });
 });
 const engine = program.command('engine').description('Manage the mastra engine');
 engine
@@ -121,24 +153,26 @@ engine
 engine
     .command('up')
     .description('Runs docker-compose up to start docker containers')
-    .action(async () => {
+    .option('-f, --file <path>', 'Path to docker-compose file')
+    .action(async (args) => {
     await analytics.trackCommandExecution({
         command: 'engine up',
-        args: {},
+        args,
         execution: async () => {
-            await up();
+            await up(args.file);
         },
     });
 });
 engine
     .command('down')
     .description('Runs docker-compose down to shut down docker containers')
-    .action(async () => {
+    .option('-f, --file <path>', 'Path to docker-compose file')
+    .action(async (args) => {
     await analytics.trackCommandExecution({
         command: 'engine down',
         args: {},
         execution: async () => {
-            await down();
+            await down(args.file);
         },
     });
 });
@@ -161,44 +195,6 @@ engine
         },
     });
 });
-const agent = program.command('agent').description('Manage Mastra agents');
-agent
-    .command('new')
-    .description('Create a new agent')
-    .option('-d, --dir <dir>', 'Path to your mastra folder')
-    .action(async (args) => {
-    await analytics.trackCommandExecution({
-        command: 'agent new',
-        args: {},
-        execution: async () => {
-            const result = await createNewAgent({ dir: args?.dir });
-            if (!result)
-                return;
-            await updateAgentIndexFile({ newAgentName: result, dir: args?.dir });
-        },
-    });
-});
-agent
-    .command('list')
-    .description('List all agents')
-    .option('-d, --dir <dir>', 'Path to your mastra folder')
-    .action(async (args) => {
-    await analytics.trackCommandExecution({
-        command: 'agent list',
-        args,
-        execution: async () => {
-            const agents = await listAgents({ dir: args?.dir });
-            if (!agents.length) {
-                p.note('No Agents...');
-                return;
-            }
-            p.intro(color.bgCyan(color.black(' Agents List ')));
-            agents.forEach((agent, index) => {
-                logger.log(`${index + 1}. ${color.blue(agent)}`);
-            });
-        },
-    });
-});
 const deploy = program.command('deploy').description('Deploy your Mastra project');
 deploy
     .command('vercel')
@@ -241,3 +237,5 @@ deploy
     });
 });
 program.parse(process.argv);
+export { create } from './commands/create/create.js';
+export { PosthogAnalytics } from './analytics/index.js';

package/dist/services/service.deps.d.ts

@@ -11,5 +11,6 @@ export declare class DepsService {
     checkDependencies(dependencies: string[]): Promise<string>;
     getProjectName(): Promise<any>;
     getPackageVersion(): Promise<string>;
+    addScriptsToPackageJson(scripts: Record<string, string>): Promise<void>;
 }
 //# sourceMappingURL=service.deps.d.ts.map

package/dist/services/service.deps.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"service.deps.d.ts","sourceRoot":"","sources":["../../src/services/service.deps.ts"],"names":[],"mappings":"AASA,qBAAa,WAAW;IACtB,OAAO,CAAC,cAAc,CAAS;;IAM/B,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,iBAAiB;IAcZ,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE;;;;;IAgBlC,iBAAiB,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAwB1D,cAAc;IAWd,iBAAiB;CAQ/B"}
+{"version":3,"file":"service.deps.d.ts","sourceRoot":"","sources":["../../src/services/service.deps.ts"],"names":[],"mappings":"AASA,qBAAa,WAAW;IACtB,OAAO,CAAC,cAAc,CAAS;;IAM/B,OAAO,CAAC,YAAY;IAcpB,OAAO,CAAC,iBAAiB;IAcZ,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE;;;;;IAgBlC,iBAAiB,CAAC,YAAY,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC;IAwB1D,cAAc;IAWd,iBAAiB;IASjB,uBAAuB,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;CAQrE"}

package/dist/services/service.deps.js

@@ -89,4 +89,12 @@ export class DepsService {
         const content = (await fsExtra.readJSON(pkgJsonPath));
         return content.version;
     }
+    async addScriptsToPackageJson(scripts) {
+        const packageJson = JSON.parse(await fsPromises.readFile('package.json', 'utf-8'));
+        packageJson.scripts = {
+            ...packageJson.scripts,
+            ...scripts,
+        };
+        await fsPromises.writeFile('package.json', JSON.stringify(packageJson, null, 2));
+    }
 }

package/dist/services/service.docker.d.ts

@@ -22,5 +22,6 @@ export declare class DockerService {
         dbUrl: string;
     }>;
     private editComposeFile;
+    getComposeFile(composePath: string): string;
 }
 //# sourceMappingURL=service.docker.d.ts.map

package/dist/services/service.docker.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"service.docker.d.ts","sourceRoot":"","sources":["../../src/services/service.docker.ts"],"names":[],"mappings":"AAOA,qBAAa,aAAa;IACxB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAqB5B,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAQ1C,SAAS,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;YAqBlD,eAAe;IAUvB,aAAa,CAAC,EAClB,mBAAmB,GACpB,GAAE;QAAE,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAAC,mBAAmB,CAAC,EAAE,MAAM,CAAA;KAAO;;;IAW1F,kBAAkB;IASlB,oBAAoB,CAAC,iBAAiB,EAAE,MAAM;IAS9C,mBAAmB,CAAC,iBAAiB,EAAE,MAAM;IAS7C,kBAAkB,CAAC,EACvB,oBAAoB,EACpB,YAAY,GACb,EAAE;QACD,oBAAoB,EAAE,MAAM,CAAC;QAC7B,YAAY,EAAE,MAAM,CAAC;KACtB;;;IAUD,OAAO,CAAC,eAAe;CAoBxB"}
+{"version":3,"file":"service.docker.d.ts","sourceRoot":"","sources":["../../src/services/service.docker.ts"],"names":[],"mappings":"AAQA,qBAAa,aAAa;IACxB,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAqB5B,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAQ1C,SAAS,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAA;KAAE,CAAC;YAqBlD,eAAe;IAUvB,aAAa,CAAC,EAClB,mBAAmB,GACpB,GAAE;QAAE,gBAAgB,CAAC,EAAE,MAAM,CAAC;QAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC;QAAC,mBAAmB,CAAC,EAAE,MAAM,CAAA;KAAO;;;IAW1F,kBAAkB;IASlB,oBAAoB,CAAC,iBAAiB,EAAE,MAAM;IAS9C,mBAAmB,CAAC,iBAAiB,EAAE,MAAM;IAS7C,kBAAkB,CAAC,EACvB,oBAAoB,EACpB,YAAY,GACb,EAAE;QACD,oBAAoB,EAAE,MAAM,CAAC;QAC7B,YAAY,EAAE,MAAM,CAAC;KACtB;;;IAUD,OAAO,CAAC,eAAe;IAqBvB,cAAc,CAAC,WAAW,EAAE,MAAM;CAmBnC"}

package/dist/services/service.docker.js

@@ -1,4 +1,5 @@
 import { execa } from 'execa';
+import fs from 'fs';
 import path from 'path';
 import { check } from 'tcp-port-used';
 import { fileURLToPath } from 'url';
@@ -108,4 +109,23 @@ export class DockerService {
             ],
         });
     }
+    getComposeFile(composePath) {
+        const fileService = new FileService();
+        let composeFile;
+        if (composePath) {
+            if (!fs.existsSync(composePath)) {
+                throw new Error(`Docker compose file not found: ${composePath}`);
+            }
+            composeFile = composePath;
+        }
+        else {
+            composeFile = fileService.getFirstExistingFile([
+                'mastra-pg.docker-compose.yaml',
+                'mastra-pg.docker-compose.yml',
+                'docker-compose.yaml',
+                'docker-compose.yml',
+            ]);
+        }
+        return composeFile;
+    }
 }

package/dist/src/starter-files/config.ts

@@ -0,0 +1,28 @@
+// @ts-ignore
+import { Config } from '@mastra/core';
+
+export const config: Config = {
+  name: 'PROJECT_NAME',
+  integrations: [],
+  db: {
+    provider: 'postgres',
+    uri: process.env.DB_URL!,
+  },
+  runner: {
+    provider: 'inngest',
+    uri: process.env.INNGEST_URL!,
+    signingKey: process.env.INNGEST_SIGNING_KEY!,
+    eventKey: process.env.INNGEST_EVENT_KEY!,
+  },
+  workflows: {
+    blueprintDirPath: '/mastra/blueprints',
+    systemEvents: {},
+    systemApis: [],
+  },
+  agents: {
+    agentDirPath: '/mastra/agents',
+    vectorProvider: [],
+  },
+  systemHostURL: process.env.APP_URL!,
+  routeRegistrationPath: '/api/mastra',
+};

package/dist/src/starter-files/mastra-pg.docker-compose.yaml

@@ -0,0 +1,15 @@
+services:
+  db:
+    image: pgvector/pgvector:pg16
+    container_name: 'REPLACE_PROJECT_NAME-db'
+    ports:
+      - 'REPLACE_DB_PORT:5432'
+    environment:
+      POSTGRES_USER: ${POSTGRES_USER:-postgres}
+      POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:-postgres}
+      POSTGRES_DB: ${POSTGRES_DB:-mastra}
+    volumes:
+      - pgdata:/var/lib/postgresql/data
+
+volumes:
+  pgdata:

package/dist/src/starter-files/tools.ts

@@ -0,0 +1,95 @@
+import { createTool } from '@mastra/core';
+import { z } from 'zod';
+
+interface WeatherResponse {
+  current: {
+    time: string;
+    temperature_2m: number;
+    apparent_temperature: number;
+    relative_humidity_2m: number;
+    wind_speed_10m: number;
+    wind_gusts_10m: number;
+    weather_code: number;
+  };
+}
+
+export const weatherTool = createTool({
+  id: 'get-weather',
+  description: 'Get current weather for a location',
+  inputSchema: z.object({
+    location: z.string().describe('City name'),
+  }),
+  outputSchema: z.object({
+    temperature: z.number(),
+    feelsLike: z.number(),
+    humidity: z.number(),
+    windSpeed: z.number(),
+    windGust: z.number(),
+    conditions: z.string(),
+    location: z.string(),
+  }),
+  execute: async ({ context }) => {
+    return await getWeather(context.location);
+  },
+});
+
+const getWeather = async (location: string) => {
+  const geocodingUrl = `https://geocoding-api.open-meteo.com/v1/search?name=${encodeURIComponent(location)}&count=1`;
+  const geocodingResponse = await fetch(geocodingUrl);
+  const geocodingData = await geocodingResponse.json();
+
+  if (!geocodingData.results?.[0]) {
+    throw new Error(`Location '${location}' not found`);
+  }
+
+  const { latitude, longitude, name } = geocodingData.results[0];
+
+  const weatherUrl = `https://api.open-meteo.com/v1/forecast?latitude=${latitude}&longitude=${longitude}&current=temperature_2m,apparent_temperature,relative_humidity_2m,wind_speed_10m,wind_gusts_10m,weather_code`;
+
+  const response = await fetch(weatherUrl);
+  const data: WeatherResponse = await response.json();
+
+  return {
+    temperature: data.current.temperature_2m,
+    feelsLike: data.current.apparent_temperature,
+    humidity: data.current.relative_humidity_2m,
+    windSpeed: data.current.wind_speed_10m,
+    windGust: data.current.wind_gusts_10m,
+    conditions: getWeatherCondition(data.current.weather_code),
+    location: name,
+  };
+};
+
+function getWeatherCondition(code: number): string {
+  const conditions: Record<number, string> = {
+    0: 'Clear sky',
+    1: 'Mainly clear',
+    2: 'Partly cloudy',
+    3: 'Overcast',
+    45: 'Foggy',
+    48: 'Depositing rime fog',
+    51: 'Light drizzle',
+    53: 'Moderate drizzle',
+    55: 'Dense drizzle',
+    56: 'Light freezing drizzle',
+    57: 'Dense freezing drizzle',
+    61: 'Slight rain',
+    63: 'Moderate rain',
+    65: 'Heavy rain',
+    66: 'Light freezing rain',
+    67: 'Heavy freezing rain',
+    71: 'Slight snow fall',
+    73: 'Moderate snow fall',
+    75: 'Heavy snow fall',
+    77: 'Snow grains',
+    80: 'Slight rain showers',
+    81: 'Moderate rain showers',
+    82: 'Violent rain showers',
+    85: 'Slight snow showers',
+    86: 'Heavy snow showers',
+    95: 'Thunderstorm',
+    96: 'Thunderstorm with slight hail',
+    99: 'Thunderstorm with heavy hail',
+  };
+  return conditions[code] || 'Unknown';
+}