@mastra/clickhouse 1.7.0 → 1.7.1-alpha.0

package/dist/index.js

@@ -687,6 +687,12 @@ function serializeJson(v) {
   if (typeof v === "object" && v != null) return JSON.stringify(v);
   return v ?? "";
 }
+function readNullableDate(val) {
+  if (val == null || val === "") return void 0;
+  const d = new Date(val);
+  if (Number.isNaN(d.getTime()) || d.getTime() === 0) return void 0;
+  return d;
+}
 function rowToTask(row) {
   const parseJson2 = (val) => {
     if (val == null || val === "") return void 0;
@@ -711,12 +717,14 @@ function rowToTask(row) {
     runId: row.run_id ?? "",
     result: parseJson2(row.result),
     error: parseJson2(row.error),
+    suspendPayload: parseJson2(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,
-    completedAt: row.completedAt ? new Date(row.completedAt) : void 0
+    startedAt: readNullableDate(row.startedAt),
+    suspendedAt: readNullableDate(row.suspendedAt),
+    completedAt: readNullableDate(row.completedAt)
   };
 }
 var BackgroundTasksStorageClickhouse = class extends BackgroundTasksStorage {
@@ -730,6 +738,11 @@ var BackgroundTasksStorageClickhouse = 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 });
@@ -750,11 +763,13 @@ var BackgroundTasksStorageClickhouse = class extends BackgroundTasksStorage {
           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() ?? "1970-01-01T00:00:00.000Z",
+          suspendedAt: task.suspendedAt?.toISOString() ?? "1970-01-01T00:00:00.000Z",
           completedAt: task.completedAt?.toISOString() ?? "1970-01-01T00:00:00.000Z"
         }
       ],
@@ -769,8 +784,10 @@ var BackgroundTasksStorageClickhouse = 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;
     await this.createTask(merged);
   }
@@ -808,7 +825,11 @@ var BackgroundTasksStorageClickhouse = class extends BackgroundTasksStorage {
       conditions.push(`tool_name = {var_tool:String}`);
       params.var_tool = filter.toolName;
     }
-    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
+    if (filter.toolCallId) {
+      conditions.push(`tool_call_id = {var_tool_call:String}`);
+      params.var_tool_call = filter.toolCallId;
+    }
+    const dateCol = filter.dateFilterBy === "startedAt" ? "startedAt" : filter.dateFilterBy === "suspendedAt" ? "suspendedAt" : filter.dateFilterBy === "completedAt" ? "completedAt" : "createdAt";
     if (filter.fromDate) {
       conditions.push(`${dateCol} >= parseDateTimeBestEffort({var_from_date:String})`);
       params.var_from_date = filter.fromDate.toISOString();
@@ -830,7 +851,7 @@ var BackgroundTasksStorageClickhouse = class extends BackgroundTasksStorage {
     });
     const countRows = await countResult.json();
     const total = Number(countRows[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 * FROM ${TABLE_BACKGROUND_TASKS} FINAL ${where} ORDER BY ${orderCol} ${direction}`;
     if (filter.perPage != null) {