@mastra/mongodb 1.8.0 → 1.8.1-alpha.0

package/dist/index.js

@@ -12,7 +12,7 @@ import { saveScorePayloadSchema } from '@mastra/core/evals';
 
 // package.json
 var package_default = {
-  version: "1.8.0"};
+  version: "1.8.1-alpha.0"};
 var MongoDBFilterTranslator = class extends BaseFilterTranslator {
   getSupportedOperators() {
     return {
@@ -1516,11 +1516,13 @@ function toDoc(task) {
     args: task.args,
     result: task.result ?? null,
     error: task.error ?? null,
+    suspend_payload: task.suspendPayload ?? null,
     retry_count: task.retryCount,
     max_retries: task.maxRetries,
     timeout_ms: task.timeoutMs,
     createdAt: task.createdAt.toISOString(),
     startedAt: task.startedAt?.toISOString() ?? null,
+    suspendedAt: task.suspendedAt?.toISOString() ?? null,
     completedAt: task.completedAt?.toISOString() ?? null
   };
 }
@@ -1537,11 +1539,13 @@ function fromDoc(doc) {
     runId: doc.run_id ?? "",
     result: doc.result ?? void 0,
     error: doc.error ?? void 0,
+    suspendPayload: doc.suspend_payload ?? void 0,
     retryCount: Number(doc.retry_count ?? 0),
     maxRetries: Number(doc.max_retries ?? 0),
     timeoutMs: Number(doc.timeout_ms ?? 3e5),
     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
   };
 }
@@ -1609,8 +1613,10 @@ var BackgroundTasksStorageMongoDB = class _BackgroundTasksStorageMongoDB extends
     if ("status" in update) $set.status = update.status;
     if ("result" in update) $set.result = update.result ?? null;
     if ("error" in update) $set.error = update.error ?? null;
+    if ("suspendPayload" in update) $set.suspend_payload = update.suspendPayload ?? null;
     if ("retryCount" in update) $set.retry_count = update.retryCount;
     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 (Object.keys($set).length === 0) return;
     const collection = await this.getCollection();
@@ -1632,14 +1638,15 @@ var BackgroundTasksStorageMongoDB = class _BackgroundTasksStorageMongoDB extends
     if (filter.resourceId) query.resource_id = filter.resourceId;
     if (filter.runId) query.run_id = filter.runId;
     if (filter.toolName) query.tool_name = filter.toolName;
-    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
+    if (filter.toolCallId) query.tool_call_id = filter.toolCallId;
+    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "suspendedAt" ? "suspendedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
     if (filter.fromDate) {
       query[dateCol] = { ...query[dateCol] || {}, $gte: filter.fromDate.toISOString() };
     }
     if (filter.toDate) {
       query[dateCol] = { ...query[dateCol] || {}, $lt: filter.toDate.toISOString() };
     }
-    const orderCol = filter.orderBy === "startedAt" ? "startedAt" : filter.orderBy === "completedAt" ? "completedAt" : "createdAt";
+    const orderCol = filter.orderBy === "startedAt" ? "startedAt" : filter.orderBy === "suspendedAt" ? "suspendedAt" : filter.orderBy === "completedAt" ? "completedAt" : "createdAt";
     const direction = filter.orderDirection === "desc" ? -1 : 1;
     const collection = await this.getCollection();
     const total = await collection.countDocuments(query);
@@ -1663,7 +1670,7 @@ var BackgroundTasksStorageMongoDB = class _BackgroundTasksStorageMongoDB extends
       const statuses = Array.isArray(filter.status) ? filter.status : [filter.status];
       query.status = statuses.length === 1 ? statuses[0] : { $in: statuses };
     }
-    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
+    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "suspendedAt" ? "suspendedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
     if (filter.fromDate) {
       query[dateCol] = { ...query[dateCol] || {}, $gte: filter.fromDate.toISOString() };
     }
@@ -1675,6 +1682,7 @@ var BackgroundTasksStorageMongoDB = class _BackgroundTasksStorageMongoDB extends
     if (filter.resourceId) query.resource_id = filter.resourceId;
     if (filter.runId) query.run_id = filter.runId;
     if (filter.toolName) query.tool_name = filter.toolName;
+    if (filter.toolCallId) query.tool_call_id = filter.toolCallId;
     if (Object.keys(query).length === 0) return;
     const collection = await this.getCollection();
     await collection.deleteMany(query);