@mastra/server 1.37.0-alpha.2 → 1.37.0-alpha.3

package/dist/{chunk-UHMBTZ44.js → chunk-VFMS237A.js}

@@ -1,6 +1,6 @@
 import { OBSERVATIONAL_MEMORY_DEFAULTS, OBSERVATION_CONTEXT_PROMPT, OBSERVATION_CONTEXT_INSTRUCTIONS, OBSERVATION_RETRIEVAL_INSTRUCTIONS, OBSERVATION_CONTINUATION_HINT } from './chunk-JMEMSVJ5.js';
 import { coreFeatures } from '@mastra/core/features';
-import { resolveModelConfig } from '@mastra/core/llm';
+import { resolveModelConfig, modelSupportsAttachments } from '@mastra/core/llm';
 import { getThreadOMMetadata, setThreadOMMetadata, parseMemoryRequestContext } from '@mastra/core/memory';
 import { MessageHistory } from '@mastra/core/processors';
 import { appendFileSync, writeFileSync, mkdirSync } from 'fs';
@@ -1040,7 +1040,7 @@ function imageSize(input) {
   throw new TypeError(`unsupported file type: ${type}`);
 }
 
-// ../memory/dist/chunk-BCYGTJF2.js
+// ../memory/dist/chunk-G72RLVRT.js
 var MINUTE = 6e4;
 var HOUR = 60 * MINUTE;
 var SHORT_TTL = 5 * MINUTE;
@@ -1683,6 +1683,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--) {
@@ -1698,7 +1699,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);
       }
     }
@@ -1717,7 +1718,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;
@@ -3273,11 +3274,15 @@ var ObservationStep = class {
     });
     if (!didThresholdCleanup) {
       const fallbackCursor = this.turn.record.threadId ? 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 = {
@@ -5093,6 +5098,41 @@ var ObserverRunner = class {
     }
     return agent;
   }
+  /**
+   * 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 = modelSupportsAttachments(routerId);
+    return supports ?? true;
+  }
   async withAbortCheck(fn, abortSignal) {
     if (abortSignal?.aborted) {
       throw new Error("The operation was aborted.");
@@ -5110,6 +5150,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",
@@ -5119,7 +5160,7 @@ var ObserverRunner = class {
         })
       },
       buildObserverHistoryMessage(messagesToObserve, {
-        attachmentFilter: this.observationConfig.observeAttachments
+        attachmentFilter
       })
     ];
     const doGenerate = async () => {
@@ -5206,6 +5247,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",
@@ -5218,7 +5260,7 @@ var ObserverRunner = class {
         )
       },
       buildMultiThreadObserverHistoryMessage(messagesByThread, threadOrder, {
-        attachmentFilter: this.observationConfig.observeAttachments
+        attachmentFilter: multiThreadAttachmentFilter
       })
     ];
     for (const msgs of messagesByThread.values()) {
@@ -10789,5 +10831,5 @@ function getObservationsAsOf(activeObservations, asOf) {
 }
 
 export { ModelByInputTokens, OBSERVER_SYSTEM_PROMPT, ObservationalMemory, ObservationalMemoryProcessor, TokenCounter, buildObserverPrompt, buildObserverSystemPrompt, combineObservationGroupRanges, deriveObservationGroupProvenance, e, estimateTokenCount, extractCurrentTask, formatMessagesForObserver, formatToolResultForObserver, getObservationsAsOf, hasCurrentTaskSection, injectAnchorIds, optimizeObservationsForContext, parseAnchorId, parseObservationGroups, parseObserverOutput, reconcileObservationGroupsFromReflection, renderObservationGroupsForReflection, resolveToolResultValue, stripEphemeralAnchorIds, stripObservationGroups, truncateStringByTokens, wrapInObservationGroup };
-//# sourceMappingURL=chunk-UHMBTZ44.js.map
-//# sourceMappingURL=chunk-UHMBTZ44.js.map
+//# sourceMappingURL=chunk-VFMS237A.js.map
+//# sourceMappingURL=chunk-VFMS237A.js.map