@mastra/memory 1.19.1-alpha.0 → 1.20.0-alpha.2

package/dist/{chunk-IRMLP4QH.cjs → chunk-SMKIROLC.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkD4J4XPGM_cjs = require('./chunk-D4J4XPGM.cjs');
+var chunk3Q3LOVQP_cjs = require('./chunk-3Q3LOVQP.cjs');
 var features = require('@mastra/core/features');
 var llm = require('@mastra/core/llm');
 var memory = require('@mastra/core/memory');
@@ -676,6 +676,7 @@ function filterObservedMessages(opts) {
   const { messageList, record } = opts;
   const allMessages = messageList.get.all.db();
   const useMarkerBoundaryPruning = opts.useMarkerBoundaryPruning ?? true;
+  const preserveMessageIds = opts.preserveMessageIds ?? /* @__PURE__ */ new Set();
   let markerMessageIndex = -1;
   let markerMessage = null;
   for (let i = allMessages.length - 1; i >= 0; i--) {
@@ -691,7 +692,7 @@ function filterObservedMessages(opts) {
     const messagesToRemove = [];
     for (let i = 0; i < markerMessageIndex; i++) {
       const msg = allMessages[i];
-      if (msg?.id && msg.id !== "om-continuation") {
+      if (msg?.id && msg.id !== "om-continuation" && !preserveMessageIds.has(msg.id)) {
         messagesToRemove.push(msg.id);
       }
     }
@@ -710,7 +711,7 @@ function filterObservedMessages(opts) {
     const lastObservedAt = record.lastObservedAt;
     const messagesToRemove = [];
     for (const msg of allMessages) {
-      if (!msg?.id || msg.id === "om-continuation") continue;
+      if (!msg?.id || msg.id === "om-continuation" || preserveMessageIds.has(msg.id)) continue;
       if (observedIds.has(msg.id)) {
         messagesToRemove.push(msg.id);
         continue;
@@ -1834,7 +1835,7 @@ var ResourceScopedObservationStrategy = class extends ObservationStrategy {
     }
   }
   async observe(_existingObservations, _messages) {
-    const maxTokensPerBatch = this.observationConfig.maxTokensPerBatch ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.maxTokensPerBatch;
+    const maxTokensPerBatch = this.observationConfig.maxTokensPerBatch ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.maxTokensPerBatch;
     const orderedThreadIds = this.threadOrder.filter((tid) => this.threadsWithMessages.has(tid));
     const batches = [];
     let currentBatch = {
@@ -2276,11 +2277,15 @@ var ObservationStep = class {
     });
     if (!didThresholdCleanup) {
       const fallbackCursor = this.turn.record.threadId ? 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).filter(Boolean)
+      );
       filterObservedMessages({
         messageList,
         record: this.turn.record,
         useMarkerBoundaryPruning: this.stepNumber === 0,
-        fallbackCursor
+        fallbackCursor,
+        preserveMessageIds: pendingMessageIds
       });
     }
     this._context = {
@@ -4108,6 +4113,41 @@ var ObserverRunner = class {
     }
     return agent$1;
   }
+  /**
+   * Extract a router-style model ID (`provider/model`) from a model config.
+   * Handles strings, LanguageModel objects, and function-based models.
+   */
+  extractModelRouterId(model, requestContext) {
+    if (typeof model === "string") return model;
+    if (typeof model === "function") {
+      if (!requestContext) return void 0;
+      try {
+        const resolved = model({ requestContext });
+        if (resolved instanceof Promise) return void 0;
+        return this.extractModelRouterId(resolved);
+      } catch {
+        return void 0;
+      }
+    }
+    const obj = model;
+    if (typeof obj.provider === "string" && typeof obj.modelId === "string") {
+      return `${obj.provider}/${obj.modelId}`;
+    }
+    return void 0;
+  }
+  /**
+   * Resolve the attachment filter for a given model. When set to `'auto'`,
+   * the provider capabilities registry is consulted to decide whether the
+   * model accepts multimodal input.
+   */
+  resolveAttachmentFilter(model, requestContext) {
+    const raw = this.observationConfig.observeAttachments;
+    if (raw !== "auto") return raw;
+    const routerId = this.extractModelRouterId(model, requestContext);
+    if (!routerId) return true;
+    const supports = llm.modelSupportsAttachments(routerId);
+    return supports ?? true;
+  }
   async withAbortCheck(fn, abortSignal) {
     if (abortSignal?.aborted) {
       throw new Error("The operation was aborted.");
@@ -4125,6 +4165,7 @@ var ObserverRunner = class {
     const inputTokens = this.tokenCounter.countMessages(messagesToObserve);
     const resolvedModel = options?.model ? { model: options.model } : this.resolveModel(inputTokens);
     const agent = this.createAgent(resolvedModel.model);
+    const attachmentFilter = this.resolveAttachmentFilter(resolvedModel.model, options?.requestContext);
     const observerMessages = [
       {
         role: "user",
@@ -4134,7 +4175,7 @@ var ObserverRunner = class {
         })
       },
       buildObserverHistoryMessage(messagesToObserve, {
-        attachmentFilter: this.observationConfig.observeAttachments
+        attachmentFilter
       })
     ];
     const doGenerate = async () => {
@@ -4221,6 +4262,7 @@ var ObserverRunner = class {
     );
     const resolvedModel = model ? { model } : this.resolveModel(inputTokens);
     const agent = this.createAgent(resolvedModel.model, true);
+    const multiThreadAttachmentFilter = this.resolveAttachmentFilter(resolvedModel.model, requestContext);
     const observerMessages = [
       {
         role: "user",
@@ -4233,7 +4275,7 @@ var ObserverRunner = class {
         )
       },
       buildMultiThreadObserverHistoryMessage(messagesByThread, threadOrder, {
-        attachmentFilter: this.observationConfig.observeAttachments
+        attachmentFilter: multiThreadAttachmentFilter
       })
     ];
     for (const msgs of messagesByThread.values()) {
@@ -7100,16 +7142,16 @@ var ObservationalMemory = class _ObservationalMemory {
     this.onIndexObservations = config.onIndexObservations;
     this.mastra = config.mastra;
     const resolveModel = (model, defaultModel) => model === "default" ? defaultModel : model;
-    const observationModel = resolveModel(config.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.observation?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.reflection?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model;
-    const reflectionModel = resolveModel(config.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.reflection?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.observation?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model;
-    const messageTokens = config.observation?.messageTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens;
-    const observationTokens = config.reflection?.observationTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.observationTokens;
+    const observationModel = resolveModel(config.model, chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.observation?.model, chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.reflection?.model, chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model;
+    const reflectionModel = resolveModel(config.model, chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.reflection?.model, chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.observation?.model, chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model;
+    const messageTokens = config.observation?.messageTokens ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens;
+    const observationTokens = config.reflection?.observationTokens ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.observationTokens;
     const isSharedBudget = config.shareTokenBudget ?? false;
     const isDefaultModelSelection = (model) => model === void 0 || model === "default" || model instanceof ModelByInputTokens;
     const observationSelectedModel = config.model ?? config.observation?.model ?? config.reflection?.model;
     const reflectionSelectedModel = config.model ?? config.reflection?.model ?? config.observation?.model;
-    const observationDefaultMaxOutputTokens = config.observation?.modelSettings?.maxOutputTokens ?? (isDefaultModelSelection(observationSelectedModel) ? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.modelSettings.maxOutputTokens : void 0);
-    const reflectionDefaultMaxOutputTokens = config.reflection?.modelSettings?.maxOutputTokens ?? (isDefaultModelSelection(reflectionSelectedModel) ? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.modelSettings.maxOutputTokens : void 0);
+    const observationDefaultMaxOutputTokens = config.observation?.modelSettings?.maxOutputTokens ?? (isDefaultModelSelection(observationSelectedModel) ? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.modelSettings.maxOutputTokens : void 0);
+    const reflectionDefaultMaxOutputTokens = config.reflection?.modelSettings?.maxOutputTokens ?? (isDefaultModelSelection(reflectionSelectedModel) ? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.modelSettings.maxOutputTokens : void 0);
     const totalBudget = messageTokens + observationTokens;
     const userExplicitlyConfiguredAsync = config.observation?.bufferTokens !== void 0 || config.observation?.bufferActivation !== void 0 || config.reflection?.bufferActivation !== void 0;
     const asyncBufferingDisabled = config.observation?.bufferTokens === false || config.scope === "resource" && !userExplicitlyConfiguredAsync;
@@ -7142,21 +7184,21 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
       messageTokens: isSharedBudget ? { min: messageTokens, max: totalBudget } : messageTokens,
       shareTokenBudget: isSharedBudget,
       modelSettings: {
-        temperature: config.observation?.modelSettings?.temperature ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.modelSettings.temperature,
+        temperature: config.observation?.modelSettings?.temperature ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.modelSettings.temperature,
         ...observationDefaultMaxOutputTokens !== void 0 ? { maxOutputTokens: observationDefaultMaxOutputTokens } : {}
       },
-      providerOptions: config.observation?.providerOptions ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.providerOptions,
-      maxTokensPerBatch: config.observation?.maxTokensPerBatch ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.maxTokensPerBatch,
+      providerOptions: config.observation?.providerOptions ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.providerOptions,
+      maxTokensPerBatch: config.observation?.maxTokensPerBatch ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.maxTokensPerBatch,
       bufferTokens: asyncBufferingDisabled ? void 0 : resolveBufferTokens(
-        config.observation?.bufferTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferTokens,
-        config.observation?.messageTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens
+        config.observation?.bufferTokens ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferTokens,
+        config.observation?.messageTokens ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens
       ),
-      bufferActivation: asyncBufferingDisabled ? void 0 : config.observation?.bufferActivation ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferActivation,
+      bufferActivation: asyncBufferingDisabled ? void 0 : config.observation?.bufferActivation ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferActivation,
       activateAfterIdle: parseActivationTTL(observationActivateAfterIdle, observationActivateAfterIdlePath),
       activateOnProviderChange: config.observation?.activateOnProviderChange ?? config.activateOnProviderChange ?? false,
       blockAfter: asyncBufferingDisabled ? void 0 : resolveBlockAfter(
-        config.observation?.blockAfter ?? (config.observation?.bufferTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferTokens ? 1.2 : void 0),
-        config.observation?.messageTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens
+        config.observation?.blockAfter ?? (config.observation?.bufferTokens ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferTokens ? 1.2 : void 0),
+        config.observation?.messageTokens ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens
       ),
       previousObserverTokens: config.observation?.previousObserverTokens ?? 2e3,
       instruction: config.observation?.instruction,
@@ -7168,16 +7210,16 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
       observationTokens,
       shareTokenBudget: isSharedBudget,
       modelSettings: {
-        temperature: config.reflection?.modelSettings?.temperature ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.modelSettings.temperature,
+        temperature: config.reflection?.modelSettings?.temperature ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.modelSettings.temperature,
         ...reflectionDefaultMaxOutputTokens !== void 0 ? { maxOutputTokens: reflectionDefaultMaxOutputTokens } : {}
       },
-      providerOptions: config.reflection?.providerOptions ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.providerOptions,
-      bufferActivation: asyncBufferingDisabled ? void 0 : config?.reflection?.bufferActivation ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.bufferActivation,
+      providerOptions: config.reflection?.providerOptions ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.providerOptions,
+      bufferActivation: asyncBufferingDisabled ? void 0 : config?.reflection?.bufferActivation ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.bufferActivation,
       activateAfterIdle: parseActivationTTL(config.reflection?.activateAfterIdle, "reflection.activateAfterIdle"),
       activateOnProviderChange: config.reflection?.activateOnProviderChange ?? false,
       blockAfter: asyncBufferingDisabled ? void 0 : resolveBlockAfter(
-        config.reflection?.blockAfter ?? (config.reflection?.bufferActivation ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.bufferActivation ? 1.2 : void 0),
-        config.reflection?.observationTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.observationTokens
+        config.reflection?.blockAfter ?? (config.reflection?.bufferActivation ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.bufferActivation ? 1.2 : void 0),
+        config.reflection?.observationTokens ?? chunk3Q3LOVQP_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.observationTokens
       ),
       instruction: config.reflection?.instruction
     };
@@ -7941,11 +7983,11 @@ ${unreflectedContent}` : bufferedReflection;
       optimized = addRelativeTimeToObservations(optimized, currentDate);
     }
     const messages = [
-      `${chunkD4J4XPGM_cjs.OBSERVATION_CONTEXT_PROMPT}
+      `${chunk3Q3LOVQP_cjs.OBSERVATION_CONTEXT_PROMPT}
 
-${chunkD4J4XPGM_cjs.OBSERVATION_CONTEXT_INSTRUCTIONS}${retrieval ? `
+${chunk3Q3LOVQP_cjs.OBSERVATION_CONTEXT_INSTRUCTIONS}${retrieval ? `
 
-${chunkD4J4XPGM_cjs.OBSERVATION_RETRIEVAL_INSTRUCTIONS}` : ""}`
+${chunk3Q3LOVQP_cjs.OBSERVATION_RETRIEVAL_INSTRUCTIONS}` : ""}`
     ];
     if (unobservedContextBlocks) {
       messages.push(
@@ -9574,7 +9616,7 @@ function injectObservationContextMessages({
     createdAt: /* @__PURE__ */ new Date(0),
     content: {
       format: 2,
-      parts: [{ type: "text", text: `<system-reminder>${chunkD4J4XPGM_cjs.OBSERVATION_CONTINUATION_HINT}</system-reminder>` }]
+      parts: [{ type: "text", text: `<system-reminder>${chunk3Q3LOVQP_cjs.OBSERVATION_CONTINUATION_HINT}</system-reminder>` }]
     },
     threadId,
     resourceId
@@ -9841,5 +9883,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-IRMLP4QH.cjs.map
-//# sourceMappingURL=chunk-IRMLP4QH.cjs.map
+//# sourceMappingURL=chunk-SMKIROLC.cjs.map
+//# sourceMappingURL=chunk-SMKIROLC.cjs.map