@mastra/mssql 1.2.1 → 1.2.2-alpha.0

package/dist/index.js

@@ -1379,11 +1379,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),
     maxRetries: Number(row.max_retries),
     timeoutMs: Number(row.timeout_ms),
     createdAt: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt),
     startedAt: row.startedAt ? row.startedAt instanceof Date ? row.startedAt : new Date(row.startedAt) : void 0,
+    suspendedAt: row.suspendedAt ? row.suspendedAt instanceof Date ? row.suspendedAt : new Date(row.suspendedAt) : void 0,
     completedAt: row.completedAt ? row.completedAt instanceof Date ? row.completedAt : new Date(row.completedAt) : void 0
   };
 }
@@ -1416,6 +1418,11 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
       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"]
+    });
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -1485,11 +1492,13 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
         args: serializeJson(task.args),
         result: serializeJson(task.result),
         error: serializeJson(task.error),
+        suspend_payload: serializeJson(task.suspendPayload),
         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
       }
     });
@@ -1510,6 +1519,10 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
       setClauses.push(`[error] = @p${idx}`);
       params[`p${idx++}`] = serializeJson(update.error);
     }
+    if ("suspendPayload" in update) {
+      setClauses.push(`[suspend_payload] = @p${idx}`);
+      params[`p${idx++}`] = serializeJson(update.suspendPayload);
+    }
     if ("retryCount" in update) {
       setClauses.push(`[retry_count] = @p${idx}`);
       params[`p${idx++}`] = update.retryCount;
@@ -1518,6 +1531,10 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
       setClauses.push(`[startedAt] = @p${idx}`);
       params[`p${idx++}`] = update.startedAt?.toISOString() ?? null;
     }
+    if ("suspendedAt" in update) {
+      setClauses.push(`[suspendedAt] = @p${idx}`);
+      params[`p${idx++}`] = update.suspendedAt?.toISOString() ?? null;
+    }
     if ("completedAt" in update) {
       setClauses.push(`[completedAt] = @p${idx}`);
       params[`p${idx++}`] = update.completedAt?.toISOString() ?? null;
@@ -1573,7 +1590,11 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
       params[`p${idx}`] = filter.toolName;
       conditions.push(`[tool_name] = @p${idx++}`);
     }
-    const dateCol = filter.dateFilterBy === "startedAt" ? "[startedAt]" : filter.dateFilterBy === "completedAt" ? "[completedAt]" : "[createdAt]";
+    if (filter.toolCallId) {
+      params[`p${idx}`] = filter.toolCallId;
+      conditions.push(`[tool_call_id] = @p${idx++}`);
+    }
+    const dateCol = filter.dateFilterBy === "startedAt" ? "[startedAt]" : filter.dateFilterBy === "suspendedAt" ? "[suspendedAt]" : filter.dateFilterBy === "completedAt" ? "[completedAt]" : "[createdAt]";
     if (filter.fromDate) {
       params[`p${idx}`] = filter.fromDate.toISOString();
       conditions.push(`${dateCol} >= @p${idx++}`);
@@ -1589,7 +1610,7 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
     }
     const countResult = await countRequest.query(`SELECT COUNT(*) as count FROM ${this.tableName()} ${where}`);
     const total = Number(countResult.recordset[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 sql5 = `SELECT * FROM ${this.tableName()} ${where} ORDER BY ${orderCol} ${direction}`;
     if (filter.perPage != null) {
@@ -1626,7 +1647,7 @@ var BackgroundTasksMSSQL = class _BackgroundTasksMSSQL extends BackgroundTasksSt
       });
       conditions.push(`[status] IN (${placeholders.join(", ")})`);
     }
-    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) {
       params[`p${idx}`] = filter.fromDate.toISOString();
       conditions.push(`${dateCol} >= @p${idx++}`);