@mastra/server 1.28.0-alpha.0 → 1.28.0-alpha.2

package/dist/{chunk-BXJTD5TI.js → chunk-HU57X24H.js}

@@ -1040,7 +1040,7 @@ function imageSize(input) {
   throw new TypeError(`unsupported file type: ${type}`);
 }
 
-// ../memory/dist/chunk-MPBMHIAQ.js
+// ../memory/dist/chunk-7742VTN5.js
 var OM_DEBUG_LOG = process.env.OM_DEBUG ? join(process.cwd(), "om-debug.log") : null;
 function omDebug(msg) {
   if (!OM_DEBUG_LOG) return;
@@ -2097,9 +2097,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 {
@@ -2605,7 +2607,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);
@@ -2621,12 +2623,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({
@@ -2638,8 +2641,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);
   }
 };
@@ -3102,7 +3107,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"
     );
@@ -3289,6 +3298,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,
@@ -3314,7 +3348,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;
@@ -3322,7 +3356,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;
@@ -3342,6 +3376,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.
    *
@@ -5399,8 +5437,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({
@@ -5413,8 +5452,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);
     }
@@ -5451,7 +5491,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);
       }
@@ -5509,8 +5549,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(
@@ -5549,7 +5594,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);
     }
@@ -5648,7 +5693,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,
@@ -5787,8 +5832,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",
@@ -5803,7 +5849,8 @@ ${unreflectedContent}` : freshRecord.bufferedReflection;
       cycleId,
       startedAt,
       recordId: record.id,
-      threadId
+      threadId,
+      resourceId: record.resourceId ?? void 0
     } : void 0;
     let reflectionUsage;
     let reflectionError;
@@ -5838,8 +5885,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",
@@ -5862,8 +5910,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) {
@@ -8790,17 +8839,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(
@@ -9460,18 +9510,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, {
@@ -9664,7 +9715,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,
@@ -10111,7 +10162,8 @@ var ObservationalMemoryProcessor = class {
           messageList,
           observabilityContext: getOmObservabilityContext(args),
           hooks: {
-            onBufferChunkSealed: rotateResponseMessageId
+            onBufferChunkSealed: rotateResponseMessageId,
+            onSyncObservationComplete: rotateResponseMessageId
           }
         });
         this.turn.writer = writer;
@@ -10122,6 +10174,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;
@@ -10257,5 +10313,5 @@ function getObservationsAsOf(activeObservations, asOf) {
 }
 
 export { ModelByInputTokens, OBSERVER_SYSTEM_PROMPT, ObservationalMemory, ObservationalMemoryProcessor, TokenCounter, buildObserverPrompt, buildObserverSystemPrompt, combineObservationGroupRanges, deriveObservationGroupProvenance, e, estimateTokenCount, extractCurrentTask, formatMessagesForObserver, formatToolResultForObserver, getObservationsAsOf, hasCurrentTaskSection, injectAnchorIds, optimizeObservationsForContext, parseAnchorId, parseObservationGroups, parseObserverOutput, reconcileObservationGroupsFromReflection, renderObservationGroupsForReflection, resolveToolResultValue, stripEphemeralAnchorIds, stripObservationGroups, truncateStringByTokens, wrapInObservationGroup };
-//# sourceMappingURL=chunk-BXJTD5TI.js.map
-//# sourceMappingURL=chunk-BXJTD5TI.js.map
+//# sourceMappingURL=chunk-HU57X24H.js.map
+//# sourceMappingURL=chunk-HU57X24H.js.map