@mastra/memory 1.27.0-alpha.1 → 1.27.0-alpha.2

package/dist/{src-_7ADtEwc.cjs → src-Drm6l_ZN.cjs}

@@ -22295,7 +22295,7 @@ var ObservationStep = class {
 				messageList.removeByIds(candidates.map((msg) => msg.id));
 				for (const msg of candidates) messageList.add(msg, "memory");
 			}
-			om.buffer({
+			om.trackBackgroundWork(om.buffer({
 				threadId,
 				resourceId,
 				messages: unobservedMessages,
@@ -22306,7 +22306,7 @@ var ObservationStep = class {
 				observabilityContext: this.turn.observabilityContext
 			}).catch((err) => {
 				omDebug(`[OM:buffer] fire-and-forget buffer failed: ${err?.message}`);
-			});
+			}));
 			buffered = true;
 		}
 		const willObserveNow = statusSnapshot.shouldObserve && !hasIncompleteToolCalls;
@@ -22674,7 +22674,7 @@ var ObservationTurn = class {
 			const allMessages = getObservableMessages(this.messageList);
 			const record = this._record;
 			const unobservedMessages = this.om.getUnobservedMessages(allMessages, record);
-			if (unobservedMessages.length > 0) this.om.buffer({
+			if (unobservedMessages.length > 0) this.om.trackBackgroundWork(this.om.buffer({
 				threadId: this.threadId,
 				resourceId: this.resourceId,
 				messages: unobservedMessages,
@@ -22688,7 +22688,7 @@ var ObservationTurn = class {
 				skipMinimumTokenCheck: true
 			}).catch((err) => {
 				omDebug(`[OM:turn.end] idle buffer failed: ${err?.message}`);
-			});
+			}));
 		}
 		return { record: this._record };
 	}
@@ -23172,6 +23172,8 @@ var ReflectorRunner = class {
 		};
 		let reflectedTokens = 0;
 		let attemptNumber = 0;
+		/** Observations from the previous attempt, used to detect a no-progress ladder. */
+		let previousObservations;
 		while (currentLevel <= maxLevel) {
 			attemptNumber++;
 			const isRetry = attemptNumber > 1;
@@ -23245,6 +23247,11 @@ var ReflectorRunner = class {
 				omDebug(`[OM:callReflector] degenerate output persists at maxLevel=${maxLevel}, breaking`);
 				break;
 			}
+			if (!parsed.degenerate && previousObservations !== void 0 && parsed.observations === previousObservations) {
+				omDebug(`[OM:callReflector] attempt #${attemptNumber} returned output identical to the previous attempt; escalating cannot help, stopping the ladder`);
+				break;
+			}
+			previousObservations = parsed.observations;
 			if (streamContext?.writer) {
 				const failedMarker = createObservationFailedMarker({
 					cycleId: streamContext.cycleId,
@@ -24223,6 +24230,39 @@ var ObservationalMemory = class ObservationalMemory {
 	*/
 	recordInitializations = /* @__PURE__ */ new Map();
 	/**
+	* In-flight fire-and-forget background cycles (buffered observation, reflection).
+	* Tracked so callers can join them via `settled()` before closing a storage
+	* connection they own — otherwise a background cycle's tail statements race the close.
+	*/
+	pendingBackgroundWork = /* @__PURE__ */ new Set();
+	/**
+	* Register fire-and-forget background work so `settled()` can join it.
+	* Returns the original promise so callers keep their own rejection handling.
+	*/
+	trackBackgroundWork(work) {
+		const tracked = work.catch(() => {}).finally(() => {
+			this.pendingBackgroundWork.delete(tracked);
+		});
+		this.pendingBackgroundWork.add(tracked);
+		return work;
+	}
+	/**
+	* Resolve once all background observational-memory work started so far has finished.
+	*
+	* Background cycles enqueue further background work (a buffered observation can
+	* trigger a reflection, which runs nested agent streams), so this drains repeatedly
+	* until nothing is left rather than joining a single snapshot.
+	*/
+	async settled() {
+		const maxDrainRounds = 100;
+		for (let round = 0; round < maxDrainRounds; round++) {
+			if (this.pendingBackgroundWork.size === 0) return;
+			await Promise.allSettled([...this.pendingBackgroundWork]);
+		}
+		if (this.pendingBackgroundWork.size === 0) return;
+		omError(`[OM:settled] background work still pending after ${maxDrainRounds} drain rounds; giving up waiting`);
+	}
+	/**
 	* Acquire a lock for the given key, execute the callback, then release.
 	* If a lock is already held, waits for it to be released before acquiring.
 	*/
@@ -25223,7 +25263,7 @@ ${formattedMessages}
 		if (!this.buffering.isAsyncObservationEnabled()) return false;
 		const lockKey = this.buffering.getLockKey(opts.threadId, opts.resourceId);
 		const shouldTrigger = this.buffering.shouldTriggerAsyncObservation(opts.pendingTokens, lockKey, opts.record, this.storage, opts.threshold);
-		if (shouldTrigger) this.startAsyncBufferedObservation(opts.record, opts.threadId, opts.unobservedMessages, lockKey, opts.writer, opts.unbufferedPendingTokens, opts.requestContext);
+		if (shouldTrigger) this.trackBackgroundWork(this.startAsyncBufferedObservation(opts.record, opts.threadId, opts.unobservedMessages, lockKey, opts.writer, opts.unbufferedPendingTokens, opts.requestContext));
 		return shouldTrigger;
 	}
 	isMessageList(value) {
@@ -27053,6 +27093,25 @@ var Memory = class extends _mastra_core_memory.MastraMemory {
 	trackVectorCleanup(cleanup) {
 		this.pendingVectorCleanup = Promise.allSettled([this.pendingVectorCleanup, cleanup]).then(() => void 0);
 	}
+	/**
+	* Resolve once all background work this Memory started has finished: observational-memory
+	* cycles (buffered observation and reflection, including the nested agent runs they spawn)
+	* and vector cleanup from `deleteThread` / `deleteMessages`.
+	*
+	* Callers that own the storage connection should await this before closing it, otherwise
+	* background statements can race the close.
+	*
+	* ```ts
+	* await agent.generate('hello', { memory: { thread, resource } });
+	* await memory.settled();
+	* await store.close();
+	* ```
+	*/
+	async settled() {
+		await this.pendingVectorCleanup;
+		await (this._omEngine ? await this._omEngine : this._omEngineInstance)?.settled();
+		await this.pendingVectorCleanup;
+	}
 	/** The shared ObservationalMemory engine. Lazily created on first access. */
 	get omEngine() {
 		if (!this._omEngine) this._omEngine = this._initOMEngine().then((engine) => {
@@ -29050,4 +29109,4 @@ Object.defineProperty(exports, "wrapInObservationGroup", {
 	}
 });
 
-//# sourceMappingURL=src-_7ADtEwc.cjs.map
+//# sourceMappingURL=src-Drm6l_ZN.cjs.map