@mastra/mongodb 1.19.0 → 1.20.0-alpha.1

package/dist/index.js

@@ -9,7 +9,7 @@ import { MessageList } from "@mastra/core/agent";
 import { saveScorePayloadSchema } from "@mastra/core/evals";
 import { skillSnapshotFieldValuesEqual } from "@mastra/core/storage/domains/skills";
 //#region package.json
-var version = "1.19.0";
+var version = "1.20.0-alpha.1";
 //#endregion
 //#region src/vector/filter.ts
 /**
@@ -2377,7 +2377,9 @@ function toDoc(task) {
 		createdAt: task.createdAt.toISOString(),
 		startedAt: task.startedAt?.toISOString() ?? null,
 		suspendedAt: task.suspendedAt?.toISOString() ?? null,
-		completedAt: task.completedAt?.toISOString() ?? null
+		completedAt: task.completedAt?.toISOString() ?? null,
+		ownerId: task.ownerId ?? null,
+		leaseExpiresAt: task.leaseExpiresAt?.toISOString() ?? null
 	};
 }
 function fromDoc(doc) {
@@ -2400,7 +2402,9 @@ function fromDoc(doc) {
 		createdAt: new Date(doc.createdAt),
 		startedAt: doc.startedAt ? new Date(doc.startedAt) : void 0,
 		suspendedAt: doc.suspendedAt ? new Date(doc.suspendedAt) : void 0,
-		completedAt: doc.completedAt ? new Date(doc.completedAt) : void 0
+		completedAt: doc.completedAt ? new Date(doc.completedAt) : void 0,
+		ownerId: doc.ownerId ?? void 0,
+		leaseExpiresAt: doc.leaseExpiresAt ? new Date(doc.leaseExpiresAt) : void 0
 	};
 }
 var BackgroundTasksStorageMongoDB = class BackgroundTasksStorageMongoDB extends BackgroundTasksStorage {
@@ -2517,11 +2521,14 @@ var BackgroundTasksStorageMongoDB = class BackgroundTasksStorageMongoDB extends
 		if ("startedAt" in update) $set.startedAt = update.startedAt?.toISOString() ?? null;
 		if ("suspendedAt" in update) $set.suspendedAt = update.suspendedAt?.toISOString() ?? null;
 		if ("completedAt" in update) $set.completedAt = update.completedAt?.toISOString() ?? null;
+		if ("ownerId" in update) $set.ownerId = update.ownerId ?? null;
+		if ("leaseExpiresAt" in update) $set.leaseExpiresAt = update.leaseExpiresAt?.toISOString() ?? null;
 		if (Object.keys($set).length === 0) return false;
-		return (await (await this.getCollection()).updateOne({
-			id: taskId,
-			...options?.expectedStatus ? { status: options.expectedStatus } : {}
-		}, { $set })).matchedCount > 0;
+		const filter = { id: taskId };
+		if (options?.expectedStatus) filter.status = options.expectedStatus;
+		if (options?.expectedOwnerId !== void 0) filter.ownerId = options.expectedOwnerId;
+		if (options?.expectedLeaseExpiresAt !== void 0) filter.leaseExpiresAt = options.expectedLeaseExpiresAt?.toISOString() ?? null;
+		return (await (await this.getCollection()).updateOne(filter, { $set })).matchedCount > 0;
 	}
 	async getTask(taskId) {
 		const doc = await (await this.getCollection()).findOne({ id: taskId });
@@ -11014,7 +11021,7 @@ var MongoDBSkillsStorage = class MongoDBSkillsStorage extends SkillsStorage {
 	}
 	async list(args) {
 		try {
-			const { page = 0, perPage: perPageInput, orderBy, authorId, visibility, metadata } = args || {};
+			const { page = 0, perPage: perPageInput, orderBy, authorId, visibility, metadata, status, entityIds } = args || {};
 			const { field, direction } = this.parseOrderBy(orderBy);
 			if (page < 0) throw new MastraError({
 				id: createStorageErrorId("MONGODB", "LIST_SKILLS", "INVALID_PAGE"),
@@ -11024,8 +11031,17 @@ var MongoDBSkillsStorage = class MongoDBSkillsStorage extends SkillsStorage {
 			}, /* @__PURE__ */ new Error("page must be >= 0"));
 			const perPage = normalizePerPage(perPageInput, 100);
 			const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
+			if (entityIds && entityIds.length === 0) return {
+				skills: [],
+				total: 0,
+				page,
+				perPage: perPageForResponse,
+				hasMore: false
+			};
 			const collection = await this.getCollection(TABLE_SKILLS);
 			const filter = {};
+			if (entityIds && entityIds.length > 0) filter.id = { $in: entityIds };
+			if (status) filter.status = status;
 			if (authorId) filter.authorId = authorId;
 			if (visibility) filter.visibility = visibility;
 			if (metadata) for (const [key, value] of Object.entries(metadata)) filter[`metadata.${key}`] = value;