@mastra/dane 0.0.3 → 0.0.4

package/dist/mastra/workflows/telephone-game.js

@@ -0,0 +1,97 @@
+import { anthropic } from '@ai-sdk/anthropic';
+import { input } from '@inquirer/prompts';
+import { Agent } from '@mastra/core/agent';
+import { Step, Workflow, getStepResult } from '@mastra/core/workflows';
+import { z } from 'zod';
+const llm = anthropic('claude-3-5-sonnet-20241022');
+const agent = new Agent({
+    name: 'telephoneGameAgent',
+    instructions: `Telephone game agent`,
+    model: llm,
+});
+export const telephoneGameWorkflow = new Workflow({
+    name: 'telephoneGame',
+});
+const stepA1 = new Step({
+    id: 'stepA1',
+    description: 'Starts the message',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async () => {
+        return {
+            message: 'Test',
+        };
+    },
+});
+const stepA2 = new Step({
+    id: 'stepA2',
+    description: 'Pass the message through',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async () => {
+        const content = await input({
+            message: 'Give me a message',
+            validate: input => input.trim().length > 0 || 'Message cannot be empty',
+        });
+        return {
+            message: content,
+        };
+    },
+});
+const stepB2 = new Step({
+    id: 'stepB2',
+    description: 'Checks if the file exists',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async ({ context }) => {
+        if (context?.steps.stepA2?.status !== 'success') {
+            throw new Error('Message not found');
+        }
+        const msg = context.steps.stepA2.output.message;
+        return {
+            message: msg,
+        };
+    },
+});
+const stepC2 = new Step({
+    id: 'stepC2',
+    description: 'Ask if you should modify the message',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async ({ suspend, context }) => {
+        const oMsg = getStepResult(context?.steps.stepA2);
+        if (context?.steps.stepC2?.status === 'success') {
+            const msg = getStepResult(context?.steps.stepC2);
+            if (msg.confirm) {
+                const result = await agent.generate(`
+            You are playing a game of telephone.
+            Here is the message the previous person sent ${oMsg.message}.
+            But you want to change the message.
+            Only return the message
+            `);
+                return {
+                    message: result.text,
+                };
+            }
+            return oMsg;
+        }
+        await suspend();
+        return { message: 'Suspended' };
+    },
+});
+const stepD2 = new Step({
+    id: 'stepD2',
+    description: 'Pass the message',
+    outputSchema: z.object({
+        message: z.string(),
+    }),
+    execute: async ({ context }) => {
+        const msg = getStepResult(context?.steps.stepC2);
+        return msg;
+    },
+});
+telephoneGameWorkflow.step(stepA1).step(stepA2).then(stepB2).then(stepC2).then(stepD2).commit();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/dane",
-  "version": "0.0.3",
+  "version": "0.0.4",
   "main": "dist/index.js",
   "type": "module",
   "files": [
@@ -42,13 +42,13 @@
     "sqlite3": "^5.1.7",
     "typescript": "^5.7.3",
     "zod": "^3.24.0",
+    "@mastra/core": "0.4.4",
+    "@mastra/memory": "0.1.7",
     "@mastra/github": "1.1.7",
     "@mastra/mcp": "0.2.6",
-    "@mastra/core": "0.4.4",
     "@mastra/rag": "0.1.7",
     "@mastra/stabilityai": "1.1.7",
-    "@mastra/upstash": "0.1.7",
-    "@mastra/memory": "0.1.7"
+    "@mastra/upstash": "0.1.7"
   },
   "scripts": {
     "build": "npx tsc",