@mastra/server 1.22.0-alpha.3 → 1.23.0-alpha.0

package/dist/{chunk-PQGBLNCU.js → chunk-O2RWEEZ3.js}

@@ -3030,6 +3030,26 @@ var ObservationStep = class {
     if (statusSnapshot.shouldBuffer && !hasIncompleteToolCalls) {
       const allMessages = messageList.get.all.db();
       const unobservedMessages = om.getUnobservedMessages(allMessages, statusSnapshot.record);
+      const candidates = om.getUnobservedMessages(unobservedMessages, statusSnapshot.record, {
+        excludeBuffered: true
+      });
+      if (candidates.length > 0) {
+        om.sealMessagesForBuffering(candidates);
+        try {
+          await this.turn.hooks?.onBufferChunkSealed?.();
+        } catch (error) {
+          omDebug(
+            `[OM:buffer] onBufferChunkSealed hook failed: ${error instanceof Error ? error.message : String(error)}`
+          );
+        }
+        if (this.turn.memory) {
+          await this.turn.memory.persistMessages(candidates);
+        }
+        messageList.removeByIds(candidates.map((msg) => msg.id));
+        for (const msg of candidates) {
+          messageList.add(msg, "memory");
+        }
+      }
       void om.buffer({
         threadId,
         resourceId,
@@ -3038,23 +3058,7 @@ var ObservationStep = class {
         record: statusSnapshot.record,
         writer: this.turn.writer,
         requestContext: this.turn.requestContext,
-        observabilityContext: this.turn.observabilityContext,
-        beforeBuffer: async (candidates) => {
-          if (candidates.length === 0) {
-            return;
-          }
-          om.sealMessagesForBuffering(candidates);
-          try {
-            await this.turn.hooks?.onBufferChunkSealed?.();
-          } catch (error) {
-            omDebug(
-              `[OM:buffer] onBufferChunkSealed hook failed: ${error instanceof Error ? error.message : String(error)}`
-            );
-          }
-          if (this.turn.memory) {
-            await this.turn.memory.persistMessages(candidates);
-          }
-        }
+        observabilityContext: this.turn.observabilityContext
       }).catch((err) => {
         omDebug(`[OM:buffer] fire-and-forget buffer failed: ${err?.message}`);
       });
@@ -9548,5 +9552,5 @@ function getObservationsAsOf(activeObservations, asOf) {
 }
 
 export { ModelByInputTokens, OBSERVER_SYSTEM_PROMPT, ObservationalMemory, ObservationalMemoryProcessor, TokenCounter, buildObserverPrompt, buildObserverSystemPrompt, combineObservationGroupRanges, deriveObservationGroupProvenance, e, estimateTokenCount, extractCurrentTask, formatMessagesForObserver, formatToolResultForObserver, getObservationsAsOf, hasCurrentTaskSection, injectAnchorIds, optimizeObservationsForContext, parseAnchorId, parseObservationGroups, parseObserverOutput, reconcileObservationGroupsFromReflection, renderObservationGroupsForReflection, resolveToolResultValue, stripEphemeralAnchorIds, stripObservationGroups, truncateStringByTokens, wrapInObservationGroup };
-//# sourceMappingURL=chunk-PQGBLNCU.js.map
-//# sourceMappingURL=chunk-PQGBLNCU.js.map
+//# sourceMappingURL=chunk-O2RWEEZ3.js.map
+//# sourceMappingURL=chunk-O2RWEEZ3.js.map