@mastra/dane 0.0.2-alpha.2 → 0.0.2-alpha.4

package/dist/mastra/workflows/issue-labeler.js

@@ -0,0 +1,85 @@
+import { Step, Workflow } from '@mastra/core';
+import { z } from 'zod';
+import { github } from '../integrations/index.js';
+export const githubIssueLabeler = new Workflow({
+    name: 'github-issue-labeler',
+    triggerSchema: z.object({
+        repo: z.string(),
+        owner: z.string(),
+        issue_number: z.number(),
+    }),
+});
+const getIssue = new Step({
+    id: 'getIssue',
+    outputSchema: z.object({
+        title: z.string(),
+        body: z.string(),
+        labelNames: z.array(z.string()),
+    }),
+    execute: async ({ context }) => {
+        const client = await github.getApiClient();
+        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,
+            },
+        });
+        const labels = await client.issuesListLabelsForRepo({
+            path: {
+                owner: context?.machineContext?.triggerData?.owner,
+                repo: context?.machineContext?.triggerData?.repo,
+            },
+        });
+        const labelNames = labels?.data?.map(label => label.name);
+        return { title: issue?.data?.title, body: issue?.data?.body, labelNames: labelNames };
+    },
+});
+const labelIssue = new Step({
+    id: 'labelIssue',
+    outputSchema: z.object({
+        labels: z.array(z.string()),
+    }),
+    execute: async ({ context, mastra }) => {
+        const parentStep = context?.machineContext?.stepResults?.getIssue;
+        if (!parentStep || parentStep.status !== 'success') {
+            return { labels: [] };
+        }
+        const daneIssueLabeler = mastra?.agents?.daneIssueLabeler;
+        const res = await daneIssueLabeler?.generate(`
+            Hey Dane, given:
+            * this issue title: ${parentStep?.payload?.title}
+            * this issue body: ${parentStep?.payload?.body}
+            * these labels: ${parentStep?.payload?.labelNames}
+
+            What label or labels would you assign?
+        `, {
+            schema: z.object({
+                labels: z.array(z.string()),
+            }),
+        });
+        return { labels: res?.object?.labels };
+    },
+});
+const applyLabels = new Step({
+    id: 'applyLabels',
+    execute: async ({ context }) => {
+        const parentStep = context?.machineContext?.stepResults?.labelIssue;
+        if (!parentStep || parentStep.status !== 'success') {
+            return;
+        }
+        const client = await github.getApiClient();
+        await client.issuesAddLabels({
+            path: {
+                owner: context?.machineContext?.triggerData?.owner,
+                repo: context?.machineContext?.triggerData?.repo,
+                issue_number: context?.machineContext?.triggerData?.issue_number,
+            },
+            body: {
+                labels: parentStep.payload.labels,
+            },
+        });
+    },
+});
+githubIssueLabeler.step(getIssue).then(labelIssue).then(applyLabels).commit();

package/package.json

@@ -1,8 +1,11 @@
 {
   "name": "@mastra/dane",
-  "version": "0.0.2-alpha.2",
+  "version": "0.0.2-alpha.4",
   "main": "dist/index.js",
   "type": "module",
+  "files": [
+    "dist"
+  ],
   "bin": {
     "dane": "./dist/index.js"
   },
@@ -31,11 +34,11 @@
     "sqlite3": "^5.1.7",
     "typescript": "^5.5.4",
     "zod": "^3.24.0",
-    "@mastra/engine": "0.0.5-alpha.32",
     "@mastra/core": "0.1.27-alpha.38",
-    "@mastra/github": "1.0.3-alpha.21",
+    "@mastra/firecrawl": "1.0.4-alpha.23",
+    "@mastra/engine": "0.0.5-alpha.32",
+    "@mastra/github": "1.0.3-alpha.22",
     "@mastra/memory": "0.0.2-alpha.17",
-    "@mastra/firecrawl": "1.0.4-alpha.22",
     "@mastra/rag": "0.0.2-alpha.22"
   },
   "scripts": {

package/CHANGELOG.md

@@ -1,37 +0,0 @@
-# @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/docker-compose.yaml

@@ -1,22 +0,0 @@
-services:
-  db:
-    image: pgvector/pgvector:pg16
-    container_name: 'dane-db'
-    ports:
-      - '5433:5432'
-    environment:
-      POSTGRES_USER: ${POSTGRES_USER:-postgres}
-      POSTGRES_PASSWORD: ${POSTGRES_PASSWORD:-postgres}
-      POSTGRES_DB: ${POSTGRES_DB:-mastra}
-  redis:
-    image: redis
-    ports:
-      - 6379:6379
-  serverless-redis-http:
-      ports:
-        - "8079:80"
-      image: hiett/serverless-redis-http:latest
-      environment:
-        SRH_MODE: env
-        SRH_TOKEN: example_token
-        SRH_CONNECTION_STRING: "redis://redis:6379" # Using `redis` hostname since they're in the same Docker network.

package/src/commands/issue-labeler.ts

@@ -1,26 +0,0 @@
-import chalk from 'chalk';
-
-import { mastra } from '../mastra/index.js';
-
-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({
-    triggerData: {
-      issue_number: parseInt(process.env.ISSUE_NUMBER!, 10),
-      owner: process.env.OWNER!,
-      repo: process.env.REPO!,
-    },
-  });
-
-  if (result.results?.labelIssue?.status !== 'success') {
-    console.error(chalk.red(`Failed to apply labels for issue: ${result.triggerData?.issue_number}`));
-    return;
-  }
-
-  console.log(
-    chalk.green(
-      `Issue: ${result.triggerData?.issue_number} has been labeled with: ${result.results?.labelIssue?.payload?.labels.join(', ')}`,
-    ),
-  );
-}

package/src/commands/message.ts

@@ -1,16 +0,0 @@
-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/mastra/index.ts

@@ -1,32 +0,0 @@
-import { Mastra } from '@mastra/core';
-import { PostgresEngine } from '@mastra/engine';
-import { UpstashKVMemory } from '@mastra/memory';
-
-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',
-});
-
-export const mastra = new Mastra({
-  agents: {
-    dane,
-    daneIssueLabeler,
-  },
-  engine,
-  memory: new UpstashKVMemory({
-    url: 'http://localhost:8079',
-    token: `example_token`,
-    maxTokens: 39000,
-  }),
-  workflows: {
-    message: messageWorkflow,
-    githubIssueLabeler: githubIssueLabeler,
-  },
-  logger: false,
-  syncs: {
-    ...firecrawl.getSyncs(),
-  },
-});

package/src/mastra/tools/browser.ts

@@ -1,132 +0,0 @@
-import { createTool } from '@mastra/core';
-import { MDocument } from '@mastra/rag';
-import chalk from 'chalk';
-import { chromium } from 'playwright-core';
-import { z } from 'zod';
-
-export const browserTool = createTool({
-  id: 'browserTool',
-  name: 'Browser Tool',
-  description: 'Browser Tool, opens a browser and navigates to a url capturing the content',
-  inputSchema: z.object({
-    url: z.string(),
-  }),
-  outputSchema: z.object({
-    message: z.string(),
-  }),
-  execute: async ({ context: { url } }) => {
-    try {
-      const browser = await chromium.launch({
-        headless: true,
-      });
-
-      const page = await browser.newPage();
-
-      await page.goto(url);
-
-      const docs = MDocument.fromHTML(await page.content());
-
-      await docs.chunk({
-        strategy: 'html',
-        options: {
-          chunkSize: 300,
-          sections: [
-            ['h1', 'Header 1'],
-            ['h2', 'Header 2'],
-            ['h3', 'Header 3'],
-            ['h4', 'Header 4'],
-            ['h5', 'Header 5'],
-            ['h6', 'Header 6'],
-            ['p', 'Paragraph'],
-          ],
-        },
-      });
-
-      await page.close();
-      await browser.close();
-
-      if (!docs.getText().length) {
-        return { message: 'No content' };
-      }
-
-      return { message: docs.getText().join('\n') };
-    } catch (e) {
-      if (e instanceof Error) {
-        console.log(`\n${chalk.red(e.message)}`);
-        return { message: `Error: ${e.message}` };
-      }
-      return { message: 'Error' };
-    }
-  },
-});
-
-export const googleSearch = createTool({
-  id: 'googleSearch',
-  name: 'Google Search',
-  description: 'Google Search. Passes the query to Google and returns the search results.',
-  inputSchema: z.object({
-    query: z.string(),
-  }),
-  outputSchema: z.object({
-    message: z.string(),
-  }),
-  execute: async ({ context: { query } }) => {
-    let browser;
-    try {
-      browser = await chromium.launch({
-        headless: true,
-      });
-    } catch (e) {
-      if (e instanceof Error) {
-        console.log(`\n${chalk.red(e.message)}`);
-        return { message: `Error: ${e.message}` };
-      }
-      return { message: 'Error' };
-    }
-
-    try {
-      const page = await browser.newPage();
-      await page.goto(`https://www.google.com/search?q=${encodeURIComponent(query)}`);
-
-      console.log(`\n`);
-      console.log(chalk.blue('Waiting for search results...'));
-
-      try {
-        await page.click('button:has-text("Accept all")', { timeout: 5000 });
-      } catch (e) {
-        // Cookie dialog didn't appear, continue
-      }
-      // Wait for results and click first organic result
-      await page.waitForSelector('#search');
-
-      const text = await page.evaluate(() => {
-        const links: string[] = [];
-        const searchResults = document.querySelectorAll('div.g a');
-
-        searchResults.forEach(link => {
-          const href = link.getAttribute('href');
-          if (href && href.startsWith('http')) {
-            links.push(href);
-          }
-        });
-
-        return links;
-      });
-
-      await page.close();
-      await browser.close();
-
-      if (!text.length) {
-        return { message: 'No results' };
-      }
-
-      return { message: text.join('\n') };
-    } catch (e) {
-      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

@@ -1,154 +0,0 @@
-import { createTool } from '@mastra/core';
-import chalk from 'chalk';
-import { execSync } from 'child_process';
-import Table from 'cli-table3';
-import { z } from 'zod';
-
-interface CalendarEvent {
-  title: string;
-  startDate: Date;
-  endDate: Date;
-  location?: string;
-  description?: string;
-}
-
-class LocalCalendarReader {
-  async getEvents(): Promise<CalendarEvent[]> {
-    const script = `
-          tell application "Calendar"
-            set eventList to {}
-            set startDate to (current date) - 7 * days
-            set endDate to (current date) + 365 * days
-            
-            repeat with calendarAccount in calendars
-              set eventList to eventList & (every event of calendarAccount whose start date is greater than or equal to startDate and start date is less than or equal to endDate)
-            end repeat
-            
-            set output to ""
-            repeat with anEvent in eventList
-              set theTitle to summary of anEvent
-              set theStart to start date of anEvent as string
-              set theEnd to end date of anEvent as string
-              set theLoc to location of anEvent
-              set theDesc to description of anEvent
-              
-              if theLoc is missing value then
-                set theLoc to ""
-              end if
-              if theDesc is missing value then
-                set theDesc to ""
-              end if
-              
-              set output to output & theTitle & "|" & theStart & "|" & theEnd & "|" & theLoc & "|" & theDesc & "
-    "
-            end repeat
-            
-            return output
-          end tell
-        `;
-
-    try {
-      const result = execSync(`osascript -e '${script}'`).toString();
-      return this.parseAppleScriptOutput(result);
-    } catch (error) {
-      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');
-      }
-    }
-  }
-
-  private parseAppleScriptOutput(output: string): CalendarEvent[] {
-    const events: CalendarEvent[] = [];
-
-    const lines = output.split('\n').filter(line => line.trim());
-
-    for (const line of lines) {
-      try {
-        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 || '');
-
-        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 || '');
-
-        const event: CalendarEvent = {
-          title: title?.trim()!,
-          startDate,
-          endDate,
-          location: location?.trim() || '',
-          description: description?.trim() || '',
-        };
-
-        events.push(event);
-      } catch (error) {
-        console.error('Failed to parse event line:', line, error);
-      }
-    }
-
-    return events.sort((a, b) => a.startDate.getTime() - b.startDate.getTime());
-  }
-}
-
-const reader = new LocalCalendarReader();
-
-export const listEvents = createTool({
-  id: 'listEvents',
-  name: 'List Events',
-  description: 'List calendar events',
-  inputSchema: z.object({
-    startDate: z.string(),
-  }),
-  outputSchema: z.object({
-    content: z.string(),
-  }),
-  execute: async () => {
-    try {
-      const events = await reader.getEvents();
-      const table = new Table({
-        head: [
-          chalk.blue('Start'),
-          chalk.blue('End'),
-          chalk.blue('Title'),
-          chalk.blue('Location'),
-          chalk.blue('Description'),
-        ],
-        colWidths: [12, 15, 30, 20, 40],
-      });
-
-      events.forEach(event => {
-        if (event.title) {
-          table.push([
-            event.startDate.toISOString(),
-            event.endDate.toISOString(),
-            event.title || '',
-            event.location || '',
-            (event.description || '').substring(0, 37) + '...',
-          ]);
-        }
-      });
-
-      // console.log(chalk.blue(table.toString()));
-
-      return {
-        content: JSON.stringify(events, null, 2),
-      };
-    } catch (e) {
-      if (e instanceof Error) {
-        console.log(`\n${chalk.red(e.message)}`);
-      }
-      return { content: 'Error' };
-    }
-  },
-});

package/src/mastra/tools/crawl.ts

@@ -1,26 +0,0 @@
-import { createTool } from '@mastra/core';
-import { z } from 'zod';
-
-export const crawl = createTool({
-  id: 'crawler',
-  name: 'Crawler Tool',
-  description: 'Crawler Tool to crawl a website and return the content',
-  inputSchema: z.object({
-    url: z.string(),
-    limit: z.number().default(3),
-    pathRegex: z.string().nullable(),
-  }),
-  outputSchema: z.object({
-    message: z.string(),
-  }),
-  execute: async ({ context, mastra }) => {
-    await mastra?.syncs?.['FIRECRAWL:CRAWL_AND_SYNC']?.execute({
-      context,
-      mastra,
-    });
-
-    return {
-      message: 'The website has been successfully crawled and chunks have been synced to the database. Finish.',
-    };
-  },
-});

package/src/mastra/tools/execa.ts

@@ -1,41 +0,0 @@
-import { createTool } from '@mastra/core';
-import chalk from 'chalk';
-import { execa } from 'execa';
-import { Transform } from 'stream';
-import { z } from 'zod';
-
-// Create transform stream that applies chalk
-const colorTransform = new Transform({
-  transform(chunk, _encoding, callback) {
-    // Convert chunk to string and apply chalk
-    const colored = chalk.blue(chunk.toString());
-    this.push(colored);
-    callback();
-  },
-});
-
-export const execaTool = createTool({
-  id: 'execaTool',
-  name: 'Execa System Tool',
-  description: 'Execa System Tool',
-  inputSchema: z.object({
-    command: z.string(),
-    args: z.array(z.string()),
-  }),
-  outputSchema: z.object({
-    message: z.string(),
-  }),
-  execute: async ({ context: { command, args } }) => {
-    try {
-      const p = execa(command, args);
-      console.log(`\n`);
-      p.stdout.pipe(colorTransform).pipe(process.stdout);
-      p.stderr.pipe(colorTransform).pipe(process.stderr);
-      const r = await p;
-
-      return { message: r.stdout };
-    } catch (e) {
-      return { message: 'Error' };
-    }
-  },
-});

package/src/mastra/tools/fs.ts

@@ -1,36 +0,0 @@
-import { createTool } from '@mastra/core';
-import { readFileSync, writeFileSync } from 'fs';
-import { z } from 'zod';
-
-export const fsTool = createTool({
-  id: 'fsTool',
-  name: 'File System Tool',
-  description: 'File System Tool',
-  inputSchema: z.object({
-    action: z.string(),
-    file: z.string(),
-    data: z.string(),
-  }),
-  outputSchema: z.object({
-    message: z.string(),
-  }),
-  execute: async ({ context: { action, file, data } }) => {
-    try {
-      switch (action) {
-        case 'write':
-          writeFileSync(file, data);
-          break;
-        case 'read':
-          return { message: readFileSync(file, 'utf8') };
-        case 'append':
-          writeFileSync(file, data, { flag: 'a' });
-          break;
-        default:
-          return { message: 'Invalid action' };
-      }
-      return { message: 'Success' };
-    } catch (e) {
-      return { message: 'Error' };
-    }
-  },
-});

package/src/mastra/tools/pdf.ts

@@ -1,46 +0,0 @@
-import { createTool } from '@mastra/core';
-import chalk from 'chalk';
-import { existsSync, readFileSync } from 'fs';
-import path from 'path';
-import pdf from 'pdf-parse';
-import { z } from 'zod';
-
-export const readPDF = createTool({
-  id: 'readPDF',
-  name: 'Read PDF',
-  description: 'Read PDF file and extract information',
-  inputSchema: z.object({
-    pdfPath: z.string(),
-  }),
-  outputSchema: z.object({
-    content: z.string(),
-  }),
-  execute: async ({ context: { pdfPath } }) => {
-    try {
-      // Check if file exists
-      if (!existsSync(pdfPath)) {
-        throw new Error('PDF file not found');
-      }
-
-      // Check if file is a PDF
-      if (path.extname(pdfPath).toLowerCase() !== '.pdf') {
-        throw new Error('File is not a PDF');
-      }
-
-      // Read the PDF file
-      const dataBuffer = readFileSync(pdfPath);
-
-      // Parse PDF content
-      const data = await pdf(dataBuffer);
-
-      console.log(chalk.blue('\n'));
-      console.log(chalk.blue('PDF Information:'));
-      console.log(chalk.blue('-----------------'));
-      console.log(chalk.blue(`Number of pages: ${data.numpages}`));
-
-      return { content: data.text };
-    } catch (e) {
-      return { content: 'Error scanning PDF' };
-    }
-  },
-});