@axiom-lattice/local-stores 2.0.4 → 2.0.6

package/dist/index.mjs

@@ -1920,6 +1920,7 @@ CREATE TABLE IF NOT EXISTS lt_eval_cases (
   output_type TEXT NOT NULL DEFAULT 'message_content',
   content_assertion TEXT NOT NULL DEFAULT '',
   rubrics TEXT,
+  interrupt_policy TEXT,
   created_at TEXT NOT NULL,
   updated_at TEXT NOT NULL,
   PRIMARY KEY (tenant_id, id)
@@ -1935,6 +1936,7 @@ CREATE TABLE IF NOT EXISTS lt_eval_runs (
   total_cases INTEGER NOT NULL DEFAULT 0,
   passed_cases INTEGER NOT NULL DEFAULT 0,
   failed_cases INTEGER NOT NULL DEFAULT 0,
+  interrupted_cases INTEGER NOT NULL DEFAULT 0,
   avg_score REAL NOT NULL DEFAULT 0,
   error TEXT,
   holdout INTEGER NOT NULL DEFAULT 0,
@@ -1960,6 +1962,7 @@ CREATE TABLE IF NOT EXISTS lt_eval_run_results (
   messages TEXT DEFAULT '[]',
   logs TEXT DEFAULT '[]',
   error TEXT,
+  interrupted INTEGER NOT NULL DEFAULT 0,
   created_at TEXT NOT NULL,
   PRIMARY KEY (run_id, id)
 );
@@ -1990,6 +1993,76 @@ var LocalEvalStore = class {
   constructor(db) {
     this.db = db;
     ensureTable(db, DDL13);
+    this.ensureColumn("lt_eval_runs", "interrupted_cases", "interrupted_cases INTEGER NOT NULL DEFAULT 0");
+    this.ensureColumn("lt_eval_run_results", "interrupted", "interrupted INTEGER NOT NULL DEFAULT 0");
+    this.ensureColumn("lt_eval_cases", "interrupt_policy", "interrupt_policy TEXT");
+    this.db.exec(`
+      UPDATE lt_eval_suites
+      SET project_id = (
+        SELECT p2.id FROM lt_eval_projects p2
+        WHERE p2.tenant_id = (
+          SELECT p.tenant_id FROM lt_eval_projects p WHERE p.id = lt_eval_suites.project_id
+        )
+        AND p2.name = (
+          SELECT p.name FROM lt_eval_projects p WHERE p.id = lt_eval_suites.project_id
+        )
+        ORDER BY p2.created_at DESC, p2.rowid DESC
+        LIMIT 1
+      )
+      WHERE project_id IN (
+        SELECT p3.id FROM lt_eval_projects p3
+        WHERE p3.id <> (
+          SELECT p2.id FROM lt_eval_projects p2
+          WHERE p2.tenant_id = p3.tenant_id AND p2.name = p3.name
+          ORDER BY p2.created_at DESC, p2.rowid DESC
+          LIMIT 1
+        )
+      )
+    `);
+    this.db.exec(`
+      UPDATE lt_eval_runs
+      SET project_id = (
+        SELECT p2.id FROM lt_eval_projects p2
+        WHERE p2.tenant_id = (
+          SELECT p.tenant_id FROM lt_eval_projects p WHERE p.id = lt_eval_runs.project_id
+        )
+        AND p2.name = (
+          SELECT p.name FROM lt_eval_projects p WHERE p.id = lt_eval_runs.project_id
+        )
+        ORDER BY p2.created_at DESC, p2.rowid DESC
+        LIMIT 1
+      )
+      WHERE project_id IN (
+        SELECT p3.id FROM lt_eval_projects p3
+        WHERE p3.id <> (
+          SELECT p2.id FROM lt_eval_projects p2
+          WHERE p2.tenant_id = p3.tenant_id AND p2.name = p3.name
+          ORDER BY p2.created_at DESC, p2.rowid DESC
+          LIMIT 1
+        )
+      )
+    `);
+    this.db.exec(`
+      DELETE FROM lt_eval_projects
+      WHERE id <> (
+        SELECT p2.id FROM lt_eval_projects p2
+        WHERE p2.tenant_id = lt_eval_projects.tenant_id
+          AND p2.name = lt_eval_projects.name
+        ORDER BY p2.created_at DESC, p2.rowid DESC
+        LIMIT 1
+      )
+    `);
+    this.db.exec(`
+      CREATE UNIQUE INDEX IF NOT EXISTS idx_lt_eval_projects_tenant_name
+      ON lt_eval_projects(tenant_id, name)
+    `);
+  }
+  /** Add a column if it does not exist (SQLite version compatible). */
+  ensureColumn(table, column, ddl) {
+    const cols = this.db.prepare(`PRAGMA table_info(${table})`).all();
+    if (!cols.some((c) => c.name === column)) {
+      this.db.exec(`ALTER TABLE ${table} ADD COLUMN ${ddl}`);
+    }
   }
   // -------------------------------------------------------------------------
   // Projects
@@ -2134,8 +2207,8 @@ var LocalEvalStore = class {
   async createCase(tenantId, suiteId, id, data) {
     const actualId = id || randomUUID();
     this.db.prepare(
-      `INSERT INTO lt_eval_cases (id, tenant_id, suite_id, input_message, input_files, steps, output_type, content_assertion, rubrics, created_at, updated_at)
-       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
+      `INSERT INTO lt_eval_cases (id, tenant_id, suite_id, input_message, input_files, steps, output_type, content_assertion, rubrics, interrupt_policy, created_at, updated_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
     ).run(
       actualId,
       tenantId,
@@ -2146,6 +2219,7 @@ var LocalEvalStore = class {
       data.outputType || "message_content",
       data.contentAssertion || "",
       data.rubrics ? JSON.stringify(data.rubrics) : null,
+      data.interruptPolicy ? JSON.stringify(data.interruptPolicy) : null,
       nowISO(),
       nowISO()
     );
@@ -2180,6 +2254,10 @@ var LocalEvalStore = class {
       set.push("rubrics = ?");
       vals.push(JSON.stringify(updates.rubrics));
     }
+    if (updates.interruptPolicy !== void 0) {
+      set.push("interrupt_policy = ?");
+      vals.push(updates.interruptPolicy ? JSON.stringify(updates.interruptPolicy) : null);
+    }
     set.push("updated_at = ?");
     vals.push(nowISO());
     vals.push(tenantId, id);
@@ -2241,6 +2319,10 @@ var LocalEvalStore = class {
       set.push("failed_cases = ?");
       vals.push(updates.failedCases);
     }
+    if (updates.interruptedCases !== void 0) {
+      set.push("interrupted_cases = ?");
+      vals.push(updates.interruptedCases);
+    }
     if (updates.avgScore !== void 0) {
       set.push("avg_score = ?");
       vals.push(updates.avgScore);
@@ -2285,8 +2367,8 @@ var LocalEvalStore = class {
   async createRunResult(tenantId, runId, id, data) {
     const actualId = id || randomUUID();
     this.db.prepare(
-      `INSERT INTO lt_eval_run_results (id, run_id, suite_name, case_id, pass, score, summary, dimension_results, duration_ms, messages, logs, error, created_at)
-       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
+      `INSERT INTO lt_eval_run_results (id, run_id, suite_name, case_id, pass, score, summary, dimension_results, duration_ms, messages, logs, error, interrupted, created_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`
     ).run(
       actualId,
       runId,
@@ -2300,6 +2382,7 @@ var LocalEvalStore = class {
       JSON.stringify(data.messages || []),
       JSON.stringify(data.logs || []),
       data.error || null,
+      data.interrupted ? 1 : 0,
       nowISO()
     );
     return await this.getRunResultById(tenantId, actualId);
@@ -2348,6 +2431,10 @@ var LocalEvalStore = class {
       set.push("error = ?");
       vals.push(updates.error || null);
     }
+    if (updates.interrupted !== void 0) {
+      set.push("interrupted = ?");
+      vals.push(updates.interrupted ? 1 : 0);
+    }
     if (set.length === 0) return this.getRunResultById(tenantId, id);
     vals.push(id, tenantId);
     this.db.prepare(
@@ -2415,6 +2502,7 @@ var LocalEvalStore = class {
       outputType: row.output_type || "message_content",
       contentAssertion: row.content_assertion || "",
       rubrics: parseOptionalJson(row.rubrics),
+      interruptPolicy: parseOptionalJson(row.interrupt_policy),
       createdAt: parseISO(row.created_at),
       updatedAt: parseISO(row.updated_at)
     };
@@ -2429,6 +2517,7 @@ var LocalEvalStore = 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 ? Boolean(row.holdout) : void 0,
@@ -2453,6 +2542,7 @@ var LocalEvalStore = class {
       messages: parseOptionalJson(row.messages),
       logs: parseOptionalJson(row.logs),
       error: row.error,
+      interrupted: row.interrupted === 1,
       createdAt: parseISO(row.created_at)
     };
   }