@mastra/mongodb 1.0.0-beta.11 → 1.0.0-beta.13

package/dist/index.cjs

@@ -13,7 +13,7 @@ var evals = require('@mastra/core/evals');
 
 // package.json
 var package_default = {
-  version: "1.0.0-beta.11"};
+  version: "1.0.0-beta.13"};
 var MongoDBFilterTranslator = class extends filter.BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -107,7 +107,7 @@ var MongoDBVector = class extends vector.MastraVector {
   client;
   db;
   collections;
-  embeddingFieldName = "embedding";
+  embeddingFieldName;
   metadataFieldName = "metadata";
   documentFieldName = "document";
   collectionForValidation = null;
@@ -116,8 +116,11 @@ var MongoDBVector = class extends vector.MastraVector {
     euclidean: "euclidean",
     dotproduct: "dotProduct"
   };
-  constructor({ id, uri, dbName, options }) {
+  constructor({ id, uri, dbName, options, embeddingFieldPath }) {
     super({ id });
+    if (!uri) {
+      throw new Error('MongoDBVector requires a connection string. Provide "uri" in the constructor options.');
+    }
     const client = new mongodb.MongoClient(uri, {
       ...options,
       driverInfo: {
@@ -128,6 +131,7 @@ var MongoDBVector = class extends vector.MastraVector {
     this.client = client;
     this.db = this.client.db(dbName);
     this.collections = /* @__PURE__ */ new Map();
+    this.embeddingFieldName = embeddingFieldPath ?? "embedding";
   }
   // Public methods
   async connect() {
@@ -859,11 +863,21 @@ function resolveMongoDBConfig(config) {
       );
     }
   }
+  const connectionString = config.uri ?? config.url;
+  if (!connectionString) {
+    throw new error.MastraError({
+      id: storage.createStorageErrorId("MONGODB", "CONSTRUCTOR", "MISSING_URI"),
+      domain: error.ErrorDomain.STORAGE,
+      category: error.ErrorCategory.USER,
+      details: { dbName: config?.dbName },
+      text: 'MongoDBStore requires a connection string. Provide "uri" (recommended) or "url" in the constructor options.'
+    });
+  }
   try {
     return MongoDBConnector.fromDatabaseConfig({
       id: "id" in config ? config.id : "domain",
       options: config.options,
-      url: config.url,
+      url: connectionString,
       dbName: config.dbName
     });
   } catch (error$1) {
@@ -872,7 +886,7 @@ function resolveMongoDBConfig(config) {
         id: storage.createStorageErrorId("MONGODB", "CONSTRUCTOR", "FAILED"),
         domain: error.ErrorDomain.STORAGE,
         category: error.ErrorCategory.USER,
-        details: { url: config?.url, dbName: config?.dbName }
+        details: { uri: config?.uri ?? "", url: config?.url ?? "", dbName: config?.dbName ?? "" }
       },
       error$1
     );
@@ -1673,25 +1687,48 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends storage.MemorySto
       );
     }
   }
-  async listThreadsByResourceId(args) {
+  async listThreads(args) {
+    const { page = 0, perPage: perPageInput, orderBy, filter } = args;
+    try {
+      this.validatePaginationInput(page, perPageInput ?? 100);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("MONGODB", "LIST_THREADS", "INVALID_PAGE"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { page, ...perPageInput !== void 0 && { perPage: perPageInput } }
+        },
+        error$1 instanceof Error ? error$1 : new Error("Invalid pagination parameters")
+      );
+    }
+    const perPage = storage.normalizePerPage(perPageInput, 100);
+    try {
+      this.validateMetadataKeys(filter?.metadata);
+    } catch (error$1) {
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("MONGODB", "LIST_THREADS", "INVALID_METADATA_KEY"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { metadataKeys: filter?.metadata ? Object.keys(filter.metadata).join(", ") : "" }
+        },
+        error$1 instanceof Error ? error$1 : new Error("Invalid metadata key")
+      );
+    }
     try {
-      const { resourceId, page = 0, perPage: perPageInput, orderBy } = args;
-      if (page < 0) {
-        throw new error.MastraError(
-          {
-            id: storage.createStorageErrorId("MONGODB", "LIST_THREADS_BY_RESOURCE_ID", "INVALID_PAGE"),
-            domain: error.ErrorDomain.STORAGE,
-            category: error.ErrorCategory.USER,
-            details: { page }
-          },
-          new Error("page must be >= 0")
-        );
-      }
-      const perPage = storage.normalizePerPage(perPageInput, 100);
       const { offset, perPage: perPageForResponse } = storage.calculatePagination(page, perPageInput, perPage);
       const { field, direction } = this.parseOrderBy(orderBy);
       const collection = await this.getCollection(storage.TABLE_THREADS);
-      const query = { resourceId };
+      const query = {};
+      if (filter?.resourceId) {
+        query.resourceId = filter.resourceId;
+      }
+      if (filter?.metadata && Object.keys(filter.metadata).length > 0) {
+        for (const [key, value] of Object.entries(filter.metadata)) {
+          query[`metadata.${key}`] = value;
+        }
+      }
       const total = await collection.countDocuments(query);
       if (perPage === 0) {
         return {
@@ -1725,10 +1762,13 @@ var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends storage.MemorySto
     } catch (error$1) {
       throw new error.MastraError(
         {
-          id: storage.createStorageErrorId("MONGODB", "LIST_THREADS_BY_RESOURCE_ID", "FAILED"),
+          id: storage.createStorageErrorId("MONGODB", "LIST_THREADS", "FAILED"),
           domain: error.ErrorDomain.STORAGE,
           category: error.ErrorCategory.THIRD_PARTY,
-          details: { resourceId: args.resourceId }
+          details: {
+            ...filter?.resourceId && { resourceId: filter.resourceId },
+            hasMetadataFilter: !!filter?.metadata
+          }
         },
         error$1
       );
@@ -2583,7 +2623,7 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends storage.ScoresSto
         };
       }
       const end = perPageInput === false ? total : start + perPage;
-      let cursor = collection.find(query).sort({ createdAt: "desc" }).skip(start);
+      let cursor = collection.find(query).sort({ createdAt: -1 }).skip(start);
       if (perPageInput !== false) {
         cursor = cursor.limit(perPage);
       }
@@ -2632,7 +2672,7 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends storage.ScoresSto
         };
       }
       const end = perPageInput === false ? total : start + perPage;
-      let cursor = collection.find({ runId }).sort({ createdAt: "desc" }).skip(start);
+      let cursor = collection.find({ runId }).sort({ createdAt: -1 }).skip(start);
       if (perPageInput !== false) {
         cursor = cursor.limit(perPage);
       }
@@ -2682,7 +2722,7 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends storage.ScoresSto
         };
       }
       const end = perPageInput === false ? total : start + perPage;
-      let cursor = collection.find({ entityId, entityType }).sort({ createdAt: "desc" }).skip(start);
+      let cursor = collection.find({ entityId, entityType }).sort({ createdAt: -1 }).skip(start);
       if (perPageInput !== false) {
         cursor = cursor.limit(perPage);
       }
@@ -2733,7 +2773,7 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends storage.ScoresSto
         };
       }
       const end = perPageInput === false ? total : start + perPage;
-      let cursor = collection.find(query).sort({ createdAt: "desc" }).skip(start);
+      let cursor = collection.find(query).sort({ createdAt: -1 }).skip(start);
       if (perPageInput !== false) {
         cursor = cursor.limit(perPage);
       }
@@ -3026,7 +3066,7 @@ var WorkflowsStorageMongoDB = class _WorkflowsStorageMongoDB extends storage.Wor
       try {
         parsedSnapshot = typeof row.snapshot === "string" ? storage.safelyParseJSON(row.snapshot) : row.snapshot;
       } catch (e) {
-        console.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
+        this.logger.warn(`Failed to parse snapshot for workflow ${row.workflow_name}: ${e}`);
       }
     }
     return {