@mastra/mongodb 1.5.3 → 1.5.4

package/dist/index.js

@@ -12,7 +12,7 @@ import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // package.json
 var package_default = {
-  version: "1.5.3"};
+  version: "1.5.4"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -3794,7 +3794,7 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
       generationCount: Number(doc.generationCount || 0),
       activeObservations: doc.activeObservations || "",
       // Handle new chunk-based structure
-      bufferedObservationChunks: doc.bufferedObservationChunks || void 0,
+      bufferedObservationChunks: Array.isArray(doc.bufferedObservationChunks) ? doc.bufferedObservationChunks : void 0,
       // Deprecated fields (for backward compatibility)
       bufferedObservations: doc.activeObservationsPendingUpdate || void 0,
       bufferedObservationTokens: doc.bufferedObservationTokens ? Number(doc.bufferedObservationTokens) : void 0,
@@ -3945,7 +3945,7 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
         totalTokensObserved: record.totalTokensObserved || 0,
         observationTokenCount: record.observationTokenCount || 0,
         observedMessageIds: record.observedMessageIds || null,
-        bufferedObservationChunks: record.bufferedObservationChunks || null,
+        bufferedObservationChunks: Array.isArray(record.bufferedObservationChunks) ? record.bufferedObservationChunks : [],
         bufferedReflection: record.bufferedReflection || null,
         bufferedReflectionTokens: record.bufferedReflectionTokens ?? null,
         bufferedReflectionInputTokens: record.bufferedReflectionInputTokens ?? null,
@@ -4286,17 +4286,17 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
         suggestedContinuation: input.chunk.suggestedContinuation,
         currentTask: input.chunk.currentTask
       };
-      const $set = { updatedAt: /* @__PURE__ */ new Date() };
+      const now = /* @__PURE__ */ new Date();
+      const setStage = {
+        updatedAt: now,
+        bufferedObservationChunks: {
+          $concatArrays: [{ $ifNull: ["$bufferedObservationChunks", []] }, [newChunk]]
+        }
+      };
       if (input.lastBufferedAtTime) {
-        $set.lastBufferedAtTime = input.lastBufferedAtTime;
+        setStage.lastBufferedAtTime = input.lastBufferedAtTime;
       }
-      const result = await collection.updateOne(
-        { id: input.id },
-        {
-          $push: { bufferedObservationChunks: newChunk },
-          $set
-        }
-      );
+      const result = await collection.updateOne({ id: input.id }, [{ $set: setStage }]);
       if (result.matchedCount === 0) {
         throw new MastraError({
           id: createStorageErrorId("MONGODB", "UPDATE_BUFFERED_OBSERVATIONS", "NOT_FOUND"),
@@ -4409,7 +4409,7 @@ ${activatedContent}` : activatedContent;
             activeObservations: newActive,
             observationTokenCount: newTokenCount,
             pendingMessageTokens: newPending,
-            bufferedObservationChunks: remainingChunks.length > 0 ? remainingChunks : null,
+            bufferedObservationChunks: remainingChunks,
             lastObservedAt,
             updatedAt: /* @__PURE__ */ new Date()
           }