@mastra/pg 1.8.4-alpha.1 → 1.8.4-alpha.3

package/dist/index.js

@@ -4330,6 +4330,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
     await this.#addColumnIfNotExists(TABLE_DATASETS, "tags", "JSONB");
     await this.#addColumnIfNotExists(TABLE_DATASETS, "targetType", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASETS, "targetIds", "JSONB");
+    await this.#addColumnIfNotExists(TABLE_DATASETS, "scorerIds", "JSONB");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "requestContext", "JSONB");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "source", "JSONB");
     await this.createDefaultIndexes();
@@ -4401,6 +4402,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       tags: row.tags ? safelyParseJSON(row.tags) : void 0,
       targetType: row.targetType || null,
       targetIds: row.targetIds || null,
+      scorerIds: row.scorerIds || null,
       version: row.version,
       createdAt: ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: ensureDate(row.updatedAtZ || row.updatedAt)
@@ -4462,6 +4464,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
           requestContextSchema: input.requestContextSchema ?? null,
           targetType: input.targetType ?? null,
           targetIds: input.targetIds !== void 0 ? JSON.stringify(input.targetIds) : null,
+          scorerIds: input.scorerIds ? JSON.stringify(input.scorerIds) : null,
           version: 0,
           createdAt: nowIso,
           updatedAt: nowIso
@@ -4477,6 +4480,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         requestContextSchema: input.requestContextSchema ?? void 0,
         targetType: input.targetType ?? null,
         targetIds: input.targetIds ?? null,
+        scorerIds: input.scorerIds ?? null,
         version: 0,
         createdAt: now,
         updatedAt: now
@@ -4560,6 +4564,10 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         setClauses.push(`"targetIds" = $${paramIndex++}`);
         values.push(args.targetIds === null ? null : JSON.stringify(args.targetIds));
       }
+      if (args.scorerIds !== void 0) {
+        setClauses.push(`"scorerIds" = $${paramIndex++}`);
+        values.push(args.scorerIds === null ? null : JSON.stringify(args.scorerIds));
+      }
       values.push(args.id);
       await this.#db.client.none(
         `UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE "id" = $${paramIndex}`,
@@ -4576,6 +4584,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         tags: (args.tags !== void 0 ? args.tags : existing.tags) ?? void 0,
         targetType: (args.targetType !== void 0 ? args.targetType : existing.targetType) ?? null,
         targetIds: (args.targetIds !== void 0 ? args.targetIds : existing.targetIds) ?? null,
+        scorerIds: (args.scorerIds !== void 0 ? args.scorerIds : existing.scorerIds) ?? null,
         updatedAt: new Date(now)
       };
     } catch (error) {
@@ -5739,6 +5748,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 });
@@ -12363,7 +12404,7 @@ function getTableName5({ indexName, schemaName }) {
   return schemaName ? `${schemaName}.${quotedIndexName}` : quotedIndexName;
 }
 function sanitizeJsonForPg(jsonString) {
-  return jsonString.replace(/\\u(0000|[Dd][89A-Fa-f][0-9A-Fa-f]{2})/g, "");
+  return jsonString.replace(/(^|[^\\])(\\(?!["\\/bfnrtu]))/g, "$1\\\\").replace(/\\u(0000|[Dd][89A-Fa-f][0-9A-Fa-f]{2})/g, "");
 }
 var WorkflowsPG = class _WorkflowsPG extends WorkflowsStorage {
   #db;