@mastra/memory 0.2.6-alpha.3 → 0.2.6-alpha.4

package/.turbo/turbo-build.log

@@ -1,27 +1,27 @@
 
-> @mastra/memory@0.2.6-alpha.3 build /home/runner/work/mastra/mastra/packages/memory
+> @mastra/memory@0.2.6-alpha.4 build /home/runner/work/mastra/mastra/packages/memory
 > pnpm run check && tsup src/index.ts --format esm,cjs --experimental-dts --clean --treeshake=smallest --splitting
 
 
-> @mastra/memory@0.2.6-alpha.3 check /home/runner/work/mastra/mastra/packages/memory
+> @mastra/memory@0.2.6-alpha.4 check /home/runner/work/mastra/mastra/packages/memory
 > tsc --noEmit
 
 CLI Building entry: src/index.ts
 CLI Using tsconfig: tsconfig.json
 CLI tsup v8.4.0
 TSC Build start
-TSC ⚡️ Build success in 8385ms
+TSC ⚡️ Build success in 13659ms
 DTS Build start
 CLI Target: es2022
-CLI Cleaning output folder
-ESM Build start
-CJS Build start
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/packages/memory/dist/_tsup-dts-rollup.d.ts
 Analysis will use the bundled TypeScript version 5.8.2
 Writing package typings: /home/runner/work/mastra/mastra/packages/memory/dist/_tsup-dts-rollup.d.cts
-DTS ⚡️ Build success in 3439ms
-CJS dist/index.cjs 13.37 KB
-CJS ⚡️ Build success in 3563ms
-ESM dist/index.js 13.34 KB
-ESM ⚡️ Build success in 3564ms
+DTS ⚡️ Build success in 5635ms
+CLI Cleaning output folder
+ESM Build start
+CJS Build start
+ESM dist/index.js 13.53 KB
+ESM ⚡️ Build success in 315ms
+CJS dist/index.cjs 13.56 KB
+CJS ⚡️ Build success in 316ms

package/CHANGELOG.md

@@ -1,5 +1,12 @@
 # @mastra/memory
 
+## 0.2.6-alpha.4
+
+### Patch Changes
+
+- 394dfad: Removed working memory tool calls from thread history after the working memory has been updated. This is to prevent updates from polluting the context history and confusing agents. They should only see the most recent copy of working memory.
+  Also made memory.getWorkingMemory() public since it's useful for testing, debugging, and building UIs.
+
 ## 0.2.6-alpha.3
 
 ### Patch Changes

package/dist/_tsup-dts-rollup.d.cts

@@ -52,7 +52,7 @@ export declare class Memory extends MastraMemory {
     }): Promise<MessageType[]>;
     protected mutateMessagesToHideWorkingMemory(messages: MessageType[]): void;
     protected parseWorkingMemory(text: string): string | null;
-    protected getWorkingMemory({ threadId }: {
+    getWorkingMemory({ threadId }: {
         threadId: string;
     }): Promise<string | null>;
     private saveWorkingMemory;

package/dist/_tsup-dts-rollup.d.ts

@@ -52,7 +52,7 @@ export declare class Memory extends MastraMemory {
     }): Promise<MessageType[]>;
     protected mutateMessagesToHideWorkingMemory(messages: MessageType[]): void;
     protected parseWorkingMemory(text: string): string | null;
-    protected getWorkingMemory({ threadId }: {
+    getWorkingMemory({ threadId }: {
         threadId: string;
     }): Promise<string | null>;
     private saveWorkingMemory;

package/dist/index.cjs

@@ -223,7 +223,7 @@ var Memory = class extends memory.MastraMemory {
   }
   mutateMessagesToHideWorkingMemory(messages) {
     const workingMemoryRegex = /<working_memory>([^]*?)<\/working_memory>/g;
-    for (const message of messages) {
+    for (const [index, message] of messages.entries()) {
       if (typeof message?.content === `string`) {
         message.content = message.content.replace(workingMemoryRegex, ``).trim();
       } else if (Array.isArray(message?.content)) {
@@ -231,6 +231,9 @@ var Memory = class extends memory.MastraMemory {
           if (content.type === `text`) {
             content.text = content.text.replace(workingMemoryRegex, ``).trim();
           }
+          if ((content.type === `tool-call` || content.type === `tool-result`) && content.toolName === `updateWorkingMemory`) {
+            delete messages[index];
+          }
         }
       }
     }

package/dist/index.js

@@ -221,7 +221,7 @@ var Memory = class extends MastraMemory {
   }
   mutateMessagesToHideWorkingMemory(messages) {
     const workingMemoryRegex = /<working_memory>([^]*?)<\/working_memory>/g;
-    for (const message of messages) {
+    for (const [index, message] of messages.entries()) {
       if (typeof message?.content === `string`) {
         message.content = message.content.replace(workingMemoryRegex, ``).trim();
       } else if (Array.isArray(message?.content)) {
@@ -229,6 +229,9 @@ var Memory = class extends MastraMemory {
           if (content.type === `text`) {
             content.text = content.text.replace(workingMemoryRegex, ``).trim();
           }
+          if ((content.type === `tool-call` || content.type === `tool-result`) && content.toolName === `updateWorkingMemory`) {
+            delete messages[index];
+          }
         }
       }
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/memory",
-  "version": "0.2.6-alpha.3",
+  "version": "0.2.6-alpha.4",
   "description": "",
   "type": "module",
   "main": "./dist/index.js",

package/src/index.ts

@@ -281,7 +281,8 @@ export class Memory extends MastraMemory {
 
   protected mutateMessagesToHideWorkingMemory(messages: MessageType[]) {
     const workingMemoryRegex = /<working_memory>([^]*?)<\/working_memory>/g;
-    for (const message of messages) {
+
+    for (const [index, message] of messages.entries()) {
       if (typeof message?.content === `string`) {
         message.content = message.content.replace(workingMemoryRegex, ``).trim();
       } else if (Array.isArray(message?.content)) {
@@ -289,6 +290,13 @@ export class Memory extends MastraMemory {
           if (content.type === `text`) {
             content.text = content.text.replace(workingMemoryRegex, ``).trim();
           }
+
+          if (
+            (content.type === `tool-call` || content.type === `tool-result`) &&
+            content.toolName === `updateWorkingMemory`
+          ) {
+            delete messages[index];
+          }
         }
       }
     }
@@ -308,7 +316,7 @@ export class Memory extends MastraMemory {
     return null;
   }
 
-  protected async getWorkingMemory({ threadId }: { threadId: string }): Promise<string | null> {
+  public async getWorkingMemory({ threadId }: { threadId: string }): Promise<string | null> {
     if (!this.threadConfig.workingMemory?.enabled) return null;
 
     // Get thread from storage