@mastra/dynamodb 1.0.6 → 1.0.7-alpha.1

package/dist/index.js

@@ -158,6 +158,11 @@ var backgroundTaskEntity = new Entity({
       required: false,
       ...jsonSetGet
     },
+    suspendPayload: {
+      type: "string",
+      required: false,
+      ...jsonSetGet
+    },
     retryCount: {
       type: "number",
       required: true
@@ -175,6 +180,10 @@ var backgroundTaskEntity = new Entity({
       type: "string",
       required: false
     },
+    suspendedAtIso: {
+      type: "string",
+      required: false
+    },
     completedAtIso: {
       type: "string",
       required: false
@@ -1180,11 +1189,13 @@ function toElectroRecord(task) {
     args: serializeJson(task.args),
     result: serializeJson(task.result),
     error: serializeJson(task.error),
+    suspendPayload: serializeJson(task.suspendPayload),
     retryCount: task.retryCount,
     maxRetries: task.maxRetries,
     timeoutMs: task.timeoutMs,
     createdAt: task.createdAt.toISOString(),
     startedAtIso: task.startedAt?.toISOString(),
+    suspendedAtIso: task.suspendedAt?.toISOString(),
     completedAtIso: task.completedAt?.toISOString()
   };
 }
@@ -1213,11 +1224,13 @@ function fromElectroRecord(data) {
     runId: String(data.runId),
     result: parseJson(data.result),
     error: parseJson(data.error),
+    suspendPayload: parseJson(data.suspendPayload),
     retryCount: Number(data.retryCount ?? 0),
     maxRetries: Number(data.maxRetries ?? 0),
     timeoutMs: Number(data.timeoutMs ?? 3e5),
     createdAt: asDate(data.createdAt) ?? /* @__PURE__ */ new Date(),
     startedAt: asDate(data.startedAtIso),
+    suspendedAt: asDate(data.suspendedAtIso),
     completedAt: asDate(data.completedAtIso)
   };
 }
@@ -1270,6 +1283,13 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
           setFields.error = serializeJson(update.error);
         }
       }
+      if ("suspendPayload" in update) {
+        if (update.suspendPayload === void 0 || update.suspendPayload === null) {
+          removeFields.push("suspendPayload");
+        } else {
+          setFields.suspendPayload = serializeJson(update.suspendPayload);
+        }
+      }
       if ("startedAt" in update) {
         if (update.startedAt === void 0 || update.startedAt === null) {
           removeFields.push("startedAtIso");
@@ -1277,6 +1297,13 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
           setFields.startedAtIso = update.startedAt.toISOString();
         }
       }
+      if ("suspendedAt" in update) {
+        if (update.suspendedAt === void 0 || update.suspendedAt === null) {
+          removeFields.push("suspendedAtIso");
+        } else {
+          setFields.suspendedAtIso = update.suspendedAt.toISOString();
+        }
+      }
       if ("completedAt" in update) {
         if (update.completedAt === void 0 || update.completedAt === null) {
           removeFields.push("completedAtIso");
@@ -1343,6 +1370,7 @@ var BackgroundTasksStorageDynamoDB = class extends BackgroundTasksStorage {
       if (filter.threadId) tasks = tasks.filter((t) => t.threadId === filter.threadId);
       if (filter.resourceId) tasks = tasks.filter((t) => t.resourceId === filter.resourceId);
       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);
       const dateCol = filter.dateFilterBy ?? "createdAt";
       if (filter.fromDate) {
@@ -1531,11 +1559,14 @@ var MemoryStorageDynamoDB = class extends MemoryStorage {
       return direction === "DESC" ? -comparison : comparison;
     });
   }
-  async getThreadById({ threadId }) {
-    this.logger.debug("Getting thread by ID", { threadId });
+  async getThreadById({
+    threadId,
+    resourceId
+  }) {
+    this.logger.debug("Getting thread by ID", { threadId, resourceId });
     try {
       const result = await this.service.entities.thread.get({ entity: "thread", id: threadId }).go();
-      if (!result.data) {
+      if (!result.data || resourceId !== void 0 && result.data.resourceId !== resourceId) {
         return null;
       }
       const data = result.data;