@mastra/memory 1.27.0 → 1.28.0-alpha.2

package/dist/{src-aTfbkxBl.js → src-kLYHhPqh.js}

@@ -18905,6 +18905,12 @@ function extractListItemsOnly(content) {
 */
 const MAX_OBSERVATION_LINE_CHARS = 1e4;
 /**
+* Minimum trimmed line length considered by the duplicate-line degenerate
+* check. Short lines (blank lines, separators, terse bullets) legitimately
+* repeat; long identical lines almost never do.
+*/
+const MIN_DUPLICATE_LINE_CHARS = 24;
+/**
 * Truncate individual observation lines that exceed the maximum length.
 */
 function sanitizeObservationLines(observations) {
@@ -18942,6 +18948,18 @@ function detectDegenerateRepetition(text) {
 	if (totalWindows > 5 && duplicateWindows / totalWindows > .4) return true;
 	const lines = text.split("\n");
 	for (const line of lines) if (line.length > 5e4) return true;
+	const seenLines = /* @__PURE__ */ new Map();
+	let duplicateLines = 0;
+	let totalCountedLines = 0;
+	for (const line of lines) {
+		const trimmed = line.trim();
+		if (trimmed.length < MIN_DUPLICATE_LINE_CHARS) continue;
+		totalCountedLines++;
+		const count = (seenLines.get(trimmed) ?? 0) + 1;
+		seenLines.set(trimmed, count);
+		if (count > 1) duplicateLines++;
+	}
+	if (totalCountedLines >= 20 && duplicateLines / totalCountedLines > .5) return true;
 	return false;
 }
 /**
@@ -18976,12 +18994,26 @@ function describeDegenerateOutput(text, snippetChars = 400) {
 		topWindow = window;
 	}
 	let longestLine = 0;
-	for (const line of text.split("\n")) if (line.length > longestLine) longestLine = line.length;
+	const seenLines = /* @__PURE__ */ new Map();
+	let duplicateLines = 0;
+	let totalCountedLines = 0;
+	for (const line of text.split("\n")) {
+		if (line.length > longestLine) longestLine = line.length;
+		const trimmed = line.trim();
+		if (trimmed.length < MIN_DUPLICATE_LINE_CHARS) continue;
+		totalCountedLines++;
+		const count = (seenLines.get(trimmed) ?? 0) + 1;
+		seenLines.set(trimmed, count);
+		if (count > 1) duplicateLines++;
+	}
 	const duplicateRatio = totalWindows > 0 ? (duplicateWindows / totalWindows).toFixed(2) : "n/a";
+	const duplicateLineRatio = totalCountedLines > 0 ? (duplicateLines / totalCountedLines).toFixed(2) : "n/a";
 	const parts = [
 		`length=${text.length}`,
 		`sampledWindows=${totalWindows}`,
 		`duplicateRatio=${duplicateRatio}`,
+		`duplicateLineRatio=${duplicateLineRatio}`,
+		`countedLines=${totalCountedLines}`,
 		`longestLine=${longestLine}`,
 		`topWindowCount=${topCount}`
 	];
@@ -19255,7 +19287,19 @@ async function withOmTracingSpan({ phase, model, inputTokens, requestContext, ob
 	});
 	const childObservabilityContext = createObservabilityContext({ currentSpan: span });
 	if (!span) return callback(childObservabilityContext);
-	return span.executeInContext(() => callback(childObservabilityContext));
+	return span.executeInContext(async () => {
+		try {
+			const result = await callback(childObservabilityContext);
+			span.end();
+			return result;
+		} catch (error) {
+			span.error({
+				error,
+				endSpan: true
+			});
+			throw error;
+		}
+	});
 }
 //#endregion
 //#region src/processors/observational-memory/observer-runner.ts
@@ -24392,10 +24436,11 @@ ObservationStrategy.create = ((om, opts) => {
 });
 //#endregion
 //#region src/processors/observational-memory/observation-turn/load-memory-context.ts
-async function loadMemoryContextMessages({ memory, messageList, threadId, resourceId }) {
+async function loadMemoryContextMessages({ memory, messageList, threadId, resourceId, runState }) {
 	const ctx = await memory.getContext({
 		threadId,
-		resourceId
+		resourceId,
+		runState
 	});
 	for (const msg of ctx.messages) if (msg.role !== "system") messageList.add(msg, "memory");
 	return ctx;
@@ -24853,10 +24898,11 @@ var ObservationTurn = class {
 	* If a MemoryContextProvider is passed, loads historical messages and adds
 	* them to the MessageList. Without a provider, only fetches/caches the record.
 	*/
-	async start(memory) {
+	async start(memory, runState) {
 		if (this._started) throw new Error("Turn already started");
 		this._started = true;
 		this._record = await this.om.getOrCreateRecord(this.threadId, this.resourceId);
+		runState?.set(`observational-memory:record:${this.threadId}:${this.resourceId ?? ""}`, this._record);
 		this._generationCountAtStart = this._record.generationCount;
 		this.memory = memory;
 		if (memory) {
@@ -24864,7 +24910,8 @@ var ObservationTurn = class {
 				memory,
 				messageList: this.messageList,
 				threadId: this.threadId,
-				resourceId: this.resourceId
+				resourceId: this.resourceId,
+				runState
 			});
 			this._context = {
 				messages: ctx.messages,
@@ -25426,8 +25473,6 @@ 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;
@@ -25501,11 +25546,6 @@ 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,
@@ -25589,52 +25629,51 @@ var ReflectorRunner = class {
 		this.storage.setBufferingReflectionFlag(record.id, true).catch((err) => {
 			omError("[OM] Failed to set buffering reflection flag", err);
 		});
-		reflectionHooks?.onReflectionStart?.();
-		const asyncOp = this.doAsyncBufferedReflection(record, bufferKey, writer, requestContext, observabilityContext, priorExtractedValues, mainAgent, sendSignal).then((outcome) => {
+		const asyncOp = (async () => {
+			let outcome;
+			let reflectionError;
 			try {
-				reflectionHooks?.onReflectionEnd?.({
-					usage: outcome?.usage,
-					...outcome?.providerMetadata ? { providerMetadata: outcome.providerMetadata } : {}
-				});
-			} catch (hookError) {
-				omError("[OM] onReflectionEnd hook failed after async buffered reflection", hookError);
-			}
-		}, async (error) => {
-			if (writer) try {
-				const failedMarker = createBufferingFailedMarker({
-					cycleId: `reflect-buf-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`,
-					operationType: "reflection",
-					startedAt: (/* @__PURE__ */ new Date()).toISOString(),
-					tokensAttempted: observationTokens,
-					error: error instanceof Error ? error.message : String(error),
-					recordId: record.id,
-					threadId: record.threadId ?? ""
-				});
-				writer.custom({
-					...failedMarker,
-					transient: true
-				}).catch(() => {});
-				await this.persistMarkerToStorage(failedMarker, record.threadId ?? "", record.resourceId ?? void 0);
-			} catch (markerError) {
-				omError("[OM] Failed to persist buffering-failed marker after async buffered reflection failure", markerError);
-			}
-			omError("[OM] Async buffered reflection failed", error);
-			try {
-				reflectionHooks?.onReflectionEnd?.({
-					usage: void 0,
-					error: error instanceof Error ? error : new Error(String(error))
+				await reflectionHooks?.onReflectionStart?.();
+				outcome = await this.doAsyncBufferedReflection(record, bufferKey, writer, requestContext, observabilityContext, priorExtractedValues, mainAgent, sendSignal);
+			} catch (error) {
+				reflectionError = error instanceof Error ? error : new Error(String(error));
+				if (writer) try {
+					const failedMarker = createBufferingFailedMarker({
+						cycleId: `reflect-buf-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`,
+						operationType: "reflection",
+						startedAt: (/* @__PURE__ */ new Date()).toISOString(),
+						tokensAttempted: observationTokens,
+						error: reflectionError.message,
+						recordId: record.id,
+						threadId: record.threadId ?? ""
+					});
+					writer.custom({
+						...failedMarker,
+						transient: true
+					}).catch(() => {});
+					await this.persistMarkerToStorage(failedMarker, record.threadId ?? "", record.resourceId ?? void 0);
+				} catch (markerError) {
+					omError("[OM] Failed to persist buffering-failed marker after async buffered reflection failure", markerError);
+				}
+				omError("[OM] Async buffered reflection failed", error);
+				BufferingCoordinator.lastBufferedBoundary.delete(bufferKey);
+			} finally {
+				try {
+					await reflectionHooks?.onReflectionEnd?.({
+						usage: outcome?.usage,
+						error: reflectionError,
+						...outcome?.providerMetadata ? { providerMetadata: outcome.providerMetadata } : {}
+					});
+				} catch (hookError) {
+					omError("[OM] onReflectionEnd hook failed after async buffered reflection", hookError);
+				}
+				BufferingCoordinator.asyncBufferingOps.delete(bufferKey);
+				unregisterOp(record.id, "bufferingReflection");
+				this.storage.setBufferingReflectionFlag(record.id, false).catch((err) => {
+					omError("[OM] Failed to clear buffering reflection flag", err);
 				});
-			} catch (hookError) {
-				omError("[OM] onReflectionEnd hook failed after async buffered reflection failure", hookError);
 			}
-			BufferingCoordinator.lastBufferedBoundary.delete(bufferKey);
-		}).finally(() => {
-			BufferingCoordinator.asyncBufferingOps.delete(bufferKey);
-			unregisterOp(record.id, "bufferingReflection");
-			this.storage.setBufferingReflectionFlag(record.id, false).catch((err) => {
-				omError("[OM] Failed to clear buffering reflection flag", err);
-			});
-		});
+		})();
 		BufferingCoordinator.asyncBufferingOps.set(bufferKey, asyncOp);
 	}
 	/**
@@ -25935,8 +25974,15 @@ var ReflectorRunner = class {
 		let reflectionUsage;
 		let reflectionProviderMetadata;
 		let reflectionError;
-		reflectionHooks?.onReflectionStart?.();
+		let lifecycleError;
 		try {
+			try {
+				await reflectionHooks?.onReflectionStart?.();
+			} catch (error) {
+				lifecycleError = error;
+				reflectionError = error instanceof Error ? error : new Error(String(error));
+				throw error;
+			}
 			const compressionStartLevel = await this.getCompressionStartLevel(requestContext);
 			const reflectResult = await this.call(record.activeObservations, void 0, streamContext, reflectThreshold, abortSignal, void 0, compressionStartLevel, requestContext, priorExtractedValues, observabilityContext, void 0, mainAgent, sendSignal);
 			reflectionUsage = reflectResult.usage;
@@ -26006,18 +26052,27 @@ var ReflectorRunner = class {
 				await this.persistMarkerToStorage(failedMarker, threadId, record.resourceId ?? void 0);
 			}
 			reflectionError = error instanceof Error ? error : new Error(String(error));
-			if (abortSignal?.aborted) throw error;
+			if (lifecycleError !== void 0 || abortSignal?.aborted) throw error;
 			omError("[OM] Reflection failed", error);
 		} finally {
-			await this.storage.setReflectingFlag(record.id, false);
 			try {
-				reflectionHooks?.onReflectionEnd?.({
+				await this.storage.setReflectingFlag(record.id, false);
+			} finally {
+				unregisterOp(record.id, "reflecting");
+			}
+			let endHookError;
+			try {
+				await reflectionHooks?.onReflectionEnd?.({
 					usage: reflectionUsage,
 					error: reflectionError,
 					...reflectionProviderMetadata ? { providerMetadata: reflectionProviderMetadata } : {}
 				});
-			} finally {
-				unregisterOp(record.id, "reflecting");
+			} catch (error) {
+				endHookError = error;
+			}
+			if (endHookError !== void 0) {
+				if (lifecycleError === void 0 && !abortSignal?.aborted) throw endHookError;
+				omDebug(`[OM:hooks] onReflectionEnd hook failed after cycle failure: ${endHookError instanceof Error ? endHookError.message : String(endHookError)}`);
 			}
 		}
 	}
@@ -26478,6 +26533,8 @@ var ObservationalMemory = class ObservationalMemory {
 	onIndexObservations;
 	/** Config-level lifecycle hooks fired for every observation/reflection cycle. */
 	hooks;
+	/** Execution policy for config-level hooks on manual and turn-synchronous cycles. */
+	hookExecution;
 	/** Observer agent runner — handles LLM calls for extracting observations. */
 	observer;
 	/** Reflector agent runner — handles LLM calls for compressing observations. */
@@ -26584,6 +26641,7 @@ var ObservationalMemory = class ObservationalMemory {
 		this.retrievalSearch = typeof config.retrieval === "object" && Boolean(config.retrieval.vector);
 		this.onIndexObservations = config.onIndexObservations;
 		this.hooks = config.hooks;
+		this.hookExecution = config.hookExecution ?? "non-blocking";
 		this.mastra = config.mastra;
 		this.memory = config.memory;
 		this.curationCadence = config.curationCadence;
@@ -28376,57 +28434,54 @@ ${formattedMessages}
 		};
 	}
 	/**
-	* Compose the config-level hooks with optional per-call hooks into a single
-	* `ObserveHooks` object, binding call context onto the config-level
-	* callbacks. Config-level hooks are guarded so a throwing consumer hook can
-	* never fail an observation/reflection cycle; per-call hooks keep their
-	* existing payloads and propagation semantics. Returns undefined when
-	* neither is configured, so call sites stay zero-cost.
-	*
-	* Used internally by every pipeline path (manual observe/reflect,
-	* turn-engine sync observation, async buffering); public only so the
-	* observation turn engine can thread config-level reflection hooks into
-	* reflector calls.
+	* Compose config-level hooks with optional per-call hooks and bind context to
+	* config-level callbacks. The configured policy applies unless a background
+	* path explicitly requests non-blocking invocation.
 	*
 	* @internal
 	*/
-	composeHooks(callHooks, context) {
+	composeHooks(callHooks, context, execution = "configured") {
 		const configHooks = this.hooks;
 		if (!configHooks) return callHooks;
 		if (!callHooks && !Object.keys(configHooks).length) return void 0;
-		const fireConfigHook = (name, arg) => {
+		const shouldAwaitConfig = execution === "configured" && this.hookExecution === "await";
+		const invokeConfigHook = async (name, arg) => {
 			const hook = configHooks[name];
 			if (!hook) return;
+			if (shouldAwaitConfig) {
+				await hook(arg);
+				return;
+			}
 			const logHookFailure = (error) => omDebug(`[OM:hooks] config-level ${name} hook failed: ${error instanceof Error ? error.message : String(error)}`);
 			try {
-				const out = hook(arg);
-				if (out && typeof out.then === "function") out.then(void 0, logHookFailure);
+				const result = hook(arg);
+				if (result && typeof result.then === "function") result.then(void 0, logHookFailure);
 			} catch (error) {
 				logHookFailure(error);
 			}
 		};
 		return {
-			onObservationStart: () => {
-				fireConfigHook("onObservationStart", context);
-				callHooks?.onObservationStart?.();
+			onObservationStart: async () => {
+				await invokeConfigHook("onObservationStart", context);
+				await callHooks?.onObservationStart?.();
 			},
-			onObservationEnd: (result) => {
-				fireConfigHook("onObservationEnd", {
+			onObservationEnd: async (result) => {
+				await invokeConfigHook("onObservationEnd", {
 					...context,
 					...result
 				});
-				callHooks?.onObservationEnd?.(result);
+				await callHooks?.onObservationEnd?.(result);
 			},
-			onReflectionStart: () => {
-				fireConfigHook("onReflectionStart", context);
-				callHooks?.onReflectionStart?.();
+			onReflectionStart: async () => {
+				await invokeConfigHook("onReflectionStart", context);
+				await callHooks?.onReflectionStart?.();
 			},
-			onReflectionEnd: (result) => {
-				fireConfigHook("onReflectionEnd", {
+			onReflectionEnd: async (result) => {
+				await invokeConfigHook("onReflectionEnd", {
 					...context,
 					...result
 				});
-				callHooks?.onReflectionEnd?.(result);
+				await callHooks?.onReflectionEnd?.(result);
 			}
 		};
 	}
@@ -28439,22 +28494,29 @@ ${formattedMessages}
 	* end hook reports `error` for failed cycles even though nothing throws.
 	*/
 	async runBufferedObservationCycle(context, run) {
-		const hooks = this.composeHooks(void 0, context);
-		hooks?.onObservationStart?.();
+		const hooks = this.composeHooks(void 0, context, "non-blocking");
 		let runResult;
 		let runError;
+		let lifecycleError;
 		try {
+			await hooks?.onObservationStart?.();
 			runResult = await run();
 			return runResult;
 		} catch (error) {
+			lifecycleError = error;
 			runError = error instanceof Error ? error : new Error(String(error));
 			throw error;
 		} finally {
-			hooks?.onObservationEnd?.({
-				usage: runResult?.usage,
-				error: runError ?? runResult?.error,
-				...runResult?.providerMetadata ? { providerMetadata: runResult.providerMetadata } : {}
-			});
+			try {
+				await hooks?.onObservationEnd?.({
+					usage: runResult?.usage,
+					error: runError ?? runResult?.error,
+					...runResult?.providerMetadata ? { providerMetadata: runResult.providerMetadata } : {}
+				});
+			} catch (endHookError) {
+				if (lifecycleError === void 0) throw endHookError;
+				omDebug(`[OM:hooks] async-buffer onObservationEnd hook failed after cycle failure: ${endHookError instanceof Error ? endHookError.message : String(endHookError)}`);
+			}
 		}
 	}
 	/**
@@ -28482,18 +28544,21 @@ ${formattedMessages}
 		let observed = false;
 		let observationUsage;
 		let observationProviderMetadata;
+		let observationError;
+		let lifecycleError;
+		let observationStarted = false;
 		let generationBefore = -1;
-		await this.withLock(lockKey, async () => {
-			const freshRecord = await this.getOrCreateRecord(threadId, resourceId);
-			generationBefore = freshRecord.generationCount;
-			const unobservedMessages = messages ? this.getUnobservedMessages(messages, freshRecord) : await this.loadMessagesFromStorage(threadId, resourceId, freshRecord.lastObservedAt ? new Date(freshRecord.lastObservedAt) : void 0);
-			if (!this.meetsObservationThreshold({
-				record: freshRecord,
-				unobservedTokens: await this.tokenCounter.countMessagesAsync(unobservedMessages)
-			})) return;
-			hooks?.onObservationStart?.();
-			let observationError;
-			try {
+		try {
+			await this.withLock(lockKey, async () => {
+				const freshRecord = await this.getOrCreateRecord(threadId, resourceId);
+				generationBefore = freshRecord.generationCount;
+				const unobservedMessages = messages ? this.getUnobservedMessages(messages, freshRecord) : await this.loadMessagesFromStorage(threadId, resourceId, freshRecord.lastObservedAt ? new Date(freshRecord.lastObservedAt) : void 0);
+				if (!this.meetsObservationThreshold({
+					record: freshRecord,
+					unobservedTokens: await this.tokenCounter.countMessagesAsync(unobservedMessages)
+				})) return;
+				observationStarted = true;
+				await hooks?.onObservationStart?.();
 				const result = await ObservationStrategy.create(this, {
 					record: freshRecord,
 					threadId,
@@ -28511,17 +28576,22 @@ ${formattedMessages}
 				observed = result.observed;
 				observationUsage = result.usage;
 				observationProviderMetadata = result.providerMetadata;
-			} catch (error) {
-				observationError = error instanceof Error ? error : new Error(String(error));
-				throw error;
-			} finally {
-				hooks?.onObservationEnd?.({
-					usage: observationUsage,
-					error: observationError,
-					...observationProviderMetadata ? { providerMetadata: observationProviderMetadata } : {}
-				});
-			}
-		});
+			});
+		} catch (error) {
+			lifecycleError = error;
+			observationError = error instanceof Error ? error : new Error(String(error));
+		}
+		if (observationStarted) try {
+			await hooks?.onObservationEnd?.({
+				usage: observationUsage,
+				error: observationError,
+				...observationProviderMetadata ? { providerMetadata: observationProviderMetadata } : {}
+			});
+		} catch (endHookError) {
+			if (lifecycleError === void 0) throw endHookError;
+			omDebug(`[OM:hooks] onObservationEnd hook failed after cycle failure: ${endHookError instanceof Error ? endHookError.message : String(endHookError)}`);
+		}
+		if (lifecycleError !== void 0) throw lifecycleError;
 		const record = await this.getOrCreateRecord(threadId, resourceId);
 		const reflected = record.generationCount > generationBefore && generationBefore >= 0;
 		if (observed) this.maybeTriggerCadenceCuration(threadId, resourceId, record, requestContext).catch((error) => {
@@ -28595,11 +28665,18 @@ ${formattedMessages}
 			resourceId,
 			trigger: "manual"
 		});
-		hooks?.onReflectionStart?.();
 		let reflectionUsage;
 		let reflectionProviderMetadata;
 		let reflectionError;
+		let lifecycleError;
 		try {
+			try {
+				await hooks?.onReflectionStart?.();
+			} catch (error) {
+				lifecycleError = error;
+				reflectionError = error instanceof Error ? error : new Error(String(error));
+				throw error;
+			}
 			const thread = await this.storage.getThreadById({ threadId });
 			const previousOmMetadata = getThreadOMMetadata(thread?.metadata);
 			const priorExtractedValues = getPriorExtractedValues(previousOmMetadata, this.reflectionConfig.extractors);
@@ -28637,6 +28714,7 @@ ${formattedMessages}
 			};
 		} catch (error) {
 			reflectionError = error instanceof Error ? error : new Error(String(error));
+			if (lifecycleError !== void 0) throw error;
 			omError("[OM] reflect() failed", error);
 			return {
 				reflected: false,
@@ -28644,13 +28722,25 @@ ${formattedMessages}
 				usage: void 0
 			};
 		} finally {
-			hooks?.onReflectionEnd?.({
-				usage: reflectionUsage,
-				error: reflectionError,
-				...reflectionProviderMetadata ? { providerMetadata: reflectionProviderMetadata } : {}
-			});
-			await this.storage.setReflectingFlag(record.id, false);
-			unregisterOp(record.id, "reflecting");
+			try {
+				await this.storage.setReflectingFlag(record.id, false);
+			} finally {
+				unregisterOp(record.id, "reflecting");
+			}
+			let endHookError;
+			try {
+				await hooks?.onReflectionEnd?.({
+					usage: reflectionUsage,
+					error: reflectionError,
+					...reflectionProviderMetadata ? { providerMetadata: reflectionProviderMetadata } : {}
+				});
+			} catch (error) {
+				endHookError = error;
+			}
+			if (endHookError !== void 0) {
+				if (lifecycleError === void 0) throw endHookError;
+				omDebug(`[OM:hooks] onReflectionEnd hook failed after cycle failure: ${endHookError instanceof Error ? endHookError.message : String(endHookError)}`);
+			}
 		}
 	}
 	/**
@@ -28946,7 +29036,9 @@ var ObservationalMemoryProcessor = class {
 			return messageList;
 		}
 		const { threadId, resourceId } = context;
-		const readOnly = parseMemoryRequestContext(requestContext)?.memoryConfig?.readOnly;
+		const memoryContext = parseMemoryRequestContext(requestContext);
+		const runState = memoryContext?.runState?.();
+		const readOnly = memoryContext?.memoryConfig?.readOnly;
 		const actorModelContext = model?.modelId ? {
 			provider: model.provider,
 			modelId: model.modelId,
@@ -28963,7 +29055,8 @@ var ObservationalMemoryProcessor = class {
 					memory: this.memory,
 					messageList,
 					threadId,
-					resourceId
+					resourceId,
+					runState
 				});
 				const systemMessages = ctx.omRecord ? await this.engine.buildContextSystemMessages({
 					threadId,
@@ -29004,7 +29097,7 @@ var ObservationalMemoryProcessor = class {
 				this.turn.sendStateSignal = args.sendStateSignal;
 				this.turn.agent = args.agent;
 				this.turn.requestContext = requestContext;
-				await this.turn.start(this.memory);
+				await this.turn.start(this.memory, runState);
 				if (stepNumber === 0 && this.temporalMarkers) await insertTemporalGapMarkers({
 					messageList,
 					sendSignal: args.sendSignal
@@ -30174,14 +30267,18 @@ ${workingMemory}`;
 		if (!this.threadConfig.workingMemory?.enabled) return null;
 		return extractWorkingMemoryContent(text)?.trim() ?? null;
 	}
-	async getWorkingMemory({ threadId, resourceId, memoryConfig }) {
+	async getWorkingMemory({ threadId, resourceId, memoryConfig, runState }) {
 		const config = this.getMergedThreadConfig(memoryConfig || {});
 		if (!config.workingMemory?.enabled) return null;
 		const scope = config.workingMemory.scope || "resource";
 		let workingMemoryData = null;
 		if (scope === "resource" && !resourceId) throw new Error("Memory error: Resource-scoped working memory is enabled but no resourceId was provided. Either provide a resourceId or explicitly set workingMemory.scope to 'thread'.");
-		if (scope === "resource" && resourceId) workingMemoryData = (await (await this.getMemoryStore()).getResourceById({ resourceId }))?.workingMemory || null;
-		else workingMemoryData = (await this.getThreadById({ threadId }))?.metadata?.workingMemory;
+		if (scope === "resource" && resourceId) {
+			const loadWorkingMemory = async () => {
+				return (await (await this.getMemoryStore()).getResourceById({ resourceId }))?.workingMemory || null;
+			};
+			workingMemoryData = runState ? await runState.load(`working-memory:resource:${resourceId}`, loadWorkingMemory) : await loadWorkingMemory();
+		} else workingMemoryData = (runState?.threadLoaded ? runState.thread : await this.getThreadById({ threadId }))?.metadata?.workingMemory;
 		if (!workingMemoryData) return null;
 		return workingMemoryData;
 	}
@@ -30213,16 +30310,18 @@ ${workingMemory}`;
 			content: (config.workingMemory.template || this.defaultWorkingMemoryTemplate).trim()
 		};
 	}
-	async getSystemMessage({ threadId, resourceId, memoryConfig }) {
+	async getSystemMessage({ threadId, resourceId, memoryConfig, runState }) {
 		const config = this.getMergedThreadConfig(memoryConfig);
 		this.assertWorkingMemoryStateSignalsCompatibility(config);
 		if (!config.workingMemory?.enabled) return null;
 		if (config.workingMemory?.useStateSignals) return null;
-		const workingMemoryTemplate = await this.getWorkingMemoryTemplate({ memoryConfig });
+		const loadTemplate = () => this.getWorkingMemoryTemplate({ memoryConfig });
+		const workingMemoryTemplate = runState ? await runState.load("working-memory:template", loadTemplate) : await loadTemplate();
 		const workingMemoryData = await this.getWorkingMemory({
 			threadId,
 			resourceId,
-			memoryConfig: config
+			memoryConfig: config,
+			runState
 		});
 		if (!workingMemoryTemplate) return null;
 		const workingMemoryConfig = config.workingMemory;
@@ -30255,7 +30354,7 @@ ${workingMemory}`;
 	* ```
 	*/
 	async getContext(opts) {
-		const { threadId, resourceId, memoryConfig } = opts;
+		const { threadId, resourceId, memoryConfig, runState } = opts;
 		const config = this.getMergedThreadConfig(memoryConfig);
 		const memoryStore = await this.getMemoryStore();
 		const systemParts = [];
@@ -30265,10 +30364,14 @@ ${workingMemory}`;
 		let otherThreadsContext;
 		const omEngine = await this.omEngine;
 		if (omEngine) {
-			omRecord = await omEngine.getRecord(threadId, resourceId);
+			const loadOmRecord = () => omEngine.getRecord(threadId, resourceId);
+			omRecord = runState ? await runState.load(`observational-memory:record:${threadId}:${resourceId ?? ""}`, loadOmRecord) : await loadOmRecord();
 			if (omRecord?.activeObservations) {
 				hasObservations = true;
-				if (omEngine.scope === "resource" && resourceId) otherThreadsContext = await omEngine.getOtherThreadsContext(resourceId, threadId);
+				if (omEngine.scope === "resource" && resourceId) {
+					const loadOtherThreadsContext = () => omEngine.getOtherThreadsContext(resourceId, threadId);
+					otherThreadsContext = runState ? await runState.load(`observational-memory:other-threads:${resourceId}:${threadId}:${omRecord.lastObservedAt ?? ""}`, loadOtherThreadsContext) : await loadOtherThreadsContext();
+				}
 				const obsSystemMessage = await omEngine.buildContextSystemMessage({
 					threadId,
 					resourceId,
@@ -30296,30 +30399,41 @@ ${workingMemory}`;
 		const workingMemoryMessage = await this.getSystemMessage({
 			threadId,
 			resourceId,
-			memoryConfig: config
+			memoryConfig: config,
+			runState
 		});
 		if (workingMemoryMessage) systemParts.push(workingMemoryMessage);
 		let messages;
 		if (omEngine && omRecord) {
 			const dateFilter = omRecord.lastObservedAt ? { dateRange: { start: new Date(new Date(omRecord.lastObservedAt).getTime() + 1) } } : void 0;
-			if (omEngine.scope === "resource" && resourceId) messages = (await memoryStore.listMessagesByResourceId({
-				resourceId,
-				orderBy: {
-					field: "createdAt",
-					direction: "ASC"
-				},
-				perPage: false,
-				filter: dateFilter
-			})).messages;
-			else messages = (await memoryStore.listMessages({
-				threadId,
-				orderBy: {
-					field: "createdAt",
-					direction: "ASC"
-				},
-				perPage: false,
-				filter: dateFilter
-			})).messages;
+			const boundary = omRecord.lastObservedAt ? new Date(omRecord.lastObservedAt).toISOString() : "";
+			if (omEngine.scope === "resource" && resourceId) {
+				const loadMessages = async () => {
+					return (await memoryStore.listMessagesByResourceId({
+						resourceId,
+						orderBy: {
+							field: "createdAt",
+							direction: "ASC"
+						},
+						perPage: false,
+						filter: dateFilter
+					})).messages;
+				};
+				messages = runState ? await runState.load(`observational-memory:messages:resource:${resourceId}:${boundary}`, loadMessages) : await loadMessages();
+			} else {
+				const loadMessages = async () => {
+					return (await memoryStore.listMessages({
+						threadId,
+						orderBy: {
+							field: "createdAt",
+							direction: "ASC"
+						},
+						perPage: false,
+						filter: dateFilter
+					})).messages;
+				};
+				messages = runState ? await runState.load(`observational-memory:messages:thread:${threadId}:${boundary}`, loadMessages) : await loadMessages();
+			}
 		} else {
 			const lastMessages = config.lastMessages;
 			if (lastMessages === false) messages = [];
@@ -31331,4 +31445,4 @@ Notes:
 //#endregion
 export { formatMessagesForObserver as A, OBSERVATION_CONTINUATION_HINT as B, SUMMARIZE_THREAD_DEFAULTS as C, buildObserverPrompt as D, OBSERVER_SYSTEM_PROMPT as E, parseAnchorId as F, ModelByInputTokens as G, KnowledgeSemanticIndexCoordinator as H, stripEphemeralAnchorIds as I, SUBCONSCIOUS_ACTIVITY_STATE_ID as J, SubconsciousCaptureExtractor as K, OBSERVATIONAL_MEMORY_DEFAULTS as L, optimizeObservationsForContext as M, parseObserverOutput as N, buildObserverSystemPrompt as O, injectAnchorIds as P, Extractor as Q, OBSERVATION_CONTEXT_INSTRUCTIONS as R, WorkingMemoryExtractor as S, TokenCounter as T, StaleKnowledgeSemanticIndexError as U, Subconscious as V, SubconsciousRemindExtractor as W, publishSubconsciousActivity as X, buildSubconsciousActivitySnapshot as Y, renderSubconsciousActivity as Z, reconcileObservationGroupsFromReflection as _, extractWorkingMemoryContent as a, wrapInObservationGroup as b, WORKING_MEMORY_STATE_ID as c, getObservationsAsOf as d, ObservationalMemoryProcessor as f, parseObservationGroups as g, deriveObservationGroupProvenance as h, WorkingMemory as i, hasCurrentTaskSection as j, extractCurrentTask as k, WORKING_MEMORY_STATE_PROCESSOR_ID as l, combineObservationGroupRanges as m, MessageHistory$1 as n, extractWorkingMemoryTags as o, ObservationalMemory as p, subconsciousCaptureSchema as q, SemanticRecall as r, removeWorkingMemoryTags as s, Memory as t, WorkingMemoryStateProcessor as u, renderObservationGroupsForReflection as v, summarizeConversation as w, deepMergeWorkingMemory as x, stripObservationGroups as y, OBSERVATION_CONTEXT_PROMPT as z };
 
-//# sourceMappingURL=src-aTfbkxBl.js.map
+//# sourceMappingURL=src-kLYHhPqh.js.map