@mastra/server 1.32.0-alpha.3 → 1.32.0-alpha.4

package/dist/{chunk-CPZEA2XG.cjs → chunk-ZW7DMFEG.cjs}

@@ -1042,7 +1042,7 @@ function imageSize(input) {
   throw new TypeError(`unsupported file type: ${type}`);
 }
 
-// ../memory/dist/chunk-Z5GWA7LB.js
+// ../memory/dist/chunk-BPJLUC2F.js
 var OM_DEBUG_LOG = process.env.OM_DEBUG ? path.join(process.cwd(), "om-debug.log") : null;
 function omDebug(msg) {
   if (!OM_DEBUG_LOG) return;
@@ -4751,6 +4751,7 @@ var ObserverRunner = class {
   observedMessageIds;
   resolveModel;
   tokenCounter;
+  mastra;
   /** Captured prompt/response from the last observer call (for repro capture). */
   lastExchange;
   constructor(opts) {
@@ -4758,9 +4759,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(
@@ -4770,6 +4775,10 @@ var ObserverRunner = class {
       ),
       model
     });
+    if (this.mastra) {
+      agent$1.__registerMastra(this.mastra);
+    }
+    return agent$1;
   }
   async withAbortCheck(fn, abortSignal) {
     if (abortSignal?.aborted) {
@@ -5285,6 +5294,7 @@ var ReflectorRunner = class {
   persistMarkerToStorage;
   persistMarkerToMessage;
   getCompressionStartLevel;
+  mastra;
   constructor(opts) {
     this.reflectionConfig = opts.reflectionConfig;
     this.observationConfig = opts.observationConfig;
@@ -5297,14 +5307,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 {
@@ -7588,6 +7606,7 @@ var ObservationalMemory = class _ObservationalMemory {
   buffering;
   shouldObscureThreadIds = false;
   hasher = e();
+  mastra;
   /**
    * Track message IDs observed during this instance's lifetime.
    * Prevents re-observing messages when per-thread lastObservedAt cursors
@@ -7654,20 +7673,10 @@ var ObservationalMemory = class _ObservationalMemory {
     this.scope = config.scope ?? "thread";
     this.retrieval = Boolean(config.retrieval);
     this.onIndexObservations = config.onIndexObservations;
-    const resolveModel = (m) => m === "default" ? chunk3D55EABW_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, chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.observation?.model, chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? resolveModel(config.reflection?.model, chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model) ?? chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.model;
+    const reflectionModel = resolveModel(config.model, chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.reflection?.model, chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? resolveModel(config.observation?.model, chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model) ?? chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.model;
     const messageTokens = config.observation?.messageTokens ?? chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.observation.messageTokens;
     const observationTokens = config.reflection?.observationTokens ?? chunk3D55EABW_cjs.OBSERVATIONAL_MEMORY_DEFAULTS.reflection.observationTokens;
     const isSharedBudget = config.shareTokenBudget ?? false;
@@ -7753,7 +7762,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,
@@ -7771,13 +7781,19 @@ Async buffering is enabled by default \u2014 this opt-out is only needed when us
       persistMarkerToStorage: (m, t2, r) => this.persistMarkerToStorage(m, t2, r),
       persistMarkerToMessage: (m, ml, t2, r) => this.persistMarkerToMessage(m, ml, t2, 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.
@@ -7886,7 +7902,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
@@ -10364,5 +10380,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-CPZEA2XG.cjs.map
-//# sourceMappingURL=chunk-CPZEA2XG.cjs.map
+//# sourceMappingURL=chunk-ZW7DMFEG.cjs.map
+//# sourceMappingURL=chunk-ZW7DMFEG.cjs.map