@axiom-lattice/local-stores 3.0.2 → 3.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/local-stores",
-  "version": "3.0.2",
+  "version": "3.1.0",
   "description": "Local SQLite-based stores for Axiom Lattice framework",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -25,8 +25,8 @@
     "@types/sql.js": "^1.4.11",
     "sql.js": "^1.14.1",
     "uuid": "^14.0.1",
-    "@axiom-lattice/core": "4.1.0",
-    "@axiom-lattice/protocols": "4.0.1"
+    "@axiom-lattice/core": "4.2.0",
+    "@axiom-lattice/protocols": "4.1.0"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/__tests__/LocalEvalStore.test.ts

@@ -8,7 +8,7 @@
 import { describe, expect, it, beforeEach, afterEach } from "@jest/globals";
 import { tmpdir } from "os";
 import path from "path";
-import { initDatabase, closeDatabase, ensureTable } from "../database";
+import { initDatabase, closeDatabase, ensureTable, type DatabaseWrapper } from "../database";
 import { LocalEvalStore, DDL } from "../stores/LocalEvalStore";
 
 // sql.js persists to disk via DatabaseWrapper.save(), so each test needs a
@@ -21,9 +21,10 @@ const nextDbPath = () => {
 
 describe("LocalEvalStore project name uniqueness", () => {
   let store: LocalEvalStore;
+  let db: DatabaseWrapper;
 
   beforeEach(async () => {
-    const db = await initDatabase({ dbPath: nextDbPath() });
+    db = await initDatabase({ dbPath: nextDbPath() });
     store = new LocalEvalStore(db);
   });
 
@@ -93,4 +94,42 @@ describe("LocalEvalStore project name uniqueness", () => {
     const runs = db.prepare(`SELECT project_id FROM lt_eval_runs WHERE id = 'run-1'`).get() as { project_id: string };
     expect(runs.project_id).toBe("dup-new");
   });
+
+  it("deletes project suites, cases, runs, and results without affecting another project", async () => {
+    await store.createProject("tenant-1", "project-delete", {
+      name: "delete-me", judgeModelConfig: {}, targetServerConfig: {},
+    });
+    await store.createProject("tenant-1", "project-keep", {
+      name: "keep-me", judgeModelConfig: {}, targetServerConfig: {},
+    });
+    await store.createSuite("tenant-1", "project-delete", "suite-delete", { name: "delete-suite" });
+    await store.createSuite("tenant-1", "project-keep", "suite-keep", { name: "keep-suite" });
+    await store.createCase("tenant-1", "suite-delete", "case-delete", {
+      inputMessage: "delete", steps: [], outputType: "message_content", contentAssertion: "delete",
+    });
+    await store.createCase("tenant-1", "suite-keep", "case-keep", {
+      inputMessage: "keep", steps: [], outputType: "message_content", contentAssertion: "keep",
+    });
+    await store.createRun("tenant-1", "project-delete", "run-delete", { concurrency: 1, totalCases: 1 });
+    await store.createRun("tenant-1", "project-keep", "run-keep", { concurrency: 1, totalCases: 1 });
+    await store.createRunResult("tenant-1", "run-delete", "result-delete", {
+      suiteName: "delete-suite", caseId: "case-delete", pass: true, score: 100,
+    });
+    await store.createRunResult("tenant-1", "run-keep", "result-keep", {
+      suiteName: "keep-suite", caseId: "case-keep", pass: true, score: 100,
+    });
+
+    await expect(store.deleteProject("tenant-1", "project-delete")).resolves.toBe(true);
+
+    for (const table of ["lt_eval_projects", "lt_eval_suites", "lt_eval_cases", "lt_eval_runs", "lt_eval_run_results"]) {
+      const deleted = db.prepare(`SELECT COUNT(*) AS count FROM ${table} WHERE id LIKE '%delete'`).get() as { count: number };
+      const kept = db.prepare(`SELECT COUNT(*) AS count FROM ${table} WHERE id LIKE '%keep'`).get() as { count: number };
+      expect(deleted.count).toBe(0);
+      expect(kept.count).toBe(1);
+    }
+  });
+
+  it("returns false when deleting a project that does not exist", async () => {
+    await expect(store.deleteProject("tenant-1", "missing-project")).resolves.toBe(false);
+  });
 });

package/src/stores/LocalEvalStore.ts

@@ -264,8 +264,34 @@ export class LocalEvalStore implements EvalStore {
   }
 
   async deleteProject(tenantId: string, id: string): Promise<boolean> {
-    const result = this.db.prepare(`DELETE FROM lt_eval_projects WHERE tenant_id = ? AND id = ?`).run(tenantId, id);
-    return result.changes > 0;
+    const db = this.db.getRawDb();
+    db.run("BEGIN TRANSACTION");
+    try {
+      db.run(
+        `DELETE FROM lt_eval_run_results
+         WHERE run_id IN (
+           SELECT id FROM lt_eval_runs WHERE tenant_id = ? AND project_id = ?
+         )`,
+        [tenantId, id]
+      );
+      db.run(
+        `DELETE FROM lt_eval_cases
+         WHERE tenant_id = ? AND suite_id IN (
+           SELECT id FROM lt_eval_suites WHERE tenant_id = ? AND project_id = ?
+         )`,
+        [tenantId, tenantId, id]
+      );
+      db.run(`DELETE FROM lt_eval_runs WHERE tenant_id = ? AND project_id = ?`, [tenantId, id]);
+      db.run(`DELETE FROM lt_eval_suites WHERE tenant_id = ? AND project_id = ?`, [tenantId, id]);
+      db.run(`DELETE FROM lt_eval_projects WHERE tenant_id = ? AND id = ?`, [tenantId, id]);
+      const deleted = db.getRowsModified() > 0;
+      db.run("COMMIT");
+      this.db.save();
+      return deleted;
+    } catch (error) {
+      db.run("ROLLBACK");
+      throw error;
+    }
   }
 
   // -------------------------------------------------------------------------