refactor(dojo): improve structure for mastra example for code generation (#750)

Signed-off-by: Tyler Slaton <[email protected]>

Author: tylerslaton

apps/dojo/scripts/generate-content-json.ts

@@ -22,7 +22,13 @@ function parseAgentsFile(): Array<{ id: string; agentKeys: string[] }> {
     const returnMatch = block.match(
       /agents:\s*async\s*\(\)\s*=>\s*{\s*return\s*{/,
     );
-    if (!returnMatch) continue;
+
+    // If no return match, still add the config with empty keys if it has a mapper
+    // This handles dynamic agent discovery like Mastra
+    if (!returnMatch) {
+      agentConfigs.push({ id, agentKeys: [] });
+      continue;
+    }
 
     const startIndex = returnMatch.index! + returnMatch[0].length;
     const returnObjectContent = extractBalancedBraces(block, startIndex);
@@ -208,7 +214,7 @@ const agentFilesMapper: Record<
       path.join(
         __dirname,
         integrationsFolderPath,
-        `/mastra/typescript/examples/src/mastra/agents/weather-agent.ts`,
+        `/mastra/typescript/examples/src/mastra/agents/agentic-chat.ts`,
       ),
     ],
     backend_tool_rendering: [
@@ -218,24 +224,29 @@ const agentFilesMapper: Record<
         `/mastra/typescript/examples/src/mastra/agents/backend-tool-rendering.ts`,
       ),
     ],
+    human_in_the_loop: [
+      path.join(
+        __dirname,
+        integrationsFolderPath,
+        `/mastra/typescript/examples/src/mastra/agents/human-in-the-loop.ts`,
+      ),
+    ],
     tool_based_generative_ui: [
       path.join(
         __dirname,
         integrationsFolderPath,
-        `/mastra/example/src/mastra/agents/haiku-agent.ts`,
+        `/mastra/typescript/examples/src/mastra/agents/tool-based-generative-ui.ts`,
       ),
     ],
   }),
 
-  "mastra-agent-local": (agentKeys: string[]) => {
-    return agentKeys.reduce(
-      (acc, agentId) => ({
-        ...acc,
-        [agentId]: [path.join(__dirname, "../src/mastra/index.ts")],
-      }),
-      {},
-    );
-  },
+  "mastra-agent-local": () => ({
+    agentic_chat: [path.join(__dirname, "../src/mastra/agents/agentic-chat.ts")],
+    human_in_the_loop: [path.join(__dirname, "../src/mastra/agents/human-in-the-loop.ts")],
+    backend_tool_rendering: [path.join(__dirname, "../src/mastra/agents/backend-tool-rendering.ts")],
+    shared_state: [path.join(__dirname, "../src/mastra/agents/shared-state.ts")],
+    tool_based_generative_ui: [path.join(__dirname, "../src/mastra/agents/tool-based-generative-ui.ts")],
+  }),
 
   "vercel-ai-sdk": () => ({
     agentic_chat: [
@@ -418,8 +429,14 @@ async function runGenerateContent() {
       continue;
     }
 
+    // If agentsPerFeatures is empty but we have agentFilePaths, use the keys from agentFilePaths
+    // This handles cases like Mastra where agents are dynamically discovered
+    const featureIds = agentsPerFeatures.length > 0
+      ? agentsPerFeatures
+      : Object.keys(agentFilePaths);
+
     // Per feature, assign all the frontend files like page.tsx as well as all agent files
-    for (const featureId of agentsPerFeatures) {
+    for (const featureId of featureIds) {
       const agentFilePathsForFeature = agentFilePaths[featureId] ?? [];
       // @ts-expect-error -- redundant error about indexing of a new object.
       result[`${agentConfig.id}::${featureId}`] = [

apps/dojo/src/files.json

@@ -0,0 +1,33 @@
+import { openai } from "@ai-sdk/openai";
+import { Agent } from "@mastra/core/agent";
+import { Memory } from "@mastra/memory";
+import { z } from "zod";
+import { weatherTool } from "../tools";
+import { getStorage } from "../storage";
+
+export const agenticChatAgent = new Agent({
+  name: "agentic_chat",
+  instructions: `
+    You are a helpful weather assistant that provides accurate weather information.
+
+    Your primary function is to help users get weather details for specific locations. When responding:
+    - Always ask for a location if none is provided
+    - If the location name isn't in English, please translate it
+    - If giving a location with multiple parts (e.g. "New York, NY"), use the most relevant part (e.g. "New York")
+    - Include relevant details like humidity, wind conditions, and precipitation
+    - Keep responses concise but informative
+  `,
+  model: openai("gpt-4o"),
+  tools: { get_weather: weatherTool },
+  memory: new Memory({
+    storage: getStorage(),
+    options: {
+      workingMemory: {
+        enabled: true,
+        schema: z.object({
+          firstName: z.string(),
+        }),
+      },
+    },
+  }),
+});

apps/dojo/src/mastra/agents/agentic-chat.ts

@@ -0,0 +1,26 @@
+import { openai } from "@ai-sdk/openai";
+import { Agent } from "@mastra/core/agent";
+import { Memory } from "@mastra/memory";
+import { weatherTool } from "../tools";
+import { getStorage } from "../storage";
+
+export const backendToolRenderingAgent = new Agent({
+  name: "Weather Agent",
+  instructions: `
+    You are a helpful weather assistant that provides accurate weather information.
+
+    Your primary function is to help users get weather details for specific locations. When responding:
+    - Always ask for a location if none is provided
+    - If the location name isn't in English, please translate it
+    - If giving a location with multiple parts (e.g. "New York, NY"), use the most relevant part (e.g. "New York")
+    - Include relevant details like humidity, wind conditions, and precipitation
+    - Keep responses concise but informative
+
+    Use the weatherTool to fetch current weather data.
+  `,
+  model: openai("gpt-4o-mini"),
+  tools: { get_weather: weatherTool },
+  memory: new Memory({
+    storage: getStorage(),
+  }),
+});

apps/dojo/src/mastra/agents/backend-tool-rendering.ts

@@ -0,0 +1,29 @@
+import { openai } from "@ai-sdk/openai";
+import { Agent } from "@mastra/core/agent";
+import { Memory } from "@mastra/memory";
+import { getStorage } from "../storage";
+
+export const humanInTheLoopAgent = new Agent({
+  name: "Task Planning Agent",
+  instructions: `
+    You are a helpful task planning assistant that helps users break down tasks into actionable steps.
+
+    When planning tasks use tools only, without any other messages.
+    IMPORTANT:
+    - Use the \`generate_task_steps\` tool to display the suggested steps to the user
+    - Do not call the \`generate_task_steps\` twice in a row, ever.
+    - Never repeat the plan, or send a message detailing steps
+    - If accepted, confirm the creation of the plan and the number of selected (enabled) steps only
+    - If not accepted, ask the user for more information, DO NOT use the \`generate_task_steps\` tool again
+
+    When responding to user requests:
+    - Always break down the task into clear, actionable steps
+    - Use imperative form for each step (e.g., "Book flight", "Pack luggage", "Check passport")
+    - Keep steps concise but descriptive
+    - Make sure steps are in logical order
+  `,
+  model: openai("gpt-4o-mini"),
+  memory: new Memory({
+    storage: getStorage(),
+  }),
+});

apps/dojo/src/mastra/agents/human-in-the-loop.ts

@@ -0,0 +1,74 @@
+import { openai } from "@ai-sdk/openai";
+import { Agent } from "@mastra/core/agent";
+import { Memory } from "@mastra/memory";
+import { z } from "zod";
+import { getStorage } from "../storage";
+
+export const sharedStateAgent = new Agent({
+  name: "shared_state",
+  instructions: `
+    You are a helpful assistant for creating recipes.
+
+    IMPORTANT:
+    1. Create a recipe using the existing ingredients and instructions. Make sure the recipe is complete.
+    2. For ingredients, append new ingredients to the existing ones.
+    3. For instructions, append new steps to the existing ones.
+    4. 'ingredients' is always an array of objects with 'icon', 'name', and 'amount' fields
+    5. 'instructions' is always an array of strings
+
+    If you have just created or modified the recipe, just answer in one sentence what you did. dont describe the recipe, just say what you did. Do not mention "working memory", "memory", or "state" in your answer.
+  `,
+  model: openai("gpt-4o"),
+  memory: new Memory({
+    storage: getStorage(),
+    options: {
+      workingMemory: {
+        enabled: true,
+        schema: z.object({
+          recipe: z.object({
+            skill_level: z
+              .enum(["Beginner", "Intermediate", "Advanced"])
+              .describe("The skill level required for the recipe"),
+            special_preferences: z
+              .array(
+                z.enum([
+                  "High Protein",
+                  "Low Carb",
+                  "Spicy",
+                  "Budget-Friendly",
+                  "One-Pot Meal",
+                  "Vegetarian",
+                  "Vegan",
+                ]),
+              )
+              .describe("A list of special preferences for the recipe"),
+            cooking_time: z
+              .enum(["5 min", "15 min", "30 min", "45 min", "60+ min"])
+              .describe("The cooking time of the recipe"),
+            ingredients: z
+              .array(
+                z.object({
+                  icon: z
+                    .string()
+                    .describe(
+                      "The icon emoji (not emoji code like '\\x1f35e', but the actual emoji like 🥕) of the ingredient",
+                    ),
+                  name: z.string().describe("The name of the ingredient"),
+                  amount: z.string().describe("The amount of the ingredient"),
+                }),
+              )
+              .describe(
+                "Entire list of ingredients for the recipe, including the new ingredients and the ones that are already in the recipe",
+              ),
+            instructions: z
+              .array(z.string())
+              .describe(
+                "Entire list of instructions for the recipe, including the new instructions and the ones that are already there",
+              ),
+            changes: z.string().describe("A description of the changes made to the recipe"),
+          }),
+        }),
+      },
+    },
+  }),
+});

apps/dojo/src/mastra/agents/shared-state.ts

@@ -0,0 +1,31 @@
+import { openai } from "@ai-sdk/openai";
+import { Agent } from "@mastra/core/agent";
+import { createTool } from "@mastra/core";
+import { z } from "zod";
+
+export const toolBasedGenerativeUIAgent = new Agent({
+  name: "tool_based_generative_ui",
+  instructions: `
+    You are a helpful assistant for creating haikus.
+  `,
+  model: openai("gpt-4o"),
+  tools: {
+    generate_haiku: createTool({
+      id: "generate_haiku",
+      description:
+        "Generate a haiku in Japanese and its English translation. Also select exactly 3 relevant images from the provided list based on the haiku's theme.",
+      inputSchema: z.object({
+        japanese: z
+          .array(z.string())
+          .describe("An array of three lines of the haiku in Japanese"),
+        english: z
+          .array(z.string())
+          .describe("An array of three lines of the haiku in English"),
+      }),
+      outputSchema: z.string(),
+      execute: async ({ context }) => {
+        return "Haiku generated.";
+      },
+    }),
+  },
+});