@mastra/upstash 1.5.3 → 1.5.4

package/dist/index.js

@@ -230,7 +230,8 @@ var BackgroundTasksUpstash = class extends BackgroundTasksStorage {
 		await this.#db.deleteData({ tableName: TABLE_BACKGROUND_TASKS });
 	}
 	async createTask(task) {
-		const { key, processedRecord } = processRecord(TABLE_BACKGROUND_TASKS, toStorageRecord(task));
+		const record = toStorageRecord(task);
+		const { key, processedRecord } = processRecord(TABLE_BACKGROUND_TASKS, record);
 		await this.client.set(key, processedRecord);
 	}
 	async updateTask(taskId, update, options) {
@@ -355,7 +356,7 @@ function runClaimKey(claims, runId) {
 	return `${claims}:run_id:${runId}`;
 }
 const RUN_FENCE_CONFLICT = "MASTRA_RUN_FENCE_CONFLICT";
-const RUN_CLAIM_RETENTION_MS = 10080 * 60 * 1e3;
+const RUN_CLAIM_RETENTION_MS = 6048e5;
 const FENCE_GUARD = `
 local fenceClaim = redis.call('HMGET', KEYS[#KEYS], 'generation', 'ownerId')
 if tonumber(fenceClaim[1]) ~= tonumber(ARGV[#ARGV - 1]) or fenceClaim[2] ~= ARGV[#ARGV] then
@@ -1367,22 +1368,24 @@ var StoreMemoryUpstash = class extends MemoryStorage {
 				}
 				for (const key in fieldsToUpdate) if (Object.prototype.hasOwnProperty.call(fieldsToUpdate, key) && key !== "content") updatedMessage[key] = fieldsToUpdate[key];
 				const key = messageIdToKey[id];
-				if (key) if (updatePayload.threadId && updatePayload.threadId !== existingMessage.threadId) {
-					const newThreadId = updatedMessage.threadId;
-					const oldThreadMessagesKey = getThreadMessagesKey(existingMessage.threadId);
-					pipeline.zrem(oldThreadMessagesKey, id);
-					pipeline.del(key);
-					const newKey = getMessageKey(newThreadId, id);
-					pipeline.set(newKey, updatedMessage);
-					pipeline.set(getMessageIndexKey(id), newThreadId);
-					messageIdToKey[id] = newKey;
-					const newThreadMessagesKey = getThreadMessagesKey(newThreadId);
-					const score = updatedMessage._index !== void 0 ? updatedMessage._index : new Date(updatedMessage.createdAt).getTime();
-					pipeline.zadd(newThreadMessagesKey, {
-						score,
-						member: id
-					});
-				} else pipeline.set(key, updatedMessage);
+				if (key) {
+					if (updatePayload.threadId && updatePayload.threadId !== existingMessage.threadId) {
+						const newThreadId = updatedMessage.threadId;
+						const oldThreadMessagesKey = getThreadMessagesKey(existingMessage.threadId);
+						pipeline.zrem(oldThreadMessagesKey, id);
+						pipeline.del(key);
+						const newKey = getMessageKey(newThreadId, id);
+						pipeline.set(newKey, updatedMessage);
+						pipeline.set(getMessageIndexKey(id), newThreadId);
+						messageIdToKey[id] = newKey;
+						const newThreadMessagesKey = getThreadMessagesKey(newThreadId);
+						const score = updatedMessage._index !== void 0 ? updatedMessage._index : new Date(updatedMessage.createdAt).getTime();
+						pipeline.zadd(newThreadMessagesKey, {
+							score,
+							member: id
+						});
+					} else pipeline.set(key, updatedMessage);
+				}
 			}
 			const now = /* @__PURE__ */ new Date();
 			for (const threadId of threadIdsToUpdate) if (threadId) {
@@ -1714,12 +1717,13 @@ var ScoresUpstash = class extends ScoresStorage {
 		const id = crypto.randomUUID();
 		const createdAt = now;
 		const updatedAt = now;
-		const { key, processedRecord } = processRecord(TABLE_SCORERS, {
+		const scoreWithId = {
 			...validatedScore,
 			id,
 			createdAt,
 			updatedAt
-		});
+		};
+		const { key, processedRecord } = processRecord(TABLE_SCORERS, scoreWithId);
 		try {
 			await this.client.set(key, processedRecord);
 			return { score: {