@mastra/memory 1.17.5-alpha.0 → 1.17.5

package/dist/{chunk-6ODYOMHN.cjs → chunk-UZDSNIGD.cjs}

@@ -3759,6 +3759,7 @@ var ObserverRunner = class {
   observedMessageIds;
   resolveModel;
   tokenCounter;
+  mastra;
   /** Captured prompt/response from the last observer call (for repro capture). */
   lastExchange;
   constructor(opts) {
@@ -3766,9 +3767,13 @@ var ObserverRunner = class {
     this.observedMessageIds = opts.observedMessageIds;
     this.resolveModel = opts.resolveModel;
     this.tokenCounter = opts.tokenCounter;
+    this.mastra = opts.mastra;
+  }
+  __registerMastra(mastra) {
+    this.mastra = mastra;
   }
   createAgent(model, isMultiThread = false) {
-    return new agent.Agent({
+    const agent$1 = new agent.Agent({
       id: isMultiThread ? "multi-thread-observer" : "observational-memory-observer",
       name: isMultiThread ? "multi-thread-observer" : "Observer",
       instructions: buildObserverSystemPrompt(
@@ -3778,6 +3783,10 @@ var ObserverRunner = class {
       ),
       model
     });
+    if (this.mastra) {
+      agent$1.__registerMastra(this.mastra);
+    }
+    return agent$1;
   }
   async withAbortCheck(fn, abortSignal) {
     if (abortSignal?.aborted) {
@@ -4297,6 +4306,7 @@ var ReflectorRunner = class {
   persistMarkerToStorage;
   persistMarkerToMessage;
   getCompressionStartLevel;
+  mastra;
   constructor(opts) {
     this.reflectionConfig = opts.reflectionConfig;
     this.observationConfig = opts.observationConfig;
@@ -4309,14 +4319,22 @@ var ReflectorRunner = class {
     this.persistMarkerToStorage = opts.persistMarkerToStorage;
     this.persistMarkerToMessage = opts.persistMarkerToMessage;
     this.getCompressionStartLevel = opts.getCompressionStartLevel;
+    this.mastra = opts.mastra;
+  }
+  __registerMastra(mastra) {
+    this.mastra = mastra;
   }
   createAgent(model) {
-    return new agent.Agent({
+    const agent$1 = new agent.Agent({
       id: "observational-memory-reflector",
       name: "Reflector",
       instructions: buildReflectorSystemPrompt(this.reflectionConfig.instruction),
       model
     });
+    if (this.mastra) {
+      agent$1.__registerMastra(this.mastra);
+    }
+    return agent$1;
   }
   getObservationMarkerConfig(record) {
     return {
@@ -6602,6 +6620,7 @@ var ObservationalMemory = class _ObservationalMemory {
   buffering;
   shouldObscureThreadIds = false;
   hasher = xxhash__default.default();
+  mastra;
   /**
    * Track message IDs observed during this instance's lifetime.
    * Prevents re-observing messages when per-thread lastObservedAt cursors
@@ -6668,20 +6687,10 @@ var ObservationalMemory = class _ObservationalMemory {
     this.scope = config.scope ?? "thread";
     this.retrieval = Boolean(config.retrieval);
     this.onIndexObservations = config.onIndexObservations;
-    const resolveModel = (m) => m === "default" ? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model : m;
-    const observationModel = resolveModel(config.model) ?? resolveModel(config.observation?.model) ?? resolveModel(config.reflection?.model);
-    const reflectionModel = resolveModel(config.model) ?? resolveModel(config.reflection?.model) ?? resolveModel(config.observation?.model);
-    if (!observationModel || !reflectionModel) {
-      throw new Error(
-        `Observational Memory requires a model to be set. Use \`observationalMemory: true\` for the default (google/gemini-2.5-flash), or set a model explicitly:
-
-  observationalMemory: {
-    model: "$provider/$model",
-  }
-
-See https://mastra.ai/docs/memory/observational-memory#models for model recommendations and alternatives.`
-      );
-    }
+    this.mastra = config.mastra;
+    const resolveModel = (model, defaultModel) => model === "default" ? defaultModel : model;
+    const observationModel = resolveModel(config.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.observation?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.reflection?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model;
+    const reflectionModel = resolveModel(config.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.reflection?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.observation?.model, chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model;
     const messageTokens = config.observation?.messageTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens;
     const observationTokens = config.reflection?.observationTokens ?? chunkD4J4XPGM_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.observationTokens;
     const isSharedBudget = config.shareTokenBudget ?? false;
@@ -6767,7 +6776,8 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
       observationConfig: this.observationConfig,
       observedMessageIds: this.observedMessageIds,
       resolveModel: (inputTokens) => this.resolveObservationModel(inputTokens),
-      tokenCounter: this.tokenCounter
+      tokenCounter: this.tokenCounter,
+      mastra: config.mastra
     });
     this.buffering = new BufferingCoordinator({
       observationConfig: this.observationConfig,
@@ -6785,13 +6795,19 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
       persistMarkerToStorage: (m, t, r) => this.persistMarkerToStorage(m, t, r),
       persistMarkerToMessage: (m, ml, t, r) => this.persistMarkerToMessage(m, ml, t, r),
       getCompressionStartLevel: (rc) => this.getCompressionStartLevel(rc),
-      resolveModel: (inputTokens) => this.resolveReflectionModel(inputTokens)
+      resolveModel: (inputTokens) => this.resolveReflectionModel(inputTokens),
+      mastra: config.mastra
     });
     this.validateBufferConfig();
     omDebug(
       `[OM:init] new ObservationalMemory instance created \u2014 scope=${this.scope}, messageTokens=${JSON.stringify(this.observationConfig.messageTokens)}, obsAsyncEnabled=${this.buffering.isAsyncObservationEnabled()}, bufferTokens=${this.observationConfig.bufferTokens}, bufferActivation=${this.observationConfig.bufferActivation}, blockAfter=${this.observationConfig.blockAfter}, reflectionTokens=${this.reflectionConfig.observationTokens}, refAsyncEnabled=${this.buffering.isAsyncReflectionEnabled()}, refAsyncActivation=${this.reflectionConfig.bufferActivation}, refBlockAfter=${this.reflectionConfig.blockAfter}`
     );
   }
+  __registerMastra(mastra) {
+    this.mastra = mastra;
+    this.observer.__registerMastra(mastra);
+    this.reflector.__registerMastra(mastra);
+  }
   /**
    * Get the current configuration for this OM instance.
    * Used by the server to expose config to the UI when OM is added via processors.
@@ -6900,7 +6916,7 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
     if (!modelToResolve) {
       return void 0;
     }
-    const resolved = await llm.resolveModelConfig(modelToResolve, requestContext);
+    const resolved = await llm.resolveModelConfig(modelToResolve, requestContext, this.mastra);
     return {
       provider: resolved.provider,
       modelId: resolved.modelId
@@ -9382,5 +9398,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-6ODYOMHN.cjs.map
-//# sourceMappingURL=chunk-6ODYOMHN.cjs.map
+//# sourceMappingURL=chunk-UZDSNIGD.cjs.map
+//# sourceMappingURL=chunk-UZDSNIGD.cjs.map