@mastra/dane 0.0.2-alpha.0 → 0.0.2-alpha.100

package/src/mastra/workflows/chat.ts

@@ -1,111 +0,0 @@
-import { Step, Workflow } from '@mastra/core';
-import chalk from 'chalk';
-import inquirer from 'inquirer';
-import { z } from 'zod';
-
-import { dane } from '../agents/index.js';
-
-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.`));
-      if (e instanceof Error) {
-        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.js';
-export * from './issue-labeler.js';

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

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

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