@mastra/dynamodb 1.4.5 → 1.5.0-alpha.0

package/dist/index.js

@@ -176,6 +176,15 @@ const backgroundTaskEntity = new Entity({
 		leaseExpiresAtIso: {
 			type: "string",
 			required: false
+		},
+		expiresAtIso: {
+			type: "string",
+			required: false
+		},
+		metadata: {
+			type: "string",
+			required: false,
+			...jsonSetGet
 		}
 	},
 	indexes: {
@@ -1303,7 +1312,9 @@ function toElectroRecord(task) {
 		suspendedAtIso: task.suspendedAt?.toISOString(),
 		completedAtIso: task.completedAt?.toISOString(),
 		ownerId: task.ownerId ?? void 0,
-		leaseExpiresAtIso: task.leaseExpiresAt?.toISOString()
+		leaseExpiresAtIso: task.leaseExpiresAt?.toISOString(),
+		expiresAtIso: task.expiresAt?.toISOString(),
+		metadata: serializeJson(task.metadata)
 	};
 }
 function fromElectroRecord(data) {
@@ -1338,10 +1349,13 @@ function fromElectroRecord(data) {
 		suspendedAt: asDate(data.suspendedAtIso),
 		completedAt: asDate(data.completedAtIso),
 		ownerId: data.ownerId != null && data.ownerId !== "" ? String(data.ownerId) : void 0,
-		leaseExpiresAt: asDate(data.leaseExpiresAtIso)
+		leaseExpiresAt: asDate(data.leaseExpiresAtIso),
+		expiresAt: asDate(data.expiresAtIso),
+		metadata: parseJson(data.metadata)
 	};
 }
 var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
+	supportsTaskExpiry = true;
 	service;
 	constructor(config) {
 		super();
@@ -1401,6 +1415,14 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
 				if (update.leaseExpiresAt === void 0 || update.leaseExpiresAt === null) removeFields.push("leaseExpiresAtIso");
 				else setFields.leaseExpiresAtIso = update.leaseExpiresAt.toISOString();
 			}
+			if ("expiresAt" in update) {
+				if (update.expiresAt === void 0 || update.expiresAt === null) removeFields.push("expiresAtIso");
+				else setFields.expiresAtIso = update.expiresAt.toISOString();
+			}
+			if ("metadata" in update) {
+				if (update.metadata === void 0 || update.metadata === null) removeFields.push("metadata");
+				else setFields.metadata = serializeJson(update.metadata);
+			}
 			if (Object.keys(setFields).length === 0 && removeFields.length === 0) return false;
 			let op = this.service.entities.background_task.patch({
 				entity: ENTITY,
@@ -1417,9 +1439,13 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
 				const expectedLeaseExpiresAt = options.expectedLeaseExpiresAt;
 				op = op.where(({ leaseExpiresAtIso }, { eq, notExists }) => expectedLeaseExpiresAt === null ? notExists(leaseExpiresAtIso) : eq(leaseExpiresAtIso, expectedLeaseExpiresAt.toISOString()));
 			}
+			if (options?.expectedSuspendedAt !== void 0) {
+				const expectedSuspendedAt = options.expectedSuspendedAt;
+				op = op.where(({ suspendedAtIso }, { eq, notExists }) => expectedSuspendedAt === null ? notExists(suspendedAtIso) : eq(suspendedAtIso, expectedSuspendedAt.toISOString()));
+			}
 			await op.go();
 		} catch (error) {
-			if ((options?.expectedStatus !== void 0 || options?.expectedOwnerId !== void 0 || options?.expectedLeaseExpiresAt !== void 0) && error?.cause?.name === "ConditionalCheckFailedException") return false;
+			if ((options?.expectedStatus !== void 0 || options?.expectedOwnerId !== void 0 || options?.expectedLeaseExpiresAt !== void 0 || options?.expectedSuspendedAt !== void 0) && error?.cause?.name === "ConditionalCheckFailedException") return false;
 			throw new MastraError({
 				id: createStorageErrorId("DYNAMODB", "BACKGROUND_TASKS_UPDATE", "FAILED"),
 				domain: ErrorDomain.STORAGE,
@@ -1473,6 +1499,7 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
 			if (filter.toolName) tasks = tasks.filter((t) => t.toolName === filter.toolName);
 			if (filter.toolCallId) tasks = tasks.filter((t) => t.toolCallId === filter.toolCallId);
 			if (filter.runId) tasks = tasks.filter((t) => t.runId === filter.runId);
+			if (filter.hasExpiresAt !== void 0) tasks = tasks.filter((t) => t.expiresAt != null === filter.hasExpiresAt);
 			const dateCol = filter.dateFilterBy ?? "createdAt";
 			if (filter.fromDate) tasks = tasks.filter((t) => {
 				const val = t[dateCol];