@mastra/core 1.3.0-alpha.1 → 1.3.0-alpha.2

package/dist/{chunk-SZH2YH74.cjs → chunk-JU6K7UDX.cjs}

@@ -931,11 +931,11 @@ var MemoryStorage = class extends StorageDomain {
     throw new Error(`Observational memory is not implemented by this storage adapter (${this.constructor.name}).`);
   }
   /**
-   * Add to the pending message token count.
-   * Called when messages are processed but observation hasn't triggered yet.
-   * This allows accumulating tokens across multiple sessions.
+   * Set the pending message token count.
+   * Called at the end of each OM processing step to persist the current
+   * context window token count so the UI can display it on page load.
    */
-  async addPendingMessageTokens(_id, _tokenCount) {
+  async setPendingMessageTokens(_id, _tokenCount) {
     throw new Error(`Observational memory is not implemented by this storage adapter (${this.constructor.name}).`);
   }
   /**
@@ -1880,12 +1880,12 @@ ${unreflectedContent}` : bufferedReflection;
     const key = this.getObservationalMemoryKey(threadId, resourceId);
     this.db.observationalMemory.delete(key);
   }
-  async addPendingMessageTokens(id, tokenCount) {
+  async setPendingMessageTokens(id, tokenCount) {
     const record = this.findObservationalMemoryRecordById(id);
     if (!record) {
       throw new Error(`Observational memory record not found: ${id}`);
     }
-    record.pendingMessageTokens = (record.pendingMessageTokens ?? 0) + tokenCount;
+    record.pendingMessageTokens = tokenCount;
     record.updatedAt = /* @__PURE__ */ new Date();
   }
   /**
@@ -3635,5 +3635,5 @@ exports.safelyParseJSON = safelyParseJSON;
 exports.serializeDate = serializeDate;
 exports.transformRow = transformRow;
 exports.transformScoreRow = transformScoreRow;
-//# sourceMappingURL=chunk-SZH2YH74.cjs.map
-//# sourceMappingURL=chunk-SZH2YH74.cjs.map
+//# sourceMappingURL=chunk-JU6K7UDX.cjs.map
+//# sourceMappingURL=chunk-JU6K7UDX.cjs.map