@mastra/libsql 1.21.1-alpha.0 → 1.21.1

package/dist/index.js

@@ -4135,8 +4135,12 @@ var DatasetsLibSQL = class extends DatasetsStorage {
 		try {
 			let result;
 			if (args.datasetVersion !== void 0) result = await this.#client.execute({
-				sql: `SELECT ${buildSelectColumns(TABLE_DATASET_ITEMS)} FROM ${TABLE_DATASET_ITEMS} WHERE id = ? AND datasetVersion = ? AND isDeleted = 0`,
-				args: [args.id, args.datasetVersion]
+				sql: `SELECT ${buildSelectColumns(TABLE_DATASET_ITEMS)} FROM ${TABLE_DATASET_ITEMS} WHERE id = ? AND datasetVersion <= ? AND (validTo IS NULL OR validTo > ?) AND isDeleted = 0 ORDER BY datasetVersion DESC LIMIT 1`,
+				args: [
+					args.id,
+					args.datasetVersion,
+					args.datasetVersion
+				]
 			});
 			else result = await this.#client.execute({
 				sql: `SELECT ${buildSelectColumns(TABLE_DATASET_ITEMS)} FROM ${TABLE_DATASET_ITEMS} WHERE id = ? AND validTo IS NULL AND isDeleted = 0`,
@@ -4600,7 +4604,8 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				"experimentSetId",
 				"comparisonId",
 				"variantId",
-				"trialIndex"
+				"trialIndex",
+				"scorerIds"
 			]
 		});
 		await this.#db.alterTable({
@@ -4612,7 +4617,8 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				"comment",
 				"toolMockReport",
 				"organizationId",
-				"projectId"
+				"projectId",
+				"attempt"
 			]
 		});
 		await this.#client.batch([
@@ -4629,7 +4635,11 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				args: []
 			},
 			{
-				sql: `CREATE UNIQUE INDEX IF NOT EXISTS idx_experiment_results_exp_item ON "${TABLE_EXPERIMENT_RESULTS}" ("experimentId", "itemId")`,
+				sql: `DROP INDEX IF EXISTS idx_experiment_results_exp_item`,
+				args: []
+			},
+			{
+				sql: `CREATE UNIQUE INDEX IF NOT EXISTS idx_experiment_results_exp_item_attempt ON "${TABLE_EXPERIMENT_RESULTS}" ("experimentId", "itemId", "attempt")`,
 				args: []
 			},
 			{
@@ -4718,8 +4728,9 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 			agentVersion: row.agentVersion ?? 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 ? safelyParseJSON(row.scorerIds) : null,
 			name: row.name ?? void 0,
 			description: row.description ?? void 0,
 			metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
@@ -4755,6 +4766,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 			startedAt: ensureDate(row.startedAt),
 			completedAt: ensureDate(row.completedAt),
 			retryCount: row.retryCount,
+			attempt: row.attempt != null ? Number(row.attempt) : 0,
 			traceId: row.traceId ?? null,
 			status: row.status ?? null,
 			tags: row.tags ? safelyParseJSON(row.tags) : null,
@@ -4777,8 +4789,9 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 					agentVersion: input.agentVersion ?? 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,
 					name: input.name ?? null,
 					description: input.description ?? null,
 					metadata: input.metadata ?? null,
@@ -4806,8 +4819,9 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				agentVersion: input.agentVersion ?? 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,
 				name: input.name,
 				description: input.description,
 				metadata: input.metadata,
@@ -5054,6 +5068,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 					startedAt: input.startedAt.toISOString(),
 					completedAt: input.completedAt.toISOString(),
 					retryCount: input.retryCount,
+					attempt: input.attempt ?? 0,
 					traceId: input.traceId ?? null,
 					status: input.status ?? null,
 					tags: input.tags !== void 0 && input.tags !== null ? JSON.stringify(input.tags) : null,
@@ -5075,6 +5090,7 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 				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,
@@ -5089,6 +5105,81 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
 			}, error);
 		}
 	}
+	async upsertExperimentResult(input) {
+		try {
+			const attempt = input.attempt ?? 0;
+			let existingId = (await this.#client.execute({
+				sql: `SELECT "id" FROM ${TABLE_EXPERIMENT_RESULTS} WHERE "experimentId" = ? AND "itemId" = ? AND COALESCE("attempt", 0) = ?`,
+				args: [
+					input.experimentId,
+					input.itemId,
+					attempt
+				]
+			})).rows[0]?.id;
+			if (!existingId) try {
+				return await this.addExperimentResult({
+					...input,
+					attempt
+				});
+			} catch (insertError) {
+				if (!hasErrorCode(insertError, /* @__PURE__ */ new Set([
+					"SQLITE_CONSTRAINT",
+					"SQLITE_CONSTRAINT_PRIMARYKEY",
+					"SQLITE_CONSTRAINT_UNIQUE"
+				]))) throw insertError;
+				existingId = (await this.#client.execute({
+					sql: `SELECT "id" FROM ${TABLE_EXPERIMENT_RESULTS} WHERE "experimentId" = ? AND "itemId" = ? AND COALESCE("attempt", 0) = ?`,
+					args: [
+						input.experimentId,
+						input.itemId,
+						attempt
+					]
+				})).rows[0]?.id;
+				if (!existingId) throw insertError;
+			}
+			await this.#client.execute({
+				sql: `UPDATE ${TABLE_EXPERIMENT_RESULTS} SET
+          "itemDatasetVersion" = ?, "organizationId" = ?, "projectId" = ?,
+          "input" = ?, "output" = ?, "groundTruth" = ?, "error" = ?,
+          "startedAt" = ?, "completedAt" = ?, "retryCount" = ?, "attempt" = ?,
+          "traceId" = ?, "status" = ?, "tags" = ?, "toolMockReport" = ?
+        WHERE "id" = ?`,
+				args: [
+					input.itemDatasetVersion ?? null,
+					input.organizationId ?? null,
+					input.projectId ?? null,
+					JSON.stringify(input.input),
+					input.output != null ? JSON.stringify(input.output) : null,
+					input.groundTruth != null ? JSON.stringify(input.groundTruth) : null,
+					input.error != null ? JSON.stringify(input.error) : null,
+					input.startedAt.toISOString(),
+					input.completedAt.toISOString(),
+					input.retryCount,
+					attempt,
+					input.traceId ?? null,
+					input.status ?? null,
+					input.tags != null ? JSON.stringify(input.tags) : null,
+					input.toolMockReport != null ? JSON.stringify(input.toolMockReport) : null,
+					existingId
+				]
+			});
+			const result = await this.getExperimentResultById({ id: existingId });
+			if (!result) throw new MastraError({
+				id: createStorageErrorId("LIBSQL", "UPSERT_EXPERIMENT_RESULT", "NOT_FOUND"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.USER,
+				details: { resultId: existingId }
+			});
+			return result;
+		} catch (error) {
+			if (error instanceof MastraError) throw error;
+			throw new MastraError({
+				id: createStorageErrorId("LIBSQL", "UPSERT_EXPERIMENT_RESULT", "FAILED"),
+				domain: ErrorDomain.STORAGE,
+				category: ErrorCategory.THIRD_PARTY
+			}, error);
+		}
+	}
 	async updateExperimentResult(input) {
 		try {
 			const setClauses = [];
@@ -11608,8 +11699,13 @@ var ScoresLibSQL = class ScoresLibSQL extends ScoresStorage {
 			}, error);
 		}
 		try {
-			const id = crypto.randomUUID();
+			const suppliedId = parsedScore.id;
+			const id = suppliedId ?? crypto.randomUUID();
 			const now = /* @__PURE__ */ new Date();
+			if (suppliedId) await this.#client.execute({
+				sql: `DELETE FROM ${TABLE_SCORERS} WHERE id = ?`,
+				args: [suppliedId]
+			});
 			await this.#db.insert({
 				tableName: TABLE_SCORERS,
 				record: {