@mastra/memory 1.16.0-alpha.1 → 1.16.0-alpha.2

package/dist/{chunk-HB6AYAFD.js → chunk-MGHUIRKN.js}

@@ -201,6 +201,19 @@ var BufferingCoordinator = class _BufferingCoordinator {
   }
 };
 
+// src/processors/observational-memory/model-context.ts
+function didProviderChange(actorModel, lastModel) {
+  if (actorModel === void 0 || lastModel === void 0) return false;
+  const actorHasSlash = actorModel.includes("/");
+  const lastHasSlash = lastModel.includes("/");
+  if (actorHasSlash && lastHasSlash) {
+    return actorModel !== lastModel;
+  }
+  const actorModelId = actorHasSlash ? actorModel.slice(actorModel.indexOf("/") + 1) : actorModel;
+  const lastModelId = lastHasSlash ? lastModel.slice(lastModel.indexOf("/") + 1) : lastModel;
+  return actorModelId !== lastModelId;
+}
+
 // src/processors/observational-memory/date-utils.ts
 function formatRelativeTime(date, currentDate) {
   const diffMs = currentDate.getTime() - date.getTime();
@@ -487,7 +500,9 @@ function createActivationMarker(params) {
       observations: params.observations,
       triggeredBy: params.triggeredBy,
       lastActivityAt: params.lastActivityAt,
-      ttlExpiredMs: params.ttlExpiredMs
+      ttlExpiredMs: params.ttlExpiredMs,
+      previousModel: params.previousModel,
+      currentModel: params.currentModel
     }
   };
 }
@@ -1982,6 +1997,7 @@ var ObservationStep = class {
         resourceId,
         checkThreshold: true,
         messages: step0Messages,
+        currentModel: this.turn.actorModelContext,
         writer: this.turn.writer,
         messageList
       });
@@ -2005,6 +2021,8 @@ var ObservationStep = class {
         observationTokens: obsTokens,
         threadId,
         writer: this.turn.writer,
+        messageList,
+        currentModel: this.turn.actorModelContext,
         requestContext: this.turn.requestContext,
         observabilityContext: this.turn.observabilityContext,
         lastActivityAt: getLastActivityFromMessages(messageList.get.all.db())
@@ -2171,6 +2189,7 @@ var ObservationStep = class {
         threadId,
         resourceId,
         messages: messageList.get.all.db(),
+        currentModel: this.turn.actorModelContext,
         writer: this.turn.writer,
         messageList
       });
@@ -2182,6 +2201,7 @@ var ObservationStep = class {
           threadId,
           writer: this.turn.writer,
           messageList,
+          currentModel: this.turn.actorModelContext,
           requestContext: this.turn.requestContext,
           observabilityContext: this.turn.observabilityContext,
           lastActivityAt: getLastActivityFromMessages(messageList.get.all.db())
@@ -2226,6 +2246,8 @@ var ObservationTurn = class {
   requestContext;
   /** Optional observability context for nested OM spans. */
   observabilityContext;
+  /** Current actor model for this step. Updated by the processor before prepare(). */
+  actorModelContext;
   /** Optional processor-provided hooks for turn/step lifecycle integration. */
   hooks;
   constructor(opts) {
@@ -4094,12 +4116,44 @@ function validateCompression(reflectedTokens, targetThreshold) {
 }
 
 // src/processors/observational-memory/reflector-runner.ts
+function formatModelContext(provider, modelId) {
+  if (provider && modelId) {
+    return `${provider}/${modelId}`;
+  }
+  return modelId;
+}
+function getCurrentModel(model) {
+  return formatModelContext(model?.provider, model?.modelId);
+}
+function getLastModelFromMessageList(messageList) {
+  const messages = messageList?.get.all.db();
+  if (!messages) return void 0;
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const message = messages[i];
+    if (!message || message.role !== "assistant" || !message.content || typeof message.content === "string") {
+      continue;
+    }
+    for (let j = message.content.parts.length - 1; j >= 0; j--) {
+      const part = message.content.parts[j];
+      if (part?.type === "step-start" && typeof part.model === "string" && part.model.length > 0) {
+        return part.model;
+      }
+    }
+    const metadata = message.content.metadata;
+    const model = formatModelContext(metadata?.provider, metadata?.modelId);
+    if (model) {
+      return model;
+    }
+  }
+  return void 0;
+}
 async function withAbortCheck(fn, abortSignal) {
   if (abortSignal?.aborted) throw new Error("The operation was aborted.");
   const result = await fn();
   if (abortSignal?.aborted) throw new Error("The operation was aborted.");
   return result;
 }
+var EARLY_ACTIVATION_SIZE_FLOOR_RATIO = 0.75;
 var ReflectorRunner = class {
   reflectionConfig;
   observationConfig;
@@ -4424,7 +4478,9 @@ var ReflectorRunner = class {
   }
   /**
    * Try to activate buffered reflection when threshold is reached.
-   * Returns true if activation succeeded.
+   * Returns a discriminated result so the caller can distinguish between
+   * "activated", "no buffer present", and "suppressed by overshoot guard"
+   * without re-deriving that state.
    */
   async tryActivateBufferedReflection(record, lockKey, writer, messageList, activationMetadata) {
     const bufferKey = this.buffering.getReflectionBufferKey(lockKey);
@@ -4447,8 +4503,8 @@ var ReflectorRunner = class {
       `[OM:reflect] tryActivateBufferedReflection: freshRecord.id=${freshRecord?.id}, freshBufferedReflection=${freshRecord?.bufferedReflection ? "present (" + freshRecord.bufferedReflection.length + " chars)" : "empty"}, freshObsTokens=${freshRecord?.observationTokenCount}`
     );
     if (!freshRecord?.bufferedReflection) {
-      omDebug(`[OM:reflect] tryActivateBufferedReflection: no buffered reflection after re-fetch, returning false`);
-      return false;
+      omDebug(`[OM:reflect] tryActivateBufferedReflection: no buffered reflection after re-fetch`);
+      return { status: "no-buffer" };
     }
     const beforeTokens = freshRecord.observationTokenCount ?? 0;
     const reflectedLineCount = freshRecord.reflectedObservationLineCount ?? 0;
@@ -4460,6 +4516,26 @@ var ReflectorRunner = class {
 
 ${unreflectedContent}` : freshRecord.bufferedReflection;
     const combinedTokenCount = this.tokenCounter.countObservations(combinedObservations);
+    if (activationMetadata?.triggeredBy === "ttl" || activationMetadata?.triggeredBy === "provider_change") {
+      const unreflectedTailTokens = unreflectedContent ? this.tokenCounter.countObservations(unreflectedContent) : 0;
+      const bufferedReflectionTokens = freshRecord.bufferedReflectionTokens ?? 0;
+      if (unreflectedTailTokens < bufferedReflectionTokens) {
+        omDebug(
+          `[OM:reflect] tryActivateBufferedReflection: suppressing early ${activationMetadata.triggeredBy} activation \u2014 unreflectedTailTokens=${unreflectedTailTokens} < bufferedReflectionTokens=${bufferedReflectionTokens}; keeping buffer for threshold activation`
+        );
+        return { status: "suppressed", reason: "composition" };
+      }
+      const bufferActivation = this.reflectionConfig.bufferActivation;
+      const reflectThreshold = getMaxThreshold(this.getEffectiveReflectionTokens(freshRecord));
+      const regularActivationTarget = reflectThreshold * (1 - bufferActivation);
+      const minCombinedTokens = Math.round(regularActivationTarget * EARLY_ACTIVATION_SIZE_FLOOR_RATIO);
+      if (combinedTokenCount < minCombinedTokens) {
+        omDebug(
+          `[OM:reflect] tryActivateBufferedReflection: suppressing early ${activationMetadata.triggeredBy} activation \u2014 combinedTokenCount=${combinedTokenCount} < minCombinedTokens=${minCombinedTokens} (${EARLY_ACTIVATION_SIZE_FLOOR_RATIO * 100}% of regular activation target ${Math.round(regularActivationTarget)}, threshold=${reflectThreshold}, bufferActivation=${bufferActivation}); keeping buffer for threshold activation`
+        );
+        return { status: "suppressed", reason: "size" };
+      }
+    }
     omDebug(
       `[OM:reflect] tryActivateBufferedReflection: activating, beforeTokens=${beforeTokens}, combinedTokenCount=${combinedTokenCount}, reflectedLineCount=${reflectedLineCount}, unreflectedLines=${unreflectedLines.length}`
     );
@@ -4489,6 +4565,8 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
         triggeredBy: activationMetadata?.triggeredBy,
         lastActivityAt: activationMetadata?.lastActivityAt,
         ttlExpiredMs: activationMetadata?.ttlExpiredMs,
+        previousModel: activationMetadata?.previousModel,
+        currentModel: activationMetadata?.currentModel,
         config: this.getObservationMarkerConfig(freshRecord)
       });
       void writer.custom(activationMarker).catch(() => {
@@ -4501,7 +4579,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       );
     }
     BufferingCoordinator.reflectionBufferCycleIds.delete(bufferKey);
-    return true;
+    return { status: "activated" };
   }
   /**
    * Check if reflection needed and trigger if so.
@@ -4515,6 +4593,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       writer,
       abortSignal,
       messageList,
+      currentModel,
       reflectionHooks,
       requestContext,
       observabilityContext,
@@ -4554,13 +4633,19 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
     const activateAfterIdle = this.reflectionConfig.activateAfterIdle;
     const ttlExpiredMs = activateAfterIdle !== void 0 && lastActivityAt !== void 0 ? Date.now() - lastActivityAt : void 0;
     const ttlExpired = ttlExpiredMs !== void 0 && activateAfterIdle !== void 0 && ttlExpiredMs >= activateAfterIdle;
-    if (observationTokens < reflectThreshold && !ttlExpired) {
+    const actorModel = getCurrentModel(currentModel);
+    const lastModel = getLastModelFromMessageList(messageList);
+    const providerChanged = this.reflectionConfig.activateOnProviderChange === true && didProviderChange(actorModel, lastModel);
+    if (observationTokens < reflectThreshold && !ttlExpired && !providerChanged) {
       return;
     }
+    const activationTriggeredBy = observationTokens >= reflectThreshold ? "threshold" : providerChanged ? "provider_change" : "ttl";
     const activationMetadata = {
-      triggeredBy: ttlExpired ? "ttl" : "threshold",
-      lastActivityAt: ttlExpired ? lastActivityAt : void 0,
-      ttlExpiredMs: ttlExpired ? ttlExpiredMs : void 0
+      triggeredBy: activationTriggeredBy,
+      lastActivityAt: activationTriggeredBy === "ttl" ? lastActivityAt : void 0,
+      ttlExpiredMs: activationTriggeredBy === "ttl" ? ttlExpiredMs : void 0,
+      previousModel: activationTriggeredBy === "provider_change" ? lastModel : void 0,
+      currentModel: activationTriggeredBy === "provider_change" ? actorModel : void 0
     };
     if (record.isReflecting) {
       if (isOpActiveInProcess(record.id, "reflecting")) {
@@ -4571,14 +4656,20 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       await this.storage.setReflectingFlag(record.id, false);
     }
     if (this.buffering.isAsyncReflectionEnabled()) {
-      const activationSuccess = await this.tryActivateBufferedReflection(
+      const activationResult = await this.tryActivateBufferedReflection(
         record,
         lockKey,
         writer,
         messageList,
         activationMetadata
       );
-      if (activationSuccess) {
+      if (activationResult.status === "activated") {
+        return;
+      }
+      if (activationResult.status === "suppressed") {
+        omDebug(
+          `[OM:reflect] skipping sync fallback / re-buffer after suppressed early ${activationMetadata.triggeredBy} activation (reason=${activationResult.reason})`
+        );
         return;
       }
       if (this.reflectionConfig.blockAfter && observationTokens >= this.reflectionConfig.blockAfter) {
@@ -6279,6 +6370,36 @@ function getLastActivityFromMessages(messages) {
   }
   return void 0;
 }
+function formatModelContext2(provider, modelId) {
+  if (provider && modelId) {
+    return `${provider}/${modelId}`;
+  }
+  return modelId;
+}
+function getLastModelFromMessages(messages) {
+  if (!messages) return void 0;
+  for (let i = messages.length - 1; i >= 0; i--) {
+    const message = messages[i];
+    if (!message || message.role !== "assistant" || !message.content || typeof message.content === "string") {
+      continue;
+    }
+    for (let j = message.content.parts.length - 1; j >= 0; j--) {
+      const part = message.content.parts[j];
+      if (part?.type === "step-start" && typeof part.model === "string" && part.model.length > 0) {
+        return part.model;
+      }
+    }
+    const metadata = message.content.metadata;
+    const model = formatModelContext2(metadata?.provider, metadata?.modelId);
+    if (model) {
+      return model;
+    }
+  }
+  return void 0;
+}
+function getCurrentModel2(model) {
+  return formatModelContext2(model?.provider, model?.modelId);
+}
 function parseActivationTTL(value, fieldPath) {
   if (value === void 0) {
     return void 0;
@@ -6455,6 +6576,7 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
       ),
       bufferActivation: asyncBufferingDisabled ? void 0 : config.observation?.bufferActivation ?? OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferActivation,
       activateAfterIdle: parseActivationTTL(config.activateAfterIdle, "activateAfterIdle"),
+      activateOnProviderChange: config.activateOnProviderChange ?? false,
       blockAfter: asyncBufferingDisabled ? void 0 : resolveBlockAfter(
         config.observation?.blockAfter ?? (config.observation?.bufferTokens ?? OBSERVATIONAL_MEMORY_DEFAULTS.observation.bufferTokens ? 1.2 : void 0),
         config.observation?.messageTokens ?? OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens
@@ -6474,6 +6596,7 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
       providerOptions: config.reflection?.providerOptions ?? OBSERVATIONAL_MEMORY_DEFAULTS.reflection.providerOptions,
       bufferActivation: asyncBufferingDisabled ? void 0 : config?.reflection?.bufferActivation ?? OBSERVATIONAL_MEMORY_DEFAULTS.reflection.bufferActivation,
       activateAfterIdle: parseActivationTTL(config.activateAfterIdle, "activateAfterIdle"),
+      activateOnProviderChange: config.activateOnProviderChange ?? false,
       blockAfter: asyncBufferingDisabled ? void 0 : resolveBlockAfter(
         config.reflection?.blockAfter ?? (config.reflection?.bufferActivation ?? OBSERVATIONAL_MEMORY_DEFAULTS.reflection.bufferActivation ? 1.2 : void 0),
         config.reflection?.observationTokens ?? OBSERVATIONAL_MEMORY_DEFAULTS.reflection.observationTokens
@@ -8370,6 +8493,8 @@ ${grouped}` : grouped;
     let activationTriggeredBy = "threshold";
     let activationLastActivityAt;
     let activateAfterIdleExpiredMs;
+    let previousModel;
+    let currentModel;
     if (opts.checkThreshold) {
       const thresholdMessages = opts.messages ?? await this.loadMessagesFromStorage(
         threadId,
@@ -8380,7 +8505,14 @@ ${grouped}` : grouped;
       const lastActivityAt = getLastActivityFromMessages(thresholdMessages);
       const ttlExpiredMs = activateAfterIdle !== void 0 && lastActivityAt !== void 0 ? Date.now() - lastActivityAt : void 0;
       const ttlExpired = ttlExpiredMs !== void 0 && activateAfterIdle !== void 0 && ttlExpiredMs >= activateAfterIdle;
-      if (ttlExpired) {
+      const actorModel = getCurrentModel2(opts.currentModel);
+      const lastModel = getLastModelFromMessages(thresholdMessages);
+      const providerChanged = this.observationConfig.activateOnProviderChange === true && didProviderChange(actorModel, lastModel);
+      if (providerChanged) {
+        activationTriggeredBy = "provider_change";
+        previousModel = lastModel;
+        currentModel = actorModel;
+      } else if (ttlExpired) {
         activationTriggeredBy = "ttl";
         activationLastActivityAt = lastActivityAt;
         activateAfterIdleExpiredMs = ttlExpiredMs;
@@ -8449,6 +8581,8 @@ ${grouped}` : grouped;
           triggeredBy: activationTriggeredBy,
           lastActivityAt: activationLastActivityAt,
           ttlExpiredMs: activateAfterIdleExpiredMs,
+          previousModel,
+          currentModel,
           config: this.getObservationMarkerConfig()
         });
         void opts.writer.custom(activationMarker).catch(() => {
@@ -8806,6 +8940,7 @@ var ObservationalMemoryProcessor = class {
       const observabilityContext = getOmObservabilityContext(args);
       state.__omObservabilityContext = observabilityContext;
       this.turn.observabilityContext = observabilityContext;
+      this.turn.actorModelContext = actorModelContext;
       {
         const step = this.turn.step(stepNumber);
         let ctx;
@@ -8939,5 +9074,5 @@ function getObservationsAsOf(activeObservations, asOf) {
 }
 
 export { ModelByInputTokens, OBSERVER_SYSTEM_PROMPT, ObservationalMemory, ObservationalMemoryProcessor, TokenCounter, buildObserverPrompt, buildObserverSystemPrompt, combineObservationGroupRanges, deriveObservationGroupProvenance, extractCurrentTask, formatMessagesForObserver, formatToolResultForObserver, getObservationsAsOf, hasCurrentTaskSection, injectAnchorIds, optimizeObservationsForContext, parseAnchorId, parseObservationGroups, parseObserverOutput, reconcileObservationGroupsFromReflection, renderObservationGroupsForReflection, resolveToolResultValue, stripEphemeralAnchorIds, stripObservationGroups, truncateStringByTokens, wrapInObservationGroup };
-//# sourceMappingURL=chunk-HB6AYAFD.js.map
-//# sourceMappingURL=chunk-HB6AYAFD.js.map
+//# sourceMappingURL=chunk-MGHUIRKN.js.map
+//# sourceMappingURL=chunk-MGHUIRKN.js.map