akm-cli 0.7.1 → 0.7.3

package/dist/indexer/db.js

@@ -9,7 +9,7 @@ import { cosineSimilarity } from "../llm/embedders/types";
 import { buildSearchFields } from "./search-fields";
 import { ensureUsageEventsSchema } from "./usage-events";
 // ── Constants ───────────────────────────────────────────────────────────────
-export const DB_VERSION = 9;
+export const DB_VERSION = 10;
 export const EMBEDDING_DIM = 384;
 // ── Database lifecycle ──────────────────────────────────────────────────────
 export function openDatabase(dbPath, options) {
@@ -29,6 +29,17 @@ export function openDatabase(dbPath, options) {
     warnIfVecMissing(db, { once: true });
     return db;
 }
+export function openExistingDatabase(dbPath) {
+    const resolvedPath = dbPath ?? getDbPath();
+    const db = new Database(resolvedPath);
+    db.exec("PRAGMA journal_mode = WAL");
+    db.exec("PRAGMA busy_timeout = 5000");
+    db.exec("PRAGMA foreign_keys = ON");
+    // Existing-DB callers must not mutate schema or embedding metadata on open,
+    // but some paths still need write access to usage_events and other tables.
+    loadVecExtension(db);
+    return db;
+}
 export function closeDatabase(db) {
     db.close();
 }
@@ -163,6 +174,15 @@ function ensureSchema(db, embeddingDim) {
       updated_at   TEXT NOT NULL DEFAULT (datetime('now')),
       FOREIGN KEY (entry_id) REFERENCES entries(id) ON DELETE CASCADE
     );
+  `);
+    db.exec(`
+    CREATE TABLE IF NOT EXISTS index_dir_state (
+      dir_path          TEXT PRIMARY KEY,
+      file_set_hash     TEXT NOT NULL,
+      file_mtime_max_ms REAL NOT NULL,
+      reason            TEXT NOT NULL,
+      updated_at        TEXT NOT NULL
+    );
   `);
     // FTS-dirty queue. Created here (not lazily on first upsert) so the
     // per-entry write path doesn't issue a CREATE TABLE IF NOT EXISTS on
@@ -261,6 +281,7 @@ function handleVersionUpgrade(db) {
     db.exec("DROP TABLE IF EXISTS embeddings");
     db.exec("DROP TABLE IF EXISTS entries_vec");
     db.exec("DROP TABLE IF EXISTS entries_fts");
+    db.exec("DROP TABLE IF EXISTS index_dir_state");
     db.exec("DROP INDEX IF EXISTS idx_entries_dir");
     db.exec("DROP INDEX IF EXISTS idx_entries_type");
     db.exec("DROP TABLE IF EXISTS entries");
@@ -332,6 +353,35 @@ export function getMeta(db, key) {
 export function setMeta(db, key, value) {
     db.prepare("INSERT OR REPLACE INTO index_meta (key, value) VALUES (?, ?)").run(key, value);
 }
+export function getIndexDirState(db, dirPath) {
+    const row = db
+        .prepare("SELECT dir_path, file_set_hash, file_mtime_max_ms, reason, updated_at FROM index_dir_state WHERE dir_path = ?")
+        .get(dirPath);
+    if (!row)
+        return undefined;
+    return {
+        dirPath: row.dir_path,
+        fileSetHash: row.file_set_hash,
+        fileMtimeMaxMs: row.file_mtime_max_ms,
+        reason: row.reason,
+        updatedAt: row.updated_at,
+    };
+}
+export function upsertIndexDirState(db, state) {
+    db.prepare(`INSERT INTO index_dir_state (dir_path, file_set_hash, file_mtime_max_ms, reason, updated_at)
+     VALUES (?, ?, ?, ?, ?)
+     ON CONFLICT(dir_path) DO UPDATE SET
+       file_set_hash = excluded.file_set_hash,
+       file_mtime_max_ms = excluded.file_mtime_max_ms,
+       reason = excluded.reason,
+       updated_at = excluded.updated_at`).run(state.dirPath, state.fileSetHash, state.fileMtimeMaxMs, state.reason, new Date().toISOString());
+}
+export function deleteIndexDirState(db, dirPath) {
+    db.prepare("DELETE FROM index_dir_state WHERE dir_path = ?").run(dirPath);
+}
+export function deleteIndexDirStatesByStashDir(db, stashDir) {
+    db.prepare("DELETE FROM index_dir_state WHERE dir_path = ? OR dir_path LIKE ?").run(stashDir, `${stashDir}${path.sep}%`);
+}
 // ── Entry operations ────────────────────────────────────────────────────────
 /**
  * Insert or update an entry in the `entries` table. Returns the row id.

package/dist/indexer/graph-extraction.js

@@ -70,7 +70,7 @@ const EMPTY_RESULT = {
  * to an empty no-op result, leaving any existing `graph.json` untouched on
  * disk.
  */
-export async function runGraphExtractionPass(config, sources) {
+export async function runGraphExtractionPass(config, sources, signal) {
     // Gate 1 — locked feature flag (§14). Defaults to enabled; only an
     // explicit `false` disables the pass entirely.
     if (config.llm?.features?.graph_extraction === false)
@@ -94,7 +94,9 @@ export async function runGraphExtractionPass(config, sources) {
     let totalEntities = 0;
     let totalRelations = 0;
     for (const candidate of eligible) {
-        const extraction = await extractGraphFromBody(llmConfig, candidate.body);
+        if (signal?.aborted)
+            break;
+        const extraction = await extractGraphFromBody(llmConfig, candidate.body, signal);
         if (extraction.entities.length === 0)
             continue;
         nodes.push({
@@ -134,7 +136,7 @@ export async function runGraphExtractionPass(config, sources) {
  * same one the rest of the indexer uses: `<stashRoot>/<type>/...`.
  *
  * Inferred-child memories (frontmatter `inferred: true`) are skipped — they
- * are atomic facts already, with no internal graph structure worth
+ * are already derived summaries, with no additional internal graph structure worth
  * extracting.
  *
  * Exported for direct unit testing.