@mastra/mongodb 1.23.2-alpha.0 → 1.24.0

package/dist/index.js

@@ -9,7 +9,7 @@ import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
 //#region package.json
-var version = "1.23.2-alpha.0";
+var version = "1.24.0";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -41,13 +41,15 @@ var MongoDBFilterTranslator = class extends BaseFilterTranslator {
 		return Object.fromEntries(translatedEntries);
 	}
 	translateOperatorValue(operator, value) {
-		if (this.isLogicalOperator(operator)) if (operator === "$not") {
-			if (typeof value !== "object" || value === null || Array.isArray(value)) throw new Error("$not operator requires an object");
-			if (this.isEmpty(value)) throw new Error("$not operator cannot be empty");
-			return this.translateNode(value);
-		} else {
-			if (!Array.isArray(value)) throw new Error(`Value for logical operator ${operator} must be an array`);
-			return value.map((item) => this.translateNode(item));
+		if (this.isLogicalOperator(operator)) {
+			if (operator === "$not") {
+				if (typeof value !== "object" || value === null || Array.isArray(value)) throw new Error("$not operator requires an object");
+				if (this.isEmpty(value)) throw new Error("$not operator cannot be empty");
+				return this.translateNode(value);
+			} else {
+				if (!Array.isArray(value)) throw new Error(`Value for logical operator ${operator} must be an array`);
+				return value.map((item) => this.translateNode(item));
+			}
 		}
 		if (this.isBasicOperator(operator) || this.isNumericOperator(operator)) {
 			if (value instanceof Date) return value.toISOString();
@@ -1531,17 +1533,19 @@ var MongoDBVector = class MongoDBVector extends MastraVector {
 		if (Array.isArray(filter)) return filter.map((item) => this.transformMetadataFilter(item, metadataMode));
 		if (metadataMode === "document") {
 			const transformed = {};
-			for (const [key, value] of Object.entries(filter)) if (key.startsWith("$")) if (Array.isArray(value)) transformed[key] = value.map((item) => this.transformMetadataFilter(item, metadataMode));
-			else if (typeof value === "object" && value !== null) transformed[key] = this.transformMetadataFilter(value, metadataMode);
-			else transformed[key] = value;
-			else transformed[key] = value;
+			for (const [key, value] of Object.entries(filter)) if (key.startsWith("$")) {
+				if (Array.isArray(value)) transformed[key] = value.map((item) => this.transformMetadataFilter(item, metadataMode));
+				else if (typeof value === "object" && value !== null) transformed[key] = this.transformMetadataFilter(value, metadataMode);
+				else transformed[key] = value;
+			} else transformed[key] = value;
 			return transformed;
 		}
 		const transformed = {};
-		for (const [key, value] of Object.entries(filter)) if (key.startsWith("$")) if (Array.isArray(value)) transformed[key] = value.map((item) => this.transformMetadataFilter(item, metadataMode));
-		else if (typeof value === "object" && value !== null) transformed[key] = this.transformMetadataFilter(value, metadataMode);
-		else transformed[key] = value;
-		else if (key.startsWith("metadata.")) transformed[key] = value;
+		for (const [key, value] of Object.entries(filter)) if (key.startsWith("$")) {
+			if (Array.isArray(value)) transformed[key] = value.map((item) => this.transformMetadataFilter(item, metadataMode));
+			else if (typeof value === "object" && value !== null) transformed[key] = this.transformMetadataFilter(value, metadataMode);
+			else transformed[key] = value;
+		} else if (key.startsWith("metadata.")) transformed[key] = value;
 		else if (this.isMetadataField(key)) transformed[`metadata.${key}`] = value;
 		else transformed[key] = value;
 		return transformed;
@@ -3210,6 +3214,7 @@ var MongoDBDatasetsStorage = class extends DatasetsStorage {
 			const { page, perPage: perPageInput } = args.pagination;
 			const collection = await this.getCollection(TABLE_DATASETS);
 			const filter = {};
+			if (args.filters?.datasetIds !== void 0) filter.id = { $in: args.filters.datasetIds };
 			if (args.filters?.organizationId !== void 0) filter.organizationId = args.filters.organizationId;
 			if (args.filters?.projectId !== void 0) filter.projectId = args.filters.projectId;
 			if (args.filters?.candidateKey !== void 0) filter.candidateKey = args.filters.candidateKey;
@@ -9575,15 +9580,13 @@ MIGRATION REQUIRED: Duplicate spans detected in ${TABLE_SPANS} collection\n=====
 							error: null
 						});
 						break;
-					case TraceStatus.SUCCESS:
-						andConditions.push({
-							endedAt: {
-								$exists: true,
-								$ne: null
-							},
-							error: null
-						});
-						break;
+					case TraceStatus.SUCCESS: andConditions.push({
+						endedAt: {
+							$exists: true,
+							$ne: null
+						},
+						error: null
+					});
 				}
 			}
 			if (andConditions.length) mongoFilter.$and = andConditions;
@@ -12365,8 +12368,10 @@ var WorkflowsStorageMongoDB = class WorkflowsStorageMongoDB extends WorkflowsSto
 			if (options.workflowName) query["workflow_name"] = options.workflowName;
 			if (options.status) query["snapshot.status"] = options.status;
 			if (options.fromDate) query["createdAt"] = { $gte: options.fromDate };
-			if (options.toDate) if (query["createdAt"]) query["createdAt"].$lte = options.toDate;
-			else query["createdAt"] = { $lte: options.toDate };
+			if (options.toDate) {
+				if (query["createdAt"]) query["createdAt"].$lte = options.toDate;
+				else query["createdAt"] = { $lte: options.toDate };
+			}
 			if (options.resourceId) query["resourceId"] = options.resourceId;
 			const collection = await this.getCollection(TABLE_WORKFLOW_SNAPSHOT);
 			let total = 0;