@mastra/libsql 1.14.1-alpha.0 → 1.14.1

package/dist/index.js

@@ -3620,9 +3620,15 @@ var DatasetsLibSQL = class extends DatasetsStorage {
     await this.#addColumnIfNotExists(TABLE_DATASETS, "targetType", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASETS, "targetIds", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASETS, "scorerIds", "TEXT");
+    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", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "source", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "expectedTrajectory", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "organizationId", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "projectId", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "toolMocks", "TEXT");
     await this.#client.batch(
       [
@@ -3645,6 +3651,18 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         {
           sql: `CREATE UNIQUE INDEX IF NOT EXISTS idx_dataset_versions_dataset_version_unique ON "${TABLE_DATASET_VERSIONS}" ("datasetId", "version")`,
           args: []
+        },
+        {
+          sql: `CREATE INDEX IF NOT EXISTS idx_datasets_tenancy_createdat ON "${TABLE_DATASETS}" ("organizationId", "projectId", "createdAt", "id")`,
+          args: []
+        },
+        {
+          sql: `CREATE INDEX IF NOT EXISTS idx_datasets_tenancy_candidate ON "${TABLE_DATASETS}" ("organizationId", "projectId", "candidateKey", "candidateId")`,
+          args: []
+        },
+        {
+          sql: `CREATE INDEX IF NOT EXISTS idx_dataset_items_tenancy_list ON "${TABLE_DATASET_ITEMS}" ("organizationId", "projectId", "datasetId", "validTo", "isDeleted")`,
+          args: []
         }
       ],
       "write"
@@ -3676,6 +3694,10 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       targetIds: row.targetIds ? safelyParseJSON(row.targetIds) : void 0,
       scorerIds: row.scorerIds ? safelyParseJSON(row.scorerIds) : void 0,
       version: row.version,
+      organizationId: row.organizationId ?? null,
+      projectId: row.projectId ?? null,
+      candidateKey: row.candidateKey ?? null,
+      candidateId: row.candidateId ?? null,
       createdAt: ensureDate(row.createdAt),
       updatedAt: ensureDate(row.updatedAt)
     };
@@ -3685,6 +3707,8 @@ var DatasetsLibSQL = class 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,
@@ -3701,6 +3725,8 @@ var DatasetsLibSQL = class 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),
@@ -3742,6 +3768,10 @@ var DatasetsLibSQL = class extends DatasetsStorage {
           targetIds: input.targetIds ? JSON.stringify(input.targetIds) : null,
           scorerIds: input.scorerIds ? JSON.stringify(input.scorerIds) : null,
           version: 0,
+          organizationId: input.organizationId ?? null,
+          projectId: input.projectId ?? null,
+          candidateKey: input.candidateKey ?? null,
+          candidateId: input.candidateId ?? null,
           createdAt: nowIso,
           updatedAt: nowIso
         }
@@ -3758,6 +3788,10 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         targetIds: input.targetIds ?? void 0,
         scorerIds: input.scorerIds ?? void 0,
         version: 0,
+        organizationId: input.organizationId ?? null,
+        projectId: input.projectId ?? null,
+        candidateKey: input.candidateKey ?? null,
+        candidateId: input.candidateId ?? null,
         createdAt: now,
         updatedAt: now
       };
@@ -3913,9 +3947,28 @@ var DatasetsLibSQL = class extends DatasetsStorage {
   async listDatasets(args) {
     try {
       const { page, perPage: perPageInput } = args.pagination;
+      const filterConditions = [];
+      const filterParams = [];
+      if (args.filters?.organizationId !== void 0) {
+        filterConditions.push("organizationId = ?");
+        filterParams.push(args.filters.organizationId);
+      }
+      if (args.filters?.projectId !== void 0) {
+        filterConditions.push("projectId = ?");
+        filterParams.push(args.filters.projectId);
+      }
+      if (args.filters?.candidateKey !== void 0) {
+        filterConditions.push("candidateKey = ?");
+        filterParams.push(args.filters.candidateKey);
+      }
+      if (args.filters?.candidateId !== void 0) {
+        filterConditions.push("candidateId = ?");
+        filterParams.push(args.filters.candidateId);
+      }
+      const whereClause = filterConditions.length > 0 ? `WHERE ${filterConditions.join(" AND ")}` : "";
       const countResult = await this.#client.execute({
-        sql: `SELECT COUNT(*) as count FROM ${TABLE_DATASETS}`,
-        args: []
+        sql: `SELECT COUNT(*) as count FROM ${TABLE_DATASETS} ${whereClause}`,
+        args: filterParams
       });
       const total = Number(countResult.rows?.[0]?.count ?? 0);
       if (total === 0) {
@@ -3929,8 +3982,8 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       const limitValue = perPageInput === false ? total : perPage;
       const end = perPageInput === false ? total : start + perPage;
       const result = await this.#client.execute({
-        sql: `SELECT ${buildSelectColumns(TABLE_DATASETS)} FROM ${TABLE_DATASETS} ORDER BY createdAt DESC, id ASC LIMIT ? OFFSET ?`,
-        args: [limitValue, start]
+        sql: `SELECT ${buildSelectColumns(TABLE_DATASETS)} FROM ${TABLE_DATASETS} ${whereClause} ORDER BY createdAt DESC, id ASC LIMIT ? OFFSET ?`,
+        args: [...filterParams, limitValue, start]
       });
       return {
         datasets: result.rows?.map((row) => this.transformDatasetRow(row)) ?? [],
@@ -3955,6 +4008,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
   // --- SCD-2 item mutations ---
   async _doAddItem(args) {
     try {
+      const dataset = await this.getDatasetById({ id: args.datasetId });
       const id = crypto.randomUUID();
       const versionId = crypto.randomUUID();
       const now = /* @__PURE__ */ new Date();
@@ -3966,11 +4020,13 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             args: [args.datasetId]
           },
           {
-            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, organizationId, projectId, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), (SELECT organizationId FROM ${TABLE_DATASETS} WHERE id = ?), (SELECT projectId FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               id,
               args.datasetId,
               args.datasetId,
+              args.datasetId,
+              args.datasetId,
               jsonbArg(args.input),
               jsonbArg(args.groundTruth),
               jsonbArg(args.expectedTrajectory),
@@ -3994,6 +4050,8 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         id,
         datasetId: args.datasetId,
         datasetVersion: newVersion,
+        organizationId: dataset?.organizationId ?? null,
+        projectId: dataset?.projectId ?? null,
         input: args.input,
         groundTruth: args.groundTruth,
         expectedTrajectory: args.expectedTrajectory,
@@ -4035,6 +4093,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
           details: { itemId: args.id, expectedDatasetId: args.datasetId, actualDatasetId: existing.datasetId }
         });
       }
+      const dataset = await this.getDatasetById({ id: args.datasetId });
       const versionId = crypto.randomUUID();
       const now = /* @__PURE__ */ new Date();
       const nowIso = now.toISOString();
@@ -4056,11 +4115,13 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             args: [args.datasetId, args.id]
           },
           {
-            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, organizationId, projectId, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), (SELECT organizationId FROM ${TABLE_DATASETS} WHERE id = ?), (SELECT projectId FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               args.id,
               args.datasetId,
               args.datasetId,
+              args.datasetId,
+              args.datasetId,
               jsonbArg(mergedInput),
               jsonbArg(mergedGroundTruth),
               jsonbArg(mergedExpectedTrajectory),
@@ -4083,6 +4144,8 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       return {
         ...existing,
         datasetVersion: newVersion,
+        organizationId: dataset?.organizationId ?? null,
+        projectId: dataset?.projectId ?? null,
         input: mergedInput,
         groundTruth: mergedGroundTruth,
         expectedTrajectory: mergedExpectedTrajectory,
@@ -4129,13 +4192,17 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             args: [datasetId, id]
           },
           {
-            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, organizationId, projectId, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), (SELECT organizationId FROM ${TABLE_DATASETS} WHERE id = ?), (SELECT projectId FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               id,
               datasetId,
               datasetId,
+              datasetId,
+              datasetId,
               jsonbArg(existing.input),
               jsonbArg(existing.groundTruth),
+              jsonbArg(existing.expectedTrajectory),
+              jsonbArg(existing.toolMocks),
               jsonbArg(existing.requestContext),
               jsonbArg(existing.metadata),
               jsonbArg(existing.source),
@@ -4236,6 +4303,14 @@ var DatasetsLibSQL = class extends DatasetsStorage {
           "isDeleted = 0"
         ];
         const queryParams2 = [args.datasetId, args.version, args.version];
+        if (args.filters?.organizationId !== void 0) {
+          conditions2.push("organizationId = ?");
+          queryParams2.push(args.filters.organizationId);
+        }
+        if (args.filters?.projectId !== void 0) {
+          conditions2.push("projectId = ?");
+          queryParams2.push(args.filters.projectId);
+        }
         if (args.search) {
           conditions2.push(`(LOWER(json(input)) LIKE ? OR LOWER(COALESCE(json(groundTruth), '')) LIKE ?)`);
           const searchPattern = `%${args.search.toLowerCase()}%`;
@@ -4273,6 +4348,14 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       }
       const conditions = ["datasetId = ?", "validTo IS NULL", "isDeleted = 0"];
       const queryParams = [args.datasetId];
+      if (args.filters?.organizationId !== void 0) {
+        conditions.push("organizationId = ?");
+        queryParams.push(args.filters.organizationId);
+      }
+      if (args.filters?.projectId !== void 0) {
+        conditions.push("projectId = ?");
+        queryParams.push(args.filters.projectId);
+      }
       if (args.search) {
         conditions.push(`(LOWER(json(input)) LIKE ? OR LOWER(COALESCE(json(groundTruth), '')) LIKE ?)`);
         const searchPattern = `%${args.search.toLowerCase()}%`;
@@ -4418,11 +4501,13 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         const id = crypto.randomUUID();
         items.push({ id, input: itemInput });
         statements.push({
-          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, organizationId, projectId, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), ?, ?, NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
           args: [
             id,
             input.datasetId,
             input.datasetId,
+            dataset.organizationId ?? null,
+            dataset.projectId ?? null,
             jsonbArg(itemInput.input),
             jsonbArg(itemInput.groundTruth),
             jsonbArg(itemInput.expectedTrajectory),
@@ -4445,6 +4530,8 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         id,
         datasetId: input.datasetId,
         datasetVersion: newVersion,
+        organizationId: dataset.organizationId ?? null,
+        projectId: dataset.projectId ?? null,
         input: itemInput.input,
         groundTruth: itemInput.groundTruth,
         expectedTrajectory: itemInput.expectedTrajectory,
@@ -4500,13 +4587,17 @@ var DatasetsLibSQL = class extends DatasetsStorage {
           args: [input.datasetId, item.id]
         });
         statements.push({
-          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, organizationId, projectId, validTo, isDeleted, input, groundTruth, expectedTrajectory, toolMocks, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), ?, ?, NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
           args: [
             item.id,
             input.datasetId,
             input.datasetId,
+            dataset.organizationId ?? null,
+            dataset.projectId ?? null,
             jsonbArg(item.input),
             jsonbArg(item.groundTruth),
+            jsonbArg(item.expectedTrajectory),
+            jsonbArg(item.toolMocks),
             jsonbArg(item.requestContext),
             jsonbArg(item.metadata),
             jsonbArg(item.source),
@@ -4551,12 +4642,12 @@ var ExperimentsLibSQL = class 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", "toolMockReport"]
+      ifNotExists: ["status", "tags", "toolMockReport", "organizationId", "projectId"]
     });
     await this.#client.batch(
       [
@@ -4571,6 +4662,15 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
         {
           sql: `CREATE UNIQUE INDEX IF NOT EXISTS idx_experiment_results_exp_item ON "${TABLE_EXPERIMENT_RESULTS}" ("experimentId", "itemId")`,
           args: []
+        },
+        // Tenancy: leading-tenant indexes for multi-tenant scans (parity with datasets domain).
+        {
+          sql: `CREATE INDEX IF NOT EXISTS idx_experiments_org_project ON "${TABLE_EXPERIMENTS}" ("organizationId", "projectId")`,
+          args: []
+        },
+        {
+          sql: `CREATE INDEX IF NOT EXISTS idx_experiment_results_org_project ON "${TABLE_EXPERIMENT_RESULTS}" ("organizationId", "projectId")`,
+          args: []
         }
       ],
       "write"
@@ -4587,6 +4687,8 @@ var ExperimentsLibSQL = class 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,
       name: row.name ?? void 0,
@@ -4610,6 +4712,8 @@ var ExperimentsLibSQL = class 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,
@@ -4637,6 +4741,8 @@ var ExperimentsLibSQL = class 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,
           name: input.name ?? null,
@@ -4658,6 +4764,8 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
         datasetId: input.datasetId,
         datasetVersion: input.datasetVersion,
         agentVersion: input.agentVersion ?? null,
+        organizationId: input.organizationId ?? null,
+        projectId: input.projectId ?? null,
         targetType: input.targetType,
         targetId: input.targetId,
         name: input.name,
@@ -4800,6 +4908,17 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
         conditions.push("status = ?");
         queryParams.push(args.status);
       }
+      if (args.filters) {
+        const { organizationId, projectId } = args.filters;
+        if (organizationId !== void 0) {
+          conditions.push("organizationId = ?");
+          queryParams.push(organizationId);
+        }
+        if (projectId !== void 0) {
+          conditions.push("projectId = ?");
+          queryParams.push(projectId);
+        }
+      }
       const whereClause = conditions.length > 0 ? `WHERE ${conditions.join(" AND ")}` : "";
       const countResult = await this.#client.execute({
         sql: `SELECT COUNT(*) as count FROM ${TABLE_EXPERIMENTS} ${whereClause}`,
@@ -4874,6 +4993,8 @@ var ExperimentsLibSQL = class 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,
           groundTruth: input.groundTruth,
@@ -4893,6 +5014,8 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
         experimentId: input.experimentId,
         itemId: input.itemId,
         itemDatasetVersion: input.itemDatasetVersion,
+        organizationId: input.organizationId ?? null,
+        projectId: input.projectId ?? null,
         input: input.input,
         output: input.output,
         groundTruth: input.groundTruth,
@@ -5012,6 +5135,17 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
         conditions.push("status = ?");
         queryParams.push(args.status);
       }
+      if (args.filters) {
+        const { organizationId, projectId } = args.filters;
+        if (organizationId !== void 0) {
+          conditions.push("organizationId = ?");
+          queryParams.push(organizationId);
+        }
+        if (projectId !== void 0) {
+          conditions.push("projectId = ?");
+          queryParams.push(projectId);
+        }
+      }
       const whereClause = `WHERE ${conditions.join(" AND ")}`;
       const countResult = await this.#client.execute({
         sql: `SELECT COUNT(*) as count FROM ${TABLE_EXPERIMENT_RESULTS} ${whereClause}`,