@mastra/dane 0.0.1 → 0.0.2-alpha.10

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' };
-    }
-  },
-});

package/src/mastra/workflows/chat.ts

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

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

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

@@ -1,103 +0,0 @@
-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();