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

package/dist/{chunk-DQIZ5FFX.cjs → chunk-VRFSEZBA.cjs}

@@ -917,6 +917,9 @@ var MessageList = class _MessageList {
   userContextMessagesPersisted = /* @__PURE__ */ new Set();
   generateMessageId;
   _agentNetworkAppend = false;
+  // Event recording for observability
+  isRecording = false;
+  recordedEvents = [];
   constructor({
     threadId,
     resourceId,
@@ -930,10 +933,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",
@@ -995,6 +1022,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;
@@ -1019,11 +1062,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;
         }
       },
@@ -1032,11 +1098,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()),
@@ -1340,12 +1441,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]) {
@@ -1404,8 +1540,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) {
@@ -1685,6 +1834,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);
@@ -1813,8 +1965,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
       };
@@ -1822,8 +1976,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
       };
@@ -1833,16 +1989,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;
@@ -2129,10 +2285,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
@@ -2185,6 +2342,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;
@@ -3173,5 +3335,5 @@ exports.DefaultGeneratedFile = DefaultGeneratedFile;
 exports.DefaultGeneratedFileWithType = DefaultGeneratedFileWithType;
 exports.MessageList = MessageList;
 exports.convertMessages = convertMessages;
-//# sourceMappingURL=chunk-DQIZ5FFX.cjs.map
-//# sourceMappingURL=chunk-DQIZ5FFX.cjs.map
+//# sourceMappingURL=chunk-VRFSEZBA.cjs.map
+//# sourceMappingURL=chunk-VRFSEZBA.cjs.map