@mastra/pg 1.8.2-alpha.0 → 1.8.3-alpha.0

package/dist/index.js

@@ -4327,7 +4327,11 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
     });
     await this.#db.createTable({ tableName: TABLE_DATASET_VERSIONS, schema: DATASET_VERSIONS_SCHEMA });
     await this.#addColumnIfNotExists(TABLE_DATASETS, "requestContextSchema", "JSONB");
+    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_DATASET_ITEMS, "requestContext", "JSONB");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "source", "JSONB");
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -4394,6 +4398,9 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       inputSchema: row.inputSchema ? safelyParseJSON(row.inputSchema) : void 0,
       groundTruthSchema: row.groundTruthSchema ? safelyParseJSON(row.groundTruthSchema) : void 0,
       requestContextSchema: row.requestContextSchema ? safelyParseJSON(row.requestContextSchema) : void 0,
+      tags: row.tags ? safelyParseJSON(row.tags) : void 0,
+      targetType: row.targetType || null,
+      targetIds: row.targetIds || null,
       version: row.version,
       createdAt: ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: ensureDate(row.updatedAtZ || row.updatedAt)
@@ -4408,6 +4415,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
       requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
+      source: row.source ? safelyParseJSON(row.source) : void 0,
       createdAt: ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: ensureDate(row.updatedAtZ || row.updatedAt)
     };
@@ -4423,6 +4431,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
       requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
+      source: row.source ? safelyParseJSON(row.source) : void 0,
       createdAt: ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: ensureDate(row.updatedAtZ || row.updatedAt)
     };
@@ -4451,6 +4460,8 @@ var DatasetsPG = class _DatasetsPG extends 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
@@ -4464,6 +4475,8 @@ var DatasetsPG = class _DatasetsPG extends 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
@@ -4535,6 +4548,18 @@ var DatasetsPG = class _DatasetsPG extends 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}`,
@@ -4548,6 +4573,9 @@ var DatasetsPG = class _DatasetsPG extends 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) {
@@ -4663,7 +4691,7 @@ var DatasetsPG = class _DatasetsPG extends 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,
@@ -4672,6 +4700,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
             jsonbArg(args.groundTruth),
             jsonbArg(args.requestContext),
             jsonbArg(args.metadata),
+            jsonbArg(args.source),
             nowIso,
             nowIso,
             nowIso,
@@ -4691,6 +4720,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         groundTruth: args.groundTruth,
         requestContext: args.requestContext,
         metadata: args.metadata,
+        source: args.source,
         createdAt: now,
         updatedAt: now
       };
@@ -4738,6 +4768,7 @@ var DatasetsPG = class _DatasetsPG extends 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(
@@ -4750,7 +4781,7 @@ var DatasetsPG = class _DatasetsPG extends 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,
@@ -4759,6 +4790,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
             jsonbArg(mergedGroundTruth),
             jsonbArg(mergedRequestContext),
             jsonbArg(mergedMetadata),
+            jsonbArg(mergedSource),
             existing.createdAt.toISOString(),
             existing.createdAt.toISOString(),
             nowIso,
@@ -4777,6 +4809,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         groundTruth: mergedGroundTruth,
         requestContext: mergedRequestContext,
         metadata: mergedMetadata,
+        source: mergedSource,
         updatedAt: now
       };
     } catch (error) {
@@ -4822,7 +4855,7 @@ var DatasetsPG = class _DatasetsPG extends 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,
@@ -4831,6 +4864,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
             jsonbArg(existing.groundTruth),
             jsonbArg(existing.requestContext),
             jsonbArg(existing.metadata),
+            jsonbArg(existing.source),
             existing.createdAt.toISOString(),
             existing.createdAt.toISOString(),
             nowIso,
@@ -4884,7 +4918,7 @@ var DatasetsPG = class _DatasetsPG extends 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,
@@ -4893,6 +4927,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
               jsonbArg(itemInput.groundTruth),
               jsonbArg(itemInput.requestContext),
               jsonbArg(itemInput.metadata),
+              jsonbArg(itemInput.source),
               nowIso,
               nowIso,
               nowIso,
@@ -4913,6 +4948,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         groundTruth: itemInput.groundTruth,
         requestContext: itemInput.requestContext,
         metadata: itemInput.metadata,
+        source: itemInput.source,
         createdAt: now,
         updatedAt: now
       }));
@@ -4967,7 +5003,7 @@ var DatasetsPG = class _DatasetsPG extends 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,
@@ -4976,6 +5012,7 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
               jsonbArg(item.groundTruth),
               jsonbArg(item.requestContext),
               jsonbArg(item.metadata),
+              jsonbArg(item.source),
               item.createdAt.toISOString(),
               item.createdAt.toISOString(),
               nowIso,
@@ -5296,6 +5333,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
       completedAt: ensureDate(row.completedAtZ || row.completedAt),
       retryCount: row.retryCount,
       traceId: row.traceId ?? null,
+      status: row.status ?? null,
+      tags: row.tags ? safelyParseJSON(row.tags) : null,
       createdAt: ensureDate(row.createdAtZ || row.createdAt)
     };
   }
@@ -5536,6 +5575,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
           completedAt: input.completedAt.toISOString(),
           retryCount: input.retryCount,
           traceId: input.traceId ?? null,
+          status: input.status ?? null,
+          tags: input.tags ?? null,
           createdAt: nowIso
         }
       });
@@ -5552,6 +5593,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
         completedAt: input.completedAt,
         retryCount: input.retryCount,
         traceId: input.traceId ?? null,
+        status: input.status ?? null,
+        tags: input.tags ?? null,
         createdAt: now
       };
     } catch (error) {
@@ -5565,6 +5608,64 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
       );
     }
   }
+  async updateExperimentResult(input) {
+    try {
+      const tableName = getTableName2({ indexName: 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 MastraError({
+            id: createStorageErrorId("PG", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+            domain: ErrorDomain.STORAGE,
+            category: 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 MastraError({
+          id: createStorageErrorId("PG", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { resultId: input.id }
+        });
+      }
+      return this.transformExperimentResultRow(row);
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "UPDATE_EXPERIMENT_RESULT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
   async getExperimentResultById({ id }) {
     try {
       const tableName = getTableName2({ indexName: TABLE_EXPERIMENT_RESULTS, schemaName: getSchemaName2(this.#schema) });