@mastra/core 1.0.0-beta.5 → 1.0.0-beta.7

package/dist/{chunk-KEXGB7FK.cjs → chunk-DGV2FWB4.cjs}

@@ -1,16 +1,16 @@
 'use strict';
 
-var chunkDSNPWVIG_cjs = require('./chunk-DSNPWVIG.cjs');
+var chunkAF74UXR5_cjs = require('./chunk-AF74UXR5.cjs');
 
 // src/base.ts
 var MastraBase = class {
-  component = chunkDSNPWVIG_cjs.RegisteredLogger.LLM;
+  component = chunkAF74UXR5_cjs.RegisteredLogger.LLM;
   logger;
   name;
   constructor({ component, name }) {
-    this.component = component || chunkDSNPWVIG_cjs.RegisteredLogger.LLM;
+    this.component = component || chunkAF74UXR5_cjs.RegisteredLogger.LLM;
     this.name = name;
-    this.logger = new chunkDSNPWVIG_cjs.ConsoleLogger({ name: `${this.component} - ${this.name}` });
+    this.logger = new chunkAF74UXR5_cjs.ConsoleLogger({ name: `${this.component} - ${this.name}` });
   }
   /**
    * Set the logger for the agent
@@ -18,12 +18,12 @@ var MastraBase = class {
    */
   __setLogger(logger) {
     this.logger = logger;
-    if (this.component !== chunkDSNPWVIG_cjs.RegisteredLogger.LLM) {
+    if (this.component !== chunkAF74UXR5_cjs.RegisteredLogger.LLM) {
       this.logger.debug(`Logger updated [component=${this.component}] [name=${this.name}]`);
     }
   }
 };
 
 exports.MastraBase = MastraBase;
-//# sourceMappingURL=chunk-KEXGB7FK.cjs.map
-//# sourceMappingURL=chunk-KEXGB7FK.cjs.map
+//# sourceMappingURL=chunk-DGV2FWB4.cjs.map
+//# sourceMappingURL=chunk-DGV2FWB4.cjs.map

package/dist/{chunk-KEXGB7FK.cjs.map → chunk-DGV2FWB4.cjs.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/base.ts"],"names":["RegisteredLogger","ConsoleLogger"],"mappings":";;;;;AAIO,IAAM,aAAN,MAAiB;AAAA,EACtB,YAA8BA,kCAAA,CAAiB,GAAA;AAAA,EACrC,MAAA;AAAA,EACV,IAAA;AAAA,EAEA,WAAA,CAAY,EAAE,SAAA,EAAW,IAAA,EAAK,EAAoD;AAChF,IAAA,IAAA,CAAK,SAAA,GAAY,aAAaA,kCAAA,CAAiB,GAAA;AAC/C,IAAA,IAAA,CAAK,IAAA,GAAO,IAAA;AACZ,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,+BAAA,CAAc,EAAE,IAAA,EAAM,CAAA,EAAG,IAAA,CAAK,SAAS,CAAA,GAAA,EAAM,IAAA,CAAK,IAAI,CAAA,CAAA,EAAI,CAAA;AAAA,EAC9E;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,YAAY,MAAA,EAAuB;AACjC,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAEd,IAAA,IAAI,IAAA,CAAK,SAAA,KAAcD,kCAAA,CAAiB,GAAA,EAAK;AAC3C,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,CAAA,0BAAA,EAA6B,IAAA,CAAK,SAAS,CAAA,QAAA,EAAW,IAAA,CAAK,IAAI,CAAA,CAAA,CAAG,CAAA;AAAA,IACtF;AAAA,EACF;AACF","file":"chunk-KEXGB7FK.cjs","sourcesContent":["import type { IMastraLogger } from './logger';\nimport { RegisteredLogger } from './logger/constants';\nimport { ConsoleLogger } from './logger/default-logger';\n\nexport class MastraBase {\n  component: RegisteredLogger = RegisteredLogger.LLM;\n  protected logger: IMastraLogger;\n  name?: string;\n\n  constructor({ component, name }: { component?: RegisteredLogger; name?: string }) {\n    this.component = component || RegisteredLogger.LLM;\n    this.name = name;\n    this.logger = new ConsoleLogger({ name: `${this.component} - ${this.name}` });\n  }\n\n  /**\n   * Set the logger for the agent\n   * @param logger\n   */\n  __setLogger(logger: IMastraLogger) {\n    this.logger = logger;\n\n    if (this.component !== RegisteredLogger.LLM) {\n      this.logger.debug(`Logger updated [component=${this.component}] [name=${this.name}]`);\n    }\n  }\n}\n\nexport * from './types';\n"]}
+{"version":3,"sources":["../src/base.ts"],"names":["RegisteredLogger","ConsoleLogger"],"mappings":";;;;;AAIO,IAAM,aAAN,MAAiB;AAAA,EACtB,YAA8BA,kCAAA,CAAiB,GAAA;AAAA,EACrC,MAAA;AAAA,EACV,IAAA;AAAA,EAEA,WAAA,CAAY,EAAE,SAAA,EAAW,IAAA,EAAK,EAAoD;AAChF,IAAA,IAAA,CAAK,SAAA,GAAY,aAAaA,kCAAA,CAAiB,GAAA;AAC/C,IAAA,IAAA,CAAK,IAAA,GAAO,IAAA;AACZ,IAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,+BAAA,CAAc,EAAE,IAAA,EAAM,CAAA,EAAG,IAAA,CAAK,SAAS,CAAA,GAAA,EAAM,IAAA,CAAK,IAAI,CAAA,CAAA,EAAI,CAAA;AAAA,EAC9E;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,YAAY,MAAA,EAAuB;AACjC,IAAA,IAAA,CAAK,MAAA,GAAS,MAAA;AAEd,IAAA,IAAI,IAAA,CAAK,SAAA,KAAcD,kCAAA,CAAiB,GAAA,EAAK;AAC3C,MAAA,IAAA,CAAK,MAAA,CAAO,MAAM,CAAA,0BAAA,EAA6B,IAAA,CAAK,SAAS,CAAA,QAAA,EAAW,IAAA,CAAK,IAAI,CAAA,CAAA,CAAG,CAAA;AAAA,IACtF;AAAA,EACF;AACF","file":"chunk-DGV2FWB4.cjs","sourcesContent":["import type { IMastraLogger } from './logger';\nimport { RegisteredLogger } from './logger/constants';\nimport { ConsoleLogger } from './logger/default-logger';\n\nexport class MastraBase {\n  component: RegisteredLogger = RegisteredLogger.LLM;\n  protected logger: IMastraLogger;\n  name?: string;\n\n  constructor({ component, name }: { component?: RegisteredLogger; name?: string }) {\n    this.component = component || RegisteredLogger.LLM;\n    this.name = name;\n    this.logger = new ConsoleLogger({ name: `${this.component} - ${this.name}` });\n  }\n\n  /**\n   * Set the logger for the agent\n   * @param logger\n   */\n  __setLogger(logger: IMastraLogger) {\n    this.logger = logger;\n\n    if (this.component !== RegisteredLogger.LLM) {\n      this.logger.debug(`Logger updated [component=${this.component}] [name=${this.name}]`);\n    }\n  }\n}\n\nexport * from './types';\n"]}

package/dist/{chunk-YDFX3JR2.js → chunk-DMPHPHMU.js}

@@ -1,4 +1,4 @@
-import { convertToCoreMessages, convertUint8ArrayToBase64 as convertUint8ArrayToBase64$1 } from './chunk-QM5SRDJX.js';
+import { convertToCoreMessages, convertUint8ArrayToBase64 as convertUint8ArrayToBase64$1 } from './chunk-ZUWJCGLM.js';
 import { MastraError } from './chunk-JJ5O45LH.js';
 import { v4 } from '@lukeed/uuid';
 import * as AIV5 from 'ai-v5';
@@ -895,6 +895,9 @@ var MessageList = class _MessageList {
   userContextMessagesPersisted = /* @__PURE__ */ new Set();
   generateMessageId;
   _agentNetworkAppend = false;
+  // Event recording for observability
+  isRecording = false;
+  recordedEvents = [];
   constructor({
     threadId,
     resourceId,
@@ -908,10 +911,34 @@ var MessageList = class _MessageList {
     this.generateMessageId = generateMessageId;
     this._agentNetworkAppend = _agentNetworkAppend || false;
   }
+  /**
+   * Start recording mutations to the MessageList for observability/tracing
+   */
+  startRecording() {
+    this.isRecording = true;
+    this.recordedEvents = [];
+  }
+  /**
+   * Stop recording and return the list of recorded events
+   */
+  stopRecording() {
+    this.isRecording = false;
+    const events = [...this.recordedEvents];
+    this.recordedEvents = [];
+    return events;
+  }
   add(messages, messageSource) {
     if (messageSource === `user`) messageSource = `input`;
     if (!messages) return this;
-    for (const message of Array.isArray(messages) ? messages : [messages]) {
+    const messageArray = Array.isArray(messages) ? messages : [messages];
+    if (this.isRecording) {
+      this.recordedEvents.push({
+        type: "add",
+        source: messageSource,
+        count: messageArray.length
+      });
+    }
+    for (const message of messageArray) {
       this.addOne(
         typeof message === `string` ? {
           role: "user",
@@ -973,6 +1000,22 @@ var MessageList = class _MessageList {
     this.userContextMessagesPersisted = this.deserializeSet(state.userContextMessagesPersisted);
     return this;
   }
+  makeMessageSourceChecker() {
+    const sources = {
+      memory: new Set(Array.from(this.memoryMessages.values()).map((m) => m.id)),
+      output: new Set(Array.from(this.newResponseMessages.values()).map((m) => m.id)),
+      input: new Set(Array.from(this.newUserMessages.values()).map((m) => m.id))
+    };
+    return {
+      ...sources,
+      getSource: (msg) => {
+        if (sources.memory.has(msg.id)) return "memory";
+        if (sources.input.has(msg.id)) return "input";
+        if (sources.output.has(msg.id)) return "response";
+        return null;
+      }
+    };
+  }
   getLatestUserContent() {
     const currentUserMessages = this.all.core().filter((m) => m.role === "user");
     const content = currentUserMessages.at(-1)?.content;
@@ -997,11 +1040,34 @@ var MessageList = class _MessageList {
   }
   get clear() {
     return {
+      all: {
+        db: () => {
+          const allMessages = [...this.messages];
+          this.messages = [];
+          this.newUserMessages.clear();
+          this.newResponseMessages.clear();
+          this.userContextMessages.clear();
+          if (this.isRecording && allMessages.length > 0) {
+            this.recordedEvents.push({
+              type: "clear",
+              count: allMessages.length
+            });
+          }
+          return allMessages;
+        }
+      },
       input: {
         db: () => {
           const userMessages = Array.from(this.newUserMessages);
           this.messages = this.messages.filter((m) => !this.newUserMessages.has(m));
           this.newUserMessages.clear();
+          if (this.isRecording && userMessages.length > 0) {
+            this.recordedEvents.push({
+              type: "clear",
+              source: "input",
+              count: userMessages.length
+            });
+          }
           return userMessages;
         }
       },
@@ -1010,11 +1076,46 @@ var MessageList = class _MessageList {
           const responseMessages = Array.from(this.newResponseMessages);
           this.messages = this.messages.filter((m) => !this.newResponseMessages.has(m));
           this.newResponseMessages.clear();
+          if (this.isRecording && responseMessages.length > 0) {
+            this.recordedEvents.push({
+              type: "clear",
+              source: "response",
+              count: responseMessages.length
+            });
+          }
           return responseMessages;
         }
       }
     };
   }
+  /**
+   * Remove messages by ID
+   * @param ids - Array of message IDs to remove
+   * @returns Array of removed messages
+   */
+  removeByIds(ids) {
+    const idsSet = new Set(ids);
+    const removed = [];
+    this.messages = this.messages.filter((m) => {
+      if (idsSet.has(m.id)) {
+        removed.push(m);
+        this.memoryMessages.delete(m);
+        this.newUserMessages.delete(m);
+        this.newResponseMessages.delete(m);
+        this.userContextMessages.delete(m);
+        return false;
+      }
+      return true;
+    });
+    if (this.isRecording && removed.length > 0) {
+      this.recordedEvents.push({
+        type: "removeByIds",
+        ids,
+        count: removed.length
+      });
+    }
+    return removed;
+  }
   all = {
     db: () => this.messages,
     v1: () => convertToV1Messages(this.all.db()),
@@ -1318,12 +1419,47 @@ var MessageList = class _MessageList {
     if (unsavedMessages.length === 0) return void 0;
     return Math.min(...unsavedMessages.map((m) => new Date(m.createdAt).getTime()));
   }
+  /**
+   * Check if a message is a new user or response message that should be saved.
+   * Checks by message ID to handle cases where the message object may be a copy.
+   */
+  isNewMessage(messageOrId) {
+    const id = typeof messageOrId === "string" ? messageOrId : messageOrId.id;
+    if (typeof messageOrId !== "string") {
+      if (this.newUserMessages.has(messageOrId) || this.newResponseMessages.has(messageOrId)) {
+        return true;
+      }
+    }
+    return Array.from(this.newUserMessages).some((m) => m.id === id) || Array.from(this.newResponseMessages).some((m) => m.id === id);
+  }
   getSystemMessages(tag) {
     if (tag) {
       return this.taggedSystemMessages[tag] || [];
     }
     return this.systemMessages;
   }
+  /**
+   * Get all system messages (both tagged and untagged)
+   * @returns Array of all system messages
+   */
+  getAllSystemMessages() {
+    return [...this.systemMessages, ...Object.values(this.taggedSystemMessages).flat()];
+  }
+  /**
+   * Replace all system messages with new ones
+   * This clears both tagged and untagged system messages and replaces them with the provided array
+   * @param messages - Array of system messages to set
+   */
+  replaceAllSystemMessages(messages) {
+    this.systemMessages = [];
+    this.taggedSystemMessages = {};
+    for (const message of messages) {
+      if (message.role === "system") {
+        this.systemMessages.push(message);
+      }
+    }
+    return this;
+  }
   addSystem(messages, tag) {
     if (!messages) return this;
     for (const message of Array.isArray(messages) ? messages : [messages]) {
@@ -1382,8 +1518,21 @@ var MessageList = class _MessageList {
     if (tag && !this.isDuplicateSystem(coreMessage, tag)) {
       this.taggedSystemMessages[tag] ||= [];
       this.taggedSystemMessages[tag].push(coreMessage);
+      if (this.isRecording) {
+        this.recordedEvents.push({
+          type: "addSystem",
+          tag,
+          message: coreMessage
+        });
+      }
     } else if (!tag && !this.isDuplicateSystem(coreMessage)) {
       this.systemMessages.push(coreMessage);
+      if (this.isRecording) {
+        this.recordedEvents.push({
+          type: "addSystem",
+          message: coreMessage
+        });
+      }
     }
   }
   isDuplicateSystem(message, tag) {
@@ -1663,6 +1812,9 @@ var MessageList = class _MessageList {
     } else if (messageSource === `response`) {
       this.newResponseMessages.add(messageV2);
       this.newResponseMessagesPersisted.add(messageV2);
+      if (this.newUserMessages.has(messageV2)) {
+        this.newUserMessages.delete(messageV2);
+      }
     } else if (messageSource === `input`) {
       this.newUserMessages.add(messageV2);
       this.newUserMessagesPersisted.add(messageV2);
@@ -1791,8 +1943,10 @@ var MessageList = class _MessageList {
     }
     if (_MessageList.isAIV5CoreMessage(message)) {
       const dbMsg = _MessageList.aiV5ModelMessageToMastraDBMessage(message, messageSource);
+      const rawCreatedAt = "metadata" in message && message.metadata && typeof message.metadata === "object" && "createdAt" in message.metadata ? message.metadata.createdAt : void 0;
       const result = {
         ...dbMsg,
+        createdAt: this.generateCreatedAt(messageSource, rawCreatedAt),
         threadId: this.memoryInfo?.threadId,
         resourceId: this.memoryInfo?.resourceId
       };
@@ -1800,8 +1954,10 @@ var MessageList = class _MessageList {
     }
     if (_MessageList.isAIV5UIMessage(message)) {
       const dbMsg = _MessageList.aiV5UIMessageToMastraDBMessage(message);
+      const rawCreatedAt = "createdAt" in message ? message.createdAt : void 0;
       return {
         ...dbMsg,
+        createdAt: this.generateCreatedAt(messageSource, rawCreatedAt),
         threadId: this.memoryInfo?.threadId,
         resourceId: this.memoryInfo?.resourceId
       };
@@ -1811,16 +1967,16 @@ var MessageList = class _MessageList {
   lastCreatedAt;
   // this makes sure messages added in order will always have a date atleast 1ms apart.
   generateCreatedAt(messageSource, start) {
-    start = start instanceof Date ? start : start ? new Date(start) : void 0;
-    if (start && !this.lastCreatedAt) {
-      this.lastCreatedAt = start.getTime();
-      return start;
+    const startDate = start instanceof Date ? start : typeof start === "string" || typeof start === "number" ? new Date(start) : void 0;
+    if (startDate && !this.lastCreatedAt) {
+      this.lastCreatedAt = startDate.getTime();
+      return startDate;
     }
-    if (start && messageSource === `memory`) {
-      return start;
+    if (startDate && messageSource === `memory`) {
+      return startDate;
     }
     const now = /* @__PURE__ */ new Date();
-    const nowTime = start?.getTime() || now.getTime();
+    const nowTime = startDate?.getTime() || now.getTime();
     const lastTime = this.messages.reduce((p, m) => {
       if (m.createdAt.getTime() > p) return m.createdAt.getTime();
       return p;
@@ -2107,10 +2263,11 @@ var MessageList = class _MessageList {
     if ("metadata" in coreMessage && coreMessage.metadata !== null && coreMessage.metadata !== void 0) {
       content.metadata = coreMessage.metadata;
     }
+    const rawCreatedAt = "metadata" in coreMessage && coreMessage.metadata && typeof coreMessage.metadata === "object" && "createdAt" in coreMessage.metadata ? coreMessage.metadata.createdAt : void 0;
     return {
       id,
       role: _MessageList.getRole(coreMessage),
-      createdAt: this.generateCreatedAt(messageSource),
+      createdAt: this.generateCreatedAt(messageSource, rawCreatedAt),
       threadId: this.memoryInfo?.threadId,
       resourceId: this.memoryInfo?.resourceId,
       content
@@ -2163,6 +2320,11 @@ var MessageList = class _MessageList {
           }
           return prev;
         }, 0);
+        const partAny = part;
+        if (partAny && Object.hasOwn(partAny, "providerMetadata") && partAny.providerMetadata && Object.hasOwn(partAny.providerMetadata, "openai") && partAny.providerMetadata.openai && Object.hasOwn(partAny.providerMetadata.openai, "itemId")) {
+          const itemId = partAny.providerMetadata.openai.itemId;
+          key += `|${itemId}`;
+        }
       }
       if (part.type === `file`) {
         key += part.data;
@@ -3148,5 +3310,5 @@ var MessageList = class _MessageList {
 };
 
 export { DefaultGeneratedFile, DefaultGeneratedFileWithType, MessageList, convertMessages };
-//# sourceMappingURL=chunk-YDFX3JR2.js.map
-//# sourceMappingURL=chunk-YDFX3JR2.js.map
+//# sourceMappingURL=chunk-DMPHPHMU.js.map
+//# sourceMappingURL=chunk-DMPHPHMU.js.map