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

package/dist/{chunk-ANKEPFC6.cjs → chunk-WNLRMAHY.cjs}

@@ -4131,13 +4131,29 @@ var ReflectorRunner = class {
       model
     });
   }
-  getObservationMarkerConfig() {
+  getObservationMarkerConfig(record) {
     return {
       messageTokens: getMaxThreshold(this.observationConfig.messageTokens),
-      observationTokens: getMaxThreshold(this.reflectionConfig.observationTokens),
+      observationTokens: getMaxThreshold(
+        record ? this.getEffectiveReflectionTokens(record) : this.reflectionConfig.observationTokens
+      ),
       scope: this.scope
     };
   }
+  /**
+   * Resolve the effective reflection observationTokens for a record.
+   * Only explicit per-record overrides (stored under `_overrides`) win;
+   * the initial config snapshot is ignored so instance-level changes
+   * still take effect for existing records.
+   */
+  getEffectiveReflectionTokens(record) {
+    const overrides = record.config?._overrides;
+    const recordTokens = overrides?.reflection?.observationTokens;
+    if (recordTokens) {
+      return recordTokens;
+    }
+    return this.reflectionConfig.observationTokens;
+  }
   /**
    * Call the Reflector agent with escalating compression levels.
    */
@@ -4326,7 +4342,7 @@ var ReflectorRunner = class {
     const freshRecord = await this.storage.getObservationalMemory(record.threadId, record.resourceId);
     const currentRecord = freshRecord ?? record;
     const observationTokens = currentRecord.observationTokenCount ?? 0;
-    const reflectThreshold = getMaxThreshold(this.reflectionConfig.observationTokens);
+    const reflectThreshold = getMaxThreshold(this.getEffectiveReflectionTokens(currentRecord));
     const bufferActivation = this.reflectionConfig.bufferActivation ?? 0.5;
     const startedAt = (/* @__PURE__ */ new Date()).toISOString();
     const cycleId = `reflect-buf-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`;
@@ -4355,7 +4371,7 @@ var ReflectorRunner = class {
         recordId: record.id,
         threadId: record.threadId ?? "",
         threadIds: record.threadId ? [record.threadId] : [],
-        config: this.getObservationMarkerConfig()
+        config: this.getObservationMarkerConfig(currentRecord)
       });
       void writer.custom(startMarker).catch(() => {
       });
@@ -4467,7 +4483,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
         threadId: freshRecord.threadId ?? "",
         generationCount: afterRecord?.generationCount ?? freshRecord.generationCount ?? 0,
         observations: afterRecord?.activeObservations,
-        config: this.getObservationMarkerConfig()
+        config: this.getObservationMarkerConfig(freshRecord)
       });
       void writer.custom(activationMarker).catch(() => {
       });
@@ -4498,7 +4514,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       observabilityContext
     } = opts;
     const lockKey = this.buffering.getLockKey(record.threadId, record.resourceId);
-    const reflectThreshold = getMaxThreshold(this.reflectionConfig.observationTokens);
+    const reflectThreshold = getMaxThreshold(this.getEffectiveReflectionTokens(record));
     if (this.buffering.isAsyncReflectionEnabled() && observationTokens < reflectThreshold) {
       const shouldTrigger = (() => {
         if (!this.buffering.isAsyncReflectionEnabled()) return false;
@@ -4577,7 +4593,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
         recordId: record.id,
         threadId,
         threadIds: [threadId],
-        config: this.getObservationMarkerConfig()
+        config: this.getObservationMarkerConfig(record)
       });
       await writer.custom(startMarker).catch(() => {
       });
@@ -8613,6 +8629,10 @@ function getOmObservabilityContext(args) {
     metrics: args.metrics
   };
 }
+var GATEWAY_STATE_KEY = "__isGatewayModel";
+function isMastraGatewayModel(model) {
+  return model instanceof llm.ModelRouterLanguageModel && model.gatewayId === "mastra";
+}
 var ObservationalMemoryProcessor = class {
   id = "observational-memory";
   name = "Observational Memory";
@@ -8648,6 +8668,11 @@ var ObservationalMemoryProcessor = class {
       omDebug(`[OM:processInputStep:NO-CONTEXT] getThreadContext returned null \u2014 returning early`);
       return messageList;
     }
+    if (isMastraGatewayModel(model)) {
+      state[GATEWAY_STATE_KEY] = true;
+      omDebug(`[OM:processInputStep:GATEWAY] gateway handles OM \u2014 skipping local processing`);
+      return messageList;
+    }
     const { threadId, resourceId } = context;
     const memoryContext = memory.parseMemoryRequestContext(requestContext);
     const readOnly = memoryContext?.memoryConfig?.readOnly;
@@ -8764,6 +8789,7 @@ var ObservationalMemoryProcessor = class {
     const state = _state ?? {};
     const context = this.engine.getThreadContext(requestContext, messageList);
     if (!context) return messageList;
+    if (state[GATEWAY_STATE_KEY]) return messageList;
     const observabilityContext = getOmObservabilityContext(args);
     state.__omObservabilityContext = observabilityContext;
     return this.engine.getTokenCounter().runWithModelContext(state.__omActorModelContext, async () => {
@@ -8840,5 +8866,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-ANKEPFC6.cjs.map
-//# sourceMappingURL=chunk-ANKEPFC6.cjs.map
+//# sourceMappingURL=chunk-WNLRMAHY.cjs.map
+//# sourceMappingURL=chunk-WNLRMAHY.cjs.map