@mastra/pg 1.8.2 → 1.8.3-alpha.0

package/CHANGELOG.md

@@ -1,5 +1,18 @@
 # @mastra/pg
 
+## 1.8.3-alpha.0
+
+### Patch Changes
+
+- Added storage support for dataset targeting and experiment status fields. ([#14470](https://github.com/mastra-ai/mastra/pull/14470))
+  - Added `targetType` (text) and `targetIds` (jsonb) columns to datasets table for entity association
+  - Added `tags` (jsonb) column to datasets table for tag vocabulary
+  - Added `status` column to experiment results for review workflow tracking
+  - Added migration logic to add new columns to existing tables
+
+- Updated dependencies [[`68ed4e9`](https://github.com/mastra-ai/mastra/commit/68ed4e9f118e8646b60a6112dabe854d0ef53902), [`085c1da`](https://github.com/mastra-ai/mastra/commit/085c1daf71b55a97b8ebad26623089e40055021c), [`4a75e10`](https://github.com/mastra-ai/mastra/commit/4a75e106bd31c283a1b3fe74c923610dcc46415b), [`085c1da`](https://github.com/mastra-ai/mastra/commit/085c1daf71b55a97b8ebad26623089e40055021c)]:
+  - @mastra/core@1.16.0-alpha.0
+
 ## 1.8.2
 
 ### Patch Changes

package/dist/docs/SKILL.md

@@ -3,7 +3,7 @@ name: mastra-pg
 description: Documentation for @mastra/pg. Use when working with @mastra/pg APIs, configuration, or implementation.
 metadata:
   package: "@mastra/pg"
-  version: "1.8.2"
+  version: "1.8.3-alpha.0"
 ---
 
 ## When to use

package/dist/docs/assets/SOURCE_MAP.json

@@ -1,5 +1,5 @@
 {
-  "version": "1.8.2",
+  "version": "1.8.3-alpha.0",
   "package": "@mastra/pg",
   "exports": {},
   "modules": {}

package/dist/index.cjs

@@ -4352,7 +4352,11 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
     });
     await this.#db.createTable({ tableName: storage.TABLE_DATASET_VERSIONS, schema: storage.DATASET_VERSIONS_SCHEMA });
     await this.#addColumnIfNotExists(storage.TABLE_DATASETS, "requestContextSchema", "JSONB");
+    await this.#addColumnIfNotExists(storage.TABLE_DATASETS, "tags", "JSONB");
+    await this.#addColumnIfNotExists(storage.TABLE_DATASETS, "targetType", "TEXT");
+    await this.#addColumnIfNotExists(storage.TABLE_DATASETS, "targetIds", "JSONB");
     await this.#addColumnIfNotExists(storage.TABLE_DATASET_ITEMS, "requestContext", "JSONB");
+    await this.#addColumnIfNotExists(storage.TABLE_DATASET_ITEMS, "source", "JSONB");
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -4419,6 +4423,9 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
       inputSchema: row.inputSchema ? storage.safelyParseJSON(row.inputSchema) : void 0,
       groundTruthSchema: row.groundTruthSchema ? storage.safelyParseJSON(row.groundTruthSchema) : void 0,
       requestContextSchema: row.requestContextSchema ? storage.safelyParseJSON(row.requestContextSchema) : void 0,
+      tags: row.tags ? storage.safelyParseJSON(row.tags) : void 0,
+      targetType: row.targetType || null,
+      targetIds: row.targetIds || null,
       version: row.version,
       createdAt: storage.ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: storage.ensureDate(row.updatedAtZ || row.updatedAt)
@@ -4433,6 +4440,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
       groundTruth: row.groundTruth ? storage.safelyParseJSON(row.groundTruth) : void 0,
       requestContext: row.requestContext ? storage.safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? storage.safelyParseJSON(row.metadata) : void 0,
+      source: row.source ? storage.safelyParseJSON(row.source) : void 0,
       createdAt: storage.ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: storage.ensureDate(row.updatedAtZ || row.updatedAt)
     };
@@ -4448,6 +4456,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
       groundTruth: row.groundTruth ? storage.safelyParseJSON(row.groundTruth) : void 0,
       requestContext: row.requestContext ? storage.safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? storage.safelyParseJSON(row.metadata) : void 0,
+      source: row.source ? storage.safelyParseJSON(row.source) : void 0,
       createdAt: storage.ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: storage.ensureDate(row.updatedAtZ || row.updatedAt)
     };
@@ -4476,6 +4485,8 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
           inputSchema: input.inputSchema ?? null,
           groundTruthSchema: input.groundTruthSchema ?? null,
           requestContextSchema: input.requestContextSchema ?? null,
+          targetType: input.targetType ?? null,
+          targetIds: input.targetIds !== void 0 ? JSON.stringify(input.targetIds) : null,
           version: 0,
           createdAt: nowIso,
           updatedAt: nowIso
@@ -4489,6 +4500,8 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         inputSchema: input.inputSchema ?? void 0,
         groundTruthSchema: input.groundTruthSchema ?? void 0,
         requestContextSchema: input.requestContextSchema ?? void 0,
+        targetType: input.targetType ?? null,
+        targetIds: input.targetIds ?? null,
         version: 0,
         createdAt: now,
         updatedAt: now
@@ -4560,6 +4573,18 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         setClauses.push(`"requestContextSchema" = $${paramIndex++}`);
         values.push(args.requestContextSchema === null ? null : JSON.stringify(args.requestContextSchema));
       }
+      if (args.tags !== void 0) {
+        setClauses.push(`"tags" = $${paramIndex++}`);
+        values.push(args.tags === null ? null : JSON.stringify(args.tags));
+      }
+      if (args.targetType !== void 0) {
+        setClauses.push(`"targetType" = $${paramIndex++}`);
+        values.push(args.targetType);
+      }
+      if (args.targetIds !== void 0) {
+        setClauses.push(`"targetIds" = $${paramIndex++}`);
+        values.push(args.targetIds === null ? null : JSON.stringify(args.targetIds));
+      }
       values.push(args.id);
       await this.#db.client.none(
         `UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE "id" = $${paramIndex}`,
@@ -4573,6 +4598,9 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         inputSchema: (args.inputSchema !== void 0 ? args.inputSchema : existing.inputSchema) ?? void 0,
         groundTruthSchema: (args.groundTruthSchema !== void 0 ? args.groundTruthSchema : existing.groundTruthSchema) ?? void 0,
         requestContextSchema: (args.requestContextSchema !== void 0 ? args.requestContextSchema : existing.requestContextSchema) ?? void 0,
+        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,
         updatedAt: new Date(now)
       };
     } catch (error$1) {
@@ -4688,7 +4716,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         );
         newVersion = row.version;
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
           [
             id,
             args.datasetId,
@@ -4697,6 +4725,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
             jsonbArg(args.groundTruth),
             jsonbArg(args.requestContext),
             jsonbArg(args.metadata),
+            jsonbArg(args.source),
             nowIso,
             nowIso,
             nowIso,
@@ -4716,6 +4745,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         groundTruth: args.groundTruth,
         requestContext: args.requestContext,
         metadata: args.metadata,
+        source: args.source,
         createdAt: now,
         updatedAt: now
       };
@@ -4763,6 +4793,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
       const mergedGroundTruth = args.groundTruth ?? existing.groundTruth;
       const mergedRequestContext = args.requestContext ?? existing.requestContext;
       const mergedMetadata = args.metadata ?? existing.metadata;
+      const mergedSource = args.source ?? existing.source;
       let newVersion;
       await this.#db.client.tx(async (t) => {
         const row = await t.one(
@@ -4775,7 +4806,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
           [newVersion, args.id]
         );
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
           [
             args.id,
             args.datasetId,
@@ -4784,6 +4815,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
             jsonbArg(mergedGroundTruth),
             jsonbArg(mergedRequestContext),
             jsonbArg(mergedMetadata),
+            jsonbArg(mergedSource),
             existing.createdAt.toISOString(),
             existing.createdAt.toISOString(),
             nowIso,
@@ -4802,6 +4834,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         groundTruth: mergedGroundTruth,
         requestContext: mergedRequestContext,
         metadata: mergedMetadata,
+        source: mergedSource,
         updatedAt: now
       };
     } catch (error$1) {
@@ -4847,7 +4880,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
           [newVersion, id]
         );
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
           [
             id,
             datasetId,
@@ -4856,6 +4889,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
             jsonbArg(existing.groundTruth),
             jsonbArg(existing.requestContext),
             jsonbArg(existing.metadata),
+            jsonbArg(existing.source),
             existing.createdAt.toISOString(),
             existing.createdAt.toISOString(),
             nowIso,
@@ -4909,7 +4943,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         newVersion = row.version;
         for (const { id, input: itemInput } of itemsWithIds) {
           await t.none(
-            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11)`,
+            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
             [
               id,
               input.datasetId,
@@ -4918,6 +4952,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
               jsonbArg(itemInput.groundTruth),
               jsonbArg(itemInput.requestContext),
               jsonbArg(itemInput.metadata),
+              jsonbArg(itemInput.source),
               nowIso,
               nowIso,
               nowIso,
@@ -4938,6 +4973,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
         groundTruth: itemInput.groundTruth,
         requestContext: itemInput.requestContext,
         metadata: itemInput.metadata,
+        source: itemInput.source,
         createdAt: now,
         updatedAt: now
       }));
@@ -4992,7 +5028,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
             [newVersion, item.id]
           );
           await t.none(
-            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11)`,
+            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12)`,
             [
               item.id,
               input.datasetId,
@@ -5001,6 +5037,7 @@ var DatasetsPG = class _DatasetsPG extends storage.DatasetsStorage {
               jsonbArg(item.groundTruth),
               jsonbArg(item.requestContext),
               jsonbArg(item.metadata),
+              jsonbArg(item.source),
               item.createdAt.toISOString(),
               item.createdAt.toISOString(),
               nowIso,
@@ -5321,6 +5358,8 @@ var ExperimentsPG = class _ExperimentsPG extends storage.ExperimentsStorage {
       completedAt: storage.ensureDate(row.completedAtZ || row.completedAt),
       retryCount: row.retryCount,
       traceId: row.traceId ?? null,
+      status: row.status ?? null,
+      tags: row.tags ? storage.safelyParseJSON(row.tags) : null,
       createdAt: storage.ensureDate(row.createdAtZ || row.createdAt)
     };
   }
@@ -5561,6 +5600,8 @@ var ExperimentsPG = class _ExperimentsPG extends storage.ExperimentsStorage {
           completedAt: input.completedAt.toISOString(),
           retryCount: input.retryCount,
           traceId: input.traceId ?? null,
+          status: input.status ?? null,
+          tags: input.tags ?? null,
           createdAt: nowIso
         }
       });
@@ -5577,6 +5618,8 @@ var ExperimentsPG = class _ExperimentsPG extends storage.ExperimentsStorage {
         completedAt: input.completedAt,
         retryCount: input.retryCount,
         traceId: input.traceId ?? null,
+        status: input.status ?? null,
+        tags: input.tags ?? null,
         createdAt: now
       };
     } catch (error$1) {
@@ -5590,6 +5633,64 @@ var ExperimentsPG = class _ExperimentsPG extends storage.ExperimentsStorage {
       );
     }
   }
+  async updateExperimentResult(input) {
+    try {
+      const tableName = getTableName2({ indexName: storage.TABLE_EXPERIMENT_RESULTS, schemaName: getSchemaName2(this.#schema) });
+      const setClauses = [];
+      const values = [];
+      let paramIndex = 1;
+      if (input.status !== void 0) {
+        setClauses.push(`"status" = $${paramIndex++}`);
+        values.push(input.status);
+      }
+      if (input.tags !== void 0) {
+        setClauses.push(`"tags" = $${paramIndex++}`);
+        values.push(JSON.stringify(input.tags));
+      }
+      if (setClauses.length === 0) {
+        const existing = await this.getExperimentResultById({ id: input.id });
+        if (!existing) {
+          throw new error.MastraError({
+            id: storage.createStorageErrorId("PG", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+            domain: error.ErrorDomain.STORAGE,
+            category: error.ErrorCategory.USER,
+            details: { resultId: input.id }
+          });
+        }
+        return existing;
+      }
+      values.push(input.id);
+      let whereClause = `"id" = $${paramIndex}`;
+      if (input.experimentId) {
+        paramIndex++;
+        values.push(input.experimentId);
+        whereClause += ` AND "experimentId" = $${paramIndex}`;
+      }
+      const row = await this.#db.client.oneOrNone(
+        `UPDATE ${tableName} SET ${setClauses.join(", ")} WHERE ${whereClause} RETURNING *`,
+        values
+      );
+      if (!row) {
+        throw new error.MastraError({
+          id: storage.createStorageErrorId("PG", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.USER,
+          details: { resultId: input.id }
+        });
+      }
+      return this.transformExperimentResultRow(row);
+    } catch (error$1) {
+      if (error$1 instanceof error.MastraError) throw error$1;
+      throw new error.MastraError(
+        {
+          id: storage.createStorageErrorId("PG", "UPDATE_EXPERIMENT_RESULT", "FAILED"),
+          domain: error.ErrorDomain.STORAGE,
+          category: error.ErrorCategory.THIRD_PARTY
+        },
+        error$1
+      );
+    }
+  }
   async getExperimentResultById({ id }) {
     try {
       const tableName = getTableName2({ indexName: storage.TABLE_EXPERIMENT_RESULTS, schemaName: getSchemaName2(this.#schema) });