@mastra/pg 1.10.0 → 1.10.1-alpha.1

package/dist/index.js

@@ -4289,11 +4289,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
   };
 }
@@ -4316,6 +4318,11 @@ var BackgroundTasksPG = class _BackgroundTasksPG 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"]
+    });
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -4402,11 +4409,13 @@ var BackgroundTasksPG = class _BackgroundTasksPG 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() ?? null,
+        suspendedAt: task.suspendedAt?.toISOString() ?? null,
         completedAt: task.completedAt?.toISOString() ?? null
       }
     });
@@ -4427,6 +4436,10 @@ var BackgroundTasksPG = class _BackgroundTasksPG extends BackgroundTasksStorage
       setClauses.push(`"error" = $${paramIdx++}`);
       params.push(serializeJson(update.error));
     }
+    if ("suspendPayload" in update) {
+      setClauses.push(`"suspend_payload" = $${paramIdx++}`);
+      params.push(serializeJson(update.suspendPayload));
+    }
     if ("retryCount" in update) {
       setClauses.push(`"retry_count" = $${paramIdx++}`);
       params.push(update.retryCount);
@@ -4435,6 +4448,10 @@ var BackgroundTasksPG = class _BackgroundTasksPG extends BackgroundTasksStorage
       setClauses.push(`"startedAt" = $${paramIdx++}`);
       params.push(update.startedAt?.toISOString() ?? null);
     }
+    if ("suspendedAt" in update) {
+      setClauses.push(`"suspendedAt" = $${paramIdx++}`);
+      params.push(update.suspendedAt?.toISOString() ?? null);
+    }
     if ("completedAt" in update) {
       setClauses.push(`"completedAt" = $${paramIdx++}`);
       params.push(update.completedAt?.toISOString() ?? null);
@@ -4480,7 +4497,11 @@ var BackgroundTasksPG = class _BackgroundTasksPG extends BackgroundTasksStorage
       conditions.push(`"tool_name" = $${paramIdx++}`);
       params.push(filter.toolName);
     }
-    const dateCol = filter.dateFilterBy === "startedAt" ? '"startedAt"' : filter.dateFilterBy === "completedAt" ? '"completedAt"' : '"createdAt"';
+    if (filter.toolCallId) {
+      conditions.push(`"tool_call_id" = $${paramIdx++}`);
+      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} >= $${paramIdx++}`);
       params.push(filter.fromDate.toISOString());
@@ -4495,7 +4516,7 @@ var BackgroundTasksPG = class _BackgroundTasksPG extends BackgroundTasksStorage
       params.slice(0, paramIdx - 1)
     );
     const total = Number(countResult?.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} ${where} ORDER BY ${orderCol} ${direction}`;
     if (filter.perPage != null) {
@@ -4524,7 +4545,7 @@ var BackgroundTasksPG = class _BackgroundTasksPG extends BackgroundTasksStorage
       conditions.push(`"status" IN (${placeholders.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} >= $${paramIdx++}`);
       params.push(filter.fromDate.toISOString());
@@ -7834,12 +7855,21 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
       resourceId: row.resourceId
     };
   }
-  async getThreadById({ threadId }) {
+  async getThreadById({
+    threadId,
+    resourceId
+  }) {
     try {
       const tableName = getTableName4({ indexName: TABLE_THREADS, schemaName: getSchemaName4(this.#schema) });
+      let query = `SELECT * FROM ${tableName} WHERE id = $1`;
+      let params = [threadId];
+      if (resourceId !== void 0) {
+        query += ` AND "resourceId" = $2`;
+        params.push(resourceId);
+      }
       const thread = await this.#db.client.oneOrNone(
-        `SELECT * FROM ${tableName} WHERE id = $1`,
-        [threadId]
+        query,
+        params
       );
       if (!thread) {
         return null;