@mastra/lance 1.3.0 → 1.3.1

package/dist/index.js

@@ -530,21 +530,23 @@ var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
 	async createTask(task) {
 		await (await this.client.openTable(TABLE_BACKGROUND_TASKS)).add([toRecord(task)], { mode: "append" });
 	}
-	async updateTask(taskId, update) {
-		const existing = await this.getTask(taskId);
-		if (!existing) return;
-		const merged = { ...existing };
-		if ("status" in update) merged.status = update.status;
-		if ("result" in update) merged.result = update.result;
-		if ("error" in update) merged.error = update.error;
-		if ("suspendPayload" in update) merged.suspendPayload = update.suspendPayload;
-		if ("retryCount" in update) merged.retryCount = update.retryCount;
-		if ("startedAt" in update) merged.startedAt = update.startedAt;
-		if ("suspendedAt" in update) merged.suspendedAt = update.suspendedAt;
-		if ("completedAt" in update) merged.completedAt = update.completedAt;
-		const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
-		await table.delete(`id = '${escapeStr(taskId)}'`);
-		await table.add([toRecord(merged)], { mode: "append" });
+	async updateTask(taskId, update, options) {
+		const values = {};
+		if ("status" in update) values.status = update.status;
+		if ("result" in update) values.result = serializeJson(update.result) ?? null;
+		if ("error" in update) values.error = serializeJson(update.error) ?? null;
+		if ("suspendPayload" in update) values.suspend_payload = serializeJson(update.suspendPayload) ?? null;
+		if ("retryCount" in update) values.retry_count = update.retryCount;
+		if ("startedAt" in update) values.startedAt = update.startedAt?.getTime() ?? 0;
+		if ("suspendedAt" in update) values.suspendedAt = update.suspendedAt?.getTime() ?? 0;
+		if ("completedAt" in update) values.completedAt = update.completedAt?.getTime() ?? 0;
+		if (Object.keys(values).length === 0) return false;
+		const conditions = [`id = '${escapeStr(taskId)}'`];
+		if (options?.expectedStatus) conditions.push(`status = '${escapeStr(options.expectedStatus)}'`);
+		return (await (await this.client.openTable(TABLE_BACKGROUND_TASKS)).update({
+			where: conditions.join(" AND "),
+			values
+		})).rowsUpdated > 0;
 	}
 	async getTask(taskId) {
 		try {