@mastra/memory 1.17.0 → 1.17.1

package/dist/{chunk-5MTY2UYL.cjs → chunk-J2XQSPAB.cjs}

@@ -1095,9 +1095,11 @@ var ObservationStrategy = class _ObservationStrategy {
   }
   async streamMarker(marker) {
     if (this.opts.writer) {
-      await this.opts.writer.custom(marker).catch(() => {
+      await this.opts.writer.custom({ ...marker, transient: true }).catch(() => {
       });
     }
+    const markerThreadId = marker.data?.threadId ?? this.opts.threadId;
+    await this.persistMarkerToStorage(marker, markerThreadId, this.opts.resourceId);
   }
   getObservationMarkerConfig() {
     return {
@@ -1603,7 +1605,7 @@ var AsyncBufferObservationStrategy = class extends ObservationStrategy {
     }
   }
   async emitEndMarkers(_cycleId, processed) {
-    if (!processed.observations || !this.opts.writer) return;
+    if (!processed.observations) return;
     const { record, threadId, messages } = this.opts;
     const tokensBuffered = await this.tokenCounter.countMessagesAsync(messages);
     const updatedRecord = await this.storage.getObservationalMemory(record.threadId, record.resourceId);
@@ -1619,12 +1621,13 @@ var AsyncBufferObservationStrategy = class extends ObservationStrategy {
       threadId,
       observations: processed.observations
     });
-    void this.opts.writer.custom(endMarker).catch(() => {
-    });
+    if (this.opts.writer) {
+      void this.opts.writer.custom({ ...endMarker, transient: true }).catch(() => {
+      });
+    }
     await this.persistMarkerToStorage(endMarker, threadId, record.resourceId ?? void 0);
   }
   async emitFailedMarkers(_cycleId, error) {
-    if (!this.opts.writer) return;
     const { record, threadId, messages } = this.opts;
     const tokensAttempted = await this.tokenCounter.countMessagesAsync(messages);
     const failedMarker = createBufferingFailedMarker({
@@ -1636,8 +1639,10 @@ var AsyncBufferObservationStrategy = class extends ObservationStrategy {
       recordId: record.id,
       threadId
     });
-    void this.opts.writer.custom(failedMarker).catch(() => {
-    });
+    if (this.opts.writer) {
+      void this.opts.writer.custom({ ...failedMarker, transient: true }).catch(() => {
+      });
+    }
     await this.persistMarkerToStorage(failedMarker, threadId, record.resourceId ?? void 0);
   }
 };
@@ -2102,7 +2107,11 @@ var ObservationStep = class {
     }
     const allMsgsForToolCheck = messageList.get.all.db();
     const lastMessage = allMsgsForToolCheck[allMsgsForToolCheck.length - 1];
-    const latestStepParts = getLatestStepParts(lastMessage?.content?.parts ?? []);
+    const pendingStepMessages = [...messageList.get.input.db(), ...messageList.get.response.db()];
+    const latestStepParts = [
+      ...getLatestStepParts(lastMessage?.content?.parts ?? []),
+      ...pendingStepMessages.flatMap((msg) => getLatestStepParts(msg.content?.parts ?? []))
+    ];
     const hasIncompleteToolCalls = latestStepParts.some(
       (part) => part?.type === "tool-invocation" && part.toolInvocation?.state === "call"
     );
@@ -2289,6 +2298,31 @@ var ObservationStep = class {
       writer: this.turn.writer,
       observabilityContext: this.turn.observabilityContext
     });
+    if (obsResult.observed) {
+      const observedMessageIds = new Set(obsResult.record.observedMessageIds ?? []);
+      const liveMessages = messageList.get.all.db();
+      let latestObservedIndex = -1;
+      for (let i = liveMessages.length - 1; i >= 0; i--) {
+        const message = liveMessages[i];
+        if (message && observedMessageIds.has(message.id)) {
+          latestObservedIndex = i;
+          break;
+        }
+      }
+      const messageToSeal = latestObservedIndex >= 0 ? liveMessages[latestObservedIndex] : void 0;
+      const messagesToSeal = messageToSeal ? [messageToSeal] : [];
+      om.sealMessagesForBuffering(messagesToSeal);
+      try {
+        await this.turn.hooks?.onSyncObservationComplete?.();
+      } catch (error) {
+        omDebug(
+          `[OM:observe] onSyncObservationComplete hook failed: ${error instanceof Error ? error.message : String(error)}`
+        );
+      }
+      if (messagesToSeal.length > 0) {
+        await om.persistMessages(messagesToSeal, threadId, resourceId);
+      }
+    }
     return {
       succeeded: obsResult.observed,
       record: obsResult.record,
@@ -2316,7 +2350,7 @@ var ObservationTurn = class {
   observabilityContext;
   /** Current actor model for this step. Updated by the processor before prepare(). */
   actorModelContext;
-  /** Optional processor-provided hooks for turn/step lifecycle integration. */
+  /** Processor-provided hooks for turn/step lifecycle integration. */
   hooks;
   constructor(opts) {
     this.om = opts.om;
@@ -2324,7 +2358,7 @@ var ObservationTurn = class {
     this.resourceId = opts.resourceId;
     this.messageList = opts.messageList;
     this.observabilityContext = opts.observabilityContext;
-    this.hooks = opts.hooks;
+    this.hooks = opts.hooks ?? {};
   }
   om;
   threadId;
@@ -2344,6 +2378,10 @@ var ObservationTurn = class {
   get currentStep() {
     return this._currentStep;
   }
+  addHooks(hooks) {
+    if (!hooks) return;
+    Object.assign(this.hooks, hooks);
+  }
   /**
    * Load context and cache the record. Call once at the start of the turn.
    *
@@ -4413,8 +4451,9 @@ var ReflectorRunner = class {
           recordId: streamContext.recordId,
           threadId: streamContext.threadId
         });
-        await streamContext.writer.custom(failedMarker).catch(() => {
+        await streamContext.writer.custom({ ...failedMarker, transient: true }).catch(() => {
         });
+        await this.persistMarkerToStorage(failedMarker, streamContext.threadId, streamContext.resourceId);
         const retryCycleId = crypto.randomUUID();
         streamContext.cycleId = retryCycleId;
         const startMarker = createObservationStartMarker({
@@ -4427,8 +4466,9 @@ var ReflectorRunner = class {
           config: this.getObservationMarkerConfig()
         });
         streamContext.startedAt = startMarker.data.startedAt;
-        await streamContext.writer.custom(startMarker).catch(() => {
+        await streamContext.writer.custom({ ...startMarker, transient: true }).catch(() => {
         });
+        await this.persistMarkerToStorage(startMarker, streamContext.threadId, streamContext.resourceId);
       }
       currentLevel = Math.min(currentLevel + 1, maxLevel);
     }
@@ -4465,7 +4505,7 @@ var ReflectorRunner = class {
           recordId: record.id,
           threadId: record.threadId ?? ""
         });
-        void writer.custom(failedMarker).catch(() => {
+        void writer.custom({ ...failedMarker, transient: true }).catch(() => {
         });
         await this.persistMarkerToStorage(failedMarker, record.threadId ?? "", record.resourceId ?? void 0);
       }
@@ -4523,8 +4563,13 @@ var ReflectorRunner = class {
         threadIds: record.threadId ? [record.threadId] : [],
         config: this.getObservationMarkerConfig(currentRecord)
       });
-      void writer.custom(startMarker).catch(() => {
+      void writer.custom({ ...startMarker, transient: true }).catch(() => {
       });
+      await this.persistMarkerToStorage(
+        startMarker,
+        currentRecord.threadId ?? "",
+        currentRecord.resourceId ?? void 0
+      );
     }
     const compressionStartLevel = await this.getCompressionStartLevel(requestContext);
     const reflectResult = await this.call(
@@ -4563,7 +4608,7 @@ var ReflectorRunner = class {
         threadId: currentRecord.threadId ?? "",
         observations: reflectResult.observations
       });
-      void writer.custom(endMarker).catch(() => {
+      void writer.custom({ ...endMarker, transient: true }).catch(() => {
       });
       await this.persistMarkerToStorage(endMarker, currentRecord.threadId ?? "", currentRecord.resourceId ?? void 0);
     }
@@ -4662,7 +4707,7 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
         currentModel: activationMetadata?.currentModel,
         config: this.getObservationMarkerConfig(freshRecord)
       });
-      void writer.custom(activationMarker).catch(() => {
+      void writer.custom({ ...activationMarker, transient: true }).catch(() => {
       });
       await this.persistMarkerToMessage(
         activationMarker,
@@ -4801,8 +4846,9 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
         threadIds: [threadId],
         config: this.getObservationMarkerConfig(record)
       });
-      await writer.custom(startMarker).catch(() => {
+      await writer.custom({ ...startMarker, transient: true }).catch(() => {
       });
+      await this.persistMarkerToStorage(startMarker, threadId, record.resourceId ?? void 0);
     }
     this.emitDebugEvent({
       type: "reflection_triggered",
@@ -4817,7 +4863,8 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       cycleId,
       startedAt,
       recordId: record.id,
-      threadId
+      threadId,
+      resourceId: record.resourceId ?? void 0
     } : void 0;
     let reflectionUsage;
     let reflectionError;
@@ -4852,8 +4899,9 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
           recordId: record.id,
           threadId
         });
-        await writer.custom(endMarker).catch(() => {
+        await writer.custom({ ...endMarker, transient: true }).catch(() => {
         });
+        await this.persistMarkerToStorage(endMarker, threadId, record.resourceId ?? void 0);
       }
       this.emitDebugEvent({
         type: "reflection_complete",
@@ -4876,8 +4924,9 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
           recordId: record.id,
           threadId
         });
-        await writer.custom(failedMarker).catch(() => {
+        await writer.custom({ ...failedMarker, transient: true }).catch(() => {
         });
+        await this.persistMarkerToStorage(failedMarker, threadId, record.resourceId ?? void 0);
       }
       reflectionError = error instanceof Error ? error : new Error(String(error));
       if (abortSignal?.aborted) {
@@ -7806,17 +7855,18 @@ ${grouped}` : grouped;
     const cycleId = `buffer-obs-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`;
     const startedAt = (/* @__PURE__ */ new Date()).toISOString();
     const tokensToBuffer = await this.tokenCounter.countMessagesAsync(messagesToBuffer);
+    const startMarker = createBufferingStartMarker({
+      cycleId,
+      operationType: "observation",
+      tokensToBuffer,
+      recordId: freshRecord.id,
+      threadId,
+      threadIds: [threadId],
+      config: this.getObservationMarkerConfig()
+    });
+    await this.persistMarkerToStorage(startMarker, threadId, freshRecord.resourceId ?? void 0);
     if (writer) {
-      const startMarker = createBufferingStartMarker({
-        cycleId,
-        operationType: "observation",
-        tokensToBuffer,
-        recordId: freshRecord.id,
-        threadId,
-        threadIds: [threadId],
-        config: this.getObservationMarkerConfig()
-      });
-      void writer.custom(startMarker).catch(() => {
+      void writer.custom({ ...startMarker, transient: true }).catch(() => {
       });
     }
     omDebug(
@@ -8476,18 +8526,19 @@ ${grouped}` : grouped;
       }
       const cycleId = `buffer-obs-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`;
       const startedAt = (/* @__PURE__ */ new Date()).toISOString();
+      const startMarker = createBufferingStartMarker({
+        cycleId,
+        operationType: "observation",
+        tokensToBuffer: newTokens,
+        recordId: record.id,
+        threadId,
+        threadIds: [threadId],
+        config: this.getObservationMarkerConfig()
+      });
+      await this.persistMarkerToStorage(startMarker, threadId, record.resourceId ?? void 0);
       const writer = opts.writer;
       if (writer) {
-        const startMarker = createBufferingStartMarker({
-          cycleId,
-          operationType: "observation",
-          tokensToBuffer: newTokens,
-          recordId: record.id,
-          threadId,
-          threadIds: [threadId],
-          config: this.getObservationMarkerConfig()
-        });
-        void writer.custom(startMarker).catch(() => {
+        void writer.custom({ ...startMarker, transient: true }).catch(() => {
         });
       }
       await ObservationStrategy.create(this, {
@@ -8680,7 +8731,7 @@ ${grouped}` : grouped;
           currentModel,
           config: this.getObservationMarkerConfig()
         });
-        void opts.writer.custom(activationMarker).catch(() => {
+        void opts.writer.custom({ ...activationMarker, transient: true }).catch(() => {
         });
         await this.persistMarkerToMessage(
           activationMarker,
@@ -9131,7 +9182,8 @@ var ObservationalMemoryProcessor = class {
           messageList,
           observabilityContext: getOmObservabilityContext(args),
           hooks: {
-            onBufferChunkSealed: rotateResponseMessageId
+            onBufferChunkSealed: rotateResponseMessageId,
+            onSyncObservationComplete: rotateResponseMessageId
           }
         });
         this.turn.writer = writer;
@@ -9142,6 +9194,10 @@ var ObservationalMemoryProcessor = class {
         }
         state.__omTurn = this.turn;
       }
+      this.turn.addHooks({
+        onBufferChunkSealed: rotateResponseMessageId,
+        onSyncObservationComplete: rotateResponseMessageId
+      });
       const observabilityContext = getOmObservabilityContext(args);
       state.__omObservabilityContext = observabilityContext;
       this.turn.observabilityContext = observabilityContext;
@@ -9304,5 +9360,5 @@ exports.stripEphemeralAnchorIds = stripEphemeralAnchorIds;
 exports.stripObservationGroups = stripObservationGroups;
 exports.truncateStringByTokens = truncateStringByTokens;
 exports.wrapInObservationGroup = wrapInObservationGroup;
-//# sourceMappingURL=chunk-5MTY2UYL.cjs.map
-//# sourceMappingURL=chunk-5MTY2UYL.cjs.map
+//# sourceMappingURL=chunk-J2XQSPAB.cjs.map
+//# sourceMappingURL=chunk-J2XQSPAB.cjs.map