@mastra/server 1.16.0 → 1.16.1-alpha.1

package/dist/{chunk-VCBFCDP3.cjs → chunk-SZBVW2WD.cjs}

@@ -1,10 +1,10 @@
 'use strict';
 
+var chunkR2ODPDKE_cjs = require('./chunk-R2ODPDKE.cjs');
 var chunk5N66PU3H_cjs = require('./chunk-5N66PU3H.cjs');
 var chunkEXKS4QPI_cjs = require('./chunk-EXKS4QPI.cjs');
 var chunkFPURK3UW_cjs = require('./chunk-FPURK3UW.cjs');
-var chunkUTQV3JAT_cjs = require('./chunk-UTQV3JAT.cjs');
-var chunkR2ODPDKE_cjs = require('./chunk-R2ODPDKE.cjs');
+var chunkR5EM57UR_cjs = require('./chunk-R5EM57UR.cjs');
 var chunk3W54ZNYP_cjs = require('./chunk-3W54ZNYP.cjs');
 var chunkDOHUOYZS_cjs = require('./chunk-DOHUOYZS.cjs');
 var chunkHITLRKIU_cjs = require('./chunk-HITLRKIU.cjs');
@@ -16193,8 +16193,8 @@ function formatTimestamp(date) {
   return date.toISOString().replace("T", " ").replace(/\.\d{3}Z$/, "Z");
 }
 function truncateByTokens(text42, maxTokens, hint) {
-  if (chunkUTQV3JAT_cjs.estimateTokenCount(text42) <= maxTokens) return { text: text42, wasTruncated: false };
-  const truncated = chunkUTQV3JAT_cjs.truncateStringByTokens(text42, maxTokens);
+  if (chunkR5EM57UR_cjs.estimateTokenCount(text42) <= maxTokens) return { text: text42, wasTruncated: false };
+  const truncated = chunkR5EM57UR_cjs.truncateStringByTokens(text42, maxTokens);
   const suffix = hint ? ` [${hint} for more]` : "";
   return { text: truncated + suffix, wasTruncated: true };
 }
@@ -16227,11 +16227,11 @@ function formatMessageParts(msg, detail) {
       } else if (partType === "tool-invocation") {
         const inv = part.toolInvocation;
         if (inv.state === "result") {
-          const { value: resultValue } = chunkUTQV3JAT_cjs.resolveToolResultValue(
+          const { value: resultValue } = chunkR5EM57UR_cjs.resolveToolResultValue(
             part,
             inv.result
           );
-          const resultStr = chunkUTQV3JAT_cjs.formatToolResultForObserver(resultValue, { maxTokens: HIGH_DETAIL_TOOL_RESULT_TOKENS });
+          const resultStr = chunkR5EM57UR_cjs.formatToolResultForObserver(resultValue, { maxTokens: HIGH_DETAIL_TOOL_RESULT_TOKENS });
           const fullText = `[Tool Result: ${inv.toolName}]
 ${resultStr}`;
           parts.push(makePart(msg, i, "tool-result", fullText, detail));
@@ -16293,9 +16293,9 @@ function expandPriority(part) {
 }
 function renderFormattedParts(parts, timestamps, options) {
   const text42 = buildRenderedText(parts, timestamps);
-  let totalTokens = chunkUTQV3JAT_cjs.estimateTokenCount(text42);
+  let totalTokens = chunkR5EM57UR_cjs.estimateTokenCount(text42);
   if (totalTokens > options.maxTokens) {
-    const truncated = chunkUTQV3JAT_cjs.truncateStringByTokens(text42, options.maxTokens);
+    const truncated = chunkR5EM57UR_cjs.truncateStringByTokens(text42, options.maxTokens);
     return { text: truncated, truncated: true, tokenOffset: totalTokens - options.maxTokens };
   }
   const truncatedIndices = parts.map((p, i) => ({ part: p, index: i })).filter(({ part }) => part.text !== part.fullText).sort((a, b) => expandPriority(a.part) - expandPriority(b.part));
@@ -16306,8 +16306,8 @@ function renderFormattedParts(parts, timestamps, options) {
   for (const { part, index } of truncatedIndices) {
     if (remaining <= 0) break;
     const maxTokens = expandLimit(part);
-    const fullTokens = chunkUTQV3JAT_cjs.estimateTokenCount(part.fullText);
-    const currentTokens = chunkUTQV3JAT_cjs.estimateTokenCount(part.text);
+    const fullTokens = chunkR5EM57UR_cjs.estimateTokenCount(part.fullText);
+    const currentTokens = chunkR5EM57UR_cjs.estimateTokenCount(part.text);
     const targetTokens = Math.min(fullTokens, maxTokens);
     const delta = targetTokens - currentTokens;
     if (delta <= 0) continue;
@@ -16318,17 +16318,17 @@ function renderFormattedParts(parts, timestamps, options) {
       const expandedLimit = Math.min(currentTokens + remaining, maxTokens);
       const hint = `recall cursor="${part.messageId}" partIndex=${part.partIndex} detail="high"`;
       const { text: expanded2 } = truncateByTokens(part.fullText, expandedLimit, hint);
-      const expandedDelta = chunkUTQV3JAT_cjs.estimateTokenCount(expanded2) - currentTokens;
+      const expandedDelta = chunkR5EM57UR_cjs.estimateTokenCount(expanded2) - currentTokens;
       parts[index] = { ...part, text: expanded2 };
       remaining -= expandedDelta;
     }
   }
   const expanded = buildRenderedText(parts, timestamps);
-  const expandedTokens = chunkUTQV3JAT_cjs.estimateTokenCount(expanded);
+  const expandedTokens = chunkR5EM57UR_cjs.estimateTokenCount(expanded);
   if (expandedTokens <= options.maxTokens) {
     return { text: expanded, truncated: false, tokenOffset: 0 };
   }
-  const hardTruncated = chunkUTQV3JAT_cjs.truncateStringByTokens(expanded, options.maxTokens);
+  const hardTruncated = chunkR5EM57UR_cjs.truncateStringByTokens(expanded, options.maxTokens);
   return { text: hardTruncated, truncated: true, tokenOffset: expandedTokens - options.maxTokens };
 }
 async function recallPart({
@@ -16363,7 +16363,7 @@ async function recallPart({
       `Part index ${partIndex} not found in message ${cursor}. Available indices: ${allParts.map((p) => p.partIndex).join(", ")}`
     );
   }
-  const truncatedText = chunkUTQV3JAT_cjs.truncateStringByTokens(target.text, maxTokens);
+  const truncatedText = chunkR5EM57UR_cjs.truncateStringByTokens(target.text, maxTokens);
   const wasTruncated = truncatedText !== target.text;
   return {
     text: truncatedText,
@@ -16784,6 +16784,14 @@ var DEFAULT_MESSAGE_RANGE = { before: 1, after: 1 };
 var DEFAULT_TOP_K = 4;
 var VECTOR_DELETE_BATCH_SIZE = 100;
 var Memory = class extends memory.MastraMemory {
+  _omEngine;
+  /** The shared ObservationalMemory engine. Lazily created on first access. */
+  get omEngine() {
+    if (!this._omEngine) {
+      this._omEngine = this._initOMEngine();
+    }
+    return this._omEngine;
+  }
   constructor(config = {}) {
     super({ name: "Memory", ...config });
     const mergedConfig = this.getMergedThreadConfig({
@@ -17185,7 +17193,7 @@ ${workingMemory}`;
     }
     return chunks;
   }
-  hasher = chunkUTQV3JAT_cjs.e();
+  hasher = chunkR5EM57UR_cjs.e();
   // embedding is computationally expensive so cache content -> embeddings/chunks
   embeddingCache = /* @__PURE__ */ new Map();
   firstEmbed;
@@ -17424,6 +17432,179 @@ ${workingMemory}`;
       data: workingMemoryData
     });
   }
+  /**
+   * Get everything needed for an LLM call in one shot.
+   *
+   * Assembles the system message (observations + working memory), loads
+   * unobserved messages from storage, and returns them ready to use.
+   *
+   * @example
+   * ```ts
+   * const ctx = await memory.getContext({ threadId });
+   * const result = await generateText({
+   *   model: openai('gpt-4o'),
+   *   system: ctx.systemMessage,
+   *   messages: ctx.messages.map(toAiSdkMessage),
+   * });
+   * ```
+   */
+  async getContext(opts) {
+    const { threadId, resourceId, memoryConfig } = opts;
+    const config = this.getMergedThreadConfig(memoryConfig);
+    const memoryStore = await this.getMemoryStore();
+    const systemParts = [];
+    let hasObservations = false;
+    let omRecord = null;
+    let continuationMessage;
+    let otherThreadsContext;
+    const omEngine = await this.omEngine;
+    if (omEngine) {
+      omRecord = await omEngine.getRecord(threadId, resourceId);
+      if (omRecord?.activeObservations) {
+        hasObservations = true;
+        if (omEngine.scope === "resource" && resourceId) {
+          otherThreadsContext = await omEngine.getOtherThreadsContext(resourceId, threadId);
+        }
+        const obsSystemMessage = await omEngine.buildContextSystemMessage({
+          threadId,
+          resourceId,
+          record: omRecord,
+          unobservedContextBlocks: otherThreadsContext
+        });
+        if (obsSystemMessage) {
+          systemParts.push(obsSystemMessage);
+        }
+        const { OBSERVATION_CONTINUATION_HINT } = await import('./constants-BDOITAO3-YWTMWUNT.cjs');
+        continuationMessage = {
+          id: "om-continuation",
+          role: "user",
+          createdAt: /* @__PURE__ */ new Date(0),
+          content: {
+            format: 2,
+            parts: [
+              {
+                type: "text",
+                text: `<system-reminder>${OBSERVATION_CONTINUATION_HINT}</system-reminder>`
+              }
+            ]
+          },
+          threadId,
+          resourceId
+        };
+      }
+    }
+    const workingMemoryMessage = await this.getSystemMessage({ threadId, resourceId, memoryConfig: config });
+    if (workingMemoryMessage) {
+      systemParts.push(workingMemoryMessage);
+    }
+    let messages;
+    if (omEngine && omRecord) {
+      const dateFilter = omRecord.lastObservedAt ? { dateRange: { start: new Date(new Date(omRecord.lastObservedAt).getTime() + 1) } } : void 0;
+      if (omEngine.scope === "resource" && resourceId) {
+        const result = await memoryStore.listMessagesByResourceId({
+          resourceId,
+          orderBy: { field: "createdAt", direction: "ASC" },
+          perPage: false,
+          filter: dateFilter
+        });
+        messages = result.messages;
+      } else {
+        const result = await memoryStore.listMessages({
+          threadId,
+          orderBy: { field: "createdAt", direction: "ASC" },
+          perPage: false,
+          filter: dateFilter
+        });
+        messages = result.messages;
+      }
+    } else {
+      const lastMessages = config.lastMessages;
+      if (lastMessages === false) {
+        messages = [];
+      } else {
+        const result = await memoryStore.listMessages({
+          threadId,
+          resourceId,
+          orderBy: { field: "createdAt", direction: "DESC" },
+          perPage: typeof lastMessages === "number" ? lastMessages : void 0
+        });
+        messages = result.messages.reverse();
+      }
+    }
+    return {
+      systemMessage: systemParts.length > 0 ? systemParts.join("\n\n") : void 0,
+      messages,
+      hasObservations,
+      omRecord,
+      continuationMessage,
+      otherThreadsContext
+    };
+  }
+  /**
+   * Raw message upsert — persist messages to storage without embedding or working memory processing.
+   * Used by the processor to save sealed messages before firing a background buffer operation.
+   */
+  async persistMessages(messages) {
+    if (messages.length === 0) return;
+    const memoryStore = await this.getMemoryStore();
+    await memoryStore.saveMessages({ messages });
+  }
+  /**
+   * One-time initialization of the shared ObservationalMemory engine.
+   * Called lazily by the `omEngine` getter on first access.
+   */
+  async _initOMEngine() {
+    const omConfig = normalizeObservationalMemoryConfig(this.threadConfig.observationalMemory);
+    if (!omConfig) return null;
+    const memoryStore = await this.storage.getStore("memory");
+    if (!memoryStore || !memoryStore.supportsObservationalMemory) return null;
+    const coreSupportsOM = features.coreFeatures.has("observationalMemory");
+    if (!coreSupportsOM) {
+      throw new Error(
+        "Observational memory is enabled but the installed version of @mastra/core does not support it. Please upgrade @mastra/core to a version that includes observational memory support."
+      );
+    }
+    if (omConfig.observation?.bufferTokens !== false && !features.coreFeatures.has("asyncBuffering")) {
+      throw new Error(
+        "Observational memory async buffering is enabled by default but the installed version of @mastra/core does not support it. Either upgrade @mastra/core, @mastra/memory, and your storage adapter (@mastra/libsql, @mastra/pg, or @mastra/mongodb) to the latest version, or explicitly disable async buffering by setting `observation: { bufferTokens: false }` in your observationalMemory config."
+      );
+    }
+    if (!features.coreFeatures.has("request-response-id-rotation")) {
+      throw new Error(
+        "Observational memory requires @mastra/core support for request-response-id-rotation. Please bump @mastra/core to a newer version."
+      );
+    }
+    const { ObservationalMemory: OMClass } = await import('./observational-memory-WMCWT577-MXWR6ZYS.cjs');
+    return new OMClass({
+      storage: memoryStore,
+      scope: omConfig.scope,
+      retrieval: omConfig.retrieval,
+      shareTokenBudget: omConfig.shareTokenBudget,
+      model: omConfig.model,
+      observation: omConfig.observation ? {
+        model: omConfig.observation.model,
+        messageTokens: omConfig.observation.messageTokens,
+        modelSettings: omConfig.observation.modelSettings,
+        maxTokensPerBatch: omConfig.observation.maxTokensPerBatch,
+        providerOptions: omConfig.observation.providerOptions,
+        bufferTokens: omConfig.observation.bufferTokens,
+        bufferActivation: omConfig.observation.bufferActivation,
+        blockAfter: omConfig.observation.blockAfter,
+        previousObserverTokens: omConfig.observation.previousObserverTokens,
+        instruction: omConfig.observation.instruction,
+        threadTitle: omConfig.observation.threadTitle
+      } : void 0,
+      reflection: omConfig.reflection ? {
+        model: omConfig.reflection.model,
+        observationTokens: omConfig.reflection.observationTokens,
+        modelSettings: omConfig.reflection.modelSettings,
+        providerOptions: omConfig.reflection.providerOptions,
+        bufferActivation: omConfig.reflection.bufferActivation,
+        blockAfter: omConfig.reflection.blockAfter,
+        instruction: omConfig.reflection.instruction
+      } : void 0
+    });
+  }
   defaultWorkingMemoryTemplate = `
 # User Information
 - **First Name**: 
@@ -18085,81 +18266,29 @@ Notes:
     return processors;
   }
   /**
-   * Creates an ObservationalMemory processor instance if configured and not already present.
-   * A new instance is created per call — processorStates (e.g., sealedIds) are shared
-   * via the ProcessorRunner's state map keyed by processor ID, not by instance identity.
+   * Creates an ObservationalMemory processor wrapping the shared engine.
+   * Returns null if OM is not configured, not supported, or already present
+   * in the user's configured processors.
    */
   async createOMProcessor(configuredProcessors = [], context2) {
     const hasObservationalMemory = configuredProcessors.some(
       (p) => !("workflow" in p) && p.id === "observational-memory"
     );
-    const memoryContext = context2?.get("MastraMemory");
-    const runtimeMemoryConfig = memoryContext?.memoryConfig;
-    const effectiveConfig = runtimeMemoryConfig ? this.getMergedThreadConfig({
-      ...runtimeMemoryConfig,
-      observationalMemory: runtimeMemoryConfig.observationalMemory
+    if (hasObservationalMemory) return null;
+    const runtimeMemory = context2?.get("MastraMemory");
+    const runtimeObservationalMemory = normalizeObservationalMemoryConfig(
+      runtimeMemory?.memoryConfig?.observationalMemory
+    );
+    const threadConfig = runtimeObservationalMemory ? this.getMergedThreadConfig({
+      ...runtimeMemory?.memoryConfig,
+      observationalMemory: runtimeObservationalMemory
     }) : this.threadConfig;
-    const omConfig = normalizeObservationalMemoryConfig(effectiveConfig.observationalMemory);
-    if (!omConfig || hasObservationalMemory) {
-      return null;
-    }
-    const coreSupportsOM = features.coreFeatures.has("observationalMemory");
-    if (!coreSupportsOM) {
-      throw new Error(
-        "Observational memory is enabled but the installed version of @mastra/core does not support it. Please upgrade @mastra/core to a version that includes observational memory support."
-      );
-    }
-    const memoryStore = await this.storage.getStore("memory");
-    if (!memoryStore) {
-      throw new Error(
-        "Using Mastra Memory observational memory requires a storage adapter but no attached adapter was detected."
-      );
-    }
-    if (!memoryStore.supportsObservationalMemory) {
-      throw new Error(
-        `Observational memory is enabled but the storage adapter (${memoryStore.constructor.name}) does not support it. If you're using @mastra/libsql, @mastra/pg, or @mastra/mongodb, upgrade to the latest version. Otherwise, use one of those adapters or disable observational memory.`
-      );
-    }
-    if (omConfig.observation?.bufferTokens !== false && !features.coreFeatures.has("asyncBuffering")) {
-      throw new Error(
-        "Observational memory async buffering is enabled by default but the installed version of @mastra/core does not support it. Either upgrade @mastra/core, @mastra/memory, and your storage adapter (@mastra/libsql, @mastra/pg, or @mastra/mongodb) to the latest version, or explicitly disable async buffering by setting `observation: { bufferTokens: false }` in your observationalMemory config."
-      );
-    }
-    if (!features.coreFeatures.has("request-response-id-rotation")) {
-      throw new Error(
-        "Observational memory requires @mastra/core support for request-response-id-rotation. Please bump @mastra/core to a newer version."
-      );
-    }
-    const { ObservationalMemory } = await import('./observational-memory-K2U3QQIO-MBBXGVVG.cjs');
-    return new ObservationalMemory({
-      storage: memoryStore,
-      scope: omConfig.scope,
-      retrieval: omConfig.retrieval,
-      shareTokenBudget: omConfig.shareTokenBudget,
-      model: omConfig.model,
-      observation: omConfig.observation ? {
-        model: omConfig.observation.model,
-        messageTokens: omConfig.observation.messageTokens,
-        modelSettings: omConfig.observation.modelSettings,
-        maxTokensPerBatch: omConfig.observation.maxTokensPerBatch,
-        providerOptions: omConfig.observation.providerOptions,
-        bufferTokens: omConfig.observation.bufferTokens,
-        bufferActivation: omConfig.observation.bufferActivation,
-        blockAfter: omConfig.observation.blockAfter,
-        previousObserverTokens: omConfig.observation.previousObserverTokens,
-        instruction: omConfig.observation.instruction,
-        threadTitle: omConfig.observation.threadTitle
-      } : void 0,
-      reflection: omConfig.reflection ? {
-        model: omConfig.reflection.model,
-        observationTokens: omConfig.reflection.observationTokens,
-        modelSettings: omConfig.reflection.modelSettings,
-        providerOptions: omConfig.reflection.providerOptions,
-        bufferActivation: omConfig.reflection.bufferActivation,
-        blockAfter: omConfig.reflection.blockAfter,
-        instruction: omConfig.reflection.instruction
-      } : void 0
-    });
+    const effectiveConfig = normalizeObservationalMemoryConfig(threadConfig.observationalMemory);
+    if (!effectiveConfig) return null;
+    const engine = await this.omEngine;
+    if (!engine) return null;
+    const { ObservationalMemoryProcessor } = await import('./observational-memory-WMCWT577-MXWR6ZYS.cjs');
+    return new ObservationalMemoryProcessor(engine, this);
   }
 };
 
@@ -29374,5 +29503,5 @@ exports.START_ASYNC_AGENT_BUILDER_ACTION_ROUTE = START_ASYNC_AGENT_BUILDER_ACTIO
 exports.STREAM_AGENT_BUILDER_ACTION_ROUTE = STREAM_AGENT_BUILDER_ACTION_ROUTE;
 exports.STREAM_LEGACY_AGENT_BUILDER_ACTION_ROUTE = STREAM_LEGACY_AGENT_BUILDER_ACTION_ROUTE;
 exports.agent_builder_exports = agent_builder_exports;
-//# sourceMappingURL=chunk-VCBFCDP3.cjs.map
-//# sourceMappingURL=chunk-VCBFCDP3.cjs.map
+//# sourceMappingURL=chunk-SZBVW2WD.cjs.map
+//# sourceMappingURL=chunk-SZBVW2WD.cjs.map