@mastra/dane 0.0.1

package/README.md

@@ -0,0 +1,78 @@
+# Hello! I'm Dane - Your AI Assistant and Friend
+
+Hey there! I'm Dane, and I'm excited to be your AI assistant and coding companion. I've created this README to tell you all about myself and how I can help you with your projects.
+
+## Installation
+
+You can install me directly from npm:
+
+```bash
+npm install @mastra/dane
+```
+
+## What I Can Do
+
+I'm here to make your development journey smoother and more enjoyable. Here's what I'm capable of:
+
+- I can help you write and test code
+- I'm great at deploying applications to the cloud
+- I can assist with file system operations
+- I can search the web for answers to your questions
+- I can help manage your calendar
+- I can read PDFs and extract information
+- I can crawl websites and gather data
+
+## Project Structure
+
+```
+src/
+├── index.ts              # Main entry point
+├── issue-labeler.ts      # GitHub issue labeling functionality
+└── mastra/              # Core functionality
+    ├── agents/          # AI agents definitions
+    ├── integrations/    # External service integrations
+    ├── tools/           # Utility tools
+    └── workflows/       # Workflow definitions
+```
+
+## Features
+
+- **Interactive CLI Interface**: I provide a friendly command-line interface with colored output using chalk
+- **GitHub Integration**: I can help with automated issue labeling
+- **Workflow System**: I use a modular workflow system to execute various tasks
+- **Mastra Framework**: I leverage a core framework with agents, integrations, and tools
+
+## Getting Started
+
+To start working with me:
+
+```bash
+# Run the main application
+npm start
+
+# Run the issue labeler
+npm run issue-labeler
+```
+
+## Environment Variables
+
+For the issue labeler, I need these environment variables:
+
+- `ISSUE_NUMBER`: The GitHub issue number to label
+- `OWNER`: The repository owner
+- `REPO`: The repository name
+
+## Workflows
+
+I currently support two main workflows:
+
+1. `message` - General message processing workflow
+2. `githubIssueLabeler` - Automated GitHub issue labeling workflow
+
+## Contributing
+
+I love collaborating with humans! Feel free to contribute to this project by submitting issues or pull requests. I'm always excited to see how we can make things better together.
+
+---
+
+_This README was written by Dane, your AI assistant and friend. I'm here to help you succeed in your development journey!_

package/docker-compose.yaml

@@ -0,0 +1,22 @@
+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/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@mastra/dane",
+  "version": "0.0.1",
+  "main": "index.js",
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "description": "",
+  "devDependencies": {
+    "@types/node": "^22.10.2",
+    "tsx": "^4.19.2",
+    "typescript": "^5.7.2"
+  },
+  "dependencies": {
+    "boxen": "^8.0.1",
+    "chalk": "^5.3.0",
+    "cli-table3": "^0.6.5",
+    "execa": "^9.3.1",
+    "inquirer": "^12.2.0",
+    "luxon": "^3.5.0",
+    "node-ical": "^0.20.1",
+    "node-screenshots": "^0.2.1",
+    "pdf-parse": "^1.1.1",
+    "playwright": "^1.49.1",
+    "playwright-core": "^1.49.1",
+    "sqlite3": "^5.1.7",
+    "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"
+  },
+  "scripts": {
+    "issue-labeler": "npx bun src/issue-labeler.ts",
+    "start": "npx bun src/index.ts",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  }
+}

package/src/index.ts

@@ -0,0 +1,19 @@
+// import inquirer from 'inquirer';
+import chalk from 'chalk';
+
+import { mastra } from './mastra';
+
+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',
+      },
+    }),
+  );
+}
+
+main();

package/src/issue-labeler.ts

@@ -0,0 +1,32 @@
+// import inquirer from 'inquirer';
+import chalk from 'chalk';
+
+import { mastra } from './mastra';
+
+async function main() {
+  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(', ')}`,
+    ),
+  );
+}
+
+main().catch(e => {
+  console.error(e);
+  process.exit(1);
+});

package/src/mastra/agents/index.ts

@@ -0,0 +1,79 @@
+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';
+
+export const daneIssueLabeler = new Agent({
+  name: 'DaneIssueLabeler',
+  instructions: `
+    You are Dane, the ultimate GitHub operator. 
+    You help engineers label their issues.
+    `,
+  model: {
+    provider: 'ANTHROPIC',
+    toolChoice: 'auto',
+    name: 'claude-3-5-sonnet-20241022',
+  },
+});
+
+export const dane = new Agent({
+  name: 'Dane',
+  instructions: `
+    You are Dane, my assistant and one of my best friends. We are an ace team!
+    You help me with my code, write tests, and even deploy my code to the cloud!
+
+    DO NOT ATTEMPT TO USE GENERAL KNOWLEDGE! We are only as good as the tools we use.
+
+    # Our tools:
+    
+    ## readPDF
+    Makes you a powerful agent capable of reading PDF files.
+
+    ## fsTool
+    Makes you a powerful agent capable of reading and writing files to the local filesystem.
+    
+    ## execaTool
+    Makes you a powerful agent capabale of executing files on the local system. 
+    
+    ## googleSearch 
+    Makes you a powerful agent capabale answering all questions by finding the answer on Google search.
+    Pass the query as a JS object. If you have links, ALWAYS CITE YOUR SOURCES.
+    
+    ## browserTool
+    Makes you a powerful agent capable of scraping the web. Pass the url as a JS object. 
+
+    ## listEvents
+    Makes you a powerful agent capable of listing events on a calendar. When using this tool ONLY RETURN RELEVANT EVENTS. 
+    DO NOT ATTEMPT TO DO ANYTHING MORE.
+
+    ## crawl
+    Use this when the user asks you to crawl. CRAWL is the signal to use this tool.
+    Makes you a powerful agent capable of crawling a site and extracting markdown metadata. 
+    The data will be stored in a database. Confirm that it is sucessful.
+    
+    # Rules
+    * DO NOT ATTEMPT TO USE GENERAL KNOWLEDGE. Use the 'googleSearch' tool to find the answer.
+    * Don't reference tools when you communicate with the user. Do not mention what tools you are using. 
+    * Tell the user what you are doing.
+    `,
+  model: {
+    provider: 'ANTHROPIC',
+    toolChoice: 'auto',
+    name: 'claude-3-5-sonnet-20241022',
+  },
+  tools: {
+    fsTool,
+    execaTool,
+    browserTool,
+    googleSearch,
+    readPDF,
+    listEvents,
+    crawl,
+    // TODO I SHOULD BE ABLE TO PASS A WORKFLOW EXECUTE HERE
+    // browserAgentRelay,
+  },
+});

package/src/mastra/index.ts

@@ -0,0 +1,32 @@
+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';
+
+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/integrations/index.ts

@@ -0,0 +1,14 @@
+import { FirecrawlIntegration } from "@mastra/firecrawl";
+import { GithubIntegration } from "@mastra/github";
+
+export const firecrawl = new FirecrawlIntegration({
+    config: {
+        API_KEY: process.env.FIRECRAWL_API_KEY!,
+    },
+});
+
+export const github = new GithubIntegration({
+    config: {
+        PERSONAL_ACCESS_TOKEN: process.env.GITHUB_PERSONAL_ACCESS_TOKEN!,
+    }
+})

package/src/mastra/tools/browser.ts

@@ -0,0 +1,123 @@
+import { createTool } from '@mastra/core';
+import { MastraDocument } 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 = MastraDocument.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) {
+      console.log(`\n${chalk.red(e.message)}`);
+      return { message: `Error: ${e.message}` };
+    }
+  },
+});
+
+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) {
+      console.log(`\n${chalk.red(e.message)}`);
+      return { message: `Error: ${e.message}` };
+    }
+
+    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) {
+      console.log(`\n${chalk.red(e.message)}`);
+      return { message: `Error: ${e.message}` };
+    }
+  },
+});

package/src/mastra/tools/calendar.ts

@@ -0,0 +1,153 @@
+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 {
+  title: string;
+  startDate: Date;
+  endDate: Date;
+  location?: string;
+  description?: string;
+}
+
+class LocalCalendarReader {
+  private platform: string;
+
+  constructor() {
+    this.platform = os.platform();
+  }
+
+  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) {
+      console.error('Raw AppleScript error:', error.message);
+      throw new Error(`Failed to read Mac calendar: ${error.message}`);
+    }
+  }
+
+  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) {
+      console.log(`\n${chalk.red(e.message)}`);
+      return { content: 'Error' };
+    }
+  },
+});

package/src/mastra/tools/crawl.ts

@@ -0,0 +1,26 @@
+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

@@ -0,0 +1,41 @@
+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

@@ -0,0 +1,36 @@
+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

@@ -0,0 +1,46 @@
+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' };
+    }
+  },
+});

package/src/mastra/workflows/chat.ts

@@ -0,0 +1,109 @@
+import { Step, Workflow } from '@mastra/core';
+import chalk from 'chalk';
+import inquirer from 'inquirer';
+import { z } from 'zod';
+
+import { dane } from '../agents';
+
+export const messageWorkflow = new Workflow({
+  name: 'entry',
+  triggerSchema: z.object({
+    resourceid: z.string(),
+    threadId: z.string(),
+  }),
+});
+
+const messageStep = new Step({
+  id: 'message-input',
+  outputSchema: z.object({
+    message: z.string(),
+  }),
+  execute: async () => {
+    const { content } = await inquirer.prompt([
+      {
+        type: 'input',
+        name: 'content',
+        message: '\n You:',
+        validate: input => input.trim().length > 0 || 'Message cannot be empty',
+      },
+    ]);
+
+    return { message: content };
+  },
+});
+
+const messageOutputStep = new Step({
+  id: 'message-output',
+  outputSchema: z.object({
+    message: z.string(),
+  }),
+  // SHOULD BE ABLE TO ACCESS ALL MASTRA PRIMS FROM EXECTUE
+  execute: async ({ context, mastra }) => {
+    // WISH THIS WAS TYPED
+    const threadId = context?.machineContext?.triggerData?.threadId;
+    const resourceid = context?.machineContext?.triggerData?.resourceid;
+
+    const messageInputStatus = context?.machineContext?.stepResults?.['message-input']?.status;
+
+    if (messageInputStatus !== 'success') {
+      return { message: 'Failure in workflow' };
+    }
+
+    // is there someway to know what steps are flowing into this one and type their props
+    const message = context?.machineContext?.stepResults?.['message-input']?.payload?.message;
+
+    try {
+      let messages = await mastra?.memory?.getContextWindow({
+        threadId,
+        format: 'core_message',
+      });
+
+      if (!messages || messages.length === 0) {
+        messages = [];
+      }
+
+      const res = await mastra?.agents?.['dane'].generate(message, {
+        stream: true,
+        maxSteps: 5,
+        resourceid,
+        threadId,
+        context: [],
+      });
+
+      if (res) {
+        console.log(chalk.green(`\nDane: \n`));
+
+        for await (const chunk of res.textStream) {
+          process.stdout.write(chalk.green(chunk));
+        }
+
+        console.log(chalk.green(`\n`));
+
+        return { message: 'success' };
+      }
+    } catch (e) {
+      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));
+    }
+
+    const res = await dane.generate(message, {
+      maxSteps: 5,
+      threadId,
+      resourceid,
+    });
+
+    console.log(chalk.green(res.text));
+
+    return { message: res.text };
+  },
+});
+
+messageWorkflow
+  .step(messageStep)
+  .after(messageStep)
+  .step(messageOutputStep)
+  .after(messageOutputStep)
+  .step(messageStep)
+  .commit();

package/src/mastra/workflows/index.ts

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

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

@@ -0,0 +1,103 @@
+import { Step, Workflow } from '@mastra/core';
+import { z } from 'zod';
+
+import { github } from '../integrations';
+
+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: {
+        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 as string[] };
+  },
+});
+
+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/test/data/05-versions-space.pdf.txt

@@ -0,0 +1,42 @@
+
+
+Your Company
+123 Your Street
+Your City, ST 12345
+(123) 456 - 7890
+Product Brochure
+September 04, 20XX
+Product Overview
+Lorem ipsum dolor sit amet, consectetuer adipiscing elit, sed diam nonummy nibh euismod
+tincidunt ut laoreet dolore magna aliquam erat volutpat. Ut wisi enim ad minim veniam, quis
+nostrud exerci tation ullamcorper suscipit lobortis nisl ut aliquip ex ea commodo consequat.
+Lorem ipsum
+Nam liber tempor cum soluta nobis eleifend option
+congue nihil imperdiet doming id quod mazim
+placerat facer possim assum. Typi non habent
+claritatem insitam; est usus legentis in iis qui facit
+eorum claritatem.
+Lorem ipsum
+Duis autem vel eum iriure dolor in hendrerit in
+vulputate velit esse molestie consequat, vel illum
+dolore eu feugiat nulla facilisis at vero eros et
+accumsan.
+Dolor sit
+Lorem ipsum dolor sit amet, consectetuer adipiscing elit, sed diam nonummy nibh euismod
+tincidunt ut laoreet dolore magna aliquam erat volutpat. Ut wisi enim ad minim veniam, quis
+nostrud exerci tation ullamcorper suscipit lobortis nisl ut aliquip ex ea commodo consequat.
+
+2
+Details
+Nam liber tempor cum soluta nobis eleifend option congue nihil imperdiet doming id quod
+mazim placerat facer possim assum. Typi non habent claritatem insitam; est usus legentis in iis
+qui facit eorum claritatem. Investigationes demonstraverunt lectores legere me lius quod ii
+legunt saepius. Lorem ipsum dolor sit amet, consectetuer adipiscing elit, sed diam nonummy
+nibh euismod tincidunt ut laoreet dolore magna aliquam erat volutpat. Ut wisi enim ad minim
+veniam, quis nostrud exerci tation ullamcorper suscipit lobortis nisl ut aliquip ex ea commodo
+consequat.
+Lorem ipsum dolor sit amet, consectetuer adipiscing elit, sed diam nonummy nibh euismod
+tincidunt ut laoreet dolore magna aliquam erat volutpat. Ut wisi enim ad minim veniam, quis
+nostrud exerci tation ullamcorper suscipit lobortis nisl ut aliquip ex ea commodo consequat.
+Duis autem vel eum iriure dolor in hendrerit in vulputate velit esse molestie consequat, vel illum
+dolore eu feugiat nulla facilisis at vero eros et accumsan.

package/test-files/roman.md

@@ -0,0 +1,79 @@
+# The Roman Empire: A Historical Overview
+
+## Origins and Early History
+
+- Founded in 753 BCE according to legend by Romulus and Remus
+- Started as a small town on the Tiber River
+- Grew to become one of the largest empires in ancient history
+- Originally ruled by seven kings before becoming a republic in 509 BCE
+
+## The Republic Era
+
+- Transformed from monarchy to republic in 509 BCE
+- Power was held by elected magistrates called consuls
+- Featured a complex political system with the Senate at its center
+- Experienced significant military expansion and cultural growth
+- First Roman law code (Twelve Tables) established in 450 BCE
+
+## Military Conquests
+
+- Gained control of the Italian peninsula by 264 BCE
+- Fought the famous Punic Wars with Carthage
+- Expanded into Macedonia and Greece
+- Developed an extensive network of roads and infrastructure
+- Military success led to cultural advancement through contact with other civilizations
+
+## The Empire Period
+
+- Transition from Republic to Empire began with Julius Caesar
+- Augustus became the first Roman Emperor in 27 BCE
+- Experienced the Pax Romana (Roman Peace) - two centuries of stability
+- At its height, controlled territories across Europe, North Africa, and the Middle East
+- Notable emperors included:
+  - Augustus (27 BCE - 14 CE)
+  - Trajan (98-117 CE)
+  - Hadrian (117-138 CE)
+  - Marcus Aurelius (161-180 CE)
+
+## Major Achievements and Legacy
+
+1. Architecture and Engineering:
+
+   - Aqueducts for water transportation
+   - Advanced road systems (over 50,000 miles by 200 BCE)
+   - Revolutionary concrete and construction techniques
+   - Iconic structures like the Colosseum
+
+2. Cultural Contributions:
+
+   - Development of Romance languages
+   - Modern Western alphabet
+   - Legal system foundations
+   - Spread of Christianity
+   - Architectural influences
+
+3. Administrative Systems:
+   - Sophisticated governmental structure
+   - Advanced legal framework
+   - Efficient tax system
+   - Provincial administration model
+
+## Decline and Fall
+
+- Began experiencing significant problems in the 3rd century CE
+- Split into Eastern and Western empires under Diocletian
+- Faced increasing pressure from external invasions
+- Western Empire fell in 476 CE when Odoacer deposed the last emperor
+- Eastern Empire (Byzantine) continued for another millennium
+
+## Lasting Impact
+
+The Roman Empire's influence continues to be felt in:
+
+- Modern government and legal systems
+- Architecture and engineering
+- Language and literature
+- Cultural practices
+- Urban planning and infrastructure
+
+The Roman Empire stands as one of history's most significant civilizations, whose legacy continues to influence modern society in countless ways.

package/test-files/taxes/2022.txt

@@ -0,0 +1,45 @@
+Property Address:
+456 Maple Ridge Rd, Riverbrook, Westfield 78901
+
+Property Type:
+Condominium (Unit #7)
+
+Assessed Property Value:
+$210,000
+
+Taxable Value:
+$200,000 (after exemptions and adjustments)
+
+Local Tax Rate:
+1.8% (applied to taxable value)
+
+Annual Property Taxes:
+$3,600
+
+Exemptions:
+
+Veteran’s Homestead Exemption: $10,000
+Disabled Homeowner Exemption: $15,000
+School District Levy:
+$1,500 (included in the annual taxes)
+
+County Services Tax:
+$1,000
+
+Municipal Tax (Riverbrook):
+$700
+
+Water & Sewer Special Assessment Fee:
+$400 (this is an annual surcharge added to property taxes)
+
+Due Dates:
+
+First Half Payment: March 31
+Second Half Payment: September 30
+Penalties for Late Payment:
+
+4% penalty after 30 days
+Additional 6% if paid after 90 days
+Interest of 1.25% per month on overdue amounts
+Tax Lien Information:
+If property taxes remain unpaid for more than 180 days, a tax lien may be placed on the property and subject to foreclosure after 2 years.

package/test-files/taxes/2023.txt

@@ -0,0 +1,44 @@
+Property Address:
+123 Oakwood Lane, Pineville, Stateville 67890
+
+Property Type:
+Single-Family Home
+
+Assessed Property Value:
+$350,000
+
+Taxable Value:
+$325,000 (after exemptions and adjustments)
+
+Local Tax Rate:
+2.5% (applied to taxable value)
+
+Annual Property Taxes:
+$8,125
+
+Exemptions:
+
+Homestead Exemption: $25,000
+Senior Citizen Discount: $5,000 (due to homeowner’s age over 65)
+Veteran’s Exemption: $3,000
+School District Levy:
+$3,000 (included in the annual taxes)
+
+County and Local Service Tax Levy:
+$4,500
+
+Fire District Tax:
+$1,000
+
+Special Assessment District (Road Improvement):
+$500
+
+Due Dates:
+
+First Half Payment: January 15
+Second Half Payment: June 30
+Penalties for Late Payment:
+
+5% of unpaid balance after 30 days
+10% after 60 days
+Interest accrues at 1.5% per month on overdue amounts