@mastra/spanner 1.6.2 → 1.6.3-alpha.0

package/dist/index.js

@@ -1967,7 +1967,7 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends BackgroundTask
 			}
 		});
 	}
-	async updateTask(taskId, update) {
+	async updateTask(taskId, update, options) {
 		const data = {};
 		if ("status" in update) data.status = update.status;
 		if ("result" in update) data.result = update.result ?? null;
@@ -1977,12 +1977,19 @@ var BackgroundTasksSpanner = class BackgroundTasksSpanner extends BackgroundTask
 		if ("startedAt" in update) data.startedAt = update.startedAt ?? null;
 		if ("suspendedAt" in update) data.suspendedAt = update.suspendedAt ?? null;
 		if ("completedAt" in update) data.completedAt = update.completedAt ?? null;
-		if (Object.keys(data).length === 0) return;
-		await this.db.update({
-			tableName: TABLE_BACKGROUND_TASKS,
-			keys: { id: taskId },
-			data
+		if (Object.keys(data).length === 0) return false;
+		const setClauses = Object.keys(data).map((field) => `${quoteIdent(field, "column name")} = @${field}`);
+		const params = {
+			...data,
+			id: taskId,
+			expectedStatus: options?.expectedStatus
+		};
+		const statusPredicate = options?.expectedStatus ? " AND status = @expectedStatus" : "";
+		const [rowCount] = await this.database.runUpdate({
+			sql: `UPDATE ${this.tableName()} SET ${setClauses.join(", ")} WHERE id = @id${statusPredicate}`,
+			params
 		});
+		return rowCount > 0;
 	}
 	async getTask(taskId) {
 		const [rows] = await this.database.run({
@@ -4015,6 +4022,7 @@ function rowToExperimentResult(row) {
 		input: t.input ?? null,
 		output: t.output ?? null,
 		groundTruth: t.groundTruth ?? null,
+		metadata: t.metadata ?? null,
 		error: t.error ?? null,
 		startedAt: toDate(t.startedAt),
 		completedAt: toDate(t.completedAt),
@@ -4081,6 +4089,7 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 			schema: TABLE_SCHEMAS[TABLE_EXPERIMENT_RESULTS],
 			ifNotExists: [
 				"comment",
+				"metadata",
 				"organizationId",
 				"projectId",
 				"attempt"
@@ -4474,6 +4483,7 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 				input: input.input ?? null,
 				output: input.output ?? null,
 				groundTruth: input.groundTruth ?? null,
+				metadata: input.metadata ?? null,
 				error: input.error ?? null,
 				startedAt: input.startedAt,
 				completedAt: input.completedAt,
@@ -4497,6 +4507,7 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 					input: result.input,
 					output: result.output,
 					groundTruth: result.groundTruth,
+					metadata: result.metadata,
 					error: result.error,
 					startedAt: result.startedAt,
 					completedAt: result.completedAt,
@@ -4568,6 +4579,7 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
                 ${quoteIdent("input", "column name")} = @input,
                 ${quoteIdent("output", "column name")} = @output,
                 ${quoteIdent("groundTruth", "column name")} = @groundTruth,
+                ${quoteIdent("metadata", "column name")} = @metadata,
                 ${quoteIdent("error", "column name")} = @error,
                 ${quoteIdent("startedAt", "column name")} = @startedAt,
                 ${quoteIdent("completedAt", "column name")} = @completedAt,
@@ -4586,6 +4598,7 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 					input: JSON.stringify(input.input),
 					output: input.output == null ? null : JSON.stringify(input.output),
 					groundTruth: input.groundTruth == null ? null : JSON.stringify(input.groundTruth),
+					metadata: input.metadata == null ? null : JSON.stringify(input.metadata),
 					error: input.error == null ? null : JSON.stringify(input.error),
 					startedAt: input.startedAt,
 					completedAt: input.completedAt,
@@ -4603,6 +4616,7 @@ var ExperimentsSpanner = class ExperimentsSpanner extends ExperimentsStorage {
 					input: "json",
 					output: "json",
 					groundTruth: "json",
+					metadata: "json",
 					error: "json",
 					traceId: "string",
 					status: "string",