@mastra/dane 0.0.1 → 0.0.2-alpha.2

package/CHANGELOG.md

@@ -0,0 +1,37 @@
+# @mastra/dane
+
+## 0.0.2-alpha.2
+
+### Patch Changes
+
+- ad58c35: Adds a shebang to the dane entry file
+
+## 0.0.2-alpha.1
+
+### Patch Changes
+
+- Updated dependencies [f031a1f]
+  - @mastra/core@0.1.27-alpha.38
+  - @mastra/rag@0.0.2-alpha.22
+  - @mastra/firecrawl@1.0.4-alpha.22
+  - @mastra/github@1.0.3-alpha.21
+  - @mastra/engine@0.0.5-alpha.32
+  - @mastra/memory@0.0.2-alpha.17
+
+## 0.0.2-alpha.0
+
+### Patch Changes
+
+- b5393f1: New example: Dane and many fixes to make it work
+- 697fbbe: Setup Dane, a cli assistant using the mastra framework.
+- Updated dependencies [45fd5b8]
+- Updated dependencies [c872875]
+- Updated dependencies [f6da688]
+- Updated dependencies [1cc6cc0]
+- Updated dependencies [b5393f1]
+  - @mastra/rag@0.0.2-alpha.21
+  - @mastra/core@0.1.27-alpha.37
+  - @mastra/engine@0.0.5-alpha.31
+  - @mastra/firecrawl@1.0.4-alpha.21
+  - @mastra/memory@0.0.2-alpha.16
+  - @mastra/github@1.0.3-alpha.20

package/package.json

@@ -1,20 +1,25 @@
 {
   "name": "@mastra/dane",
-  "version": "0.0.1",
-  "main": "index.js",
+  "version": "0.0.2-alpha.2",
+  "main": "dist/index.js",
+  "type": "module",
+  "bin": {
+    "dane": "./dist/index.js"
+  },
   "keywords": [],
   "author": "",
   "license": "ISC",
   "description": "",
   "devDependencies": {
     "@types/node": "^22.10.2",
-    "tsx": "^4.19.2",
-    "typescript": "^5.7.2"
+    "@types/pdf-parse": "^1.1.4",
+    "tsx": "^4.19.2"
   },
   "dependencies": {
     "boxen": "^8.0.1",
     "chalk": "^5.3.0",
     "cli-table3": "^0.6.5",
+    "commander": "^12.1.0",
     "execa": "^9.3.1",
     "inquirer": "^12.2.0",
     "luxon": "^3.5.0",
@@ -24,15 +29,17 @@
     "playwright": "^1.49.1",
     "playwright-core": "^1.49.1",
     "sqlite3": "^5.1.7",
+    "typescript": "^5.5.4",
     "zod": "^3.24.0",
-    "@mastra/core": "0.1.27-alpha.36",
-    "@mastra/github": "1.0.3-alpha.19",
-    "@mastra/rag": "0.0.2-alpha.20",
-    "@mastra/memory": "0.0.2-alpha.15",
-    "@mastra/firecrawl": "1.0.4-alpha.20",
-    "@mastra/engine": "0.0.5-alpha.30"
+    "@mastra/engine": "0.0.5-alpha.32",
+    "@mastra/core": "0.1.27-alpha.38",
+    "@mastra/github": "1.0.3-alpha.21",
+    "@mastra/memory": "0.0.2-alpha.17",
+    "@mastra/firecrawl": "1.0.4-alpha.22",
+    "@mastra/rag": "0.0.2-alpha.22"
   },
   "scripts": {
+    "build": "npx tsc",
     "issue-labeler": "npx bun src/issue-labeler.ts",
     "start": "npx bun src/index.ts",
     "test": "echo \"Error: no test specified\" && exit 1"

package/src/{issue-labeler.ts → commands/issue-labeler.ts}

@@ -1,9 +1,8 @@
-// import inquirer from 'inquirer';
 import chalk from 'chalk';
 
-import { mastra } from './mastra';
+import { mastra } from '../mastra/index.js';
 
-async function main() {
+export async function issueLabelerCommand() {
   console.log(chalk.green("Hi! I'm Dane!"));
   console.log(chalk.green('Let me label this for you..\n'));
   const result = await mastra.getWorkflow('githubIssueLabeler').execute({
@@ -25,8 +24,3 @@ async function main() {
     ),
   );
 }
-
-main().catch(e => {
-  console.error(e);
-  process.exit(1);
-});

package/src/commands/message.ts

@@ -0,0 +1,16 @@
+import chalk from 'chalk';
+
+import { mastra } from '../mastra/index.js';
+
+export async function message() {
+  console.log(chalk.green("Hi! I'm Dane!"));
+  console.log(chalk.green('What would you like to do today?\n'));
+  console.log(
+    await mastra.getWorkflow('message').execute({
+      triggerData: {
+        resourceid: 'f8b5c3a1-d6e7-4f9c-b2a3-1d8e4c7f9b5a',
+        threadId: '2d9e8c7f-6b5a-4d3c-8f1e-9b7d5c3a2e8h',
+      },
+    }),
+  );
+}

package/src/index.ts

@@ -1,19 +1,13 @@
-// import inquirer from 'inquirer';
-import chalk from 'chalk';
+#! /usr/bin/env node
+import { Command } from 'commander';
 
-import { mastra } from './mastra';
+import { issueLabelerCommand } from './commands/issue-labeler.js';
+import { message } from './commands/message.js';
 
-async function main() {
-  console.log(chalk.green("Hi! I'm Dane!"));
-  console.log(chalk.green('What would you like to do today?\n'));
-  console.log(
-    await mastra.getWorkflow('message').execute({
-      triggerData: {
-        resourceid: 'f8b5c3a1-d6e7-4f9c-b2a3-1d8e4c7f9b5a',
-        threadId: '2d9e8c7f-6b5a-4d3c-8f1e-9b7d5c3a2e8h',
-      },
-    }),
-  );
-}
+const program = new Command();
 
-main();
+program.command('chat').action(message);
+
+program.command('issue-labeler').action(issueLabelerCommand);
+
+program.parse(process.argv);

package/src/mastra/agents/index.ts

@@ -1,11 +1,11 @@
 import { Agent } from '@mastra/core';
 
-import { browserTool, googleSearch } from '../tools/browser';
-import { listEvents } from '../tools/calendar';
-import { crawl } from '../tools/crawl';
-import { execaTool } from '../tools/execa';
-import { fsTool } from '../tools/fs';
-import { readPDF } from '../tools/pdf';
+import { browserTool, googleSearch } from '../tools/browser.js';
+import { listEvents } from '../tools/calendar.js';
+import { crawl } from '../tools/crawl.js';
+import { execaTool } from '../tools/execa.js';
+import { fsTool } from '../tools/fs.js';
+import { readPDF } from '../tools/pdf.js';
 
 export const daneIssueLabeler = new Agent({
   name: 'DaneIssueLabeler',

package/src/mastra/index.ts

@@ -2,9 +2,9 @@ import { Mastra } from '@mastra/core';
 import { PostgresEngine } from '@mastra/engine';
 import { UpstashKVMemory } from '@mastra/memory';
 
-import { dane, daneIssueLabeler } from './agents';
-import { firecrawl } from './integrations';
-import { messageWorkflow, githubIssueLabeler } from './workflows';
+import { dane, daneIssueLabeler } from './agents/index.js';
+import { firecrawl } from './integrations/index.js';
+import { messageWorkflow, githubIssueLabeler } from './workflows/index.js';
 
 const engine = new PostgresEngine({
   url: 'postgres://postgres:postgres@localhost:5433/mastra',

package/src/mastra/tools/browser.ts

@@ -1,5 +1,5 @@
 import { createTool } from '@mastra/core';
-import { MastraDocument } from '@mastra/rag';
+import { MDocument } from '@mastra/rag';
 import chalk from 'chalk';
 import { chromium } from 'playwright-core';
 import { z } from 'zod';
@@ -24,7 +24,7 @@ export const browserTool = createTool({
 
       await page.goto(url);
 
-      const docs = MastraDocument.fromHTML(await page.content());
+      const docs = MDocument.fromHTML(await page.content());
 
       await docs.chunk({
         strategy: 'html',
@@ -51,8 +51,11 @@ export const browserTool = createTool({
 
       return { message: docs.getText().join('\n') };
     } catch (e) {
-      console.log(`\n${chalk.red(e.message)}`);
-      return { message: `Error: ${e.message}` };
+      if (e instanceof Error) {
+        console.log(`\n${chalk.red(e.message)}`);
+        return { message: `Error: ${e.message}` };
+      }
+      return { message: 'Error' };
     }
   },
 });
@@ -74,8 +77,11 @@ export const googleSearch = createTool({
         headless: true,
       });
     } catch (e) {
-      console.log(`\n${chalk.red(e.message)}`);
-      return { message: `Error: ${e.message}` };
+      if (e instanceof Error) {
+        console.log(`\n${chalk.red(e.message)}`);
+        return { message: `Error: ${e.message}` };
+      }
+      return { message: 'Error' };
     }
 
     try {
@@ -116,8 +122,11 @@ export const googleSearch = createTool({
 
       return { message: text.join('\n') };
     } catch (e) {
-      console.log(`\n${chalk.red(e.message)}`);
-      return { message: `Error: ${e.message}` };
+      if (e instanceof Error) {
+        console.log(`\n${chalk.red(e.message)}`);
+        return { message: `Error: ${e.message}` };
+      }
+      return { message: `Error` };
     }
   },
 });

package/src/mastra/tools/calendar.ts

@@ -2,7 +2,6 @@ import { createTool } from '@mastra/core';
 import chalk from 'chalk';
 import { execSync } from 'child_process';
 import Table from 'cli-table3';
-import * as os from 'os';
 import { z } from 'zod';
 
 interface CalendarEvent {
@@ -14,12 +13,6 @@ interface CalendarEvent {
 }
 
 class LocalCalendarReader {
-  private platform: string;
-
-  constructor() {
-    this.platform = os.platform();
-  }
-
   async getEvents(): Promise<CalendarEvent[]> {
     const script = `
           tell application "Calendar"
@@ -58,8 +51,13 @@ class LocalCalendarReader {
       const result = execSync(`osascript -e '${script}'`).toString();
       return this.parseAppleScriptOutput(result);
     } catch (error) {
-      console.error('Raw AppleScript error:', error.message);
-      throw new Error(`Failed to read Mac calendar: ${error.message}`);
+      if (error instanceof Error) {
+        console.error('Raw AppleScript error:', error.message);
+        throw new Error(`Failed to read Mac calendar: ${error.message}`);
+      } else {
+        console.error('An unknown error occurred:', error);
+        throw new Error('Failed to read Mac calendar');
+      }
     }
   }
 
@@ -73,19 +71,20 @@ class LocalCalendarReader {
         const [title, startDateStr, endDateStr, location, description] = line.split('|');
 
         const startStandardized = startDateStr
-          .split(',')[1] // Remove day name
-          .replace(' at ', ' ') // Remove 'at'
-          .trim(); // 'January 3, 2025 9:00:00 AM'
-        const startDate = new Date(startStandardized);
+          ?.split(',')?.[1] // Remove day name
+          ?.replace(' at ', ' ') // Remove 'at'
+          ?.trim(); // 'January 3, 2025 9:00:00 AM'
+
+        const startDate = new Date(startStandardized || '');
 
         const endStandardized = endDateStr
-          .split(',')[1] // Remove day name
-          .replace(' at ', ' ') // Remove 'at'
-          .trim(); // 'January 3, 2025 9:00:00 AM'
-        const endDate = new Date(endStandardized);
+          ?.split(',')?.[1] // Remove day name
+          ?.replace(' at ', ' ') // Remove 'at'
+          ?.trim(); // 'January 3, 2025 9:00:00 AM'
+        const endDate = new Date(endStandardized || '');
 
         const event: CalendarEvent = {
-          title: title.trim(),
+          title: title?.trim()!,
           startDate,
           endDate,
           location: location?.trim() || '',
@@ -146,7 +145,9 @@ export const listEvents = createTool({
         content: JSON.stringify(events, null, 2),
       };
     } catch (e) {
-      console.log(`\n${chalk.red(e.message)}`);
+      if (e instanceof Error) {
+        console.log(`\n${chalk.red(e.message)}`);
+      }
       return { content: 'Error' };
     }
   },

package/src/mastra/tools/crawl.ts

@@ -14,7 +14,7 @@ export const crawl = createTool({
     message: z.string(),
   }),
   execute: async ({ context, mastra }) => {
-    await mastra?.syncs?.['FIRECRAWL:CRAWL_AND_SYNC'].execute({
+    await mastra?.syncs?.['FIRECRAWL:CRAWL_AND_SYNC']?.execute({
       context,
       mastra,
     });

package/src/mastra/tools/execa.ts

@@ -6,7 +6,7 @@ import { z } from 'zod';
 
 // Create transform stream that applies chalk
 const colorTransform = new Transform({
-  transform(chunk, encoding, callback) {
+  transform(chunk, _encoding, callback) {
     // Convert chunk to string and apply chalk
     const colored = chalk.blue(chunk.toString());
     this.push(colored);

package/src/mastra/workflows/chat.ts

@@ -3,7 +3,7 @@ import chalk from 'chalk';
 import inquirer from 'inquirer';
 import { z } from 'zod';
 
-import { dane } from '../agents';
+import { dane } from '../agents/index.js';
 
 export const messageWorkflow = new Workflow({
   name: 'entry',
@@ -62,7 +62,7 @@ const messageOutputStep = new Step({
         messages = [];
       }
 
-      const res = await mastra?.agents?.['dane'].generate(message, {
+      const res = await mastra?.agents?.['dane']?.generate(message, {
         stream: true,
         maxSteps: 5,
         resourceid,
@@ -85,7 +85,9 @@ const messageOutputStep = new Step({
       console.log(chalk.red(`\n`));
       console.log(chalk.red(`\n`));
       console.log(chalk.red(`Error streaming results. Let's try again.`));
-      console.log(chalk.red(e.message));
+      if (e instanceof Error) {
+        console.log(chalk.red(e.message));
+      }
     }
 
     const res = await dane.generate(message, {
@@ -94,9 +96,9 @@ const messageOutputStep = new Step({
       resourceid,
     });
 
-    console.log(chalk.green(res.text));
+    console.log(chalk.green(res?.text));
 
-    return { message: res.text };
+    return { message: res?.text };
   },
 });
 

package/src/mastra/workflows/index.ts

@@ -1,2 +1,2 @@
-export * from './chat';
-export * from './issue-labeler';
+export * from './chat.js';
+export * from './issue-labeler.js';

package/src/mastra/workflows/issue-labeler.ts

@@ -1,7 +1,7 @@
 import { Step, Workflow } from '@mastra/core';
 import { z } from 'zod';
 
-import { github } from '../integrations';
+import { github } from '../integrations/index.js';
 
 export const githubIssueLabeler = new Workflow({
   name: 'github-issue-labeler',
@@ -24,6 +24,7 @@ const getIssue = new Step({
 
     const issue = await client.issuesGet({
       path: {
+        // TODO: Type triggerData in machineContext to the triggerSchema
         owner: context?.machineContext?.triggerData?.owner,
         repo: context?.machineContext?.triggerData?.repo,
         issue_number: context?.machineContext?.triggerData?.issue_number,
@@ -59,10 +60,10 @@ const labelIssue = new Step({
     const res = await daneIssueLabeler?.generate(
       `
             Hey Dane, given:
-            * this issue title: ${parentStep?.payload?.title}    
+            * this issue title: ${parentStep?.payload?.title}
             * this issue body: ${parentStep?.payload?.body}
             * these labels: ${parentStep?.payload?.labelNames}
-            
+
             What label or labels would you assign?
         `,
       {

package/tsconfig.json

@@ -0,0 +1,11 @@
+{
+  "extends": "../../tsconfig.json",
+  "compilerOptions": {
+    "moduleResolution": "bundler",
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "noEmit": false
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "**/*.test.ts"]
+}