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

package/dist/{chunk-GXDPND6K.js → chunk-ZEKCVX4E.js}

@@ -4,15 +4,15 @@ import { resolveModelConfig } from '@mastra/core/llm';
 import { getThreadOMMetadata, setThreadOMMetadata, parseMemoryRequestContext } from '@mastra/core/memory';
 import { MessageHistory } from '@mastra/core/processors';
 import xxhash from 'xxhash-wasm';
-import { appendFileSync, mkdirSync, writeFileSync } from 'fs';
+import { appendFileSync, writeFileSync, mkdirSync } from 'fs';
 import { join } from 'path';
 import { randomBytes, createHash, randomUUID } from 'crypto';
 import { estimateTokenCount } from 'tokenx';
 import { Agent } from '@mastra/core/agent';
 import { getOrCreateSpan, EntityType, SpanType, createObservabilityContext } from '@mastra/core/observability';
+import { inspect } from 'util';
 import { AsyncLocalStorage } from 'async_hooks';
 import imageSize from 'image-size';
-import { inspect } from 'util';
 
 var OM_DEBUG_LOG = process.env.OM_DEBUG ? join(process.cwd(), "om-debug.log") : null;
 function omDebug(msg) {
@@ -2922,11 +2922,13 @@ var OBSERVER_IMAGE_FILE_EXTENSIONS = /* @__PURE__ */ new Set([
   "heif",
   "avif"
 ]);
-function formatObserverTimestamp(createdAt) {
-  return createdAt ? new Date(createdAt).toLocaleString("en-US", {
-    year: "numeric",
-    month: "short",
-    day: "numeric",
+function formatObserverDate(createdAt) {
+  return createdAt ? `${createdAt.toLocaleDateString("en-US", {
+    month: "short"
+  })} ${createdAt.getDate()} ${createdAt.getFullYear()}` : "";
+}
+function formatObserverTime(createdAt) {
+  return createdAt ? createdAt.toLocaleTimeString("en-US", {
     hour: "numeric",
     minute: "2-digit",
     hour12: true
@@ -3017,19 +3019,72 @@ function formatObserverAttachmentPlaceholder(part, counter) {
   const label = resolveObserverAttachmentLabel(part);
   return label ? `[${attachmentType} #${attachmentId}: ${label}]` : `[${attachmentType} #${attachmentId}]`;
 }
+function formatObserverPartLine(title, body, time, previousTime) {
+  const timeLabel = time && time !== previousTime ? ` (${time})` : "";
+  return `${title}${timeLabel}: ${body}`;
+}
+function normalizeObserverCreatedAt(createdAt) {
+  if (createdAt instanceof Date) {
+    if (Number.isNaN(createdAt.getTime())) {
+      return void 0;
+    }
+    return createdAt;
+  }
+  if (typeof createdAt === "number" || typeof createdAt === "string") {
+    const date = new Date(createdAt);
+    if (Number.isNaN(date.getTime())) {
+      return void 0;
+    }
+    return date;
+  }
+  return void 0;
+}
+function formatObserverLines(lines, context = {}) {
+  const output = [];
+  let previousDate = context.previousDate;
+  let previousTime = context.previousTime;
+  for (const line of lines) {
+    if (line.date && line.date !== previousDate) {
+      output.push(`${line.date}:`);
+      previousDate = line.date;
+      previousTime = void 0;
+    }
+    output.push(formatObserverPartLine(line.title, line.body, line.time, previousTime));
+    previousTime = line.time || previousTime;
+  }
+  return {
+    text: output.join("\n"),
+    context: { previousDate, previousTime }
+  };
+}
 function formatObserverMessage(msg, counter, options) {
   const maxLen = options?.maxPartLength;
   const maxToolResultTokens = options?.maxToolResultTokens ?? DEFAULT_OBSERVER_TOOL_RESULT_MAX_TOKENS;
-  const timestamp = formatObserverTimestamp(msg.createdAt);
   const role = msg.role.charAt(0).toUpperCase() + msg.role.slice(1);
-  const timestampStr = timestamp ? ` (${timestamp})` : "";
   const attachments = [];
-  let content = "";
+  const messageCreatedAt = normalizeObserverCreatedAt(msg.createdAt);
+  let lines = [];
+  const pushLine = (title, body, createdAt) => {
+    if (!body) {
+      return;
+    }
+    const normalizedCreatedAt = normalizeObserverCreatedAt(createdAt) ?? messageCreatedAt;
+    lines.push({
+      date: formatObserverDate(normalizedCreatedAt),
+      time: formatObserverTime(normalizedCreatedAt),
+      title,
+      body
+    });
+  };
   if (typeof msg.content === "string") {
-    content = maybeTruncate(msg.content, maxLen);
+    pushLine(role, maybeTruncate(msg.content, maxLen), messageCreatedAt);
   } else if (msg.content?.parts && Array.isArray(msg.content.parts) && msg.content.parts.length > 0) {
-    content = msg.content.parts.map((part) => {
-      if (part.type === "text") return maybeTruncate(part.text, maxLen);
+    msg.content.parts.forEach((part) => {
+      const partCreatedAt = normalizeObserverCreatedAt(part.createdAt) ?? messageCreatedAt;
+      if (part.type === "text") {
+        pushLine(role, maybeTruncate(part.text, maxLen), partCreatedAt);
+        return;
+      }
       if (part.type === "tool-invocation") {
         const inv = part.toolInvocation;
         if (inv.state === "result") {
@@ -3037,19 +3092,24 @@ function formatObserverMessage(msg, counter, options) {
             part,
             inv.result
           );
-          const resultStr = formatToolResultForObserver(resultForObserver, { maxTokens: maxToolResultTokens });
-          return `[Tool Result: ${inv.toolName}]
-${maybeTruncate(resultStr, maxLen)}`;
+          pushLine(
+            `Tool Result ${inv.toolName}`,
+            maybeTruncate(formatToolResultForObserver(resultForObserver, { maxTokens: maxToolResultTokens }), maxLen),
+            partCreatedAt
+          );
+          return;
         }
-        const argsStr = JSON.stringify(inv.args, null, 2);
-        return `[Tool Call: ${inv.toolName}]
-${maybeTruncate(argsStr, maxLen)}`;
+        pushLine(`Tool Call ${inv.toolName}`, maybeTruncate(JSON.stringify(inv.args, null, 2), maxLen), partCreatedAt);
+        return;
       }
       const partType = part.type;
       if (partType === "reasoning") {
         const reasoning = part.reasoning;
-        if (reasoning) return maybeTruncate(reasoning, maxLen);
-        return "";
+        if (!reasoning) {
+          return;
+        }
+        pushLine("Reasoning", maybeTruncate(reasoning, maxLen), partCreatedAt);
+        return;
       }
       if (partType === "image" || partType === "file") {
         const attachment = part;
@@ -3057,40 +3117,58 @@ ${maybeTruncate(argsStr, maxLen)}`;
         if (inputAttachment) {
           attachments.push(inputAttachment);
         }
-        return formatObserverAttachmentPlaceholder(attachment, counter);
+        pushLine(
+          partType === "image" ? "Image" : "File",
+          formatObserverAttachmentPlaceholder(attachment, counter),
+          partCreatedAt
+        );
       }
-      if (partType?.startsWith("data-")) return "";
-      return "";
-    }).filter(Boolean).join("\n");
+    });
   } else if (msg.content?.content) {
-    content = maybeTruncate(msg.content.content, maxLen);
+    pushLine(role, maybeTruncate(msg.content.content, maxLen), messageCreatedAt);
   }
-  if (!content && attachments.length === 0) {
-    return { text: "", attachments };
+  if (lines.length === 0 && attachments.length === 0) {
+    return { lines: [], attachments };
   }
   return {
-    text: `**${role}${timestampStr}:**
-${content}`,
+    lines,
     attachments
   };
 }
 function formatMessagesForObserver(messages, options) {
   const counter = { nextImageId: 1, nextFileId: 1 };
-  return messages.map((msg) => formatObserverMessage(msg, counter, options).text).filter(Boolean).join("\n\n---\n\n");
+  const sections = [];
+  let context = {};
+  for (const message of messages) {
+    const formatted = formatObserverMessage(message, counter, options);
+    if (formatted.lines.length === 0) {
+      continue;
+    }
+    const rendered = formatObserverLines(formatted.lines, context);
+    if (!rendered.text) {
+      continue;
+    }
+    sections.push(rendered.text);
+    context = rendered.context;
+  }
+  return sections.join("\n");
+}
+function appendFormattedObserverMessage(content, formatted, context) {
+  const rendered = formatObserverLines(formatted.lines, context);
+  if (rendered.text) {
+    content.push({ type: "text", text: rendered.text });
+  }
+  content.push(...formatted.attachments);
+  return rendered.context;
 }
 function buildObserverHistoryMessage(messages, options) {
   const counter = { nextImageId: 1, nextFileId: 1 };
   const content = [{ type: "text", text: "## New Message History to Observe\n\n" }];
-  let visibleCount = 0;
+  let context = {};
   messages.forEach((message) => {
     const formatted = formatObserverMessage(message, counter, options);
-    if (!formatted.text && formatted.attachments.length === 0) return;
-    if (visibleCount > 0) {
-      content.push({ type: "text", text: "\n\n---\n\n" });
-    }
-    content.push({ type: "text", text: formatted.text });
-    content.push(...formatted.attachments);
-    visibleCount++;
+    if (formatted.lines.length === 0 && formatted.attachments.length === 0) return;
+    context = appendFormattedObserverMessage(content, formatted, context);
   });
   return {
     role: "user",
@@ -3120,18 +3198,15 @@ The following messages are from ${threadOrder.length} different conversation thr
     const messages = messagesByThread.get(threadId);
     if (!messages || messages.length === 0) return;
     const threadContent = [];
-    let visibleCount = 0;
+    let context = {};
+    let hasVisibleContent = false;
     messages.forEach((message) => {
       const formatted = formatObserverMessage(message, counter, options);
-      if (!formatted.text && formatted.attachments.length === 0) return;
-      if (visibleCount > 0) {
-        threadContent.push({ type: "text", text: "\n\n---\n\n" });
-      }
-      threadContent.push({ type: "text", text: formatted.text });
-      threadContent.push(...formatted.attachments);
-      visibleCount++;
+      if (formatted.lines.length === 0 && formatted.attachments.length === 0) return;
+      context = appendFormattedObserverMessage(threadContent, formatted, context);
+      hasVisibleContent = true;
     });
-    if (visibleCount === 0) return;
+    if (!hasVisibleContent) return;
     content.push({ type: "text", text: `<thread id="${threadId}">
 ` });
     content.push(...threadContent);
@@ -4587,101 +4662,423 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
     }
   }
 };
-var IMAGE_FILE_EXTENSIONS = /* @__PURE__ */ new Set([
-  "png",
-  "jpg",
-  "jpeg",
-  "webp",
-  "gif",
-  "bmp",
-  "tiff",
-  "tif",
-  "heic",
-  "heif",
-  "avif"
-]);
-var TOKEN_ESTIMATE_CACHE_VERSION = 6;
-var DEFAULT_IMAGE_ESTIMATOR = {
-  baseTokens: 85,
-  tileTokens: 170,
-  fallbackTiles: 4
-};
-var GOOGLE_LEGACY_IMAGE_TOKENS_PER_TILE = 258;
-var GOOGLE_GEMINI_3_IMAGE_TOKENS_BY_RESOLUTION = {
-  low: 280,
-  medium: 560,
-  high: 1120,
-  ultra_high: 2240,
-  unspecified: 1120
-};
-var ANTHROPIC_IMAGE_TOKENS_PER_PIXEL = 1 / 750;
-var ANTHROPIC_IMAGE_MAX_LONG_EDGE = 1568;
-var GOOGLE_MEDIA_RESOLUTION_VALUES = /* @__PURE__ */ new Set([
-  "low",
-  "medium",
-  "high",
-  "ultra_high",
-  "unspecified"
-]);
-var ATTACHMENT_COUNT_TIMEOUT_MS = 2e4;
-var REMOTE_IMAGE_PROBE_TIMEOUT_MS = 2500;
-var PROVIDER_API_KEY_ENV_VARS = {
-  openai: ["OPENAI_API_KEY"],
-  google: ["GOOGLE_GENERATIVE_AI_API_KEY", "GOOGLE_API_KEY"],
-  anthropic: ["ANTHROPIC_API_KEY"]
-};
-function getPartMastraMetadata(part) {
-  return part.providerMetadata?.mastra;
+function getOmReproCaptureDir() {
+  return process.env.OM_REPRO_CAPTURE_DIR ?? ".mastra-om-repro";
 }
-function ensurePartMastraMetadata(part) {
-  const typedPart = part;
-  typedPart.providerMetadata ??= {};
-  typedPart.providerMetadata.mastra ??= {};
-  return typedPart.providerMetadata.mastra;
+function sanitizeCapturePathSegment(value) {
+  const sanitized = value.replace(/[\\/]+/g, "_").replace(/\.{2,}/g, "_").trim();
+  return sanitized.length > 0 ? sanitized : "unknown-thread";
 }
-function getContentMastraMetadata(content) {
-  if (!content || typeof content !== "object") {
-    return void 0;
+function isOmReproCaptureEnabled() {
+  return process.env.OM_REPRO_CAPTURE === "1";
+}
+function safeCaptureJson(value) {
+  return JSON.parse(
+    JSON.stringify(value, (_key, current) => {
+      if (typeof current === "bigint") return current.toString();
+      if (typeof current === "function") return "[function]";
+      if (typeof current === "symbol") return current.toString();
+      if (current instanceof Error) return { name: current.name, message: current.message, stack: current.stack };
+      if (current instanceof Set) return { __type: "Set", values: Array.from(current.values()) };
+      if (current instanceof Map) return { __type: "Map", entries: Array.from(current.entries()) };
+      return current;
+    })
+  );
+}
+function safeCaptureJsonOrError(value) {
+  try {
+    return { ok: true, value: safeCaptureJson(value) };
+  } catch (error) {
+    return {
+      ok: false,
+      error: safeCaptureJson({
+        message: error instanceof Error ? error.message : String(error),
+        stack: error instanceof Error ? error.stack : void 0,
+        inspected: inspect(value, { depth: 3, maxArrayLength: 20, breakLength: 120 })
+      })
+    };
   }
-  return content.metadata?.mastra;
 }
-function ensureContentMastraMetadata(content) {
-  if (!content || typeof content !== "object") {
+function formatCaptureDate(value) {
+  if (!value) return void 0;
+  if (value instanceof Date) return value.toISOString();
+  try {
+    return new Date(value).toISOString();
+  } catch {
     return void 0;
   }
-  const typedContent = content;
-  typedContent.metadata ??= {};
-  typedContent.metadata.mastra ??= {};
-  return typedContent.metadata.mastra;
 }
-function getMessageMastraMetadata(message) {
-  return message.metadata?.mastra;
+function summarizeOmTurn(value) {
+  if (!value || typeof value !== "object") {
+    return value;
+  }
+  const turn = value;
+  return {
+    __type: "ObservationTurn",
+    threadId: turn.threadId,
+    resourceId: turn.resourceId,
+    started: turn._started,
+    ended: turn._ended,
+    generationCountAtStart: turn._generationCountAtStart,
+    record: turn._record ? {
+      id: turn._record.id,
+      scope: turn._record.scope,
+      threadId: turn._record.threadId,
+      resourceId: turn._record.resourceId,
+      createdAt: formatCaptureDate(turn._record.createdAt),
+      updatedAt: formatCaptureDate(turn._record.updatedAt),
+      lastObservedAt: formatCaptureDate(turn._record.lastObservedAt),
+      generationCount: turn._record.generationCount,
+      observationTokenCount: turn._record.observationTokenCount,
+      pendingMessageTokens: turn._record.pendingMessageTokens,
+      isBufferingObservation: turn._record.isBufferingObservation,
+      isBufferingReflection: turn._record.isBufferingReflection
+    } : void 0,
+    context: turn._context ? {
+      messageCount: Array.isArray(turn._context.messages) ? turn._context.messages.length : void 0,
+      hasSystemMessage: Array.isArray(turn._context.systemMessage) ? turn._context.systemMessage.length > 0 : Boolean(turn._context.systemMessage),
+      continuationId: turn._context.continuation?.id,
+      hasOtherThreadsContext: Boolean(turn._context.otherThreadsContext),
+      recordId: turn._context.record?.id
+    } : void 0,
+    currentStep: turn._currentStep ? {
+      stepNumber: turn._currentStep.stepNumber,
+      prepared: turn._currentStep._prepared,
+      context: turn._currentStep._context ? {
+        activated: turn._currentStep._context.activated,
+        observed: turn._currentStep._context.observed,
+        buffered: turn._currentStep._context.buffered,
+        reflected: turn._currentStep._context.reflected,
+        didThresholdCleanup: turn._currentStep._context.didThresholdCleanup,
+        messageCount: Array.isArray(turn._currentStep._context.messages) ? turn._currentStep._context.messages.length : void 0,
+        systemMessageCount: Array.isArray(turn._currentStep._context.systemMessage) ? turn._currentStep._context.systemMessage.length : void 0
+      } : void 0
+    } : void 0
+  };
 }
-function ensureMessageMastraMetadata(message) {
-  const typedMessage = message;
-  typedMessage.metadata ??= {};
-  typedMessage.metadata.mastra ??= {};
-  return typedMessage.metadata.mastra;
+function sanitizeCaptureState(rawState) {
+  return Object.fromEntries(
+    Object.entries(rawState).map(([key, value]) => {
+      if (key === "__omTurn") {
+        return [key, summarizeOmTurn(value)];
+      }
+      return [key, value];
+    })
+  );
 }
-function buildEstimateKey(kind, text) {
-  const payloadHash = createHash("sha1").update(text).digest("hex");
-  return `${kind}:${payloadHash}`;
+function buildReproMessageFingerprint(message) {
+  const createdAt = message.createdAt instanceof Date ? message.createdAt.toISOString() : message.createdAt ? new Date(message.createdAt).toISOString() : "";
+  return JSON.stringify({
+    role: message.role,
+    createdAt,
+    content: message.content
+  });
 }
-function resolveEstimatorId() {
-  return "tokenx";
+function inferReproIdRemap(preMessages, postMessages) {
+  const preByFingerprint = /* @__PURE__ */ new Map();
+  const postByFingerprint = /* @__PURE__ */ new Map();
+  for (const message of preMessages) {
+    if (!message.id) continue;
+    const fingerprint = buildReproMessageFingerprint(message);
+    const list = preByFingerprint.get(fingerprint) ?? [];
+    list.push(message.id);
+    preByFingerprint.set(fingerprint, list);
+  }
+  for (const message of postMessages) {
+    if (!message.id) continue;
+    const fingerprint = buildReproMessageFingerprint(message);
+    const list = postByFingerprint.get(fingerprint) ?? [];
+    list.push(message.id);
+    postByFingerprint.set(fingerprint, list);
+  }
+  const remap = [];
+  for (const [fingerprint, preIds] of preByFingerprint.entries()) {
+    const postIds = postByFingerprint.get(fingerprint);
+    if (!postIds || preIds.length !== 1 || postIds.length !== 1) continue;
+    const fromId = preIds[0];
+    const toId = postIds[0];
+    if (!fromId || !toId || fromId === toId) {
+      continue;
+    }
+    remap.push({ fromId, toId, fingerprint });
+  }
+  return remap;
 }
-function isTokenEstimateEntry(value) {
-  if (!value || typeof value !== "object") return false;
-  const entry = value;
-  return typeof entry.v === "number" && typeof entry.source === "string" && typeof entry.key === "string" && typeof entry.tokens === "number";
+function createOmReproCaptureDir(threadId, label) {
+  const sanitizedThreadId = sanitizeCapturePathSegment(threadId);
+  const captureDir = join(
+    process.cwd(),
+    getOmReproCaptureDir(),
+    sanitizedThreadId,
+    `${Date.now()}-${label}-${randomUUID()}`
+  );
+  mkdirSync(captureDir, { recursive: true });
+  return captureDir;
 }
-function getCacheEntry(cache, key) {
-  if (!cache || typeof cache !== "object") return void 0;
-  if (isTokenEstimateEntry(cache)) {
-    return cache.key === key ? cache : void 0;
+function writeObserverExchangeReproCapture(params) {
+  if (!isOmReproCaptureEnabled() || !params.observerExchange) {
+    return;
+  }
+  try {
+    const captureDir = createOmReproCaptureDir(params.threadId, params.label);
+    const payloads = [
+      {
+        fileName: "input.json",
+        data: {
+          threadId: params.threadId,
+          resourceId: params.resourceId,
+          label: params.label
+        }
+      },
+      {
+        fileName: "output.json",
+        data: {
+          details: params.details ?? {}
+        }
+      },
+      {
+        fileName: "observer-exchange.json",
+        data: params.observerExchange
+      }
+    ];
+    const captureErrors = [];
+    for (const payload of payloads) {
+      const serialized = safeCaptureJsonOrError(payload.data);
+      if (serialized.ok) {
+        writeFileSync(join(captureDir, payload.fileName), `${JSON.stringify(serialized.value, null, 2)}
+`);
+        continue;
+      }
+      captureErrors.push({ fileName: payload.fileName, error: serialized.error });
+      writeFileSync(
+        join(captureDir, payload.fileName),
+        `${JSON.stringify({ __captureError: serialized.error }, null, 2)}
+`
+      );
+    }
+    if (captureErrors.length > 0) {
+      writeFileSync(join(captureDir, "capture-error.json"), `${JSON.stringify(captureErrors, null, 2)}
+`);
+      params.debug?.(
+        `[OM:repro-capture] wrote ${params.label} capture with ${captureErrors.length} serialization error(s) to ${captureDir}`
+      );
+      return;
+    }
+    params.debug?.(`[OM:repro-capture] wrote ${params.label} capture to ${captureDir}`);
+  } catch (error) {
+    params.debug?.(`[OM:repro-capture] failed to write ${params.label} capture: ${String(error)}`);
   }
-  const keyedEntry = cache[key];
-  return isTokenEstimateEntry(keyedEntry) ? keyedEntry : void 0;
+}
+function writeProcessInputStepReproCapture(params) {
+  if (!isOmReproCaptureEnabled()) {
+    return;
+  }
+  try {
+    const captureDir = createOmReproCaptureDir(params.threadId, `step-${params.stepNumber}`);
+    const contextMessages = params.messageList.get.all.db();
+    const memoryContext = parseMemoryRequestContext(params.args.requestContext);
+    const preMessageIds = new Set(params.preMessages.map((message) => message.id));
+    const postMessageIds = new Set(contextMessages.map((message) => message.id));
+    const removedMessageIds = params.preMessages.map((message) => message.id).filter((id) => Boolean(id) && !postMessageIds.has(id));
+    const addedMessageIds = contextMessages.map((message) => message.id).filter((id) => Boolean(id) && !preMessageIds.has(id));
+    const idRemap = inferReproIdRemap(params.preMessages, contextMessages);
+    const rawState = params.args.state ?? {};
+    const sanitizedState = sanitizeCaptureState(rawState);
+    const payloads = [
+      {
+        fileName: "input.json",
+        data: {
+          stepNumber: params.stepNumber,
+          threadId: params.threadId,
+          resourceId: params.resourceId,
+          readOnly: memoryContext?.memoryConfig?.readOnly,
+          messageCount: contextMessages.length,
+          messageIds: contextMessages.map((message) => message.id),
+          stateKeys: Object.keys(rawState),
+          state: sanitizedState,
+          args: {
+            messages: params.args.messages,
+            steps: params.args.steps,
+            systemMessages: params.args.systemMessages,
+            retryCount: params.args.retryCount,
+            toolChoice: params.args.toolChoice,
+            activeTools: params.args.activeTools,
+            modelSettings: params.args.modelSettings,
+            structuredOutput: params.args.structuredOutput
+          }
+        }
+      },
+      {
+        fileName: "pre-state.json",
+        data: {
+          record: params.preRecord,
+          bufferedChunks: params.preBufferedChunks,
+          contextTokenCount: params.preContextTokenCount,
+          messages: params.preMessages,
+          messageList: params.preSerializedMessageList
+        }
+      },
+      {
+        fileName: "output.json",
+        data: {
+          details: params.details,
+          messageDiff: {
+            removedMessageIds,
+            addedMessageIds,
+            idRemap
+          }
+        }
+      },
+      {
+        fileName: "post-state.json",
+        data: {
+          record: params.postRecord,
+          bufferedChunks: params.postBufferedChunks,
+          contextTokenCount: params.postContextTokenCount,
+          messageCount: contextMessages.length,
+          messageIds: contextMessages.map((message) => message.id),
+          messages: contextMessages,
+          messageList: params.messageList.serialize()
+        }
+      }
+    ];
+    const captureErrors = [];
+    for (const payload of payloads) {
+      const serialized = safeCaptureJsonOrError(payload.data);
+      if (serialized.ok) {
+        writeFileSync(join(captureDir, payload.fileName), `${JSON.stringify(serialized.value, null, 2)}
+`);
+        continue;
+      }
+      captureErrors.push({ fileName: payload.fileName, error: serialized.error });
+      writeFileSync(
+        join(captureDir, payload.fileName),
+        `${JSON.stringify({ __captureError: serialized.error }, null, 2)}
+`
+      );
+    }
+    if (params.observerExchange) {
+      const serialized = safeCaptureJsonOrError(params.observerExchange);
+      if (serialized.ok) {
+        writeFileSync(join(captureDir, "observer-exchange.json"), `${JSON.stringify(serialized.value, null, 2)}
+`);
+      } else {
+        captureErrors.push({ fileName: "observer-exchange.json", error: serialized.error });
+        writeFileSync(
+          join(captureDir, "observer-exchange.json"),
+          `${JSON.stringify({ __captureError: serialized.error }, null, 2)}
+`
+        );
+      }
+    }
+    if (captureErrors.length > 0) {
+      writeFileSync(join(captureDir, "capture-error.json"), `${JSON.stringify(captureErrors, null, 2)}
+`);
+      params.debug?.(
+        `[OM:repro-capture] wrote processInputStep capture with ${captureErrors.length} serialization error(s) to ${captureDir}`
+      );
+      return;
+    }
+    params.debug?.(`[OM:repro-capture] wrote processInputStep capture to ${captureDir}`);
+  } catch (error) {
+    params.debug?.(`[OM:repro-capture] failed to write processInputStep capture: ${String(error)}`);
+  }
+}
+var IMAGE_FILE_EXTENSIONS = /* @__PURE__ */ new Set([
+  "png",
+  "jpg",
+  "jpeg",
+  "webp",
+  "gif",
+  "bmp",
+  "tiff",
+  "tif",
+  "heic",
+  "heif",
+  "avif"
+]);
+var TOKEN_ESTIMATE_CACHE_VERSION = 6;
+var DEFAULT_IMAGE_ESTIMATOR = {
+  baseTokens: 85,
+  tileTokens: 170,
+  fallbackTiles: 4
+};
+var GOOGLE_LEGACY_IMAGE_TOKENS_PER_TILE = 258;
+var GOOGLE_GEMINI_3_IMAGE_TOKENS_BY_RESOLUTION = {
+  low: 280,
+  medium: 560,
+  high: 1120,
+  ultra_high: 2240,
+  unspecified: 1120
+};
+var ANTHROPIC_IMAGE_TOKENS_PER_PIXEL = 1 / 750;
+var ANTHROPIC_IMAGE_MAX_LONG_EDGE = 1568;
+var GOOGLE_MEDIA_RESOLUTION_VALUES = /* @__PURE__ */ new Set([
+  "low",
+  "medium",
+  "high",
+  "ultra_high",
+  "unspecified"
+]);
+var ATTACHMENT_COUNT_TIMEOUT_MS = 2e4;
+var REMOTE_IMAGE_PROBE_TIMEOUT_MS = 2500;
+var PROVIDER_API_KEY_ENV_VARS = {
+  openai: ["OPENAI_API_KEY"],
+  google: ["GOOGLE_GENERATIVE_AI_API_KEY", "GOOGLE_API_KEY"],
+  anthropic: ["ANTHROPIC_API_KEY"]
+};
+function getPartMastraMetadata(part) {
+  return part.providerMetadata?.mastra;
+}
+function ensurePartMastraMetadata(part) {
+  const typedPart = part;
+  typedPart.providerMetadata ??= {};
+  typedPart.providerMetadata.mastra ??= {};
+  return typedPart.providerMetadata.mastra;
+}
+function getContentMastraMetadata(content) {
+  if (!content || typeof content !== "object") {
+    return void 0;
+  }
+  return content.metadata?.mastra;
+}
+function ensureContentMastraMetadata(content) {
+  if (!content || typeof content !== "object") {
+    return void 0;
+  }
+  const typedContent = content;
+  typedContent.metadata ??= {};
+  typedContent.metadata.mastra ??= {};
+  return typedContent.metadata.mastra;
+}
+function getMessageMastraMetadata(message) {
+  return message.metadata?.mastra;
+}
+function ensureMessageMastraMetadata(message) {
+  const typedMessage = message;
+  typedMessage.metadata ??= {};
+  typedMessage.metadata.mastra ??= {};
+  return typedMessage.metadata.mastra;
+}
+function buildEstimateKey(kind, text) {
+  const payloadHash = createHash("sha1").update(text).digest("hex");
+  return `${kind}:${payloadHash}`;
+}
+function resolveEstimatorId() {
+  return "tokenx";
+}
+function isTokenEstimateEntry(value) {
+  if (!value || typeof value !== "object") return false;
+  const entry = value;
+  return typeof entry.v === "number" && typeof entry.source === "string" && typeof entry.key === "string" && typeof entry.tokens === "number";
+}
+function getCacheEntry(cache, key) {
+  if (!cache || typeof cache !== "object") return void 0;
+  if (isTokenEstimateEntry(cache)) {
+    return cache.key === key ? cache : void 0;
+  }
+  const keyedEntry = cache[key];
+  return isTokenEstimateEntry(keyedEntry) ? keyedEntry : void 0;
 }
 function mergeCacheEntry(cache, key, entry) {
   if (isTokenEstimateEntry(cache)) {
@@ -7786,6 +8183,23 @@ ${grouped}` : grouped;
         requestContext,
         observabilityContext
       }).run();
+      if (isOmReproCaptureEnabled()) {
+        writeObserverExchangeReproCapture({
+          threadId,
+          resourceId: record.resourceId ?? void 0,
+          label: `buffer-${cycleId}`,
+          observerExchange: this.observer.lastExchange,
+          details: {
+            cycleId,
+            startedAt,
+            buffered: true,
+            candidateMessageIds: candidateMessages.map((message) => message.id),
+            candidateMessageCount: candidateMessages.length,
+            pendingTokens: currentTokens,
+            newTokens
+          }
+        });
+      }
       await this.storage.setBufferingObservationFlag(record.id, false, newTokens).catch(() => {
       });
       flagCleared = true;
@@ -8181,266 +8595,6 @@ ${grouped}` : grouped;
     });
   }
 };
-var OM_REPRO_CAPTURE_DIR = process.env.OM_REPRO_CAPTURE_DIR ?? ".mastra-om-repro";
-function sanitizeCapturePathSegment(value) {
-  const sanitized = value.replace(/[\\/]+/g, "_").replace(/\.{2,}/g, "_").trim();
-  return sanitized.length > 0 ? sanitized : "unknown-thread";
-}
-function isOmReproCaptureEnabled() {
-  return process.env.OM_REPRO_CAPTURE === "1";
-}
-function safeCaptureJson(value) {
-  return JSON.parse(
-    JSON.stringify(value, (_key, current) => {
-      if (typeof current === "bigint") return current.toString();
-      if (typeof current === "function") return "[function]";
-      if (typeof current === "symbol") return current.toString();
-      if (current instanceof Error) return { name: current.name, message: current.message, stack: current.stack };
-      if (current instanceof Set) return { __type: "Set", values: Array.from(current.values()) };
-      if (current instanceof Map) return { __type: "Map", entries: Array.from(current.entries()) };
-      return current;
-    })
-  );
-}
-function safeCaptureJsonOrError(value) {
-  try {
-    return { ok: true, value: safeCaptureJson(value) };
-  } catch (error) {
-    return {
-      ok: false,
-      error: safeCaptureJson({
-        message: error instanceof Error ? error.message : String(error),
-        stack: error instanceof Error ? error.stack : void 0,
-        inspected: inspect(value, { depth: 3, maxArrayLength: 20, breakLength: 120 })
-      })
-    };
-  }
-}
-function formatCaptureDate(value) {
-  if (!value) return void 0;
-  if (value instanceof Date) return value.toISOString();
-  try {
-    return new Date(value).toISOString();
-  } catch {
-    return void 0;
-  }
-}
-function summarizeOmTurn(value) {
-  if (!value || typeof value !== "object") {
-    return value;
-  }
-  const turn = value;
-  return {
-    __type: "ObservationTurn",
-    threadId: turn.threadId,
-    resourceId: turn.resourceId,
-    started: turn._started,
-    ended: turn._ended,
-    generationCountAtStart: turn._generationCountAtStart,
-    record: turn._record ? {
-      id: turn._record.id,
-      scope: turn._record.scope,
-      threadId: turn._record.threadId,
-      resourceId: turn._record.resourceId,
-      createdAt: formatCaptureDate(turn._record.createdAt),
-      updatedAt: formatCaptureDate(turn._record.updatedAt),
-      lastObservedAt: formatCaptureDate(turn._record.lastObservedAt),
-      generationCount: turn._record.generationCount,
-      observationTokenCount: turn._record.observationTokenCount,
-      pendingMessageTokens: turn._record.pendingMessageTokens,
-      isBufferingObservation: turn._record.isBufferingObservation,
-      isBufferingReflection: turn._record.isBufferingReflection
-    } : void 0,
-    context: turn._context ? {
-      messageCount: Array.isArray(turn._context.messages) ? turn._context.messages.length : void 0,
-      hasSystemMessage: Array.isArray(turn._context.systemMessage) ? turn._context.systemMessage.length > 0 : Boolean(turn._context.systemMessage),
-      continuationId: turn._context.continuation?.id,
-      hasOtherThreadsContext: Boolean(turn._context.otherThreadsContext),
-      recordId: turn._context.record?.id
-    } : void 0,
-    currentStep: turn._currentStep ? {
-      stepNumber: turn._currentStep.stepNumber,
-      prepared: turn._currentStep._prepared,
-      context: turn._currentStep._context ? {
-        activated: turn._currentStep._context.activated,
-        observed: turn._currentStep._context.observed,
-        buffered: turn._currentStep._context.buffered,
-        reflected: turn._currentStep._context.reflected,
-        didThresholdCleanup: turn._currentStep._context.didThresholdCleanup,
-        messageCount: Array.isArray(turn._currentStep._context.messages) ? turn._currentStep._context.messages.length : void 0,
-        systemMessageCount: Array.isArray(turn._currentStep._context.systemMessage) ? turn._currentStep._context.systemMessage.length : void 0
-      } : void 0
-    } : void 0
-  };
-}
-function sanitizeCaptureState(rawState) {
-  return Object.fromEntries(
-    Object.entries(rawState).map(([key, value]) => {
-      if (key === "__omTurn") {
-        return [key, summarizeOmTurn(value)];
-      }
-      return [key, value];
-    })
-  );
-}
-function buildReproMessageFingerprint(message) {
-  const createdAt = message.createdAt instanceof Date ? message.createdAt.toISOString() : message.createdAt ? new Date(message.createdAt).toISOString() : "";
-  return JSON.stringify({
-    role: message.role,
-    createdAt,
-    content: message.content
-  });
-}
-function inferReproIdRemap(preMessages, postMessages) {
-  const preByFingerprint = /* @__PURE__ */ new Map();
-  const postByFingerprint = /* @__PURE__ */ new Map();
-  for (const message of preMessages) {
-    if (!message.id) continue;
-    const fingerprint = buildReproMessageFingerprint(message);
-    const list = preByFingerprint.get(fingerprint) ?? [];
-    list.push(message.id);
-    preByFingerprint.set(fingerprint, list);
-  }
-  for (const message of postMessages) {
-    if (!message.id) continue;
-    const fingerprint = buildReproMessageFingerprint(message);
-    const list = postByFingerprint.get(fingerprint) ?? [];
-    list.push(message.id);
-    postByFingerprint.set(fingerprint, list);
-  }
-  const remap = [];
-  for (const [fingerprint, preIds] of preByFingerprint.entries()) {
-    const postIds = postByFingerprint.get(fingerprint);
-    if (!postIds || preIds.length !== 1 || postIds.length !== 1) continue;
-    const fromId = preIds[0];
-    const toId = postIds[0];
-    if (!fromId || !toId || fromId === toId) {
-      continue;
-    }
-    remap.push({ fromId, toId, fingerprint });
-  }
-  return remap;
-}
-function writeProcessInputStepReproCapture(params) {
-  if (!isOmReproCaptureEnabled()) {
-    return;
-  }
-  try {
-    const sanitizedThreadId = sanitizeCapturePathSegment(params.threadId);
-    const runId = `${Date.now()}-step-${params.stepNumber}-${randomUUID()}`;
-    const captureDir = join(process.cwd(), OM_REPRO_CAPTURE_DIR, sanitizedThreadId, runId);
-    mkdirSync(captureDir, { recursive: true });
-    const contextMessages = params.messageList.get.all.db();
-    const memoryContext = parseMemoryRequestContext(params.args.requestContext);
-    const preMessageIds = new Set(params.preMessages.map((message) => message.id));
-    const postMessageIds = new Set(contextMessages.map((message) => message.id));
-    const removedMessageIds = params.preMessages.map((message) => message.id).filter((id) => Boolean(id) && !postMessageIds.has(id));
-    const addedMessageIds = contextMessages.map((message) => message.id).filter((id) => Boolean(id) && !preMessageIds.has(id));
-    const idRemap = inferReproIdRemap(params.preMessages, contextMessages);
-    const rawState = params.args.state ?? {};
-    const sanitizedState = sanitizeCaptureState(rawState);
-    const payloads = [
-      {
-        fileName: "input.json",
-        data: {
-          stepNumber: params.stepNumber,
-          threadId: params.threadId,
-          resourceId: params.resourceId,
-          readOnly: memoryContext?.memoryConfig?.readOnly,
-          messageCount: contextMessages.length,
-          messageIds: contextMessages.map((message) => message.id),
-          stateKeys: Object.keys(rawState),
-          state: sanitizedState,
-          args: {
-            messages: params.args.messages,
-            steps: params.args.steps,
-            systemMessages: params.args.systemMessages,
-            retryCount: params.args.retryCount,
-            toolChoice: params.args.toolChoice,
-            activeTools: params.args.activeTools,
-            modelSettings: params.args.modelSettings,
-            structuredOutput: params.args.structuredOutput
-          }
-        }
-      },
-      {
-        fileName: "pre-state.json",
-        data: {
-          record: params.preRecord,
-          bufferedChunks: params.preBufferedChunks,
-          contextTokenCount: params.preContextTokenCount,
-          messages: params.preMessages,
-          messageList: params.preSerializedMessageList
-        }
-      },
-      {
-        fileName: "output.json",
-        data: {
-          details: params.details,
-          messageDiff: {
-            removedMessageIds,
-            addedMessageIds,
-            idRemap
-          }
-        }
-      },
-      {
-        fileName: "post-state.json",
-        data: {
-          record: params.postRecord,
-          bufferedChunks: params.postBufferedChunks,
-          contextTokenCount: params.postContextTokenCount,
-          messageCount: contextMessages.length,
-          messageIds: contextMessages.map((message) => message.id),
-          messages: contextMessages,
-          messageList: params.messageList.serialize()
-        }
-      }
-    ];
-    const captureErrors = [];
-    for (const payload of payloads) {
-      const serialized = safeCaptureJsonOrError(payload.data);
-      if (serialized.ok) {
-        writeFileSync(join(captureDir, payload.fileName), `${JSON.stringify(serialized.value, null, 2)}
-`);
-        continue;
-      }
-      captureErrors.push({ fileName: payload.fileName, error: serialized.error });
-      writeFileSync(
-        join(captureDir, payload.fileName),
-        `${JSON.stringify({ __captureError: serialized.error }, null, 2)}
-`
-      );
-    }
-    if (params.observerExchange) {
-      const serialized = safeCaptureJsonOrError(params.observerExchange);
-      if (serialized.ok) {
-        writeFileSync(join(captureDir, "observer-exchange.json"), `${JSON.stringify(serialized.value, null, 2)}
-`);
-      } else {
-        captureErrors.push({ fileName: "observer-exchange.json", error: serialized.error });
-        writeFileSync(
-          join(captureDir, "observer-exchange.json"),
-          `${JSON.stringify({ __captureError: serialized.error }, null, 2)}
-`
-        );
-      }
-    }
-    if (captureErrors.length > 0) {
-      writeFileSync(join(captureDir, "capture-error.json"), `${JSON.stringify(captureErrors, null, 2)}
-`);
-      params.debug?.(
-        `[OM:repro-capture] wrote processInputStep capture with ${captureErrors.length} serialization error(s) to ${captureDir}`
-      );
-      return;
-    }
-    params.debug?.(`[OM:repro-capture] wrote processInputStep capture to ${captureDir}`);
-  } catch (error) {
-    params.debug?.(`[OM:repro-capture] failed to write processInputStep capture: ${String(error)}`);
-  }
-}
-
-// src/processors/observational-memory/processor.ts
 function getOmObservabilityContext(args) {
   if (!args.tracing || !args.tracingContext || !args.loggerVNext || !args.metrics) {
     return void 0;
@@ -8654,5 +8808,5 @@ function getObservationsAsOf(activeObservations, asOf) {
 }
 
 export { ModelByInputTokens, OBSERVER_SYSTEM_PROMPT, ObservationalMemory, ObservationalMemoryProcessor, TokenCounter, buildObserverPrompt, buildObserverSystemPrompt, combineObservationGroupRanges, deriveObservationGroupProvenance, extractCurrentTask, formatMessagesForObserver, formatToolResultForObserver, getObservationsAsOf, hasCurrentTaskSection, injectAnchorIds, optimizeObservationsForContext, parseAnchorId, parseObservationGroups, parseObserverOutput, reconcileObservationGroupsFromReflection, renderObservationGroupsForReflection, resolveToolResultValue, stripEphemeralAnchorIds, stripObservationGroups, truncateStringByTokens, wrapInObservationGroup };
-//# sourceMappingURL=chunk-GXDPND6K.js.map
-//# sourceMappingURL=chunk-GXDPND6K.js.map
+//# sourceMappingURL=chunk-ZEKCVX4E.js.map
+//# sourceMappingURL=chunk-ZEKCVX4E.js.map