@mastra/memory 1.26.1-alpha.2 → 1.26.1-alpha.4

package/dist/{src-DY9-_lul.cjs → src-kjoZv97r.cjs}

@@ -15635,6 +15635,16 @@ function sanitizeToolResultValue(value, seen = /* @__PURE__ */ new WeakMap()) {
 	}
 	return sanitizedObject;
 }
+/**
+* Serializes a tool result without truncating it.
+*
+* Token accounting must see the full result: the truncation applied by
+* {@link formatToolResultForObserver} exists to bound what the Observer LLM reads,
+* not to describe what the agent's provider context actually holds.
+*/
+function serializeToolResultForTokenCounting(value) {
+	return stringifyToolResult(value);
+}
 function stringifyToolResult(value) {
 	if (typeof value === "string") return value;
 	const sanitized = sanitizeToolResultValue(value);
@@ -18240,7 +18250,7 @@ var TokenCounter = class TokenCounter {
 		let tokens = 0;
 		const cacheParts = [];
 		const countJsonContentPart = (contentPart) => {
-			const formatted = formatToolResultForObserver(contentPart);
+			const formatted = serializeToolResultForTokenCounting(contentPart);
 			tokens += this.countString(formatted);
 			cacheParts.push({
 				type: "json",
@@ -18542,39 +18552,68 @@ var TokenCounter = class TokenCounter {
 		if (isImageAttachment) await resolveImageDimensionsAsync(part);
 		return this.countAttachmentPartSync(part);
 	}
+	/**
+	* Count the name and the arguments of a tool call. Every state before the tool produces an
+	* output holds the same call signature in the context window, so all of those states share
+	* these cache kinds. `buildEstimateKey` hashes the text, so a shared kind stays correct and
+	* keeps the estimate warm while the invocation moves from one state to the next.
+	*/
+	countToolCallSignature(part, invocation) {
+		let tokens = 0;
+		let overheadDelta = 0;
+		if (invocation.toolName) tokens += this.readOrPersistPartEstimate(part, "tool-call-name", invocation.toolName);
+		if (invocation.args) if (typeof invocation.args === "string") tokens += this.readOrPersistPartEstimate(part, "tool-call-args", invocation.args);
+		else {
+			const argsJson = JSON.stringify(invocation.args);
+			tokens += this.readOrPersistPartEstimate(part, "tool-call-args-json", argsJson);
+			overheadDelta -= 12;
+		}
+		return {
+			tokens,
+			overheadDelta
+		};
+	}
 	countNonAttachmentPart(part) {
 		let overheadDelta = 0;
-		let toolResultDelta = 0;
+		let extraMessageDelta = 0;
 		if (part.type === "text") return {
 			tokens: this.readOrPersistPartEstimate(part, "text", part.text),
 			overheadDelta,
-			toolResultDelta
+			extraMessageDelta
 		};
 		if (part.type === "tool-invocation") {
 			const invocation = part.toolInvocation;
+			const state = invocation.state;
 			let tokens = 0;
-			if (invocation.state === "call" || invocation.state === "partial-call") {
-				if (invocation.toolName) tokens += this.readOrPersistPartEstimate(part, `tool-${invocation.state}-name`, invocation.toolName);
-				if (invocation.args) if (typeof invocation.args === "string") tokens += this.readOrPersistPartEstimate(part, `tool-${invocation.state}-args`, invocation.args);
-				else {
-					const argsJson = JSON.stringify(invocation.args);
-					tokens += this.readOrPersistPartEstimate(part, `tool-${invocation.state}-args-json`, argsJson);
-					overheadDelta -= 12;
-				}
+			if (state === "call" || state === "partial-call" || state === "approval-requested") {
+				const signature = this.countToolCallSignature(part, invocation);
+				return {
+					tokens: signature.tokens,
+					overheadDelta: overheadDelta + signature.overheadDelta,
+					extraMessageDelta
+				};
+			}
+			if (state === "approval-responded") {
+				extraMessageDelta++;
+				const signature = this.countToolCallSignature(part, invocation);
+				tokens += signature.tokens;
+				overheadDelta += signature.overheadDelta;
+				const reason = invocation.approval?.reason;
+				if (reason) tokens += this.readOrPersistPartEstimate(part, "tool-approval-reason", reason);
 				return {
 					tokens,
 					overheadDelta,
-					toolResultDelta
+					extraMessageDelta
 				};
 			}
-			if (invocation.state === "result") {
-				toolResultDelta++;
+			if (state === "result") {
+				extraMessageDelta++;
 				const { value: resultForCounting, usingStoredModelOutput } = this.resolveToolResultForTokenCounting(part, invocation.result);
 				if (resultForCounting !== void 0) {
 					const contentTokens = this.countMultimodalToolResultContent(part, resultForCounting);
 					if (contentTokens !== void 0) tokens += contentTokens;
 					else {
-						const formattedResult = formatToolResultForObserver(resultForCounting);
+						const formattedResult = serializeToolResultForTokenCounting(resultForCounting);
 						tokens += this.readOrPersistPartEstimate(part, usingStoredModelOutput ? "tool-result-model-output-json" : "tool-result-json", formattedResult);
 					}
 					if (typeof resultForCounting !== "string") overheadDelta -= 12;
@@ -18582,47 +18621,45 @@ var TokenCounter = class TokenCounter {
 				return {
 					tokens,
 					overheadDelta,
-					toolResultDelta
+					extraMessageDelta
 				};
 			}
-			if (invocation.state === "output-denied") {
-				toolResultDelta++;
+			if (state === "output-denied") {
+				extraMessageDelta++;
 				const reason = invocation.approval?.reason ?? "Tool call was not approved by the user";
 				tokens += this.readOrPersistPartEstimate(part, "tool-result-denied", reason);
 				return {
 					tokens,
 					overheadDelta,
-					toolResultDelta
+					extraMessageDelta
 				};
 			}
-			if (invocation.state === "output-error") {
-				toolResultDelta++;
-				const errorText = invocation.errorText;
-				const errorMessage = typeof errorText === "string" ? errorText : "Tool execution failed";
+			if (state === "output-error") {
+				extraMessageDelta++;
+				const errorMessage = typeof invocation.errorText === "string" ? invocation.errorText : "Tool execution failed";
 				tokens += this.readOrPersistPartEstimate(part, "tool-result-error", errorMessage);
 				return {
 					tokens,
 					overheadDelta,
-					toolResultDelta
+					extraMessageDelta
 				};
 			}
-			throw new Error(`Unhandled tool-invocation state '${part.toolInvocation?.state}' in token counting for part type '${part.type}'`);
 		}
 		if (typeof part.type === "string" && part.type.startsWith("data-")) return {
 			tokens: 0,
 			overheadDelta,
-			toolResultDelta
+			extraMessageDelta
 		};
 		if (part.type === "reasoning") return {
 			tokens: 0,
 			overheadDelta,
-			toolResultDelta
+			extraMessageDelta
 		};
 		const serialized = serializePartForTokenCounting(part);
 		return {
 			tokens: this.readOrPersistPartEstimate(part, `part-${part.type}`, serialized),
 			overheadDelta,
-			toolResultDelta
+			extraMessageDelta
 		};
 	}
 	/**
@@ -18631,7 +18668,7 @@ var TokenCounter = class TokenCounter {
 	countMessage(message) {
 		let payloadTokens = this.countString(message.role);
 		let overhead = TokenCounter.TOKENS_PER_MESSAGE;
-		let toolResultCount = 0;
+		let extraMessageCount = 0;
 		if (typeof message.content === "string") payloadTokens += this.readOrPersistMessageEstimate(message, "message-content", message.content);
 		else if (message.content && typeof message.content === "object") {
 			if (message.content.content && !Array.isArray(message.content.parts)) payloadTokens += this.readOrPersistMessageEstimate(message, "content-content", message.content.content);
@@ -18644,16 +18681,16 @@ var TokenCounter = class TokenCounter {
 				const result = this.countNonAttachmentPart(part);
 				payloadTokens += result.tokens;
 				overhead += result.overheadDelta;
-				toolResultCount += result.toolResultDelta;
+				extraMessageCount += result.extraMessageDelta;
 			}
 		}
-		if (toolResultCount > 0) overhead += toolResultCount * TokenCounter.TOKENS_PER_MESSAGE;
+		if (extraMessageCount > 0) overhead += extraMessageCount * TokenCounter.TOKENS_PER_MESSAGE;
 		return Math.round(payloadTokens + overhead);
 	}
 	async countMessageAsync(message) {
 		let payloadTokens = this.countString(message.role);
 		let overhead = TokenCounter.TOKENS_PER_MESSAGE;
-		let toolResultCount = 0;
+		let extraMessageCount = 0;
 		if (typeof message.content === "string") payloadTokens += this.readOrPersistMessageEstimate(message, "message-content", message.content);
 		else if (message.content && typeof message.content === "object") {
 			if (message.content.content && !Array.isArray(message.content.parts)) payloadTokens += this.readOrPersistMessageEstimate(message, "content-content", message.content.content);
@@ -18666,10 +18703,10 @@ var TokenCounter = class TokenCounter {
 				const result = this.countNonAttachmentPart(part);
 				payloadTokens += result.tokens;
 				overhead += result.overheadDelta;
-				toolResultCount += result.toolResultDelta;
+				extraMessageCount += result.extraMessageDelta;
 			}
 		}
-		if (toolResultCount > 0) overhead += toolResultCount * TokenCounter.TOKENS_PER_MESSAGE;
+		if (extraMessageCount > 0) overhead += extraMessageCount * TokenCounter.TOKENS_PER_MESSAGE;
 		return Math.round(payloadTokens + overhead);
 	}
 	/**
@@ -19772,6 +19809,7 @@ function deepMergeWorkingMemory(existing, update) {
 	for (const key of Object.keys(update)) {
 		const updateValue = update[key];
 		const existingValue = result[key];
+		if (updateValue === void 0) continue;
 		if (updateValue === null) delete result[key];
 		else if (Array.isArray(updateValue)) result[key] = updateValue;
 		else if (typeof updateValue === "object" && updateValue !== null && typeof existingValue === "object" && existingValue !== null && !Array.isArray(existingValue)) result[key] = deepMergeWorkingMemory(existingValue, updateValue);
@@ -19824,7 +19862,7 @@ const updateWorkingMemoryTool = (memoryConfig) => {
 			version: 1,
 			vendor: "mastra",
 			validate: (value) => {
-				const memoryValue = !!value && typeof value === "object" && !Array.isArray(value) && "memory" in value ? value.memory : stripNullsFromOptional(value, jsonSchema);
+				const memoryValue = stripNullsFromOptional(!!value && typeof value === "object" && !Array.isArray(value) && "memory" in value ? value.memory : value, jsonSchema);
 				const result = validateMemory(memoryValue);
 				return result instanceof Promise ? result.then(toWrappedResult) : toWrappedResult(result);
 			},
@@ -19841,6 +19879,7 @@ const updateWorkingMemoryTool = (memoryConfig) => {
 		id: "update-working-memory",
 		description: schema ? useStateSignals ? `${stateSignalsPreamble} Data is merged with existing memory — only include fields you want to add or update.` : `Update the working memory with new information. Data is merged with existing memory - only include fields you want to add or update. To preserve existing data, omit the field entirely. Arrays are replaced entirely when provided, so pass the complete array or omit it to keep the existing values.` : useStateSignals ? `${stateSignalsPreamble} Pass the full updated Markdown blob as a string in the memory field.` : `Update the working memory with new information. Any data not included will be overwritten. Always pass data as string to the memory field. Never pass an object.`,
 		inputSchema,
+		...usesMergeSemantics ? { strict: false } : {},
 		execute: async (inputData, context) => {
 			const workingMemoryInput = inputData;
 			const threadId = context?.agent?.threadId;
@@ -21363,7 +21402,7 @@ var SyncObservationStrategy = class extends ObservationStrategy {
 			});
 			await this.storage.updateThread({
 				id: threadId,
-				title: shouldUpdateThreadTitle ? newTitle : thread.title ?? "",
+				...shouldUpdateThreadTitle ? { title: newTitle } : {},
 				metadata: newMetadata
 			});
 			if (shouldUpdateThreadTitle) threadUpdateMarker = createThreadUpdateMarker({
@@ -21555,7 +21594,7 @@ var AsyncBufferObservationStrategy = class extends ObservationStrategy {
 				});
 				await this.storage.updateThread({
 					id: threadId,
-					title: shouldUpdateThreadTitle ? newTitle : thread.title ?? "",
+					...shouldUpdateThreadTitle ? { title: newTitle } : {},
 					metadata: newMetadata
 				});
 				if (shouldUpdateThreadTitle) {
@@ -21900,7 +21939,7 @@ var ResourceScopedObservationStrategy = class extends ObservationStrategy {
 				});
 				await this.storage.updateThread({
 					id: update.threadId,
-					title: shouldUpdateThreadTitle ? newTitle : thread.title ?? "",
+					...shouldUpdateThreadTitle ? { title: newTitle } : {},
 					metadata: newMetadata
 				});
 				if (shouldUpdateThreadTitle) threadUpdateMarkers.push(createThreadUpdateMarker({
@@ -22843,7 +22882,6 @@ async function persistThreadExtractedValues(storage, extractors, threadId, value
 	});
 	await storage.updateThread({
 		id: threadId,
-		title: thread.title ?? "",
 		metadata: newMetadata
 	});
 }
@@ -25815,7 +25853,7 @@ ${formattedMessages}
 				const shouldUpdateThreadTitle = !!newTitle && newTitle.length >= 3 && newTitle !== oldTitle;
 				await this.storage.updateThread({
 					id: threadId,
-					title: shouldUpdateThreadTitle ? newTitle : thread.title ?? "",
+					...shouldUpdateThreadTitle ? { title: newTitle } : {},
 					metadata: newMetadata
 				});
 			}
@@ -26033,7 +26071,6 @@ ${formattedMessages}
 				});
 				await this.storage.updateThread({
 					id: threadId,
-					title: thread.title ?? "",
 					metadata: newMetadata
 				});
 			}
@@ -27179,7 +27216,6 @@ var Memory = class extends _mastra_core_memory.MastraMemory {
 					if (!thread) throw new Error(`Thread ${threadId} not found`);
 					await memoryStore.updateThread({
 						id: threadId,
-						title: thread.title || "",
 						metadata: {
 							...thread.metadata,
 							workingMemory
@@ -27268,7 +27304,6 @@ ${workingMemory}`;
 				if (!thread) throw new Error(`Thread ${threadId} not found`);
 				await memoryStore.updateThread({
 					id: threadId,
-					title: thread.title || "",
 					metadata: {
 						...thread.metadata,
 						workingMemory
@@ -28838,4 +28873,4 @@ Object.defineProperty(exports, "wrapInObservationGroup", {
 	}
 });
 
-//# sourceMappingURL=src-DY9-_lul.cjs.map
+//# sourceMappingURL=src-kjoZv97r.cjs.map