@axiom-lattice/pg-stores 2.0.3 → 2.0.5

package/dist/index.mjs

@@ -3918,6 +3918,103 @@ var addEnvToEvalRuns = {
     `);
   }
 };
+var addInterruptColumnsToEval = {
+  version: 111,
+  name: "add_interrupt_columns_to_eval",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_run_results
+        ADD COLUMN IF NOT EXISTS interrupted BOOLEAN NOT NULL DEFAULT FALSE
+    `);
+    await client.query(`
+      ALTER TABLE lattice_eval_runs
+        ADD COLUMN IF NOT EXISTS interrupted_cases INTEGER NOT NULL DEFAULT 0
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_run_results DROP COLUMN IF EXISTS interrupted
+    `);
+    await client.query(`
+      ALTER TABLE lattice_eval_runs DROP COLUMN IF EXISTS interrupted_cases
+    `);
+  }
+};
+var addInterruptPolicyToEvalCases = {
+  version: 112,
+  name: "add_interrupt_policy_to_eval_cases",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_cases
+        ADD COLUMN IF NOT EXISTS interrupt_policy JSONB
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_cases DROP COLUMN IF EXISTS interrupt_policy
+    `);
+  }
+};
+var enforceUniqueEvalProjectName = {
+  version: 163,
+  name: "enforce_unique_eval_project_name",
+  up: async (client) => {
+    await client.query(`
+      UPDATE lattice_eval_suites s
+      SET project_id = (
+        SELECT p2.id FROM lattice_eval_projects p2
+        WHERE p2.tenant_id = p.tenant_id AND p2.name = p.name
+        ORDER BY p2.created_at DESC, p2.id DESC
+        LIMIT 1
+      )
+      FROM lattice_eval_projects p
+      WHERE s.project_id = p.id
+        AND s.project_id <> (
+          SELECT p2.id FROM lattice_eval_projects p2
+          WHERE p2.tenant_id = p.tenant_id AND p2.name = p.name
+          ORDER BY p2.created_at DESC, p2.id DESC
+          LIMIT 1
+        )
+    `);
+    await client.query(`
+      UPDATE lattice_eval_runs r
+      SET project_id = (
+        SELECT p2.id FROM lattice_eval_projects p2
+        WHERE p2.tenant_id = p.tenant_id AND p2.name = p.name
+        ORDER BY p2.created_at DESC, p2.id DESC
+        LIMIT 1
+      )
+      FROM lattice_eval_projects p
+      WHERE r.project_id = p.id
+        AND r.project_id <> (
+          SELECT p2.id FROM lattice_eval_projects p2
+          WHERE p2.tenant_id = p.tenant_id AND p2.name = p.name
+          ORDER BY p2.created_at DESC, p2.id DESC
+          LIMIT 1
+        )
+    `);
+    await client.query(`
+      DELETE FROM lattice_eval_projects p
+      USING lattice_eval_projects p2
+      WHERE p.tenant_id = p2.tenant_id
+        AND p.name = p2.name
+        AND (
+          p.created_at < p2.created_at
+          OR (p.created_at = p2.created_at AND p.id < p2.id)
+        )
+    `);
+    await client.query(`
+      ALTER TABLE lattice_eval_projects
+        ADD CONSTRAINT lattice_eval_projects_tenant_name_unique UNIQUE (tenant_id, name)
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_eval_projects
+        DROP CONSTRAINT IF EXISTS lattice_eval_projects_tenant_name_unique
+    `);
+  }
+};
 var evalMigrations = [
   createEvalProjectsTable,
   createEvalSuitesTable,
@@ -3925,7 +4022,10 @@ var evalMigrations = [
   createEvalRunsTable,
   createEvalRunResultsTable,
   addHoldoutToEvalRuns,
-  addEnvToEvalRuns
+  addEnvToEvalRuns,
+  addInterruptColumnsToEval,
+  addInterruptPolicyToEvalCases,
+  enforceUniqueEvalProjectName
 ];
 
 // src/stores/PostgreSQLEvalStore.ts
@@ -4032,6 +4132,7 @@ var PostgreSQLEvalStore = class {
       outputType: row.output_type || "message_content",
       contentAssertion: row.content_assertion || "",
       rubrics: this.parseOptionalJson(row.rubrics),
+      interruptPolicy: this.parseOptionalJson(row.interrupt_policy),
       createdAt: new Date(row.created_at),
       updatedAt: new Date(row.updated_at)
     };
@@ -4046,6 +4147,7 @@ var PostgreSQLEvalStore = class {
       totalCases: row.total_cases ?? 0,
       passedCases: row.passed_cases ?? 0,
       failedCases: row.failed_cases ?? 0,
+      interruptedCases: row.interrupted_cases ?? 0,
       avgScore: row.avg_score ?? 0,
       error: row.error,
       holdout: row.holdout,
@@ -4070,6 +4172,7 @@ var PostgreSQLEvalStore = class {
       messages: this.parseOptionalJson(row.messages),
       logs: this.parseOptionalJson(row.logs),
       error: row.error,
+      interrupted: row.interrupted ?? false,
       createdAt: new Date(row.created_at)
     };
   }
@@ -4289,7 +4392,7 @@ var PostgreSQLEvalStore = class {
     const { rows } = await this.pool.query(
       `SELECT id, tenant_id, suite_id,
               input_message, input_files, steps,
-              output_type, content_assertion, rubrics,
+              output_type, content_assertion, rubrics, interrupt_policy,
               created_at, updated_at
        FROM lattice_eval_cases
        WHERE tenant_id = $1 AND suite_id = $2
@@ -4304,7 +4407,7 @@ var PostgreSQLEvalStore = class {
     const { rows } = await this.pool.query(
       `SELECT id, tenant_id, suite_id,
               input_message, input_files, steps,
-              output_type, content_assertion, rubrics,
+              output_type, content_assertion, rubrics, interrupt_policy,
               created_at, updated_at
        FROM lattice_eval_cases
        WHERE id = $1 AND tenant_id = $2`,
@@ -4318,11 +4421,11 @@ var PostgreSQLEvalStore = class {
     const actualId = id || uuidv4();
     const { rows } = await this.pool.query(
       `INSERT INTO lattice_eval_cases
-        (id, tenant_id, suite_id, input_message, input_files, steps, output_type, content_assertion, rubrics)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9)
+        (id, tenant_id, suite_id, input_message, input_files, steps, output_type, content_assertion, rubrics, interrupt_policy)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
        RETURNING id, tenant_id, suite_id,
                  input_message, input_files, steps,
-                 output_type, content_assertion, rubrics,
+                 output_type, content_assertion, rubrics, interrupt_policy,
                  created_at, updated_at`,
       [
         actualId,
@@ -4333,7 +4436,8 @@ var PostgreSQLEvalStore = class {
         JSON.stringify(data.steps),
         data.outputType,
         data.contentAssertion || null,
-        JSON.stringify(data.rubrics || [])
+        JSON.stringify(data.rubrics || []),
+        data.interruptPolicy ? JSON.stringify(data.interruptPolicy) : null
       ]
     );
     return this.mapRowToCase(rows[0]);
@@ -4368,6 +4472,10 @@ var PostgreSQLEvalStore = class {
       set.push(`rubrics = $${i++}`);
       vals.push(JSON.stringify(updates.rubrics));
     }
+    if (updates.interruptPolicy !== void 0) {
+      set.push(`interrupt_policy = $${i++}`);
+      vals.push(updates.interruptPolicy ? JSON.stringify(updates.interruptPolicy) : null);
+    }
     if (set.length === 0) return this.getCaseById(tenantId, id);
     set.push(`updated_at = NOW()`);
     vals.push(id, tenantId);
@@ -4376,7 +4484,7 @@ var PostgreSQLEvalStore = class {
        WHERE id = $${i} AND tenant_id = $${i + 1}
        RETURNING id, tenant_id, suite_id,
                  input_message, input_files, steps,
-                 output_type, content_assertion, rubrics,
+                 output_type, content_assertion, rubrics, interrupt_policy,
                  created_at, updated_at`,
       vals
     );
@@ -4411,7 +4519,7 @@ var PostgreSQLEvalStore = class {
     const { rows } = await this.pool.query(
       `SELECT id, project_id, tenant_id,
               status, concurrency, total_cases,
-              passed_cases, failed_cases, avg_score,
+              passed_cases, failed_cases, interrupted_cases, avg_score,
               error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at
        FROM lattice_eval_runs
        WHERE ${conditions.join(" AND ")}
@@ -4426,7 +4534,7 @@ var PostgreSQLEvalStore = class {
     const { rows } = await this.pool.query(
       `SELECT id, project_id, tenant_id,
               status, concurrency, total_cases,
-              passed_cases, failed_cases, avg_score,
+              passed_cases, failed_cases, interrupted_cases, avg_score,
               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`,
@@ -4444,7 +4552,7 @@ var PostgreSQLEvalStore = class {
        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,
+                 passed_cases, failed_cases, interrupted_cases, avg_score,
                  error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at`,
       [
         actualId,
@@ -4482,6 +4590,10 @@ var PostgreSQLEvalStore = class {
       set.push(`failed_cases = $${i++}`);
       vals.push(updates.failedCases);
     }
+    if (updates.interruptedCases !== void 0) {
+      set.push(`interrupted_cases = $${i++}`);
+      vals.push(updates.interruptedCases);
+    }
     if (updates.avgScore !== void 0) {
       set.push(`avg_score = $${i++}`);
       vals.push(updates.avgScore);
@@ -4501,7 +4613,7 @@ var PostgreSQLEvalStore = class {
        WHERE id = $${i} AND tenant_id = $${i + 1}
        RETURNING id, project_id, tenant_id,
                  status, concurrency, total_cases,
-                 passed_cases, failed_cases, avg_score,
+                 passed_cases, failed_cases, interrupted_cases, avg_score,
                  error, holdout, env_project_id, env_workspace_id, created_at, started_at, completed_at`,
       vals
     );
@@ -4526,7 +4638,7 @@ var PostgreSQLEvalStore = class {
     const { rows } = await this.pool.query(
       `SELECT rr.id, rr.run_id, rr.suite_name, rr.case_id,
               rr.pass, rr.score, rr.summary, rr.dimension_results,
-              rr.duration_ms, rr.messages, rr.logs, rr.error, rr.created_at
+              rr.duration_ms, rr.messages, rr.logs, rr.error, rr.interrupted, rr.created_at
        FROM lattice_eval_run_results rr
        INNER JOIN lattice_eval_runs r ON r.id = rr.run_id
        WHERE r.tenant_id = $1 AND rr.run_id = $2
@@ -4541,11 +4653,11 @@ var PostgreSQLEvalStore = class {
     const actualId = id || uuidv4();
     const { rows } = await this.pool.query(
       `INSERT INTO lattice_eval_run_results
-        (id, run_id, suite_name, case_id, pass, score, summary, dimension_results, duration_ms, messages, logs, error)
-       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12)
+        (id, run_id, suite_name, case_id, pass, score, summary, dimension_results, duration_ms, messages, logs, error, interrupted)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13)
        RETURNING id, run_id, suite_name, case_id,
                  pass, score, summary, dimension_results,
-                 duration_ms, messages, logs, error, created_at`,
+                 duration_ms, messages, logs, error, interrupted, created_at`,
       [
         actualId,
         runId,
@@ -4558,7 +4670,8 @@ var PostgreSQLEvalStore = class {
         data.durationMs ?? null,
         JSON.stringify(data.messages || []),
         JSON.stringify(data.logs || []),
-        data.error || null
+        data.error || null,
+        data.interrupted ?? false
       ]
     );
     return this.mapRowToRunResult(rows[0]);
@@ -4612,6 +4725,10 @@ var PostgreSQLEvalStore = class {
       set.push(`error = $${i++}`);
       vals.push(updates.error);
     }
+    if (updates.interrupted !== void 0) {
+      set.push(`interrupted = $${i++}`);
+      vals.push(updates.interrupted);
+    }
     if (set.length === 0) return this.getRunResultById(tenantId, id);
     vals.push(id, tenantId);
     const { rows } = await this.pool.query(
@@ -4642,7 +4759,7 @@ var PostgreSQLEvalStore = class {
     const { rows } = await this.pool.query(
       `SELECT rr.id, rr.run_id, rr.suite_name, rr.case_id,
               rr.pass, rr.score, rr.summary, rr.dimension_results,
-              rr.duration_ms, rr.messages, rr.logs, rr.error, rr.created_at
+              rr.duration_ms, rr.messages, rr.logs, rr.error, rr.interrupted, rr.created_at
        FROM lattice_eval_run_results rr
        INNER JOIN lattice_eval_runs r ON r.id = rr.run_id
        WHERE rr.id = $1 AND r.tenant_id = $2`,
@@ -8462,6 +8579,8 @@ export {
   addEnvToEvalRuns,
   addFileContentType,
   addHoldoutToEvalRuns,
+  addInterruptColumnsToEval,
+  addInterruptPolicyToEvalCases,
   addProjectConfigColumn,
   addScheduleTenantId,
   addSkillTenantId,
@@ -8499,6 +8618,7 @@ export {
   createUsersTable,
   createWorkflowTrackingTables,
   createWorkspacesTable,
+  enforceUniqueEvalProjectName,
   evalMigrations,
   mapRowToRunStep,
   mapRowToWorkflowRun,