@mastra/agent-builder 1.0.0-beta.2 → 1.0.0-beta.3

package/CHANGELOG.md

@@ -1,5 +1,13 @@
 # @mastra/agent-builder
 
+## 1.0.0-beta.3
+
+### Patch Changes
+
+- Updated dependencies [[`ac0d2f4`](https://github.com/mastra-ai/mastra/commit/ac0d2f4ff8831f72c1c66c2be809706d17f65789), [`1a0d3fc`](https://github.com/mastra-ai/mastra/commit/1a0d3fc811482c9c376cdf79ee615c23bae9b2d6), [`85a628b`](https://github.com/mastra-ai/mastra/commit/85a628b1224a8f64cd82ea7f033774bf22df7a7e), [`c237233`](https://github.com/mastra-ai/mastra/commit/c23723399ccedf7f5744b3f40997b79246bfbe64), [`15f9e21`](https://github.com/mastra-ai/mastra/commit/15f9e216177201ea6e3f6d0bfb063fcc0953444f), [`ff94dea`](https://github.com/mastra-ai/mastra/commit/ff94dea935f4e34545c63bcb6c29804732698809), [`5b2ff46`](https://github.com/mastra-ai/mastra/commit/5b2ff4651df70c146523a7fca773f8eb0a2272f8), [`db41688`](https://github.com/mastra-ai/mastra/commit/db4168806d007417e2e60b4f68656dca4e5f40c9), [`5ca599d`](https://github.com/mastra-ai/mastra/commit/5ca599d0bb59a1595f19f58473fcd67cc71cef58), [`bff1145`](https://github.com/mastra-ai/mastra/commit/bff114556b3cbadad9b2768488708f8ad0e91475), [`5c8ca24`](https://github.com/mastra-ai/mastra/commit/5c8ca247094e0cc2cdbd7137822fb47241f86e77), [`e191844`](https://github.com/mastra-ai/mastra/commit/e1918444ca3f80e82feef1dad506cd4ec6e2875f), [`22553f1`](https://github.com/mastra-ai/mastra/commit/22553f11c63ee5e966a9c034a349822249584691), [`7237163`](https://github.com/mastra-ai/mastra/commit/72371635dbf96a87df4b073cc48fc655afbdce3d), [`2500740`](https://github.com/mastra-ai/mastra/commit/2500740ea23da067d6e50ec71c625ab3ce275e64), [`873ecbb`](https://github.com/mastra-ai/mastra/commit/873ecbb517586aa17d2f1e99283755b3ebb2863f), [`97fee86`](https://github.com/mastra-ai/mastra/commit/97fee863b3dcca8b998faa016741e57ddff0acc9), [`4f9bbe5`](https://github.com/mastra-ai/mastra/commit/4f9bbe5968f42c86f4930b8193de3c3c17e5bd36), [`02e51fe`](https://github.com/mastra-ai/mastra/commit/02e51feddb3d4155cfbcc42624fd0d0970d032c0), [`8f3fa3a`](https://github.com/mastra-ai/mastra/commit/8f3fa3a652bb77da092f913ec51ae46e3a7e27dc), [`cd29ad2`](https://github.com/mastra-ai/mastra/commit/cd29ad23a255534e8191f249593849ed29160886), [`bdf4d8c`](https://github.com/mastra-ai/mastra/commit/bdf4d8cdc656d8a2c21d81834bfa3bfa70f56c16), [`854e3da`](https://github.com/mastra-ai/mastra/commit/854e3dad5daac17a91a20986399d3a51f54bf68b), [`ce18d38`](https://github.com/mastra-ai/mastra/commit/ce18d38678c65870350d123955014a8432075fd9), [`cccf9c8`](https://github.com/mastra-ai/mastra/commit/cccf9c8b2d2dfc1a5e63919395b83d78c89682a0), [`61a5705`](https://github.com/mastra-ai/mastra/commit/61a570551278b6743e64243b3ce7d73de915ca8a), [`db70a48`](https://github.com/mastra-ai/mastra/commit/db70a48aeeeeb8e5f92007e8ede52c364ce15287), [`f0fdc14`](https://github.com/mastra-ai/mastra/commit/f0fdc14ee233d619266b3d2bbdeea7d25cfc6d13), [`db18bc9`](https://github.com/mastra-ai/mastra/commit/db18bc9c3825e2c1a0ad9a183cc9935f6691bfa1), [`9b37b56`](https://github.com/mastra-ai/mastra/commit/9b37b565e1f2a76c24f728945cc740c2b09be9da), [`41a23c3`](https://github.com/mastra-ai/mastra/commit/41a23c32f9877d71810f37e24930515df2ff7a0f), [`5d171ad`](https://github.com/mastra-ai/mastra/commit/5d171ad9ef340387276b77c2bb3e83e83332d729), [`f03ae60`](https://github.com/mastra-ai/mastra/commit/f03ae60500fe350c9d828621006cdafe1975fdd8), [`d1e74a0`](https://github.com/mastra-ai/mastra/commit/d1e74a0a293866dece31022047f5dbab65a304d0), [`39e7869`](https://github.com/mastra-ai/mastra/commit/39e7869bc7d0ee391077ce291474d8a84eedccff), [`5761926`](https://github.com/mastra-ai/mastra/commit/57619260c4a2cdd598763abbacd90de594c6bc76), [`c900fdd`](https://github.com/mastra-ai/mastra/commit/c900fdd504c41348efdffb205cfe80d48c38fa33), [`604a79f`](https://github.com/mastra-ai/mastra/commit/604a79fecf276e26a54a3fe01bb94e65315d2e0e), [`887f0b4`](https://github.com/mastra-ai/mastra/commit/887f0b4746cdbd7cb7d6b17ac9f82aeb58037ea5), [`2562143`](https://github.com/mastra-ai/mastra/commit/256214336b4faa78646c9c1776612393790d8784), [`ef11a61`](https://github.com/mastra-ai/mastra/commit/ef11a61920fa0ed08a5b7ceedd192875af119749), [`ce32589`](https://github.com/mastra-ai/mastra/commit/ce325896358daa1088715ea13aac9eca596a4871)]:
+  - @mastra/core@1.0.0-beta.6
+  - @mastra/memory@1.0.0-beta.3
+
 ## 1.0.0-beta.2
 
 ### Patch Changes

package/dist/agent/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,KAAK,EAIV,qBAAqB,EACtB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAExE,OAAO,KAAK,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAK3E,OAAO,KAAK,EAAE,kBAAkB,EAAwB,MAAM,UAAU,CAAC;AA+BzE,qBAAa,YAAa,SAAQ,KAAK;IACrC,OAAO,CAAC,aAAa,CAAqB;IAE1C;;OAEG;gBACS,MAAM,EAAE,kBAAkB;IAiCtC;;;OAGG;IACH,cAAc,EAAE,KAAK,CAAC,gBAAgB,CAAC,CA6BrC;IAEF;;;OAGG;IACH,YAAY,EAAE,KAAK,CAAC,cAAc,CAAC,CA4BjC;IAEF;;;OAGG;IACG,MAAM,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAClD,QAAQ,EAAE,gBAAgB,EAC1B,aAAa,CAAC,EAAE,qBAAqB,CAAC,MAAM,CAAC,GAC5C,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IA2B/B,QAAQ,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EACpD,QAAQ,EAAE,gBAAgB,EAC1B,OAAO,CAAC,EAAE,qBAAqB,CAAC,MAAM,CAAC,GACtC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;CA0B5E"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAC3C,OAAO,KAAK,EAIV,qBAAqB,EACtB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAExE,OAAO,KAAK,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAI3E,OAAO,KAAK,EAAE,kBAAkB,EAAwB,MAAM,UAAU,CAAC;AA+BzE,qBAAa,YAAa,SAAQ,KAAK;IACrC,OAAO,CAAC,aAAa,CAAqB;IAE1C;;OAEG;gBACS,MAAM,EAAE,kBAAkB;IAgCtC;;;OAGG;IACH,cAAc,EAAE,KAAK,CAAC,gBAAgB,CAAC,CA6BrC;IAEF;;;OAGG;IACH,YAAY,EAAE,KAAK,CAAC,cAAc,CAAC,CA4BjC;IAEF;;;OAGG;IACG,MAAM,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EAClD,QAAQ,EAAE,gBAAgB,EAC1B,aAAa,CAAC,EAAE,qBAAqB,CAAC,MAAM,CAAC,GAC5C,OAAO,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;IA2B/B,QAAQ,CAAC,MAAM,SAAS,YAAY,GAAG,SAAS,EACpD,QAAQ,EAAE,gBAAgB,EAC1B,OAAO,CAAC,EAAE,qBAAqB,CAAC,MAAM,CAAC,GACtC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;CA0B5E"}

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';
@@ -3084,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",
@@ -3125,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
               });
@@ -3166,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]`;
         }
       });
     }
@@ -3201,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;