@mastra/dane 0.0.2-alpha.112 → 0.0.2-alpha.114

package/dist/mastra/workflows/publish-packages.js

@@ -0,0 +1,387 @@
+import { Step, Workflow } from '@mastra/core/workflows';
+import chalk from 'chalk';
+import { existsSync } from 'fs';
+import path from 'path';
+import { z } from 'zod';
+import { PACKAGES_LIST_PROMPT, PUBLISH_PACKAGES_PROMPT } from '../agents/package-publisher.js';
+import { pnpmBuild } from '../tools/pnpm.js';
+export const packagePublisher = new Workflow({
+    name: 'pnpm-changset-publisher',
+});
+const outputSchema = z.object({
+    packages: z.array(z.string()),
+    integrations: z.array(z.string()),
+    deployers: z.array(z.string()),
+    vector_stores: z.array(z.string()),
+    stores: z.array(z.string()),
+    speech: z.array(z.string()),
+});
+const defaultSet = {
+    packages: [],
+    deployers: [],
+    integrations: [],
+    vector_stores: [],
+    stores: [],
+    speech: [],
+};
+const getPacakgesToPublish = new Step({
+    id: 'getPacakgesToPublish',
+    outputSchema,
+    execute: async ({ mastra }) => {
+        const agent = mastra?.agents?.['danePackagePublisher'];
+        if (!agent) {
+            throw new Error('Agent not found');
+        }
+        const result = await agent.generate(PACKAGES_LIST_PROMPT);
+        console.log(chalk.green(`\n${result.text}`));
+        const resultObj = await agent.generate(`
+      Please convert this into a structured object:
+
+      Input Text: ${result.text}
+
+      1. Order Requirements:
+         - @mastra/core MUST be first within packages
+         - @mastra/deployer MUST be second within packages
+         - Group parallel builds by directory type
+
+      2. Output Format:
+         - Group into: packages[], integrations[], deployers[], vector_stores[], stores[]
+         - Place create-mastra in packages[] array
+         - Maintain correct order within each group
+
+      3. Critical Rules:
+         - Never publish without building first
+         - Only include packages that need updates
+         - Follow dependency order strictly
+    `, {
+            output: z.object({
+                packages: z.array(z.string()),
+                integrations: z.array(z.string()),
+                deployers: z.array(z.string()),
+                vector_stores: z.array(z.string()),
+                stores: z.array(z.string()),
+                speech: z.array(z.string()),
+            }),
+        });
+        return {
+            packages: resultObj?.object?.packages,
+            integrations: resultObj?.object?.integrations,
+            deployers: resultObj?.object?.deployers,
+            vector_stores: resultObj?.object?.vector_stores,
+            stores: resultObj?.object?.stores,
+            speech: resultObj?.object?.speech,
+        };
+    },
+});
+const assemblePackages = new Step({
+    id: 'assemblePackages',
+    outputSchema,
+    execute: async ({ context }) => {
+        if (context.machineContext?.stepResults.getPacakgesToPublish?.status !== 'success') {
+            return {
+                packages: [],
+                integrations: [],
+                deployers: [],
+                vector_stores: [],
+                stores: [],
+                speech: [],
+            };
+        }
+        const payload = context.machineContext.stepResults.getPacakgesToPublish.payload;
+        const packagesToBuild = new Set();
+        const deployersToBuild = new Set();
+        const integrationsToBuild = new Set();
+        const vector_storesToBuild = new Set();
+        const storesToBuild = new Set();
+        const speechToBuild = new Set();
+        if (payload?.packages) {
+            payload.packages.forEach((pkg) => {
+                let pkgName = pkg.replace('@mastra/', '');
+                if (pkgName === 'mastra') {
+                    pkgName = 'cli';
+                }
+                const pkgPath = path.join(process.cwd(), 'packages', pkgName);
+                packagesToBuild.add(pkgPath);
+            });
+        }
+        if (payload?.deployers) {
+            payload.deployers.forEach((pkg) => {
+                let pkgName = pkg.replace('@mastra/deployer-', '');
+                if (pkgName === 'mastra') {
+                    pkgName = 'cli';
+                }
+                const pkgPath = path.join(process.cwd(), 'deployers', pkgName);
+                deployersToBuild.add(pkgPath);
+            });
+        }
+        if (payload?.vector_stores) {
+            payload.vector_stores.forEach((pkg) => {
+                let pkgName = pkg.replace('@mastra/vector-', '');
+                const pkgPath = path.join(process.cwd(), 'vector-stores', pkgName);
+                vector_storesToBuild.add(pkgPath);
+            });
+        }
+        if (payload?.stores) {
+            payload.stores.forEach((pkg) => {
+                let pkgName = pkg.replace('@mastra/store-', '');
+                const pkgPath = path.join(process.cwd(), 'storage', pkgName);
+                storesToBuild.add(pkgPath);
+            });
+        }
+        if (payload?.integrations) {
+            const integrations = payload.integrations;
+            integrations.forEach((integration) => {
+                let pkgName = integration.replace('@mastra/', '');
+                const integrationPath = path.join(process.cwd(), 'integrations', pkgName);
+                integrationsToBuild.add(integrationPath);
+            });
+        }
+        if (payload?.speech) {
+            const speecs = payload.speech;
+            speecs.forEach((speech) => {
+                let pkgName = speech.replace('@mastra/speech-', '');
+                const speechPath = path.join(process.cwd(), 'speech', pkgName);
+                speechToBuild.add(speechPath);
+            });
+        }
+        const pkgSet = Array.from(packagesToBuild.keys());
+        const deploySet = Array.from(deployersToBuild.keys());
+        const integrationSet = Array.from(integrationsToBuild.keys());
+        const vectorStoreSet = Array.from(vector_storesToBuild.keys());
+        const storeSet = Array.from(storesToBuild.keys());
+        const speechSet = Array.from(speechToBuild.keys());
+        if (!packagesToBuild.size &&
+            !deployersToBuild.size &&
+            !integrationsToBuild.size &&
+            !vector_storesToBuild.size &&
+            !storesToBuild.size) {
+            console.error(chalk.red('No packages to build.'));
+            return defaultSet;
+        }
+        console.log(chalk.green(`\nBuilding packages:\n`));
+        pkgSet.forEach((pkg) => {
+            console.log(chalk.green(pkg));
+        });
+        if (deploySet.length > 0) {
+            console.log(chalk.green(`\nBuilding deployers:\n`));
+            deploySet.forEach((pkg) => {
+                console.log(chalk.green(pkg));
+            });
+        }
+        if (integrationSet.length > 0) {
+            console.log(chalk.green(`\nBuilding integrations:\n`));
+            integrationSet.forEach((pkg) => {
+                console.log(chalk.green(pkg));
+            });
+        }
+        if (vectorStoreSet.length > 0) {
+            console.log(chalk.green(`\nBuilding vector stores:\n`));
+            vectorStoreSet.forEach((pkg) => {
+                console.log(chalk.green(pkg));
+            });
+        }
+        if (storeSet.length > 0) {
+            console.log(chalk.green(`\nBuilding storage packages:\n`));
+            storeSet.forEach((pkg) => {
+                console.log(chalk.green(pkg));
+            });
+        }
+        if (speechSet.length > 0) {
+            console.log(chalk.green(`\nBuilding speech:\n`));
+            speechSet.forEach((pkg) => {
+                console.log(chalk.green(pkg));
+            });
+        }
+        return {
+            packages: pkgSet,
+            deployers: deploySet,
+            integrations: integrationSet,
+            vector_stores: vectorStoreSet,
+            stores: storeSet,
+            speech: speechSet,
+        };
+    },
+});
+const buildPackages = new Step({
+    id: 'buildPackages',
+    outputSchema,
+    execute: async ({ context, mastra }) => {
+        if (context.machineContext?.stepResults.assemblePackages?.status !== 'success') {
+            return defaultSet;
+        }
+        const pkgSet = context.machineContext.stepResults.assemblePackages.payload.packages;
+        const deploySet = context.machineContext.stepResults.assemblePackages.payload.deployers;
+        const integrationSet = context.machineContext.stepResults.assemblePackages.payload.integrations;
+        const vectorStoreSet = context.machineContext.stepResults.assemblePackages.payload.vector_stores;
+        const storeSet = context.machineContext.stepResults.assemblePackages.payload.stores;
+        const speechSet = context.machineContext.stepResults.assemblePackages.payload.speech;
+        console.log({
+            pkgSet,
+            deploySet,
+            integrationSet,
+            vectorStoreSet,
+            storeSet,
+            speechSet,
+        });
+        const agent = mastra?.agents?.['danePackagePublisher'];
+        if (!agent) {
+            throw new Error('Agent not found');
+        }
+        async function buildSet(list) {
+            for (const pkg of list) {
+                await pnpmBuild.execute({
+                    context: {
+                        name: pkg,
+                        packagePath: pkg,
+                    },
+                    suspend: async () => { },
+                });
+            }
+        }
+        let built = false;
+        if (pkgSet.length > 0) {
+            built = true;
+            await buildSet(pkgSet);
+        }
+        if (deploySet.length > 0) {
+            await buildSet(deploySet);
+            built = true;
+        }
+        if (integrationSet.length > 0) {
+            await buildSet(integrationSet);
+            built = true;
+        }
+        if (vectorStoreSet.length > 0) {
+            await buildSet(vectorStoreSet);
+            built = true;
+        }
+        if (speechSet.length > 0) {
+            await buildSet(speechSet);
+            built = true;
+        }
+        if (!built) {
+            console.error(chalk.red('Failed to build one or more packages'));
+            throw new Error('Failed to build one or more packages');
+        }
+        return {
+            packages: pkgSet,
+            deployers: deploySet,
+            integrations: integrationSet,
+            vector_stores: vectorStoreSet,
+            stores: storeSet,
+            speech: speechSet,
+        };
+    },
+});
+const verifyBuild = new Step({
+    id: 'verifyBuild',
+    outputSchema: z.object({
+        packages: z.array(z.string()),
+    }),
+    execute: async ({ context }) => {
+        if (context.machineContext?.stepResults.buildPackages?.status !== 'success') {
+            return {
+                packages: [],
+            };
+        }
+        const pkgSet = context.machineContext.stepResults.buildPackages.payload.packages;
+        const deploySet = context.machineContext.stepResults.buildPackages.payload.deployers;
+        const integrationSet = context.machineContext.stepResults.buildPackages.payload.integrations;
+        const vectorStoreSet = context.machineContext.stepResults.buildPackages.payload.vector_stores;
+        const storeSet = context.machineContext.stepResults.buildPackages.payload.stores;
+        const speechSet = context.machineContext.stepResults.buildPackages.payload.speech;
+        const allPackages = [...pkgSet, ...deploySet, ...integrationSet, ...vectorStoreSet, ...storeSet, ...speechSet];
+        function checkMissingPackages(pkgSet) {
+            const missingPackages = [];
+            for (const pkg of pkgSet) {
+                if (!existsSync(`${pkg}/dist`)) {
+                    console.error(chalk.red(`We did not find the dist folder for ${pkg}.`));
+                    missingPackages.push(pkg);
+                }
+            }
+            return missingPackages;
+        }
+        console.log('Verifying the output for:', context.machineContext.stepResults.buildPackages.payload.allPackages);
+        const missingPackages = checkMissingPackages(allPackages);
+        if (missingPackages.length > 0) {
+            console.error(chalk.red(`Missing packages: ${missingPackages.join(', ')}`));
+            throw new Error('Failed to build one or more packages');
+        }
+        return {
+            packages: allPackages,
+        };
+    },
+});
+const publishChangeset = new Step({
+    id: 'publishChangeset',
+    outputSchema: z.object({
+        packages: z.array(z.string()),
+    }),
+    execute: async ({ context, mastra }) => {
+        if (context.machineContext?.stepResults.buildPackages?.status !== 'success') {
+            return {
+                packages: [],
+            };
+        }
+        const pkgSet = context.machineContext.stepResults.buildPackages.payload.packages;
+        const agent = mastra?.agents?.['danePackagePublisher'];
+        if (!agent) {
+            throw new Error('Agent not found');
+        }
+        const res = await agent.generate(PUBLISH_PACKAGES_PROMPT);
+        console.log(chalk.green(res.text));
+        return { packages: pkgSet };
+    },
+});
+const setLatestDistTag = new Step({
+    id: 'setLatestDistTag',
+    outputSchema: z.object({
+        packages: z.array(z.string()),
+    }),
+    execute: async ({ context, mastra }) => {
+        if (context.machineContext?.stepResults.publishChangeset?.status !== 'success') {
+            return {
+                packages: [],
+            };
+        }
+        const pkgSet = context.machineContext.stepResults.publishChangeset.payload.packages;
+        const agent = mastra?.agents?.['danePackagePublisher'];
+        if (!agent) {
+            throw new Error('Agent not found');
+        }
+        let res = await agent.generate(`
+            Set the active tag for these packages ${pkgSet.join(',')}.
+        `);
+        console.log(chalk.green(res.text));
+        return { packages: pkgSet };
+    },
+});
+packagePublisher
+    .step(getPacakgesToPublish)
+    .then(assemblePackages)
+    .then(buildPackages, {
+    when: async ({ context }) => {
+        return (context.stepResults.assemblePackages?.status === 'success' &&
+            context.stepResults.assemblePackages?.payload?.packages.length > 0);
+    },
+})
+    .then(verifyBuild, {
+    when: async ({ context }) => {
+        return (context.stepResults.buildPackages?.status === 'success' &&
+            context.stepResults.buildPackages?.payload?.packages.length > 0);
+    },
+})
+    .after(verifyBuild)
+    .step(publishChangeset, {
+    when: async ({ context }) => {
+        return (context.stepResults.buildPackages?.status === 'success' &&
+            context.stepResults.buildPackages?.payload?.packages.length > 0);
+    },
+})
+    .then(setLatestDistTag, {
+    when: async ({ context }) => {
+        return (context.stepResults.publishChangeset?.status === 'success' &&
+            context.stepResults.publishChangeset?.payload?.packages.length > 0);
+    },
+})
+    .commit();

package/dist/mastra/workflows/telephone-game.d.ts

@@ -0,0 +1,3 @@
+import { Workflow } from '@mastra/core/workflows';
+export declare const telephoneGameWorkflow: Workflow<any, any>;
+//# sourceMappingURL=telephone-game.d.ts.map

package/dist/mastra/workflows/telephone-game.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"telephone-game.d.ts","sourceRoot":"","sources":["../../../src/mastra/workflows/telephone-game.ts"],"names":[],"mappings":"AACA,OAAO,EAAQ,QAAQ,EAAiB,MAAM,wBAAwB,CAAC;AAGvE,eAAO,MAAM,qBAAqB,oBAEhC,CAAC"}

package/dist/mastra/workflows/telephone-game.js

@@ -0,0 +1,95 @@
+import { input } from '@inquirer/prompts';
+import { Step, Workflow, getStepResult } from '@mastra/core/workflows';
+import { z } from 'zod';
+export const telephoneGameWorkflow = new Workflow({
+    name: 'telephoneGame',
+});
+const stepA1 = new Step({
+    id: 'stepA1',
+    description: 'Starts the message',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async () => {
+        return {
+            message: 'Test',
+        };
+    },
+});
+const stepA2 = new Step({
+    id: 'stepA2',
+    description: 'Pass the message through',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async () => {
+        const content = await input({
+            message: 'Give me a message',
+            validate: input => input.trim().length > 0 || 'Message cannot be empty',
+        });
+        return {
+            message: content,
+        };
+    },
+});
+const stepB2 = new Step({
+    id: 'stepB2',
+    description: 'Checks if the file exists',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async ({ context }) => {
+        if (context.machineContext?.stepResults.stepA2?.status !== 'success') {
+            throw new Error('Message not found');
+        }
+        const msg = context.machineContext.stepResults.stepA2.payload.message;
+        return {
+            message: msg,
+        };
+    },
+});
+const stepC2 = new Step({
+    id: 'stepC2',
+    description: 'Ask if you should modify the message',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async ({ suspend, context, mastra }) => {
+        const oMsg = getStepResult(context?.machineContext?.stepResults.stepA2);
+        if (context?.machineContext?.stepResults.stepC2?.status === 'success') {
+            const msg = getStepResult(context?.machineContext?.stepResults.stepC2);
+            if (msg.confirm) {
+                if (mastra?.llm) {
+                    const llm = mastra?.llm({
+                        provider: 'ANTHROPIC',
+                        name: 'claude-3-5-haiku-20241022',
+                    });
+                    const result = await llm.generate(`
+            You are playing a game of telephone.
+            Here is the message the previous person sent ${oMsg.message}. 
+            But you want to change the message. 
+            Only return the message
+            `);
+                    return {
+                        message: result.text,
+                    };
+                }
+            }
+            return oMsg;
+        }
+        await suspend();
+        return { message: 'Suspended' };
+    },
+});
+const stepD2 = new Step({
+    id: 'stepD2',
+    description: 'Pass the message',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async ({ context }) => {
+        const msg = getStepResult(context?.machineContext?.stepResults.stepC2);
+        return msg;
+    },
+});
+telephoneGameWorkflow.step(stepA1).step(stepA2).then(stepB2).then(stepC2).then(stepD2).commit();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/dane",
-  "version": "0.0.2-alpha.112",
+  "version": "0.0.2-alpha.114",
   "main": "dist/index.js",
   "type": "module",
   "files": [
@@ -20,7 +20,7 @@
     "@types/node": "^22.10.2",
     "@types/pdf-parse": "^1.1.4",
     "tsx": "^4.19.2",
-    "mastra": "0.2.0-alpha.141"
+    "mastra": "0.2.0-alpha.142"
   },
   "dependencies": {
     "@inquirer/prompts": "^7.2.1",
@@ -41,14 +41,14 @@
     "sqlite3": "^5.1.7",
     "typescript": "^5.7.3",
     "zod": "^3.24.0",
-    "@mastra/core": "0.2.0-alpha.88",
-    "@mastra/firecrawl": "1.1.0-alpha.82",
-    "@mastra/github": "1.0.3-alpha.73",
-    "@mastra/mcp": "0.1.0-alpha.30",
-    "@mastra/memory": "0.1.0-alpha.70",
-    "@mastra/rag": "0.1.0-alpha.80",
-    "@mastra/stabilityai": "1.0.1-alpha.63",
-    "@mastra/store-upstash": "0.0.0-alpha.6"
+    "@mastra/core": "0.2.0-alpha.89",
+    "@mastra/firecrawl": "1.1.0-alpha.83",
+    "@mastra/github": "1.0.3-alpha.74",
+    "@mastra/memory": "0.1.0-alpha.71",
+    "@mastra/stabilityai": "1.0.1-alpha.64",
+    "@mastra/mcp": "0.1.0-alpha.31",
+    "@mastra/rag": "0.1.0-alpha.81",
+    "@mastra/store-upstash": "0.0.0-alpha.7"
   },
   "scripts": {
     "build": "npx tsc",