@mastra/memory 1.39.1-alpha.0 → 1.39.1-alpha.1

package/dist/{src-DQ-kLMpB.js → src-BUWL69LT.js}

@@ -36647,7 +36647,6 @@ var ObservationStep = class {
 		let observerExchange;
 		const step0Messages = this.stepNumber === 0 ? getObservableMessages(messageList) : [];
 		if (this.stepNumber === 0) {
-			this.turn.inputMessageIds = messageList.get.input.db().map((msg) => msg.id);
 			const activation = await om.activate({
 				threadId,
 				resourceId,
@@ -36778,7 +36777,7 @@ var ObservationStep = class {
 					drainedKeys.set(msg.id, /* @__PURE__ */ new Set([...seen ?? [], ...keys]));
 					return !seen || keys.some((key) => !seen.has(key));
 				}).map((msg) => msg.id);
-				inFlightIds = [.../* @__PURE__ */ new Set([...this.turn.inputMessageIds, ...unreadIds])];
+				inFlightIds = [...new Set(unreadIds)];
 				if (messagesToSave.length > 0) {
 					for (const msg of messagesToSave) messageList.add(msg, "memory");
 					try {
@@ -36855,7 +36854,7 @@ var ObservationStep = class {
 		});
 		if (!didThresholdCleanup) {
 			const fallbackCursor = this.turn.record.threadId ? getThreadOMMetadata((await om.getStorage().getThreadById({ threadId: this.turn.record.threadId }))?.metadata)?.lastObservedMessageCursor : void 0;
-			const pendingMessageIds = new Set([...messageList.get.input.db(), ...messageList.get.response.db()].map((msg) => msg.id).concat(inFlightIds ?? [], this.turn.inputMessageIds, this.changedSinceDrain(messageList)).filter(Boolean));
+			const pendingMessageIds = new Set([...messageList.get.input.db(), ...messageList.get.response.db()].map((msg) => msg.id).concat(inFlightIds ?? [], this.changedSinceDrain(messageList)).filter(Boolean));
 			filterObservedMessages({
 				messageList,
 				record: this.turn.record,
@@ -37141,12 +37140,6 @@ var ObservationTurn = class {
 	/** Generation count at turn start — used to detect if reflection happened during the turn. */
 	_generationCountAtStart = -1;
 	/**
-	* Ids of the messages this turn was started with (the user's request). Mid-turn
-	* observation may record them, but they stay in the live prompt until the turn ends.
-	* @internal Set by step 0.
-	*/
-	inputMessageIds = [];
-	/**
 	* Fingerprints of the non-marker parts each message had when a step last drained it. A message
 	* is unread step output only if it gained a part the model has not seen — including a tool result
 	* merged in place into an existing tool-invocation part — not when it only gained OM markers.
@@ -41917,7 +41910,6 @@ var ObservationalMemoryProcessor = class {
 				this.turn.requestContext = requestContext;
 				await this.turn.start(this.memory, runState);
 				if (stepNumber > 0) {
-					this.turn.inputMessageIds = messageList.getPersisted.input.db().map((message) => message.id);
 					if (this.memory?.getPersistedMessages) {
 						const responseIds = messageList.getPersisted.response.db().map((message) => message.id);
 						const snapshotMessages = new Map(messageList.get.all.db().map((message) => [message.id, message]));
@@ -42084,6 +42076,7 @@ var ObservationalMemoryProcessor = class {
 			writer: args.writer,
 			observabilityContext: getOmObservabilityContext(args)
 		}));
+		if (observed) args.rotateResponseMessageId?.();
 		return { retry: activated || observed };
 	}
 	get config() {
@@ -43169,9 +43162,7 @@ ${workingMemory}`;
 	async saveMessages({ messages, memoryConfig, observabilityContext }) {
 		const span = this.createMemorySpan("save", observabilityContext, void 0, { messageCount: messages.length });
 		try {
-			const updatedMessages = messages.filter((m) => m.role !== "system" && !isTransientSignalMessage(m)).map((m) => {
-				return this.updateMessageToHideWorkingMemoryV2(m);
-			}).filter((m) => Boolean(m));
+			const updatedMessages = this.prepareMessagesForPersistence(messages);
 			const config = this.getMergedThreadConfig(memoryConfig);
 			const dbMessages = new MessageList({ generateMessageId: () => this.generateId() }).add(updatedMessages, "memory").get.all.db();
 			const memoryStore = await this.getMemoryStore();
@@ -43277,12 +43268,17 @@ ${workingMemory}`;
 			throw error;
 		}
 	}
+	prepareMessagesForPersistence(messages) {
+		return messages.filter((m) => m.role !== "system" && !isTransientSignalMessage(m)).map((m) => this.updateMessageToHideWorkingMemoryV2(m)).filter((m) => m !== null);
+	}
 	updateMessageToHideWorkingMemoryV2(message) {
 		const newMessage = { ...message };
 		if (message.content && typeof message.content === "object" && !Array.isArray(message.content)) newMessage.content = { ...message.content };
 		if (typeof newMessage.content?.content === "string" && newMessage.content.content.length > 0) newMessage.content.content = removeWorkingMemoryTags(newMessage.content.content).trim();
 		if (Array.isArray(newMessage.content?.parts)) {
 			if (Array.isArray(newMessage.content.toolInvocations)) newMessage.content.toolInvocations = removeWorkingMemoryToolInvocations(newMessage.content.toolInvocations);
+			const lastOriginalPart = newMessage.content.parts.at(-1);
+			const sealMetadata = lastOriginalPart && "metadata" in lastOriginalPart ? lastOriginalPart.metadata : void 0;
 			newMessage.content.parts = removeWorkingMemoryToolInvocationParts(newMessage.content.parts).map((part) => {
 				if (part?.type === "text") {
 					const text = typeof part.text === "string" ? part.text : "";
@@ -43293,6 +43289,26 @@ ${workingMemory}`;
 				}
 				return part;
 			});
+			const lastIndex = newMessage.content.parts.length - 1;
+			const lastPart = newMessage.content.parts[lastIndex];
+			if (lastPart && sealMetadata && typeof sealMetadata === "object" && "mastra" in sealMetadata) {
+				const mastra = sealMetadata.mastra;
+				if (mastra && typeof mastra === "object" && "sealedAt" in mastra && typeof mastra.sealedAt === "number") {
+					const existingMetadata = "metadata" in lastPart ? lastPart.metadata : void 0;
+					const metadata = existingMetadata && typeof existingMetadata === "object" ? existingMetadata : {};
+					const existingMastra = "mastra" in metadata && metadata.mastra && typeof metadata.mastra === "object" ? metadata.mastra : {};
+					newMessage.content.parts[lastIndex] = {
+						...lastPart,
+						metadata: {
+							...metadata,
+							mastra: {
+								...existingMastra,
+								sealedAt: mastra.sealedAt
+							}
+						}
+					};
+				}
+			}
 			if (newMessage.content.parts.length === 0) {
 				if (!(typeof newMessage.content.content === "string" && newMessage.content.content.trim().length > 0)) return null;
 			}
@@ -43542,12 +43558,12 @@ ${workingMemory}`;
 		};
 	}
 	/**
-	* Raw message upsert — persist messages to storage without embedding or working memory processing.
+	* Raw message upsert — apply the persistence projection without embedding or extracting working memory.
 	* Used by the processor to save sealed messages before firing a background buffer operation.
 	*/
 	async persistMessages(messages) {
 		if (messages.length === 0) return;
-		const persistableMessages = messages.filter((m) => m.role !== "system" && !isTransientSignalMessage(m));
+		const persistableMessages = this.prepareMessagesForPersistence(messages);
 		if (persistableMessages.length === 0) return;
 		await (await this.getMemoryStore()).saveMessages({ messages: persistableMessages });
 	}
@@ -44029,27 +44045,75 @@ Notes:
 		if (messages.length === 0) return [];
 		const memoryStore = await this.getMemoryStore();
 		const config = this.getMergedThreadConfig(memoryConfig);
+		const existing = await memoryStore.listMessagesById({ messageIds: messages.map((m) => m.id) });
+		const existingById = new Map(existing.messages.map((m) => [m.id, m]));
+		const textForEmbeddingById = /* @__PURE__ */ new Map();
+		const textFields = ["content", "parts"];
+		const textOf = (content, fields) => {
+			for (const field of fields) {
+				const text = field === "content" ? content.content?.trim() ? content.content : null : (content.parts ?? []).filter((part) => part?.type === "text").map((part) => part.text).join(" ").trim();
+				if (text) return text;
+			}
+			return null;
+		};
+		messages = messages.flatMap((update) => {
+			const stored = existingById.get(update.id);
+			if (!stored) return [];
+			const effective = {
+				...stored,
+				...update,
+				content: update.content === void 0 ? stored.content : {
+					...stored.content,
+					...update.content,
+					metadata: {
+						...stored.content.metadata,
+						...update.content.metadata
+					}
+				}
+			};
+			const [projected] = this.prepareMessagesForPersistence([effective]);
+			if (!projected) return [];
+			if (update.content === void 0) {
+				existingById.set(update.id, effective);
+				return [update];
+			}
+			const content = {
+				...update.content,
+				...update.content.parts !== void 0 ? { parts: projected.content.parts } : {},
+				...update.content.content !== void 0 ? { content: projected.content.content } : {},
+				...update.content.toolInvocations !== void 0 ? { toolInvocations: projected.content.toolInvocations } : {}
+			};
+			existingById.set(update.id, {
+				...effective,
+				content: {
+					...effective.content,
+					...content
+				}
+			});
+			const patched = textFields.filter((field) => update.content?.[field] !== void 0);
+			if (patched.length > 0) {
+				const omitted = textFields.filter((field) => !patched.includes(field));
+				const text = textOf(projected.content, patched) ?? (textOf(update.content, patched) ? textOf(projected.content, omitted) : null);
+				textForEmbeddingById.set(update.id, text);
+			}
+			return [{
+				...update,
+				content
+			}];
+		});
+		if (messages.length === 0) return [];
 		if (this.vector && config.semanticRecall) {
-			const messagesWithContent = messages.filter((m) => m.content !== void 0);
-			if (messagesWithContent.length > 0) {
-				const existingMessagesResult = await memoryStore.listMessagesById({ messageIds: messagesWithContent.map((m) => m.id) });
-				const existingMessagesMap = new Map(existingMessagesResult.messages.map((m) => [m.id, m]));
+			const messagesWithTextUpdates = [...textForEmbeddingById.keys()].map((id) => ({ id }));
+			if (messagesWithTextUpdates.length > 0) {
+				const existingMessagesMap = new Map(existing.messages.map((m) => [m.id, m]));
 				const embeddingData = [];
 				let dimension;
 				const messageIdsWithNewEmbeddings = /* @__PURE__ */ new Set();
 				const messageIdsWithClearedContent = /* @__PURE__ */ new Set();
-				await Promise.all(messagesWithContent.map(async (message) => {
+				await Promise.all(messagesWithTextUpdates.map(async (message) => {
 					const existingMessage = existingMessagesMap.get(message.id);
 					if (!existingMessage) return;
-					let textForEmbedding = null;
-					const content = message.content;
-					if (content) {
-						if ("content" in content && content.content && typeof content.content === "string" && content.content.trim() !== "") textForEmbedding = content.content;
-						else if ("parts" in content && content.parts && Array.isArray(content.parts) && content.parts.length > 0) {
-							const joined = content.parts.filter((part) => part?.type === "text").map((part) => part.text).join(" ").trim();
-							if (joined) textForEmbedding = joined;
-						}
-					}
+					const textForEmbedding = textForEmbeddingById.get(message.id) ?? null;
 					if (textForEmbedding) {
 						let embeddings = [];
 						let documents;
@@ -44796,4 +44860,4 @@ Notes:
 //#endregion
 export { renderSubconsciousActivity as $, buildObserverSystemPrompt as A, OBSERVATION_CONTEXT_INSTRUCTIONS as B, WorkingMemoryExtractor as C, TokenCounter as D, summarizeConversation as E, parseObserverOutput as F, Subconscious as G, OBSERVATION_CONTEXT_PROMPT_THREAD as H, injectAnchorIds as I, SubconsciousRemindExtractor as J, KnowledgeSemanticIndexCoordinator as K, parseAnchorId as L, formatMessagesForObserver as M, hasCurrentTaskSection as N, OBSERVER_SYSTEM_PROMPT as O, optimizeObservationsForContext as P, publishSubconsciousActivity as Q, stripEphemeralAnchorIds as R, wrapInObservationGroup as S, SUMMARIZE_THREAD_DEFAULTS as T, OBSERVATION_CONTINUATION_HINT as U, OBSERVATION_CONTEXT_PROMPT as V, getObservationContextPrompt as W, SUBCONSCIOUS_ACTIVITY_STATE_ID as X, ModelByInputTokens as Y, buildSubconsciousActivitySnapshot as Z, deriveObservationGroupProvenance as _, extractWorkingMemoryContent as a, renderObservationGroupsForReflection as b, getObservationsAsOf as c, AUTO_MODEL_BY_PROVIDER as d, Extractor as et, resolveAutoModelId as f, combineObservationGroupRanges as g, WorkingMemoryStateProcessor as h, WorkingMemory as i, extractCurrentTask as j, buildObserverPrompt as k, ObservationalMemoryProcessor as l, WORKING_MEMORY_STATE_PROCESSOR_ID as m, MessageHistory$1 as n, extractWorkingMemoryTags as o, WORKING_MEMORY_STATE_ID as p, StaleKnowledgeSemanticIndexError as q, SemanticRecall as r, removeWorkingMemoryTags as s, Memory as t, ObservationalMemory as u, parseObservationGroups as v, deepMergeWorkingMemory as w, stripObservationGroups as x, reconcileObservationGroupsFromReflection as y, OBSERVATIONAL_MEMORY_DEFAULTS as z };
 
-//# sourceMappingURL=src-DQ-kLMpB.js.map
+//# sourceMappingURL=src-BUWL69LT.js.map