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

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

@@ -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;
@@ -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()) {
@@ -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-PNKMNTE6.cjs.map
+//# sourceMappingURL=chunk-PNKMNTE6.cjs.map