@mastra/cloudflare-d1 1.0.5 → 1.0.6-alpha.0

package/dist/index.js

@@ -753,11 +753,13 @@ function rowToTask(row) {
     runId: row.run_id ?? "",
     result: parseJson(row.result),
     error: parseJson(row.error),
+    suspendPayload: parseJson(row.suspend_payload),
     retryCount: Number(row.retry_count ?? 0),
     maxRetries: Number(row.max_retries ?? 0),
     timeoutMs: Number(row.timeout_ms ?? 3e5),
     createdAt: new Date(row.createdAt),
     startedAt: row.startedAt ? new Date(row.startedAt) : void 0,
+    suspendedAt: row.suspendedAt ? new Date(row.suspendedAt) : void 0,
     completedAt: row.completedAt ? new Date(row.completedAt) : void 0
   };
 }
@@ -770,6 +772,11 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
   }
   async init() {
     await this.#db.createTable({ tableName: TABLE_BACKGROUND_TASKS, schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS] });
+    await this.#db.alterTable({
+      tableName: TABLE_BACKGROUND_TASKS,
+      schema: TABLE_SCHEMAS[TABLE_BACKGROUND_TASKS],
+      ifNotExists: ["suspend_payload", "suspendedAt"]
+    });
   }
   async dangerouslyClearAll() {
     await this.#db.clearTable({ tableName: TABLE_BACKGROUND_TASKS });
@@ -790,11 +797,13 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
         "args",
         "result",
         "error",
+        "suspend_payload",
         "retry_count",
         "max_retries",
         "timeout_ms",
         "createdAt",
         "startedAt",
+        "suspendedAt",
         "completedAt"
       ],
       [
@@ -809,11 +818,13 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
         serializeJson(task.args),
         serializeJson(task.result),
         serializeJson(task.error),
+        serializeJson(task.suspendPayload),
         task.retryCount,
         task.maxRetries,
         task.timeoutMs,
         task.createdAt.toISOString(),
         task.startedAt?.toISOString() ?? null,
+        task.suspendedAt?.toISOString() ?? null,
         task.completedAt?.toISOString() ?? null
       ]
     ).build();
@@ -834,6 +845,10 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
       sets.push("error = ?");
       params.push(serializeJson(update.error));
     }
+    if ("suspendPayload" in update) {
+      sets.push("suspend_payload = ?");
+      params.push(serializeJson(update.suspendPayload));
+    }
     if ("retryCount" in update) {
       sets.push("retry_count = ?");
       params.push(update.retryCount);
@@ -842,6 +857,10 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
       sets.push("startedAt = ?");
       params.push(update.startedAt?.toISOString() ?? null);
     }
+    if ("suspendedAt" in update) {
+      sets.push("suspendedAt = ?");
+      params.push(update.suspendedAt?.toISOString() ?? null);
+    }
     if ("completedAt" in update) {
       sets.push("completedAt = ?");
       params.push(update.completedAt?.toISOString() ?? null);
@@ -885,7 +904,11 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
       builder = builder.whereAnd("tool_name = ?", filter.toolName);
       countBuilder = countBuilder.whereAnd("tool_name = ?", filter.toolName);
     }
-    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
+    if (filter.toolCallId) {
+      builder = builder.whereAnd("tool_call_id = ?", filter.toolCallId);
+      countBuilder = countBuilder.whereAnd("tool_call_id = ?", filter.toolCallId);
+    }
+    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "suspendedAt" ? "suspendedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
     if (filter.fromDate) {
       builder = builder.whereAnd(`${dateCol} >= ?`, filter.fromDate.toISOString());
       countBuilder = countBuilder.whereAnd(`${dateCol} >= ?`, filter.fromDate.toISOString());
@@ -897,7 +920,7 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
     const { sql: countSql, params: countParams } = countBuilder.build();
     const countRow = await this.#db.executeQuery({ sql: countSql, params: countParams, first: true });
     const total = Number(countRow?.count ?? 0);
-    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";
     builder = builder.orderBy(orderCol, filter.orderDirection === "desc" ? "DESC" : "ASC");
     if (filter.perPage != null) {
       builder = builder.limit(filter.perPage);
@@ -921,7 +944,7 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
       conditions.push(`status IN (${statuses.map(() => "?").join(",")})`);
       params.push(...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) {
       conditions.push(`${dateCol} >= ?`);
       params.push(filter.fromDate.toISOString());
@@ -938,6 +961,10 @@ var BackgroundTasksStorageD1 = class extends BackgroundTasksStorage {
       conditions.push("run_id = ?");
       params.push(filter.runId);
     }
+    if (filter.toolCallId) {
+      conditions.push("tool_call_id = ?");
+      params.push(filter.toolCallId);
+    }
     if (conditions.length === 0) return;
     const fullTableName = this.#db.getTableName(TABLE_BACKGROUND_TASKS);
     await this.#db.executeQuery({