@mastra/pg 1.8.4-alpha.0 → 1.8.4-alpha.2

package/dist/index.js

@@ -5739,6 +5739,38 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
       );
     }
   }
+  // --- Aggregation ---
+  async getReviewSummary() {
+    try {
+      const tableName = getTableName2({ indexName: TABLE_EXPERIMENT_RESULTS, schemaName: getSchemaName2(this.#schema) });
+      const rows = await this.#db.client.manyOrNone(
+        `SELECT
+          "experimentId",
+          COUNT(*)::int as total,
+          SUM(CASE WHEN status = 'needs-review' THEN 1 ELSE 0 END)::int as "needsReview",
+          SUM(CASE WHEN status = 'reviewed' THEN 1 ELSE 0 END)::int as reviewed,
+          SUM(CASE WHEN status = 'complete' THEN 1 ELSE 0 END)::int as complete
+        FROM ${tableName}
+        GROUP BY "experimentId"`
+      );
+      return (rows || []).map((row) => ({
+        experimentId: row.experimentId,
+        total: Number(row.total ?? 0),
+        needsReview: Number(row.needsReview ?? 0),
+        reviewed: Number(row.reviewed ?? 0),
+        complete: Number(row.complete ?? 0)
+      }));
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "GET_REVIEW_SUMMARY", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
   // --- Clear ---
   async dangerouslyClearAll() {
     await this.#db.clearTable({ tableName: TABLE_EXPERIMENT_RESULTS });