@mastra/agent-builder 0.0.0-main-test-05-11-2025-2-20251106025330 → 0.0.0-main-test-2-20251127211532

package/dist/index.js

@@ -1,6 +1,5 @@
 import { Agent, tryGenerateWithJsonFallback, tryStreamWithJsonFallback } from '@mastra/core/agent';
 import { Memory } from '@mastra/memory';
-import { TokenLimiter } from '@mastra/memory/processors';
 import { exec as exec$1, execFile as execFile$1, spawn as spawn$1 } from 'child_process';
 import { mkdtemp, rm, readFile, writeFile, readdir, mkdir, copyFile, stat } from 'fs/promises';
 import { join, resolve, basename, extname, dirname, isAbsolute, relative } from 'path';
@@ -11,7 +10,6 @@ import { existsSync, readFileSync } from 'fs';
 import { createRequire } from 'module';
 import { promisify } from 'util';
 import { ModelRouterLanguageModel } from '@mastra/core/llm';
-import { MemoryProcessor } from '@mastra/core/memory';
 import { tmpdir } from 'os';
 import { openai } from '@ai-sdk/openai';
 import { createStep, createWorkflow } from '@mastra/core/workflows';
@@ -913,6 +911,7 @@ export const weatherAgent = new Agent({
   tools: { weatherTool },
   memory: new Memory({
     storage: new LibSQLStore({
+      id: 'mastra-memory-storage',
       url: 'file:../mastra.db', // ask user what database to use, use this as the default
     }),
   }),
@@ -1080,6 +1079,7 @@ export const mastra = new Mastra({
   workflows: { weatherWorkflow },
   agents: { weatherAgent },
   storage: new LibSQLStore({
+    id: 'mastra-storage',
     // stores observability, evals, ... into memory storage, if it needs to persist, change to file:../mastra.db
     url: ":memory:",
   }),
@@ -3082,11 +3082,12 @@ export const mastra = new Mastra({
     }
   }
 };
-var ToolSummaryProcessor = class extends MemoryProcessor {
+var ToolSummaryProcessor = class {
+  id = "tool-summary-processor";
+  name = "ToolSummaryProcessor";
   summaryAgent;
   summaryCache = /* @__PURE__ */ new Map();
   constructor({ summaryModel }) {
-    super({ name: "ToolSummaryProcessor" });
     this.summaryAgent = new Agent({
       id: "tool-summary-agent",
       name: "Tool Summary Agent",
@@ -3123,30 +3124,37 @@ var ToolSummaryProcessor = class extends MemoryProcessor {
       keys: Array.from(this.summaryCache.keys())
     };
   }
-  async process(messages) {
+  async processInput({
+    messages,
+    messageList: _messageList
+  }) {
     const summaryTasks = [];
     for (const message of messages) {
-      if (message.role === "tool" && Array.isArray(message.content) && message.content.length > 0 && message.content?.some((content) => content.type === "tool-result")) {
-        for (const content of message.content) {
-          if (content.type === "tool-result") {
-            const assistantMessageWithToolCall = messages.find(
-              (message2) => message2.role === "assistant" && Array.isArray(message2.content) && message2.content.length > 0 && message2.content?.some(
-                (assistantContent) => assistantContent.type === "tool-call" && assistantContent.toolCallId === content.toolCallId
-              )
-            );
-            const toolCall = Array.isArray(assistantMessageWithToolCall?.content) ? assistantMessageWithToolCall?.content.find(
-              (assistantContent) => assistantContent.type === "tool-call" && assistantContent.toolCallId === content.toolCallId
-            ) : null;
-            const cacheKey = this.createCacheKey(toolCall);
+      if (message.content.format === 2 && message.content.parts) {
+        for (let partIndex = 0; partIndex < message.content.parts.length; partIndex++) {
+          const part = message.content.parts[partIndex];
+          if (part && part.type === "tool-invocation" && part.toolInvocation?.state === "result") {
+            const cacheKey = this.createCacheKey(part.toolInvocation);
             const cachedSummary = this.summaryCache.get(cacheKey);
             if (cachedSummary) {
-              content.result = `Tool call summary: ${cachedSummary}`;
+              message.content.parts[partIndex] = {
+                type: "tool-invocation",
+                toolInvocation: {
+                  state: "result",
+                  step: part.toolInvocation.step,
+                  toolCallId: part.toolInvocation.toolCallId,
+                  toolName: part.toolInvocation.toolName,
+                  args: part.toolInvocation.args,
+                  result: `Tool call summary: ${cachedSummary}`
+                }
+              };
             } else {
               const summaryPromise = this.summaryAgent.generate(
-                `Summarize the following tool call: ${JSON.stringify(toolCall)} and result: ${JSON.stringify(content)}`
+                `Summarize the following tool call: ${JSON.stringify(part.toolInvocation)}`
               );
               summaryTasks.push({
-                content,
+                message,
+                partIndex,
                 promise: summaryPromise,
                 cacheKey
               });
@@ -3164,10 +3172,24 @@ var ToolSummaryProcessor = class extends MemoryProcessor {
           const summaryResult = result.value;
           const summaryText = summaryResult.text;
           this.summaryCache.set(task.cacheKey, summaryText);
-          task.content.result = `Tool call summary: ${summaryText}`;
+          if (task.message.content.format === 2 && task.message.content.parts) {
+            const part = task.message.content.parts[task.partIndex];
+            if (part && part.type === "tool-invocation" && part.toolInvocation?.state === "result") {
+              task.message.content.parts[task.partIndex] = {
+                type: "tool-invocation",
+                toolInvocation: {
+                  state: "result",
+                  step: part.toolInvocation.step,
+                  toolCallId: part.toolInvocation.toolCallId,
+                  toolName: part.toolInvocation.toolName,
+                  args: part.toolInvocation.args,
+                  result: `Tool call summary: ${summaryText}`
+                }
+              };
+            }
+          }
         } else if (result.status === "rejected") {
           console.warn(`Failed to generate summary for tool call:`, result.reason);
-          task.content.result = `Tool call summary: [Summary generation failed]`;
         }
       });
     }
@@ -3187,6 +3209,7 @@ var AgentBuilder = class extends Agent {
 ${config.instructions}` : "";
     const combinedInstructions = additionalInstructions + AgentBuilderDefaults.DEFAULT_INSTRUCTIONS(config.projectPath);
     const agentConfig = {
+      id: "agent-builder",
       name: "agent-builder",
       description: "An AI agent specialized in generating Mastra agents, tools, and workflows from natural language requirements.",
       instructions: combinedInstructions,
@@ -3198,15 +3221,14 @@ ${config.instructions}` : "";
         };
       },
       memory: new Memory({
-        options: AgentBuilderDefaults.DEFAULT_MEMORY_CONFIG,
-        processors: [
-          // use the write to disk processor to debug the agent's context
-          // new WriteToDiskProcessor({ prefix: 'before-filter' }),
-          new ToolSummaryProcessor({ summaryModel: config.summaryModel || config.model }),
-          new TokenLimiter(1e5)
-          // new WriteToDiskProcessor({ prefix: 'after-filter' }),
-        ]
-      })
+        options: AgentBuilderDefaults.DEFAULT_MEMORY_CONFIG
+      }),
+      inputProcessors: [
+        // use the write to disk processor to debug the agent's context
+        // new WriteToDiskProcessor({ prefix: 'before-filter' }),
+        new ToolSummaryProcessor({ summaryModel: config.summaryModel || config.model })
+        // new WriteToDiskProcessor({ prefix: 'after-filter' }),
+      ]
     };
     super(agentConfig);
     this.builderConfig = config;
@@ -5070,6 +5092,7 @@ var planningIterationStep = createStep({
     try {
       const model = await resolveModel({ requestContext });
       const planningAgent = new Agent({
+        id: "workflow-planning-agent",
         model,
         instructions: taskPlanningPrompts.planningAgent.instructions({
           storedQAPairs
@@ -5410,7 +5433,7 @@ export const mastra = new Mastra({
     sendEmailWorkflow,      // Use camelCase for keys
     dataProcessingWorkflow
   },
-  storage: new LibSQLStore({ url: 'file:./mastra.db' }), // Required for suspend/resume
+  storage: new LibSQLStore({ id: 'mastra-storage', url: 'file:./mastra.db' }), // Required for suspend/resume
 });
 \`\`\`
 
@@ -5800,6 +5823,7 @@ var workflowResearchStep = createStep({
     try {
       const model = await resolveModel({ requestContext });
       const researchAgent = new Agent({
+        id: "workflow-research-agent",
         model,
         instructions: workflowBuilderPrompts.researchAgent.instructions,
         name: "Workflow Research Agent"