@axiom-lattice/pg-stores 2.0.2 → 2.0.4

package/dist/index.mjs

@@ -3816,6 +3816,9 @@ var createEvalRunsTable = {
         failed_cases INTEGER NOT NULL DEFAULT 0,
         avg_score FLOAT NOT NULL DEFAULT 0,
         error TEXT,
+        holdout BOOLEAN NOT NULL DEFAULT FALSE,
+        env_project_id TEXT,
+        env_workspace_id TEXT,
         created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
         started_at TIMESTAMPTZ,
         completed_at TIMESTAMPTZ
@@ -3882,12 +3885,47 @@ var createEvalRunResultsTable = {
     await client.query("DROP TABLE IF EXISTS lattice_eval_run_results");
   }
 };
+var addHoldoutToEvalRuns = {
+  version: 109,
+  name: "add_holdout_to_eval_runs",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_runs
+        ADD COLUMN IF NOT EXISTS holdout BOOLEAN NOT NULL DEFAULT FALSE
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_runs DROP COLUMN IF EXISTS holdout
+    `);
+  }
+};
+var addEnvToEvalRuns = {
+  version: 110,
+  name: "add_env_to_eval_runs",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_runs
+        ADD COLUMN IF NOT EXISTS env_project_id TEXT,
+        ADD COLUMN IF NOT EXISTS env_workspace_id TEXT
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_runs
+        DROP COLUMN IF EXISTS env_project_id,
+        DROP COLUMN IF EXISTS env_workspace_id
+    `);
+  }
+};
 var evalMigrations = [
   createEvalProjectsTable,
   createEvalSuitesTable,
   createEvalCasesTable,
   createEvalRunsTable,
-  createEvalRunResultsTable
+  createEvalRunResultsTable,
+  addHoldoutToEvalRuns,
+  addEnvToEvalRuns
 ];
 
 // src/stores/PostgreSQLEvalStore.ts
@@ -4010,6 +4048,9 @@ var PostgreSQLEvalStore = class {
       failedCases: row.failed_cases ?? 0,
       avgScore: row.avg_score ?? 0,
       error: row.error,
+      holdout: row.holdout,
+      envProjectId: row.env_project_id,
+      envWorkspaceId: row.env_workspace_id,
       createdAt: new Date(row.created_at),
       startedAt: row.started_at ? new Date(row.started_at) : void 0,
       completedAt: row.completed_at ? new Date(row.completed_at) : void 0
@@ -4371,7 +4412,7 @@ var PostgreSQLEvalStore = class {
       `SELECT id, project_id, tenant_id,
               status, concurrency, total_cases,
               passed_cases, failed_cases, avg_score,
-              error, created_at, started_at, completed_at
+              error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at
        FROM lattice_eval_runs
        WHERE ${conditions.join(" AND ")}
        ORDER BY created_at DESC`,
@@ -4386,7 +4427,7 @@ var PostgreSQLEvalStore = class {
       `SELECT id, project_id, tenant_id,
               status, concurrency, total_cases,
               passed_cases, failed_cases, avg_score,
-              error, created_at, started_at, completed_at
+              error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at
        FROM lattice_eval_runs
        WHERE id = $1 AND tenant_id = $2`,
       [id, tenantId]
@@ -4399,12 +4440,12 @@ var PostgreSQLEvalStore = class {
     const actualId = id || uuidv4();
     const { rows } = await this.pool.query(
       `INSERT INTO lattice_eval_runs
-        (id, project_id, tenant_id, status, concurrency, total_cases, passed_cases, failed_cases, avg_score, started_at)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
+        (id, project_id, tenant_id, status, concurrency, total_cases, passed_cases, failed_cases, avg_score, holdout, env_project_id, env_workspace_id, started_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)
        RETURNING id, project_id, tenant_id,
                  status, concurrency, total_cases,
                  passed_cases, failed_cases, avg_score,
-                 error, created_at, started_at, completed_at`,
+                 error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at`,
       [
         actualId,
         projectId,
@@ -4415,6 +4456,9 @@ var PostgreSQLEvalStore = class {
         0,
         0,
         0,
+        data.holdout ?? false,
+        data.envProjectId || null,
+        data.envWorkspaceId || null,
         /* @__PURE__ */ new Date()
       ]
     );
@@ -4458,7 +4502,7 @@ var PostgreSQLEvalStore = class {
        RETURNING id, project_id, tenant_id,
                  status, concurrency, total_cases,
                  passed_cases, failed_cases, avg_score,
-                 error, created_at, started_at, completed_at`,
+                 error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at`,
       vals
     );
     return rows.length ? this.mapRowToRun(rows[0]) : null;
@@ -8415,7 +8459,9 @@ export {
   PostgresSharedResourceStore,
   ThreadMessageQueueStore,
   addAssistantTenantId,
+  addEnvToEvalRuns,
   addFileContentType,
+  addHoldoutToEvalRuns,
   addProjectConfigColumn,
   addScheduleTenantId,
   addSkillTenantId,