@mastra/memory 1.17.6-alpha.1 → 1.18.0-alpha.2

package/dist/{chunk-QZGJY67D.js → chunk-PBZHHKPE.js}

@@ -2022,6 +2022,22 @@ ObservationStrategy.create = ((om, opts) => {
   if (deps.scope === "resource" && opts.resourceId) return new ResourceScopedObservationStrategy(deps, opts);
   return new SyncObservationStrategy(deps, opts);
 });
+
+// src/processors/observational-memory/observation-turn/load-memory-context.ts
+async function loadMemoryContextMessages({
+  memory,
+  messageList,
+  threadId,
+  resourceId
+}) {
+  const ctx = await memory.getContext({ threadId, resourceId });
+  for (const msg of ctx.messages) {
+    if (msg.role !== "system") {
+      messageList.add(msg, "memory");
+    }
+  }
+  return ctx;
+}
 var ObservationStep = class {
   constructor(turn, stepNumber) {
     this.turn = turn;
@@ -2389,12 +2405,12 @@ var ObservationTurn = class {
     this._generationCountAtStart = this._record.generationCount;
     this.memory = memory;
     if (memory) {
-      const ctx = await memory.getContext({ threadId: this.threadId, resourceId: this.resourceId });
-      for (const msg of ctx.messages) {
-        if (msg.role !== "system") {
-          this.messageList.add(msg, "memory");
-        }
-      }
+      const ctx = await loadMemoryContextMessages({
+        memory,
+        messageList: this.messageList,
+        threadId: this.threadId,
+        resourceId: this.resourceId
+      });
       this._context = {
         messages: ctx.messages,
         systemMessage: ctx.systemMessage,
@@ -9216,6 +9232,33 @@ var GATEWAY_STATE_KEY = "__isGatewayModel";
 function isMastraGatewayModel(model) {
   return typeof model === "object" && model !== null && "gatewayId" in model && model.gatewayId === "mastra";
 }
+function injectObservationContextMessages({
+  messageList,
+  systemMessages,
+  continuationMessage,
+  threadId,
+  resourceId
+}) {
+  if (!systemMessages?.length) {
+    return;
+  }
+  messageList.clearSystemMessages("observational-memory");
+  for (const msg of systemMessages) {
+    messageList.addSystem(msg, "observational-memory");
+  }
+  const contMsg = 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
+  };
+  messageList.add(contMsg, "memory");
+}
 var ObservationalMemoryProcessor = class {
   id = "observational-memory";
   name = "Observational Memory";
@@ -9270,6 +9313,25 @@ var ObservationalMemoryProcessor = class {
       const preMessagesSnapshot = reproCaptureEnabled ? safeCaptureJson(messageList.get.all.db()) : null;
       const preSerializedMessageList = reproCaptureEnabled ? safeCaptureJson(messageList.serialize()) : null;
       if (readOnly) {
+        const ctx = await loadMemoryContextMessages({
+          memory: this.memory,
+          messageList,
+          threadId,
+          resourceId
+        });
+        const systemMessages = ctx.hasObservations && ctx.omRecord ? await this.engine.buildContextSystemMessages({
+          threadId,
+          resourceId,
+          record: ctx.omRecord,
+          unobservedContextBlocks: ctx.otherThreadsContext
+        }) : void 0;
+        injectObservationContextMessages({
+          messageList,
+          systemMessages,
+          continuationMessage: ctx.continuationMessage,
+          threadId,
+          resourceId
+        });
         return messageList;
       }
       const activeTurn = state.__omTurn ?? this.turn;
@@ -9325,26 +9387,13 @@ var ObservationalMemoryProcessor = class {
           }
           throw err;
         }
-        if (ctx.systemMessage) {
-          messageList.clearSystemMessages("observational-memory");
-          for (const msg of ctx.systemMessage) {
-            messageList.addSystem(msg, "observational-memory");
-          }
-          const contMsg = this.turn.context.continuation ?? {
-            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
-          };
-          messageList.add(contMsg, "memory");
-        }
+        injectObservationContextMessages({
+          messageList,
+          systemMessages: ctx.systemMessage,
+          continuationMessage: this.turn.context.continuation,
+          threadId,
+          resourceId
+        });
         const freshRecord = await this.engine.getOrCreateRecord(threadId, resourceId);
         await this.engine.emitProgress({
           record: freshRecord,
@@ -9445,5 +9494,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-QZGJY67D.js.map
-//# sourceMappingURL=chunk-QZGJY67D.js.map
+//# sourceMappingURL=chunk-PBZHHKPE.js.map
+//# sourceMappingURL=chunk-PBZHHKPE.js.map