@mastra/libsql 1.10.0 → 1.10.1-alpha.0

package/dist/index.js

@@ -2991,11 +2991,13 @@ function rowToTask(row) {
     runId: String(row.run_id),
     result: parseJson(row.result),
     error: parseJson(row.error),
+    suspendPayload: parseJson(row.suspend_payload),
     retryCount: Number(row.retry_count),
     maxRetries: Number(row.max_retries),
     timeoutMs: Number(row.timeout_ms),
     createdAt: new Date(String(row.createdAt)),
     startedAt: row.startedAt ? new Date(String(row.startedAt)) : void 0,
+    suspendedAt: row.suspendedAt ? new Date(String(row.suspendedAt)) : void 0,
     completedAt: row.completedAt ? new Date(String(row.completedAt)) : void 0
   };
 }
@@ -3013,6 +3015,11 @@ var BackgroundTasksLibSQL = class extends BackgroundTasksStorage {
       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.deleteData({ tableName: TABLE_BACKGROUND_TASKS });
@@ -3032,11 +3039,13 @@ var BackgroundTasksLibSQL = class extends BackgroundTasksStorage {
         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
       }
     });
@@ -3056,6 +3065,10 @@ var BackgroundTasksLibSQL = class extends BackgroundTasksStorage {
       setClauses.push("error = jsonb(?)");
       params.push(serializeJson(update.error));
     }
+    if ("suspendPayload" in update) {
+      setClauses.push("suspend_payload = jsonb(?)");
+      params.push(serializeJson(update.suspendPayload));
+    }
     if ("retryCount" in update) {
       setClauses.push("retry_count = ?");
       params.push(update.retryCount);
@@ -3064,6 +3077,10 @@ var BackgroundTasksLibSQL = class extends BackgroundTasksStorage {
       setClauses.push("startedAt = ?");
       params.push(update.startedAt?.toISOString() ?? null);
     }
+    if ("suspendedAt" in update) {
+      setClauses.push("suspendedAt = ?");
+      params.push(update.suspendedAt?.toISOString() ?? null);
+    }
     if ("completedAt" in update) {
       setClauses.push("completedAt = ?");
       params.push(update.completedAt?.toISOString() ?? null);
@@ -3111,7 +3128,11 @@ var BackgroundTasksLibSQL = class extends BackgroundTasksStorage {
       conditions.push("tool_name = ?");
       params.push(filter.toolName);
     }
-    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
+    if (filter.toolCallId) {
+      conditions.push("tool_call_id = ?");
+      params.push(filter.toolCallId);
+    }
+    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());
@@ -3126,7 +3147,7 @@ var BackgroundTasksLibSQL = class extends BackgroundTasksStorage {
       args: [...params]
     });
     const total = Number(countResult.rows[0]?.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";
     const direction = filter.orderDirection === "desc" ? "DESC" : "ASC";
     let sql = `SELECT ${buildSelectColumns(TABLE_BACKGROUND_TASKS)} FROM ${TABLE_BACKGROUND_TASKS} ${where} ORDER BY ${orderCol} ${direction}`;
     if (filter.perPage != null) {
@@ -3154,7 +3175,7 @@ var BackgroundTasksLibSQL = 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());