@mastra/mongodb 1.6.2 → 1.7.0

package/dist/index.js

@@ -12,7 +12,7 @@ import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // package.json
 var package_default = {
-  version: "1.6.2"};
+  version: "1.7.0"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -5264,11 +5264,22 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
       );
     }
   }
-  async getObservationalMemoryHistory(threadId, resourceId, limit = 10) {
+  async getObservationalMemoryHistory(threadId, resourceId, limit = 10, options) {
     try {
       const lookupKey = this.getOMKey(threadId, resourceId);
       const collection = await this.getCollection(OM_TABLE);
-      const docs = await collection.find({ lookupKey }).sort({ generationCount: -1 }).limit(limit).toArray();
+      const filter = { lookupKey };
+      if (options?.from || options?.to) {
+        const createdAtFilter = {};
+        if (options.from) createdAtFilter["$gte"] = options.from;
+        if (options.to) createdAtFilter["$lte"] = options.to;
+        filter["createdAt"] = createdAtFilter;
+      }
+      let cursor = collection.find(filter).sort({ generationCount: -1 });
+      if (options?.offset != null) {
+        cursor = cursor.skip(options.offset);
+      }
+      const docs = await cursor.limit(limit).toArray();
       return docs.map((doc) => this.parseOMDocument(doc));
     } catch (error) {
       throw new MastraError(
@@ -5694,6 +5705,37 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
       );
     }
   }
+  async updateObservationalMemoryConfig(input) {
+    try {
+      const collection = await this.getCollection(OM_TABLE);
+      const doc = await collection.findOne({ id: input.id }, { projection: { config: 1 } });
+      if (!doc) {
+        throw new MastraError({
+          id: createStorageErrorId("MONGODB", "UPDATE_OM_CONFIG", "NOT_FOUND"),
+          text: `Observational memory record not found: ${input.id}`,
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { id: input.id }
+        });
+      }
+      const existing = doc.config ?? {};
+      const merged = this.deepMergeConfig(existing, input.config);
+      await collection.updateOne({ id: input.id }, { $set: { config: merged, updatedAt: /* @__PURE__ */ new Date() } });
+    } catch (error) {
+      if (error instanceof MastraError) {
+        throw error;
+      }
+      throw new MastraError(
+        {
+          id: createStorageErrorId("MONGODB", "UPDATE_OM_CONFIG", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { id: input.id }
+        },
+        error
+      );
+    }
+  }
   // ============================================
   // Async Buffering Methods
   // ============================================