@mastra/server 1.80.0-alpha.2 → 1.80.0-alpha.3

package/dist/{dist-bT2343kX.cjs → dist-C9Jwt1f0.cjs}

@@ -1,3 +1,3 @@
 require("./rolldown-runtime-BwC_XMyb.cjs");
-const require_dist = require("./dist-U8ZBR8z5.cjs");
+const require_dist = require("./dist-CukR8zi9.cjs");
 exports.openai = require_dist.openai;

package/dist/{dist-U8ZBR8z5.cjs → dist-CukR8zi9.cjs}

@@ -8619,7 +8619,7 @@ function splitLines(text) {
 	return result;
 }
 //#endregion
-//#region ../memory/dist/src-DQ-kLMpB.js
+//#region ../memory/dist/src-BUWL69LT.js
 var __defProp$3 = Object.defineProperty;
 var __exportAll = (all, no_symbols) => {
 	let target = {};
@@ -38944,7 +38944,7 @@ async function resolveImageDimensionsAsync(part) {
 	const url = resolveHttpAssetUrl(getObjectValue(part, "image") ?? getObjectValue(part, "data"));
 	if (!url) return existing;
 	try {
-		const probeImageSize = (await Promise.resolve().then(() => /* @__PURE__ */ require_rolldown_runtime.__toESM(require("./probe-image-size-mZ2ldE98.cjs").default, 1))).default;
+		const probeImageSize = (await Promise.resolve().then(() => /* @__PURE__ */ require_rolldown_runtime.__toESM(require("./probe-image-size-l1uuhbob.cjs").default, 1))).default;
 		const probed = await probeImageSize(url, {
 			open_timeout: REMOTE_IMAGE_PROBE_TIMEOUT_MS,
 			response_timeout: REMOTE_IMAGE_PROBE_TIMEOUT_MS,
@@ -45086,7 +45086,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,
@@ -45217,7 +45216,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 {
@@ -45294,7 +45293,7 @@ var ObservationStep = class {
 		});
 		if (!didThresholdCleanup) {
 			const fallbackCursor = this.turn.record.threadId ? (0, _mastra_core_memory.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,
@@ -45578,12 +45577,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.
@@ -50340,7 +50333,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]));
@@ -50507,6 +50499,7 @@ var ObservationalMemoryProcessor = class {
 			writer: args.writer,
 			observabilityContext: getOmObservabilityContext(args)
 		}));
+		if (observed) args.rotateResponseMessageId?.();
 		return { retry: activated || observed };
 	}
 	get config() {
@@ -51571,9 +51564,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 _mastra_core_agent.MessageList({ generateMessageId: () => this.generateId() }).add(updatedMessages, "memory").get.all.db();
 			const memoryStore = await this.getMemoryStore();
@@ -51679,12 +51670,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 : "";
@@ -51695,6 +51691,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;
 			}
@@ -51944,12 +51960,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 });
 	}
@@ -52431,27 +52447,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;
@@ -53988,7 +54052,7 @@ const createModelInstance = async (provider, modelId, version = "v2") => {
 	try {
 		const providerFn = version === `v1` ? { v1: {
 			openai: async () => {
-				const { openai } = await Promise.resolve().then(() => require("./dist-bT2343kX.cjs"));
+				const { openai } = await Promise.resolve().then(() => require("./dist-C9Jwt1f0.cjs"));
 				return openai(modelId);
 			},
 			anthropic: async () => {
@@ -61725,4 +61789,4 @@ Object.defineProperty(exports, "require_sync", {
 	}
 });
 
-//# sourceMappingURL=dist-U8ZBR8z5.cjs.map
+//# sourceMappingURL=dist-CukR8zi9.cjs.map