@mastra/memory 1.19.0 → 1.20.0-alpha.1

package/dist/{chunk-5IJQOXJM.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()) {
@@ -9473,6 +9515,16 @@ function getTemporalGapReminderMetadata(message, gapText, gapMs, timestamp) {
     }
   };
 }
+function getTemporalGapReminderAttributes(message, gapText, gapMs, timestamp) {
+  return {
+    type: TEMPORAL_GAP_REMINDER_TYPE,
+    gapText,
+    gapMs,
+    timestamp: formatTemporalTimestamp(new Date(timestamp)),
+    timestampMs: timestamp,
+    precedesMessageId: message.id
+  };
+}
 function isTemporalGapMarkerForMessage(message, targetMessageId) {
   if (!isTemporalGapMarker(message)) {
     return false;
@@ -9483,29 +9535,9 @@ function isTemporalGapMarkerForMessage(message, targetMessageId) {
   }
   return metadata?.systemReminder?.type === TEMPORAL_GAP_REMINDER_TYPE && metadata.systemReminder.precedesMessageId === targetMessageId;
 }
-function createTemporalGapMarker(message, gapText, gapMs, timestamp) {
-  const metadata = getTemporalGapReminderMetadata(message, gapText, gapMs, timestamp);
-  return {
-    id: `__temporal_gap_${crypto.randomUUID()}`,
-    role: "user",
-    createdAt: new Date(timestamp - 1),
-    threadId: message.threadId,
-    resourceId: message.resourceId,
-    content: {
-      format: 2,
-      parts: [
-        {
-          type: "text",
-          text: `<system-reminder type="${TEMPORAL_GAP_REMINDER_TYPE}" precedesMessageId="${message.id}">${getTemporalGapReminderText(gapText, timestamp)}</system-reminder>`
-        }
-      ],
-      metadata
-    }
-  };
-}
 async function insertTemporalGapMarkers({
   messageList,
-  writer
+  sendSignal
 }) {
   const inputMessages = messageList.get.input.db().filter((message) => Boolean(message));
   const latestInputMessage = inputMessages.at(-1);
@@ -9537,17 +9569,15 @@ async function insertTemporalGapMarkers({
   if (!gapText) {
     return;
   }
-  const reminderMetadata = getTemporalGapReminderMetadata(latestInputMessage, gapText, gapMs, timestamp);
-  await writer?.custom({
-    type: "data-system-reminder",
-    data: {
-      message: getTemporalGapReminderText(gapText, timestamp),
-      ...reminderMetadata
-    },
-    transient: true
+  await sendSignal?.({
+    id: `__temporal_gap_${crypto.randomUUID()}`,
+    type: "system-reminder",
+    contents: getTemporalGapReminderText(gapText, timestamp),
+    createdAt: new Date(timestamp - 1),
+    acceptedAt: new Date(timestamp),
+    attributes: getTemporalGapReminderAttributes(latestInputMessage, gapText, gapMs, timestamp),
+    metadata: getTemporalGapReminderMetadata(latestInputMessage, gapText, gapMs, timestamp)
   });
-  const marker = createTemporalGapMarker(latestInputMessage, gapText, gapMs, timestamp);
-  messageList.add(marker, "input");
 }
 
 // src/processors/observational-memory/processor.ts
@@ -9696,7 +9726,7 @@ var ObservationalMemoryProcessor = class {
         this.turn.requestContext = requestContext;
         await this.turn.start(this.memory);
         if (stepNumber === 0 && this.temporalMarkers) {
-          await insertTemporalGapMarkers({ messageList, writer });
+          await insertTemporalGapMarkers({ messageList, sendSignal: args.sendSignal });
         }
         state.__omTurn = this.turn;
       }
@@ -9853,5 +9883,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-5IJQOXJM.cjs.map
-//# sourceMappingURL=chunk-5IJQOXJM.cjs.map
+//# sourceMappingURL=chunk-PNKMNTE6.cjs.map
+//# sourceMappingURL=chunk-PNKMNTE6.cjs.map