@mastra/mongodb 1.18.1-alpha.0 → 1.18.1

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.18.1-alpha.0";
+var version = "1.18.1";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -3677,8 +3677,9 @@ function transformExperimentRow(row) {
 		datasetVersion: row.datasetVersion != null ? Number(row.datasetVersion) : null,
 		organizationId: row.organizationId ?? null,
 		projectId: row.projectId ?? null,
-		targetType: row.targetType,
-		targetId: row.targetId,
+		targetType: row.targetType ?? null,
+		targetId: row.targetId ?? null,
+		scorerIds: row.scorerIds ?? null,
 		status: row.status,
 		totalItems: Number(row.totalItems ?? 0),
 		succeededCount: Number(row.succeededCount ?? 0),
@@ -3706,6 +3707,7 @@ function transformExperimentResultRow(row) {
 		startedAt: toDate(row.startedAt),
 		completedAt: toDate(row.completedAt),
 		retryCount: Number(row.retryCount ?? 0),
+		attempt: row.attempt != null ? Number(row.attempt) : 0,
 		traceId: row.traceId ?? null,
 		status: row.status ?? null,
 		tags: Array.isArray(row.tags) ? row.tags : parseJsonField(row.tags) ?? null,
@@ -3848,7 +3850,8 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends Experime
 				collection: TABLE_EXPERIMENT_RESULTS,
 				keys: {
 					experimentId: 1,
-					itemId: 1
+					itemId: 1,
+					attempt: 1
 				},
 				options: { unique: true }
 			},
@@ -3875,6 +3878,9 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends Experime
 	}
 	async createDefaultIndexes() {
 		if (this.#skipDefaultIndexes) return;
+		try {
+			await (await this.getCollection(TABLE_EXPERIMENT_RESULTS)).dropIndex("experimentId_1_itemId_1");
+		} catch {}
 		for (const indexDef of this.getDefaultIndexDefinitions()) try {
 			await (await this.getCollection(indexDef.collection)).createIndex(indexDef.keys, indexDef.options);
 		} catch (error) {
@@ -3911,8 +3917,9 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends Experime
 			datasetVersion: input.datasetVersion ?? null,
 			organizationId: input.organizationId ?? null,
 			projectId: input.projectId ?? null,
-			targetType: input.targetType,
-			targetId: input.targetId,
+			targetType: input.targetType ?? null,
+			targetId: input.targetId ?? null,
+			scorerIds: input.scorerIds ?? null,
 			status: "pending",
 			totalItems: input.totalItems,
 			succeededCount: 0,
@@ -4083,6 +4090,7 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends Experime
 			startedAt: input.startedAt,
 			completedAt: input.completedAt,
 			retryCount: input.retryCount,
+			attempt: input.attempt ?? 0,
 			traceId: input.traceId ?? null,
 			status: input.status ?? null,
 			tags: input.tags ?? null,
@@ -4105,6 +4113,7 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends Experime
 				startedAt: input.startedAt,
 				completedAt: input.completedAt,
 				retryCount: input.retryCount,
+				attempt: input.attempt ?? 0,
 				traceId: input.traceId ?? null,
 				status: input.status ?? null,
 				tags: input.tags ?? null,
@@ -4120,6 +4129,48 @@ var MongoDBExperimentsStorage = class MongoDBExperimentsStorage extends Experime
 			}, error);
 		}
 	}
+	async upsertExperimentResult(input) {
+		const attempt = input.attempt ?? 0;
+		try {
+			return transformExperimentResultRow(await (await this.getCollection(TABLE_EXPERIMENT_RESULTS)).findOneAndUpdate({
+				experimentId: input.experimentId,
+				itemId: input.itemId,
+				$or: [{ attempt }, ...attempt === 0 ? [{ attempt: null }, { attempt: { $exists: false } }] : []]
+			}, {
+				$set: {
+					itemDatasetVersion: input.itemDatasetVersion ?? null,
+					organizationId: input.organizationId ?? null,
+					projectId: input.projectId ?? null,
+					input: input.input,
+					output: input.output ?? null,
+					groundTruth: input.groundTruth ?? null,
+					error: input.error ?? null,
+					startedAt: input.startedAt,
+					completedAt: input.completedAt,
+					retryCount: input.retryCount,
+					attempt,
+					traceId: input.traceId ?? null,
+					status: input.status ?? null,
+					tags: input.tags ?? null,
+					toolMockReport: input.toolMockReport ?? null
+				},
+				$setOnInsert: {
+					id: randomUUID(),
+					createdAt: /* @__PURE__ */ new Date()
+				}
+			}, {
+				upsert: true,
+				returnDocument: "after"
+			}));
+		} catch (error) {
+			throw new MastraError({
+				id: createStorageErrorId("MONGODB", "UPSERT_EXPERIMENT_RESULT", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY,
+				details: { experimentId: input.experimentId }
+			}, error);
+		}
+	}
 	async updateExperimentResult(input) {
 		const updateFields = {};
 		if (input.status !== void 0) updateFields.status = input.status;
@@ -10059,7 +10110,8 @@ var ScoresStorageMongoDB = class ScoresStorageMongoDB extends ScoresStorage {
 		}
 		try {
 			const now = /* @__PURE__ */ new Date();
-			const scoreId = randomUUID();
+			const suppliedId = validatedScore.id;
+			const scoreId = suppliedId ?? randomUUID();
 			const scorer = typeof validatedScore.scorer === "string" ? safelyParseJSON(validatedScore.scorer) : validatedScore.scorer;
 			const preprocessStepResult = typeof validatedScore.preprocessStepResult === "string" ? safelyParseJSON(validatedScore.preprocessStepResult) : validatedScore.preprocessStepResult;
 			const analyzeStepResult = typeof validatedScore.analyzeStepResult === "string" ? safelyParseJSON(validatedScore.analyzeStepResult) : validatedScore.analyzeStepResult;
@@ -10082,7 +10134,9 @@ var ScoresStorageMongoDB = class ScoresStorageMongoDB extends ScoresStorage {
 				createdAt,
 				updatedAt
 			};
-			await (await this.getCollection(TABLE_SCORERS)).insertOne(dataToSave);
+			const collection = await this.getCollection(TABLE_SCORERS);
+			if (suppliedId) await collection.replaceOne({ id: scoreId }, dataToSave, { upsert: true });
+			else await collection.insertOne(dataToSave);
 			return { score: dataToSave };
 		} catch (error) {
 			throw new MastraError({