@mastra/mongodb 1.12.0-alpha.1 → 1.12.0-alpha.2

package/dist/index.js

@@ -13,7 +13,7 @@ import { skillSnapshotFieldValuesEqual } from '@mastra/core/storage/domains/skil
 
 // package.json
 var package_default = {
-  version: "1.12.0-alpha.1"};
+  version: "1.12.0-alpha.2"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -2149,6 +2149,16 @@ var MongoDBBlobStore = class _MongoDBBlobStore extends BlobStore {
     };
   }
 };
+function formatDateForMongoDB(date) {
+  return typeof date === "string" ? new Date(date) : date;
+}
+function applyTenancyFilter(filter, filters) {
+  if (!filters) return;
+  if (filters.organizationId !== void 0) filter.organizationId = filters.organizationId;
+  if (filters.projectId !== void 0) filter.projectId = filters.projectId;
+}
+
+// src/storage/domains/datasets/index.ts
 var MANAGED_COLLECTIONS = [TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS];
 var MongoDBDatasetsStorage = class extends DatasetsStorage {
   #connector;
@@ -2343,10 +2353,15 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
       );
     }
   }
-  async getDatasetById({ id }) {
+  async getDatasetById({
+    id,
+    filters
+  }) {
     try {
       const collection = await this.getCollection(TABLE_DATASETS);
-      const row = await collection.findOne({ id });
+      const query = { id };
+      applyTenancyFilter(query, filters);
+      const row = await collection.findOne(query);
       return row ? this.transformDatasetRow(row) : null;
     } catch (error) {
       throw new MastraError(
@@ -2361,7 +2376,7 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
   }
   async _doUpdateDataset(args) {
     try {
-      const existing = await this.getDatasetById({ id: args.id });
+      const existing = await this.getDatasetById({ id: args.id, filters: args.filters });
       if (!existing) {
         throw new MastraError({
           id: createStorageErrorId("MONGODB", "UPDATE_DATASET", "NOT_FOUND"),
@@ -2398,8 +2413,13 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
       );
     }
   }
-  async deleteDataset({ id }) {
+  async deleteDataset({ id, filters }) {
     try {
+      const datasetsCollectionForGate = await this.getCollection(TABLE_DATASETS);
+      const gateQuery = { id };
+      applyTenancyFilter(gateQuery, filters);
+      const gateHit = await datasetsCollectionForGate.findOne(gateQuery, { projection: { id: 1 } });
+      if (!gateHit) return;
       try {
         const experimentsCollection = await this.getCollection(TABLE_EXPERIMENTS);
         const experimentIds = await experimentsCollection.find({ datasetId: id }, { projection: { id: 1 } }).toArray();
@@ -2419,7 +2439,9 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
       const datasetsCollection = await this.getCollection(TABLE_DATASETS);
       await versionsCollection.deleteMany({ datasetId: id });
       await itemsCollection.deleteMany({ datasetId: id });
-      await datasetsCollection.deleteOne({ id });
+      const parentDeleteQuery = { id };
+      applyTenancyFilter(parentDeleteQuery, filters);
+      await datasetsCollection.deleteOne(parentDeleteQuery);
     } catch (error) {
       if (error instanceof MastraError) throw error;
       throw new MastraError(
@@ -2441,6 +2463,14 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
       if (args.filters?.projectId !== void 0) filter.projectId = args.filters.projectId;
       if (args.filters?.candidateKey !== void 0) filter.candidateKey = args.filters.candidateKey;
       if (args.filters?.candidateId !== void 0) filter.candidateId = args.filters.candidateId;
+      if (args.filters?.targetType !== void 0) filter.targetType = args.filters.targetType;
+      if (args.filters?.targetIds !== void 0 && args.filters.targetIds.length > 0) {
+        filter.targetIds = { $in: args.filters.targetIds };
+      }
+      if (args.filters?.name !== void 0 && args.filters.name.length > 0) {
+        const escapedName = args.filters.name.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+        filter.name = { $regex: escapedName, $options: "i" };
+      }
       const total = await collection.countDocuments(filter);
       if (total === 0) {
         return { datasets: [], pagination: { total: 0, page, perPage: perPageInput, hasMore: false } };
@@ -3441,10 +3471,15 @@ var MongoDBExperimentsStorage = class _MongoDBExperimentsStorage extends Experim
       );
     }
   }
-  async getExperimentById({ id }) {
+  async getExperimentById({
+    id,
+    filters
+  }) {
     try {
       const collection = await this.getCollection(TABLE_EXPERIMENTS);
-      const doc = await collection.findOne({ id });
+      const query = { id };
+      applyTenancyFilter(query, filters);
+      const doc = await collection.findOne(query);
       if (!doc) return null;
       return transformExperimentRow(doc);
     } catch (error) {
@@ -3519,12 +3554,20 @@ var MongoDBExperimentsStorage = class _MongoDBExperimentsStorage extends Experim
       );
     }
   }
-  async deleteExperiment({ id }) {
+  async deleteExperiment({ id, filters }) {
     try {
-      const resultsCollection = await this.getCollection(TABLE_EXPERIMENT_RESULTS);
-      await resultsCollection.deleteMany({ experimentId: id });
       const experimentsCollection = await this.getCollection(TABLE_EXPERIMENTS);
-      await experimentsCollection.deleteOne({ id });
+      const gateQuery = { id };
+      applyTenancyFilter(gateQuery, filters);
+      const existing = await experimentsCollection.findOne(gateQuery);
+      if (!existing) return;
+      const resultsCollection = await this.getCollection(TABLE_EXPERIMENT_RESULTS);
+      const resultsQuery = { experimentId: id };
+      applyTenancyFilter(resultsQuery, filters);
+      await resultsCollection.deleteMany(resultsQuery);
+      const parentDeleteQuery = { id };
+      applyTenancyFilter(parentDeleteQuery, filters);
+      await experimentsCollection.deleteOne(parentDeleteQuery);
     } catch (error) {
       throw new MastraError(
         {
@@ -3643,10 +3686,15 @@ var MongoDBExperimentsStorage = class _MongoDBExperimentsStorage extends Experim
       );
     }
   }
-  async getExperimentResultById({ id }) {
+  async getExperimentResultById({
+    id,
+    filters
+  }) {
     try {
       const collection = await this.getCollection(TABLE_EXPERIMENT_RESULTS);
-      const doc = await collection.findOne({ id });
+      const query = { id };
+      applyTenancyFilter(query, filters);
+      const doc = await collection.findOne(query);
       if (!doc) return null;
       return transformExperimentResultRow(doc);
     } catch (error) {
@@ -3713,10 +3761,22 @@ var MongoDBExperimentsStorage = class _MongoDBExperimentsStorage extends Experim
       );
     }
   }
-  async deleteExperimentResults({ experimentId }) {
+  async deleteExperimentResults({
+    experimentId,
+    filters
+  }) {
     try {
+      if (filters?.organizationId !== void 0 || filters?.projectId !== void 0) {
+        const experimentsCollection = await this.getCollection(TABLE_EXPERIMENTS);
+        const gateQuery = { id: experimentId };
+        applyTenancyFilter(gateQuery, filters);
+        const parent = await experimentsCollection.findOne(gateQuery);
+        if (!parent) return;
+      }
       const collection = await this.getCollection(TABLE_EXPERIMENT_RESULTS);
-      await collection.deleteMany({ experimentId });
+      const deleteQuery = { experimentId };
+      applyTenancyFilter(deleteQuery, filters);
+      await collection.deleteMany(deleteQuery);
     } catch (error) {
       throw new MastraError(
         {
@@ -4878,11 +4938,6 @@ var MongoDBMCPServersStorage = class _MongoDBMCPServersStorage extends MCPServer
     return result;
   }
 };
-function formatDateForMongoDB(date) {
-  return typeof date === "string" ? new Date(date) : date;
-}
-
-// src/storage/domains/memory/index.ts
 var OM_TABLE = "mastra_observational_memory";
 var MemoryStorageMongoDB = class _MemoryStorageMongoDB extends MemoryStorage {
   supportsObservationalMemory = true;
@@ -9267,6 +9322,14 @@ function transformScoreRow(row) {
     convertTimestamps: true
   });
 }
+function applyTenancyFilters(query, filters) {
+  if (filters?.organizationId !== void 0) {
+    query.organizationId = filters.organizationId;
+  }
+  if (filters?.projectId !== void 0) {
+    query.projectId = filters.projectId;
+  }
+}
 var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
   #connector;
   #skipDefaultIndexes;
@@ -9435,7 +9498,8 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
     pagination,
     entityId,
     entityType,
-    source
+    source,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
@@ -9451,6 +9515,7 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
       if (source) {
         query.source = source;
       }
+      applyTenancyFilters(query, filters);
       const collection = await this.getCollection(TABLE_SCORERS);
       const total = await collection.countDocuments(query);
       if (total === 0) {
@@ -9494,14 +9559,17 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
   }
   async listScoresByRunId({
     runId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+      const query = { runId };
+      applyTenancyFilters(query, filters);
       const collection = await this.getCollection(TABLE_SCORERS);
-      const total = await collection.countDocuments({ runId });
+      const total = await collection.countDocuments(query);
       if (total === 0) {
         return {
           scores: [],
@@ -9514,7 +9582,7 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
         };
       }
       const end = perPageInput === false ? total : start + perPage;
-      let cursor = collection.find({ runId }).sort({ createdAt: -1 }).skip(start);
+      let cursor = collection.find(query).sort({ createdAt: -1 }).skip(start);
       if (perPageInput !== false) {
         cursor = cursor.limit(perPage);
       }
@@ -9544,14 +9612,17 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
   async listScoresByEntityId({
     entityId,
     entityType,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+      const query = { entityId, entityType };
+      applyTenancyFilters(query, filters);
       const collection = await this.getCollection(TABLE_SCORERS);
-      const total = await collection.countDocuments({ entityId, entityType });
+      const total = await collection.countDocuments(query);
       if (total === 0) {
         return {
           scores: [],
@@ -9564,7 +9635,7 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
         };
       }
       const end = perPageInput === false ? total : start + perPage;
-      let cursor = collection.find({ entityId, entityType }).sort({ createdAt: -1 }).skip(start);
+      let cursor = collection.find(query).sort({ createdAt: -1 }).skip(start);
       if (perPageInput !== false) {
         cursor = cursor.limit(perPage);
       }
@@ -9594,13 +9665,15 @@ var ScoresStorageMongoDB = class _ScoresStorageMongoDB extends ScoresStorage {
   async listScoresBySpan({
     traceId,
     spanId,
-    pagination
+    pagination,
+    filters
   }) {
     try {
       const { page, perPage: perPageInput } = pagination;
       const perPage = normalizePerPage(perPageInput, 100);
       const { offset: start, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const query = { traceId, spanId };
+      applyTenancyFilters(query, filters);
       const collection = await this.getCollection(TABLE_SCORERS);
       const total = await collection.countDocuments(query);
       if (total === 0) {