@mastra/pg 1.14.0 → 1.14.1-alpha.1

package/dist/index.js

@@ -5181,9 +5181,16 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
     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_DATASETS, "organizationId", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASETS, "projectId", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASETS, "candidateKey", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASETS, "candidateId", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "requestContext", "JSONB");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "source", "JSONB");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "expectedTrajectory", "JSONB");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "organizationId", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "projectId", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "toolMocks", "JSONB");
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
   }
@@ -5217,6 +5224,22 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         table: TABLE_DATASET_VERSIONS,
         columns: ["datasetId", "version"],
         unique: true
+      },
+      // Tenancy: leading-tenant indexes for multi-tenant scans (parity with observability storage).
+      {
+        name: "idx_datasets_org_project",
+        table: TABLE_DATASETS,
+        columns: ["organizationId", "projectId"]
+      },
+      {
+        name: "idx_datasets_candidate",
+        table: TABLE_DATASETS,
+        columns: ["candidateKey", "candidateId"]
+      },
+      {
+        name: "idx_dataset_items_org_project",
+        table: TABLE_DATASET_ITEMS,
+        columns: ["organizationId", "projectId"]
       }
     ];
   }
@@ -5254,6 +5277,10 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       targetType: row.targetType || null,
       targetIds: row.targetIds || null,
       scorerIds: row.scorerIds || null,
+      organizationId: row.organizationId ?? null,
+      projectId: row.projectId ?? null,
+      candidateKey: row.candidateKey ?? null,
+      candidateId: row.candidateId ?? null,
       version: row.version,
       createdAt: ensureDate(row.createdAtZ || row.createdAt),
       updatedAt: ensureDate(row.updatedAtZ || row.updatedAt)
@@ -5264,9 +5291,12 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       id: row.id,
       datasetId: row.datasetId,
       datasetVersion: row.datasetVersion,
+      organizationId: row.organizationId ?? null,
+      projectId: row.projectId ?? null,
       input: safelyParseJSON(row.input),
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
       expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
+      toolMocks: row.toolMocks ? safelyParseJSON(row.toolMocks) : 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,
@@ -5279,11 +5309,14 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       id: row.id,
       datasetId: row.datasetId,
       datasetVersion: row.datasetVersion,
+      organizationId: row.organizationId ?? null,
+      projectId: row.projectId ?? null,
       validTo: row.validTo,
       isDeleted: Boolean(row.isDeleted),
       input: safelyParseJSON(row.input),
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
       expectedTrajectory: row.expectedTrajectory ? safelyParseJSON(row.expectedTrajectory) : void 0,
+      toolMocks: row.toolMocks ? safelyParseJSON(row.toolMocks) : 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,
@@ -5318,6 +5351,10 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
           targetType: input.targetType ?? null,
           targetIds: input.targetIds !== void 0 ? JSON.stringify(input.targetIds) : null,
           scorerIds: input.scorerIds ? JSON.stringify(input.scorerIds) : null,
+          organizationId: input.organizationId ?? null,
+          projectId: input.projectId ?? null,
+          candidateKey: input.candidateKey ?? null,
+          candidateId: input.candidateId ?? null,
           version: 0,
           createdAt: nowIso,
           updatedAt: nowIso
@@ -5334,6 +5371,10 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         targetType: input.targetType ?? null,
         targetIds: input.targetIds ?? null,
         scorerIds: input.scorerIds ?? null,
+        organizationId: input.organizationId ?? null,
+        projectId: input.projectId ?? null,
+        candidateKey: input.candidateKey ?? null,
+        candidateId: input.candidateId ?? null,
         version: 0,
         createdAt: now,
         updatedAt: now
@@ -5438,6 +5479,10 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         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,
+        organizationId: existing.organizationId ?? null,
+        projectId: existing.projectId ?? null,
+        candidateKey: existing.candidateKey ?? null,
+        candidateId: existing.candidateId ?? null,
         updatedAt: new Date(now)
       };
     } catch (error) {
@@ -5500,7 +5545,33 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
     try {
       const { page, perPage: perPageInput } = args.pagination;
       const tableName = getTableName2({ indexName: TABLE_DATASETS, schemaName: getSchemaName2(this.#schema) });
-      const countResult = await this.#db.client.one(`SELECT COUNT(*) as count FROM ${tableName}`);
+      const conditions = [];
+      const queryParams = [];
+      let paramIndex = 1;
+      if (args.filters) {
+        const { organizationId, projectId, candidateKey, candidateId } = args.filters;
+        if (organizationId !== void 0) {
+          conditions.push(`"organizationId" = $${paramIndex++}`);
+          queryParams.push(organizationId);
+        }
+        if (projectId !== void 0) {
+          conditions.push(`"projectId" = $${paramIndex++}`);
+          queryParams.push(projectId);
+        }
+        if (candidateKey !== void 0) {
+          conditions.push(`"candidateKey" = $${paramIndex++}`);
+          queryParams.push(candidateKey);
+        }
+        if (candidateId !== void 0) {
+          conditions.push(`"candidateId" = $${paramIndex++}`);
+          queryParams.push(candidateId);
+        }
+      }
+      const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
+      const countResult = await this.#db.client.one(
+        `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`,
+        queryParams
+      );
       const total = parseInt(countResult.count, 10);
       if (total === 0) {
         return { datasets: [], pagination: { total: 0, page, perPage: perPageInput, hasMore: false } };
@@ -5509,8 +5580,8 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       const { offset, perPage: perPageForResponse } = calculatePagination(page, perPageInput, perPage);
       const limitValue = perPageInput === false ? total : perPage;
       const rows = await this.#db.client.manyOrNone(
-        `SELECT * FROM ${tableName} ORDER BY "createdAt" DESC, "id" ASC LIMIT $1 OFFSET $2`,
-        [limitValue, offset]
+        `SELECT * FROM ${tableName} ${whereClause} ORDER BY "createdAt" DESC, "id" ASC LIMIT $${paramIndex} OFFSET $${paramIndex + 1}`,
+        [...queryParams, limitValue, offset]
       );
       return {
         datasets: (rows || []).map((row) => this.transformDatasetRow(row)),
@@ -5546,21 +5617,28 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       const now = /* @__PURE__ */ new Date();
       const nowIso = now.toISOString();
       let newVersion;
+      let parentOrganizationId = null;
+      let parentProjectId = null;
       await this.#db.client.tx(async (t) => {
         const row = await t.one(
-          `UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version"`,
+          `UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version", "organizationId", "projectId"`,
           [args.datasetId]
         );
         newVersion = row.version;
+        parentOrganizationId = row.organizationId ?? null;
+        parentProjectId = row.projectId ?? null;
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,NULL,false,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16)`,
           [
             id,
             args.datasetId,
             newVersion,
+            parentOrganizationId,
+            parentProjectId,
             JSON.stringify(args.input),
             jsonbArg(args.groundTruth),
             jsonbArg(args.expectedTrajectory),
+            jsonbArg(args.toolMocks),
             jsonbArg(args.requestContext),
             jsonbArg(args.metadata),
             jsonbArg(args.source),
@@ -5579,9 +5657,12 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         id,
         datasetId: args.datasetId,
         datasetVersion: newVersion,
+        organizationId: parentOrganizationId,
+        projectId: parentProjectId,
         input: args.input,
         groundTruth: args.groundTruth,
         expectedTrajectory: args.expectedTrajectory,
+        toolMocks: args.toolMocks,
         requestContext: args.requestContext,
         metadata: args.metadata,
         source: args.source,
@@ -5631,29 +5712,37 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       const mergedInput = args.input !== void 0 ? args.input : existing.input;
       const mergedGroundTruth = args.groundTruth !== void 0 ? args.groundTruth : existing.groundTruth;
       const mergedExpectedTrajectory = args.expectedTrajectory !== void 0 ? args.expectedTrajectory : existing.expectedTrajectory;
+      const mergedToolMocks = args.toolMocks !== void 0 ? args.toolMocks : existing.toolMocks;
       const mergedRequestContext = args.requestContext !== void 0 ? args.requestContext : existing.requestContext;
       const mergedMetadata = args.metadata !== void 0 ? args.metadata : existing.metadata;
       const mergedSource = args.source !== void 0 ? args.source : existing.source;
       let newVersion;
+      let parentOrganizationId = null;
+      let parentProjectId = null;
       await this.#db.client.tx(async (t) => {
         const row = await t.one(
-          `UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version"`,
+          `UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version", "organizationId", "projectId"`,
           [args.datasetId]
         );
         newVersion = row.version;
+        parentOrganizationId = row.organizationId ?? null;
+        parentProjectId = row.projectId ?? null;
         await t.none(
           `UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`,
           [newVersion, args.id]
         );
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,NULL,false,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16)`,
           [
             args.id,
             args.datasetId,
             newVersion,
+            parentOrganizationId,
+            parentProjectId,
             JSON.stringify(mergedInput),
             jsonbArg(mergedGroundTruth),
             jsonbArg(mergedExpectedTrajectory),
+            jsonbArg(mergedToolMocks),
             jsonbArg(mergedRequestContext),
             jsonbArg(mergedMetadata),
             jsonbArg(mergedSource),
@@ -5671,9 +5760,12 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       return {
         ...existing,
         datasetVersion: newVersion,
+        organizationId: parentOrganizationId,
+        projectId: parentProjectId,
         input: mergedInput,
         groundTruth: mergedGroundTruth,
         expectedTrajectory: mergedExpectedTrajectory,
+        toolMocks: mergedToolMocks,
         requestContext: mergedRequestContext,
         metadata: mergedMetadata,
         source: mergedSource,
@@ -5713,23 +5805,28 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       const nowIso = (/* @__PURE__ */ new Date()).toISOString();
       await this.#db.client.tx(async (t) => {
         const row = await t.one(
-          `UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version"`,
+          `UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version", "organizationId", "projectId"`,
           [datasetId]
         );
         const newVersion = row.version;
+        const parentOrganizationId = row.organizationId ?? null;
+        const parentProjectId = row.projectId ?? null;
         await t.none(
           `UPDATE ${itemsTable} SET "validTo" = $1 WHERE "id" = $2 AND "validTo" IS NULL AND "isDeleted" = false`,
           [newVersion, id]
         );
         await t.none(
-          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
+          `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,NULL,true,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16)`,
           [
             id,
             datasetId,
             newVersion,
+            parentOrganizationId,
+            parentProjectId,
             JSON.stringify(existing.input),
             jsonbArg(existing.groundTruth),
             jsonbArg(existing.expectedTrajectory),
+            jsonbArg(existing.toolMocks),
             jsonbArg(existing.requestContext),
             jsonbArg(existing.metadata),
             jsonbArg(existing.source),
@@ -5777,6 +5874,8 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       const nowIso = now.toISOString();
       const versionId = crypto.randomUUID();
       const itemsWithIds = input.items.map((itemInput) => ({ id: crypto.randomUUID(), input: itemInput }));
+      const parentOrganizationId = dataset.organizationId ?? null;
+      const parentProjectId = dataset.projectId ?? null;
       let newVersion;
       await this.#db.client.tx(async (t) => {
         const row = await t.one(
@@ -5786,14 +5885,17 @@ 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","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,false,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
+            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,NULL,false,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16)`,
             [
               id,
               input.datasetId,
               newVersion,
+              parentOrganizationId,
+              parentProjectId,
               JSON.stringify(itemInput.input),
               jsonbArg(itemInput.groundTruth),
               jsonbArg(itemInput.expectedTrajectory),
+              jsonbArg(itemInput.toolMocks),
               jsonbArg(itemInput.requestContext),
               jsonbArg(itemInput.metadata),
               jsonbArg(itemInput.source),
@@ -5813,9 +5915,12 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         id,
         datasetId: input.datasetId,
         datasetVersion: newVersion,
+        organizationId: parentOrganizationId,
+        projectId: parentProjectId,
         input: itemInput.input,
         groundTruth: itemInput.groundTruth,
         expectedTrajectory: itemInput.expectedTrajectory,
+        toolMocks: itemInput.toolMocks,
         requestContext: itemInput.requestContext,
         metadata: itemInput.metadata,
         source: itemInput.source,
@@ -5861,6 +5966,8 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
       });
       const nowIso = (/* @__PURE__ */ new Date()).toISOString();
       const versionId = crypto.randomUUID();
+      const parentOrganizationId = dataset.organizationId ?? null;
+      const parentProjectId = dataset.projectId ?? null;
       await this.#db.client.tx(async (t) => {
         const row = await t.one(
           `UPDATE ${datasetsTable} SET "version" = "version" + 1 WHERE "id" = $1 RETURNING "version"`,
@@ -5873,14 +5980,17 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
             [newVersion, item.id]
           );
           await t.none(
-            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","validTo","isDeleted","input","groundTruth","expectedTrajectory","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,NULL,true,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13)`,
+            `INSERT INTO ${itemsTable} ("id","datasetId","datasetVersion","organizationId","projectId","validTo","isDeleted","input","groundTruth","expectedTrajectory","toolMocks","requestContext","metadata","source","createdAt","createdAtZ","updatedAt","updatedAtZ") VALUES ($1,$2,$3,$4,$5,NULL,true,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,$16)`,
             [
               item.id,
               input.datasetId,
               newVersion,
+              parentOrganizationId,
+              parentProjectId,
               JSON.stringify(item.input),
               jsonbArg(item.groundTruth),
               jsonbArg(item.expectedTrajectory),
+              jsonbArg(item.toolMocks),
               jsonbArg(item.requestContext),
               jsonbArg(item.metadata),
               jsonbArg(item.source),
@@ -5998,6 +6108,17 @@ var DatasetsPG = class _DatasetsPG extends DatasetsStorage {
         queryParams.push(`%${args.search}%`);
         paramIndex++;
       }
+      if (args.filters) {
+        const { organizationId, projectId } = args.filters;
+        if (organizationId !== void 0) {
+          conditions.push(`"organizationId" = $${paramIndex++}`);
+          queryParams.push(organizationId);
+        }
+        if (projectId !== void 0) {
+          conditions.push(`"projectId" = $${paramIndex++}`);
+          queryParams.push(projectId);
+        }
+      }
       const whereClause = `WHERE ${conditions.join(" AND ")}`;
       const countResult = await this.#db.client.one(
         `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`,
@@ -6136,12 +6257,12 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
     await this.#db.alterTable({
       tableName: TABLE_EXPERIMENTS,
       schema: EXPERIMENTS_SCHEMA,
-      ifNotExists: ["agentVersion"]
+      ifNotExists: ["agentVersion", "organizationId", "projectId"]
     });
     await this.#db.alterTable({
       tableName: TABLE_EXPERIMENT_RESULTS,
       schema: EXPERIMENT_RESULTS_SCHEMA,
-      ifNotExists: ["status", "tags"]
+      ifNotExists: ["status", "tags", "toolMockReport", "organizationId", "projectId"]
     });
     await this.createDefaultIndexes();
     await this.createCustomIndexes();
@@ -6155,6 +6276,17 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
         table: TABLE_EXPERIMENT_RESULTS,
         columns: ["experimentId", "itemId"],
         unique: true
+      },
+      // Tenancy: leading-tenant indexes for multi-tenant scans (parity with datasets domain).
+      {
+        name: "idx_experiments_org_project",
+        table: TABLE_EXPERIMENTS,
+        columns: ["organizationId", "projectId"]
+      },
+      {
+        name: "idx_experiment_results_org_project",
+        table: TABLE_EXPERIMENT_RESULTS,
+        columns: ["organizationId", "projectId"]
       }
     ];
   }
@@ -6188,6 +6320,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
       datasetId: row.datasetId ?? null,
       datasetVersion: row.datasetVersion != null ? row.datasetVersion : null,
       agentVersion: row.agentVersion ?? null,
+      organizationId: row.organizationId ?? null,
+      projectId: row.projectId ?? null,
       targetType: row.targetType,
       targetId: row.targetId,
       status: row.status,
@@ -6207,6 +6341,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
       experimentId: row.experimentId,
       itemId: row.itemId,
       itemDatasetVersion: row.itemDatasetVersion != null ? row.itemDatasetVersion : null,
+      organizationId: row.organizationId ?? null,
+      projectId: row.projectId ?? null,
       input: safelyParseJSON(row.input),
       output: row.output ? safelyParseJSON(row.output) : null,
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : null,
@@ -6217,6 +6353,7 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
       traceId: row.traceId ?? null,
       status: row.status ?? null,
       tags: row.tags ? safelyParseJSON(row.tags) : null,
+      toolMockReport: row.toolMockReport ? safelyParseJSON(row.toolMockReport) : null,
       createdAt: ensureDate(row.createdAtZ || row.createdAt)
     };
   }
@@ -6236,6 +6373,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
           datasetId: input.datasetId ?? null,
           datasetVersion: input.datasetVersion ?? null,
           agentVersion: input.agentVersion ?? null,
+          organizationId: input.organizationId ?? null,
+          projectId: input.projectId ?? null,
           targetType: input.targetType,
           targetId: input.targetId,
           status: "pending",
@@ -6257,6 +6396,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
         datasetId: input.datasetId ?? null,
         datasetVersion: input.datasetVersion ?? null,
         agentVersion: input.agentVersion ?? null,
+        organizationId: input.organizationId ?? null,
+        projectId: input.projectId ?? null,
         targetType: input.targetType,
         targetId: input.targetId,
         status: "pending",
@@ -6398,6 +6539,17 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
         conditions.push(`"status" = $${paramIndex++}`);
         queryParams.push(args.status);
       }
+      if (args.filters) {
+        const { organizationId, projectId } = args.filters;
+        if (organizationId !== void 0) {
+          conditions.push(`"organizationId" = $${paramIndex++}`);
+          queryParams.push(organizationId);
+        }
+        if (projectId !== void 0) {
+          conditions.push(`"projectId" = $${paramIndex++}`);
+          queryParams.push(projectId);
+        }
+      }
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
       const countResult = await this.#db.client.one(
         `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`,
@@ -6467,6 +6619,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
           experimentId: input.experimentId,
           itemId: input.itemId,
           itemDatasetVersion: input.itemDatasetVersion ?? null,
+          organizationId: input.organizationId ?? null,
+          projectId: input.projectId ?? null,
           input: input.input,
           output: input.output ?? null,
           groundTruth: input.groundTruth ?? null,
@@ -6477,6 +6631,7 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
           traceId: input.traceId ?? null,
           status: input.status ?? null,
           tags: input.tags ?? null,
+          toolMockReport: input.toolMockReport ?? null,
           createdAt: nowIso
         }
       });
@@ -6485,6 +6640,8 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
         experimentId: input.experimentId,
         itemId: input.itemId,
         itemDatasetVersion: input.itemDatasetVersion ?? null,
+        organizationId: input.organizationId ?? null,
+        projectId: input.projectId ?? null,
         input: input.input,
         output: input.output ?? null,
         groundTruth: input.groundTruth ?? null,
@@ -6495,6 +6652,7 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
         traceId: input.traceId ?? null,
         status: input.status ?? null,
         tags: input.tags ?? null,
+        toolMockReport: input.toolMockReport ?? null,
         createdAt: now
       };
     } catch (error) {
@@ -6597,6 +6755,17 @@ var ExperimentsPG = class _ExperimentsPG extends ExperimentsStorage {
         conditions.push(`"status" = $${paramIndex++}`);
         queryParams.push(args.status);
       }
+      if (args.filters) {
+        const { organizationId, projectId } = args.filters;
+        if (organizationId !== void 0) {
+          conditions.push(`"organizationId" = $${paramIndex++}`);
+          queryParams.push(organizationId);
+        }
+        if (projectId !== void 0) {
+          conditions.push(`"projectId" = $${paramIndex++}`);
+          queryParams.push(projectId);
+        }
+      }
       const whereClause = `WHERE ${conditions.join(" AND ")}`;
       const countResult = await this.#db.client.one(
         `SELECT COUNT(*) as count FROM ${tableName} ${whereClause}`,
@@ -8284,12 +8453,7 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
     await this.#db.createTable({ tableName: TABLE_THREADS, schema: TABLE_SCHEMAS[TABLE_THREADS] });
     await this.#db.createTable({ tableName: TABLE_MESSAGES, schema: TABLE_SCHEMAS[TABLE_MESSAGES] });
     await this.#db.createTable({ tableName: TABLE_RESOURCES, schema: TABLE_SCHEMAS[TABLE_RESOURCES] });
-    let omSchema;
-    try {
-      const { OBSERVATIONAL_MEMORY_TABLE_SCHEMA } = await import('@mastra/core/storage');
-      omSchema = OBSERVATIONAL_MEMORY_TABLE_SCHEMA?.[OM_TABLE];
-    } catch {
-    }
+    const omSchema = _omTableSchema?.[OM_TABLE];
     if (omSchema) {
       await this.#db.createTable({
         tableName: OM_TABLE,
@@ -19147,6 +19311,9 @@ var PostgresStore = class extends MastraCompositeStore {
   #poolClosed = false;
   schema;
   isInitialized = false;
+  // Caches the in-flight init() so concurrent callers share one initialization
+  // instead of each acquiring + pinning a client. See init() / issue #18282.
+  #initPromise = null;
   stores;
   constructor(config) {
     try {
@@ -19213,16 +19380,25 @@ var PostgresStore = class extends MastraCompositeStore {
     throw new Error("PostgresStore: invalid config");
   }
   async init() {
+    if (this.disableInit || process.env.MASTRA_DISABLE_STORAGE_INIT === "true") {
+      return;
+    }
     if (this.isInitialized) {
       return;
     }
-    const pinnedClient = await this.#pool.connect();
-    const pinned = new PinnedClientAdapter(this.#pool, pinnedClient);
+    this.#initPromise ??= this.#runPinnedInit();
+    await this.#initPromise;
+  }
+  async #runPinnedInit() {
+    let pinnedClient;
     try {
+      pinnedClient = await this.#pool.connect();
+      const pinned = new PinnedClientAdapter(this.#pool, pinnedClient);
       this.#db.pin(pinned);
       await super.init();
       this.isInitialized = true;
     } catch (error) {
+      this.#initPromise = null;
       if (error instanceof MastraError) {
         throw error;
       }
@@ -19235,8 +19411,10 @@ var PostgresStore = class extends MastraCompositeStore {
         error
       );
     } finally {
-      this.#db.unpin();
-      pinnedClient.release();
+      if (pinnedClient) {
+        this.#db.unpin();
+        pinnedClient.release();
+      }
     }
   }
   /**