@mastra/pg 1.19.0-alpha.2 → 1.19.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.js CHANGED
@@ -5275,6 +5275,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5275
5275
  await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "projectId", "TEXT");
5276
5276
  await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "toolMocks", "JSONB");
5277
5277
  await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "unmockedToolPolicy", "TEXT");
5278
+ await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "scorerIds", "JSONB");
5278
5279
  await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "externalId", "TEXT");
5279
5280
  await this.createDefaultIndexes();
5280
5281
  await this.createCustomIndexes();
@@ -5398,6 +5399,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5398
5399
  expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
5399
5400
  toolMocks: row.toolMocks ? safelyParseJSON(row.toolMocks) : void 0,
5400
5401
  unmockedToolPolicy: row.unmockedToolPolicy ?? void 0,
5402
+ scorerIds: row.scorerIds ? safelyParseJSON(row.scorerIds) : void 0,
5401
5403
  requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
5402
5404
  metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
5403
5405
  source: row.source ? safelyParseJSON(row.source) : void 0,
@@ -5420,6 +5422,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5420
5422
  expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
5421
5423
  toolMocks: row.toolMocks ? safelyParseJSON(row.toolMocks) : void 0,
5422
5424
  unmockedToolPolicy: row.unmockedToolPolicy ?? void 0,
5425
+ scorerIds: row.scorerIds ? safelyParseJSON(row.scorerIds) : void 0,
5423
5426
  requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
5424
5427
  metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
5425
5428
  source: row.source ? safelyParseJSON(row.source) : void 0,
@@ -5751,7 +5754,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5751
5754
  newVersion = row.version;
5752
5755
  parentOrganizationId = row.organizationId ?? null;
5753
5756
  parentProjectId = row.projectId ?? null;
5754
- await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
5757
+ await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","scorerIds","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19)`, [
5755
5758
  id,
5756
5759
  args.datasetId,
5757
5760
  newVersion,
@@ -5763,6 +5766,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5763
5766
  jsonbArg(args.expectedTrajectory),
5764
5767
  jsonbArg(args.toolMocks),
5765
5768
  args.unmockedToolPolicy ?? null,
5769
+ jsonbArg(args.scorerIds),
5766
5770
  jsonbArg(args.requestContext),
5767
5771
  jsonbArg(args.metadata),
5768
5772
  jsonbArg(args.source),
@@ -5790,6 +5794,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5790
5794
  expectedTrajectory: args.expectedTrajectory,
5791
5795
  toolMocks: args.toolMocks,
5792
5796
  unmockedToolPolicy: args.unmockedToolPolicy,
5797
+ scorerIds: args.scorerIds,
5793
5798
  requestContext: args.requestContext,
5794
5799
  metadata: args.metadata,
5795
5800
  source: args.source,
@@ -5844,6 +5849,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5844
5849
  const mergedExpectedTrajectory = args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory;
5845
5850
  const mergedToolMocks = args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks;
5846
5851
  const mergedUnmockedToolPolicy = args.unmockedToolPolicy !== void 0 ? args.unmockedToolPolicy : existing.unmockedToolPolicy;
5852
+ const mergedScorerIds = args.scorerIds !== void 0 ? args.scorerIds ?? void 0 : existing.scorerIds;
5847
5853
  const mergedRequestContext = args.requestContext !== void 0 ? args.requestContext : existing.requestContext;
5848
5854
  const mergedMetadata = args.metadata !== void 0 ? args.metadata : existing.metadata;
5849
5855
  const mergedSource = args.source !== void 0 ? args.source : existing.source;
@@ -5856,7 +5862,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5856
5862
  parentOrganizationId = row.organizationId ?? null;
5857
5863
  parentProjectId = row.projectId ?? null;
5858
5864
  await t.none(`UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`, [newVersion, args.id]);
5859
- await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
5865
+ await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","scorerIds","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19)`, [
5860
5866
  args.id,
5861
5867
  args.datasetId,
5862
5868
  newVersion,
@@ -5868,6 +5874,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5868
5874
  jsonbArg(mergedExpectedTrajectory),
5869
5875
  jsonbArg(mergedToolMocks),
5870
5876
  mergedUnmockedToolPolicy ?? null,
5877
+ jsonbArg(mergedScorerIds),
5871
5878
  jsonbArg(mergedRequestContext),
5872
5879
  jsonbArg(mergedMetadata),
5873
5880
  jsonbArg(mergedSource),
@@ -5894,6 +5901,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5894
5901
  expectedTrajectory: mergedExpectedTrajectory,
5895
5902
  toolMocks: mergedToolMocks,
5896
5903
  unmockedToolPolicy: mergedUnmockedToolPolicy,
5904
+ scorerIds: mergedScorerIds,
5897
5905
  requestContext: mergedRequestContext,
5898
5906
  metadata: mergedMetadata,
5899
5907
  source: mergedSource,
@@ -5942,7 +5950,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5942
5950
  const parentOrganizationId = row.organizationId ?? null;
5943
5951
  const parentProjectId = row.projectId ?? null;
5944
5952
  await t.none(`UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`, [newVersion, id]);
5945
- await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
5953
+ await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","scorerIds","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19)`, [
5946
5954
  id,
5947
5955
  datasetId,
5948
5956
  newVersion,
@@ -5954,6 +5962,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
5954
5962
  jsonbArg(existing.expectedTrajectory),
5955
5963
  jsonbArg(existing.toolMocks),
5956
5964
  existing.unmockedToolPolicy ?? null,
5965
+ jsonbArg(existing.scorerIds),
5957
5966
  jsonbArg(existing.requestContext),
5958
5967
  jsonbArg(existing.metadata),
5959
5968
  jsonbArg(existing.source),
@@ -6013,7 +6022,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
6013
6022
  const nowIso = now.toISOString();
6014
6023
  await t.none(`UPDATE ${datasetsTable} SET "version" = $2 WHERE "id" = $1`, [input.datasetId, newVersion]);
6015
6024
  for (const { id, item } of plan.inserts) {
6016
- await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
6025
+ await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","scorerIds","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,false,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19)`, [
6017
6026
  id,
6018
6027
  input.datasetId,
6019
6028
  newVersion,
@@ -6025,6 +6034,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
6025
6034
  jsonbArg(item.expectedTrajectory),
6026
6035
  jsonbArg(item.toolMocks),
6027
6036
  item.unmockedToolPolicy ?? null,
6037
+ jsonbArg(item.scorerIds),
6028
6038
  jsonbArg(item.requestContext),
6029
6039
  jsonbArg(item.metadata),
6030
6040
  jsonbArg(item.source),
@@ -6045,6 +6055,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
6045
6055
  expectedTrajectory: item.expectedTrajectory,
6046
6056
  toolMocks: item.toolMocks,
6047
6057
  unmockedToolPolicy: item.unmockedToolPolicy,
6058
+ scorerIds: item.scorerIds,
6048
6059
  requestContext: item.requestContext,
6049
6060
  metadata: item.metadata,
6050
6061
  source: item.source,
@@ -6107,7 +6118,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
6107
6118
  const newVersion = (await t.one(`UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version"`, [input.datasetId])).version;
6108
6119
  for (const item of currentItems) {
6109
6120
  await t.none(`UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`, [newVersion, item.id]);
6110
- await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18)`, [
6121
+ await t.none(`INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","externalId","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","unmockedToolPolicy","scorerIds","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,$6,NULL,true,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16,$17,$18,$19)`, [
6111
6122
  item.id,
6112
6123
  input.datasetId,
6113
6124
  newVersion,
@@ -6119,6 +6130,7 @@ var DatasetsPG = class DatasetsPG extends DatasetsStorage {
6119
6130
  jsonbArg(item.expectedTrajectory),
6120
6131
  jsonbArg(item.toolMocks),
6121
6132
  item.unmockedToolPolicy ?? null,
6133
+ jsonbArg(item.scorerIds),
6122
6134
  jsonbArg(item.requestContext),
6123
6135
  jsonbArg(item.metadata),
6124
6136
  jsonbArg(item.source),
@@ -14228,6 +14240,7 @@ async function listLogsDelta(client, table, filters, after, limit) {
14228
14240
  */
14229
14241
  function applyMetricFilters(acc, filters) {
14230
14242
  applyCommonFilters(acc, filters);
14243
+ applySingleOrArrayFilter(acc, "traceId", filters?.traceIds);
14231
14244
  applySingleOrArrayFilter(acc, "name", filters?.name);
14232
14245
  applySingleOrArrayFilter(acc, "provider", filters?.provider);
14233
14246
  applySingleOrArrayFilter(acc, "model", filters?.model);