@mastra/lance 1.0.5-alpha.0 → 1.0.6-alpha.0

package/dist/index.js

@@ -605,11 +605,13 @@ function toRecord(task) {
     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,
     startedAt: task.startedAt ?? /* @__PURE__ */ new Date(0),
+    suspendedAt: task.suspendedAt ?? /* @__PURE__ */ new Date(0),
     completedAt: task.completedAt ?? /* @__PURE__ */ new Date(0)
   };
 }
@@ -626,6 +628,7 @@ function fromRecord(row) {
     return val;
   };
   const startedAt = row.startedAt instanceof Date ? row.startedAt : row.startedAt ? new Date(row.startedAt) : void 0;
+  const suspendedAt = row.suspendedAt instanceof Date ? row.suspendedAt : row.suspendedAt ? new Date(row.suspendedAt) : void 0;
   const completedAt = row.completedAt instanceof Date ? row.completedAt : row.completedAt ? new Date(row.completedAt) : void 0;
   return {
     id: String(row.id),
@@ -639,11 +642,13 @@ function fromRecord(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: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt),
     startedAt: startedAt && startedAt.getTime() > 0 ? startedAt : void 0,
+    suspendedAt: suspendedAt && suspendedAt.getTime() > 0 ? suspendedAt : void 0,
     completedAt: completedAt && completedAt.getTime() > 0 ? completedAt : void 0
   };
 }
@@ -664,6 +669,11 @@ var StoreBackgroundTasksLance = 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.clearTable({ tableName: TABLE_BACKGROUND_TASKS });
@@ -679,8 +689,10 @@ var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
     if ("status" in update) merged.status = update.status;
     if ("result" in update) merged.result = update.result;
     if ("error" in update) merged.error = update.error;
+    if ("suspendPayload" in update) merged.suspendPayload = update.suspendPayload;
     if ("retryCount" in update) merged.retryCount = update.retryCount;
     if ("startedAt" in update) merged.startedAt = update.startedAt;
+    if ("suspendedAt" in update) merged.suspendedAt = update.suspendedAt;
     if ("completedAt" in update) merged.completedAt = update.completedAt;
     const table = await this.client.openTable(TABLE_BACKGROUND_TASKS);
     await table.delete(`id = '${escapeStr(taskId)}'`);
@@ -710,6 +722,7 @@ var StoreBackgroundTasksLance = class extends BackgroundTasksStorage {
       if (filter.resourceId) conditions.push(`resource_id = '${escapeStr(filter.resourceId)}'`);
       if (filter.runId) conditions.push(`run_id = '${escapeStr(filter.runId)}'`);
       if (filter.toolName) conditions.push(`tool_name = '${escapeStr(filter.toolName)}'`);
+      if (filter.toolCallId) conditions.push(`tool_call_id = '${escapeStr(filter.toolCallId)}'`);
       let query = table.query();
       if (conditions.length > 0) {
         query = query.where(conditions.join(" AND "));