akm-cli 0.9.0-beta.52 → 0.9.0-beta.54

package/dist/scripts/migrate-storage.js

@@ -468,21 +468,11 @@ function scanKeys(text) {
   }
   return keys;
 }
-function scanComments(text) {
-  const comments = [];
-  for (const line of text.split(/\r?\n/)) {
-    const trimmed = line.trimStart();
-    if (trimmed.startsWith("#")) {
-      comments.push(trimmed.slice(1).trimStart());
-    }
-  }
-  return comments;
-}
 function listKeys(envPath) {
   if (!fs.existsSync(envPath))
-    return { keys: [], comments: [] };
+    return { keys: [] };
   const text = fs.readFileSync(envPath, "utf8");
-  return { keys: scanKeys(text), comments: scanComments(text) };
+  return { keys: scanKeys(text) };
 }
 var import_dotenv, ASSIGN_RE;
 var init_env = __esm(() => {
@@ -7605,23 +7595,35 @@ function appendToLogFile(level, args) {
   }
 }
 function warn(...args) {
+  if (sinkOverride) {
+    sinkOverride("warn", args);
+    return;
+  }
   appendToLogFile("WARN", args);
   if (!quiet) {
     console.warn(...args);
   }
 }
 function error(...args) {
+  if (sinkOverride) {
+    sinkOverride("error", args);
+    return;
+  }
   appendToLogFile("ERROR", args);
   if (!quiet) {
     console.error(...args);
   }
 }
 function warnVerbose(...args) {
+  if (sinkOverride) {
+    sinkOverride("warnVerbose", args);
+    return;
+  }
   if (isVerbose()) {
     warn(...args);
   }
 }
-var quiet = false, verbose = false, logFilePath;
+var quiet = false, verbose = false, logFilePath, sinkOverride;
 var init_warn = () => {};
 
 // src/indexer/walk/file-context.ts
@@ -8325,12 +8327,7 @@ function applyScriptMetadata(entry, ctx) {
   }
 }
 function applyEnvMetadata(entry, ctx) {
-  const { keys, comments } = listKeys(ctx.absPath);
-  if (comments.length > 0 && !entry.description) {
-    entry.description = comments.join(" ").slice(0, 500);
-    entry.source = "comments";
-    entry.confidence = 0.7;
-  }
+  const { keys } = listKeys(ctx.absPath);
   if (keys.length > 0) {
     entry.searchHints = keys;
   }
@@ -9401,6 +9398,44 @@ var init_migrations = __esm(() => {
       up: `
       ALTER TABLE asset_salience ADD COLUMN encoding_source TEXT DEFAULT NULL;
     `
+    },
+    {
+      id: "016-collapse-churn-detector",
+      up: `
+      CREATE TABLE IF NOT EXISTS canary_queries (
+        id            INTEGER PRIMARY KEY AUTOINCREMENT,
+        canary_set_id TEXT    NOT NULL,
+        anchor_ref    TEXT    NOT NULL,
+        query         TEXT    NOT NULL,
+        source        TEXT    NOT NULL DEFAULT 'auto',
+        active        INTEGER NOT NULL DEFAULT 1,
+        created_at    TEXT    NOT NULL
+      );
+      CREATE INDEX IF NOT EXISTS idx_canary_queries_active
+        ON canary_queries(active, canary_set_id);
+
+      CREATE TABLE IF NOT EXISTS improve_cycle_metrics (
+        id                      INTEGER PRIMARY KEY AUTOINCREMENT,
+        run_id                  TEXT    NOT NULL,
+        ts                      TEXT    NOT NULL,
+        pass                    TEXT    NOT NULL,
+        canary_set_id           TEXT    NOT NULL,
+        mean_recall             REAL    NOT NULL,
+        mean_ndcg               REAL    NOT NULL,
+        mean_mrr                REAL    NOT NULL,
+        canary_ranks_json       TEXT    NOT NULL,
+        store_total             INTEGER NOT NULL,
+        store_by_type_json      TEXT    NOT NULL,
+        distinct_content_ratio  REAL    NOT NULL,
+        mean_bigram_diversity   REAL    NOT NULL,
+        over_generation_count   INTEGER NOT NULL,
+        accepted_actions        INTEGER NOT NULL,
+        merge_floor_violations  INTEGER NOT NULL DEFAULT 0,
+        alerts_json             TEXT    NOT NULL DEFAULT '[]'
+      );
+      CREATE INDEX IF NOT EXISTS idx_improve_cycle_metrics_ts
+        ON improve_cycle_metrics(ts);
+    `
     }
   ];
 });
@@ -9423,10 +9458,12 @@ __export(exports_state_db, {
   recordFsProposalsImport: () => recordFsProposalsImport,
   readStateEvents: () => readStateEvents,
   queryTaskHistory: () => queryTaskHistory,
+  queryRecentCycleMetrics: () => queryRecentCycleMetrics,
   queryImproveRuns: () => queryImproveRuns,
   queryCompletedTaskIntervals: () => queryCompletedTaskIntervals,
   purgeOldImproveRuns: () => purgeOldImproveRuns,
   purgeOldEvents: () => purgeOldEvents,
+  purgeOldCycleMetrics: () => purgeOldCycleMetrics,
   proposalToRowValues: () => proposalToRowValues,
   proposalRowToProposal: () => proposalRowToProposal,
   persistPhaseThreshold: () => persistPhaseThreshold,
@@ -9436,8 +9473,11 @@ __export(exports_state_db, {
   listStateProposalIdsByPrefix: () => listStateProposalIdsByPrefix,
   listProposalGateDecisions: () => listProposalGateDecisions,
   listExistingTableNames: () => listExistingTableNames,
+  listActiveCanarySetIds: () => listActiveCanarySetIds,
   insertProposalIfAbsent: () => insertProposalIfAbsent,
   insertEvent: () => insertEvent,
+  insertCycleMetrics: () => insertCycleMetrics,
+  insertCanaries: () => insertCanaries,
   importEventsJsonl: () => importEventsJsonl,
   hasImportedFsProposals: () => hasImportedFsProposals,
   getTaskHistoryRuns: () => getTaskHistoryRuns,
@@ -9446,15 +9486,19 @@ __export(exports_state_db, {
   getStateDbPath: () => getStateDbPath,
   getRecombineHypothesis: () => getRecombineHypothesis,
   getPhaseThreshold: () => getPhaseThreshold,
+  getLatestCycleMetrics: () => getLatestCycleMetrics,
   getLastExtractRunAt: () => getLastExtractRunAt,
   getExtractedSessionsMap: () => getExtractedSessionsMap,
   getExtractedSession: () => getExtractedSession,
   getConsolidationJudgedMap: () => getConsolidationJudgedMap,
+  getCanariesBySetId: () => getCanariesBySetId,
   getBodyEmbeddings: () => getBodyEmbeddings,
+  getActiveCanaries: () => getActiveCanaries,
   findMatchingRecombineHypothesis: () => findMatchingRecombineHypothesis,
   eventRowToEnvelope: () => eventRowToEnvelope,
   embeddingToBlob: () => embeddingToBlob,
   decayUnseenRecombineHypotheses: () => decayUnseenRecombineHypotheses,
+  deactivateCanarySet: () => deactivateCanarySet,
   computeImproveRunMetrics: () => computeImproveRunMetrics,
   blobToEmbedding: () => blobToEmbedding
 });
@@ -9686,7 +9730,7 @@ function insertProposalIfAbsent(db, proposal, stashDir) {
 }
 function isRetryableBeginError(err) {
   const msg = (err instanceof Error ? err.message : String(err)).toLowerCase();
-  return msg.includes("within a transaction") || msg.includes("database is locked") || msg.includes("database table is locked") || msg.includes("did not open a transaction");
+  return msg.includes("database is locked") || msg.includes("database table is locked") || msg.includes("did not open a transaction");
 }
 function sleepSyncMs(ms) {
   if (ms <= 0)
@@ -9694,6 +9738,9 @@ function sleepSyncMs(ms) {
   Atomics.wait(new Int32Array(new SharedArrayBuffer(4)), 0, 0, ms);
 }
 function withImmediateTransaction(db, fn) {
+  if (db.inTransaction) {
+    return fn();
+  }
   let lastBeginErr;
   for (let attempt = 1;attempt <= WITH_IMMEDIATE_TX_MAX_ATTEMPTS; attempt++) {
     try {
@@ -9704,9 +9751,11 @@ function withImmediateTransaction(db, fn) {
     } catch (err) {
       lastBeginErr = err;
       if (isRetryableBeginError(err) && attempt < WITH_IMMEDIATE_TX_MAX_ATTEMPTS) {
-        try {
-          db.exec("ROLLBACK");
-        } catch {}
+        if (db.inTransaction) {
+          try {
+            db.exec("ROLLBACK");
+          } catch {}
+        }
         sleepSyncMs(2 ** (attempt - 1));
         continue;
       }
@@ -9714,12 +9763,17 @@ function withImmediateTransaction(db, fn) {
     }
     try {
       const result = fn();
+      if (!db.inTransaction) {
+        throw new Error("withImmediateTransaction invariant violated: transaction opened by BEGIN IMMEDIATE was no longer active after the transaction body ran; refusing to COMMIT (writes may have escaped serialization)");
+      }
       db.exec("COMMIT");
       return result;
     } catch (err) {
-      try {
-        db.exec("ROLLBACK");
-      } catch {}
+      if (db.inTransaction) {
+        try {
+          db.exec("ROLLBACK");
+        } catch {}
+      }
       throw err;
     }
   }
@@ -10091,6 +10145,75 @@ function upsertBodyEmbeddings(db, entries) {
     }
   })();
 }
+function insertCanaries(db, canarySetId, canaries, now) {
+  if (canaries.length === 0)
+    return;
+  const ts = now ?? new Date().toISOString();
+  const stmt = db.prepare(`
+    INSERT INTO canary_queries (canary_set_id, anchor_ref, query, source, active, created_at)
+    VALUES (?, ?, ?, ?, 1, ?)
+  `);
+  db.transaction(() => {
+    for (const c of canaries) {
+      stmt.run(canarySetId, c.anchorRef, c.query, c.source ?? "auto", ts);
+    }
+  })();
+}
+function getActiveCanaries(db) {
+  return db.prepare(`SELECT * FROM canary_queries
+       WHERE active = 1 AND canary_set_id = (
+         SELECT canary_set_id FROM canary_queries WHERE active = 1
+         ORDER BY created_at DESC, id DESC LIMIT 1
+       )
+       ORDER BY id`).all();
+}
+function getCanariesBySetId(db, canarySetId) {
+  return db.prepare(`SELECT * FROM canary_queries WHERE canary_set_id = ? ORDER BY id`).all(canarySetId);
+}
+function listActiveCanarySetIds(db) {
+  const rows = db.prepare(`SELECT DISTINCT canary_set_id FROM canary_queries WHERE active = 1`).all();
+  return rows.map((r) => r.canary_set_id);
+}
+function deactivateCanarySet(db, canarySetId) {
+  const result = db.prepare(`UPDATE canary_queries SET active = 0 WHERE canary_set_id = ? AND active = 1`).run(canarySetId);
+  const changes = result.changes ?? 0;
+  return typeof changes === "bigint" ? Number(changes) : changes;
+}
+function insertCycleMetrics(db, row) {
+  db.prepare(`
+    INSERT INTO improve_cycle_metrics
+      (run_id, ts, pass, canary_set_id, mean_recall, mean_ndcg, mean_mrr,
+       canary_ranks_json, store_total, store_by_type_json, distinct_content_ratio,
+       mean_bigram_diversity, over_generation_count, accepted_actions,
+       merge_floor_violations, alerts_json)
+    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+  `).run(row.run_id, row.ts, row.pass, row.canary_set_id, row.mean_recall, row.mean_ndcg, row.mean_mrr, row.canary_ranks_json, row.store_total, row.store_by_type_json, row.distinct_content_ratio, row.mean_bigram_diversity, row.over_generation_count, row.accepted_actions, row.merge_floor_violations, row.alerts_json);
+}
+function queryRecentCycleMetrics(db, canarySetId, limit) {
+  const rows = db.prepare(`SELECT run_id, ts, pass, canary_set_id, mean_recall, mean_ndcg, mean_mrr,
+              canary_ranks_json, store_total, store_by_type_json, distinct_content_ratio,
+              mean_bigram_diversity, over_generation_count, accepted_actions,
+              merge_floor_violations, alerts_json
+       FROM improve_cycle_metrics WHERE canary_set_id = ?
+       ORDER BY ts DESC, id DESC LIMIT ?`).all(canarySetId, Math.max(0, limit));
+  return rows.reverse();
+}
+function getLatestCycleMetrics(db) {
+  const row = db.prepare(`SELECT run_id, ts, pass, canary_set_id, mean_recall, mean_ndcg, mean_mrr,
+              canary_ranks_json, store_total, store_by_type_json, distinct_content_ratio,
+              mean_bigram_diversity, over_generation_count, accepted_actions,
+              merge_floor_violations, alerts_json
+       FROM improve_cycle_metrics ORDER BY ts DESC, id DESC LIMIT 1`).get();
+  return row == null ? undefined : row;
+}
+function purgeOldCycleMetrics(db, retentionDays = 365) {
+  if (!Number.isFinite(retentionDays) || retentionDays <= 0)
+    return 0;
+  const cutoff = new Date(Date.now() - retentionDays * 86400000).toISOString();
+  const result = db.prepare("DELETE FROM improve_cycle_metrics WHERE ts < ?").run(cutoff);
+  const changes = result.changes ?? 0;
+  return typeof changes === "bigint" ? Number(changes) : changes;
+}
 var WITH_IMMEDIATE_TX_MAX_ATTEMPTS = 5;
 var init_state_db = __esm(() => {
   init_managed_db();
@@ -15790,7 +15913,7 @@ var init_config_types = __esm(() => {
 });
 
 // src/core/config/config-schema.ts
-var positiveInt, nonNegativeNumber, nonEmptyString, httpUrl, LlmCapabilitiesSchema, LlmConnectionConfigSchema, LlmProfileConfigSchema, EmbeddingOllamaOptionsSchema, EmbeddingConnectionConfigSchema, AgentPlatformSchema, AgentProfileConfigSchema, ImproveProcessConfigSchema, ImproveProfileProcessesSchema, ImproveProfileConfigSchema, ProfilesSchema, DefaultsSchema, SourceConfigEntryOptionsSchema, SourceConfigEntrySchema, RegistryConfigEntrySchema, KitSourceSchema, InstalledStashEntrySchema, OutputConfigSchema, SearchGraphBoostSchema, SearchConfigSchema, FeedbackConfigSchema, ImproveUtilityDecaySchema, ImproveCalibrationSchema, ImproveExplorationSchema, ImproveSalienceSchema, ImproveConfigSchema, GRAPH_EXTRACTION_INCLUDE_TYPES_ALLOWED, INDEX_PASS_PROVIDER_KEYS, INDEX_PASS_KNOWN_KEYS, IndexPassConfigSchema, MetadataEnhanceSchema, StalenessDetectionSchema, IndexConfigSchema, SetupTaskSchedulesSchema, SetupConfigSchema, AkmConfigShape, AkmConfigBaseSchema, AkmConfigSchema;
+var positiveInt, nonNegativeNumber, nonEmptyString, httpUrl, LlmCapabilitiesSchema, LlmConnectionConfigSchema, LlmProfileConfigSchema, EmbeddingOllamaOptionsSchema, EmbeddingConnectionConfigSchema, AgentPlatformSchema, AgentProfileConfigSchema, ImproveProcessConfigSchema, ImproveProfileProcessesSchema, ImproveProfileConfigSchema, ProfilesSchema, DefaultsSchema, SourceConfigEntryOptionsSchema, SourceConfigEntrySchema, RegistryConfigEntrySchema, KitSourceSchema, InstalledStashEntrySchema, OutputConfigSchema, SearchGraphBoostSchema, SearchConfigSchema, FeedbackConfigSchema, ImproveUtilityDecaySchema, ImproveCalibrationSchema, ImproveExplorationSchema, ImproveSalienceSchema, ImproveCollapseDetectorSchema, ImproveConfigSchema, GRAPH_EXTRACTION_INCLUDE_TYPES_ALLOWED, INDEX_PASS_PROVIDER_KEYS, INDEX_PASS_KNOWN_KEYS, IndexPassConfigSchema, MetadataEnhanceSchema, StalenessDetectionSchema, IndexConfigSchema, SetupTaskSchedulesSchema, SetupConfigSchema, AkmConfigShape, AkmConfigBaseSchema, AkmConfigSchema;
 var init_config_schema = __esm(() => {
   init_zod();
   init_config_types();
@@ -15883,11 +16006,6 @@ var init_config_schema = __esm(() => {
     indexSessions: exports_external.boolean().optional(),
     minSessionDuration: exports_external.number().min(0).optional(),
     p90ChunkSecondsDefault: exports_external.number().finite().positive().optional(),
-    homeostaticDemotion: exports_external.object({
-      enabled: exports_external.boolean().optional(),
-      staleDays: exports_external.number().int().min(0).optional(),
-      demotionFactor: exports_external.number().min(0).max(1).optional()
-    }).passthrough().optional(),
     schemaSimilarity: exports_external.object({
       enabled: exports_external.boolean().optional(),
       epsilon: exports_external.number().min(0).max(1).optional(),
@@ -15900,7 +16018,9 @@ var init_config_schema = __esm(() => {
       enabled: exports_external.boolean().optional(),
       maxGeneration: exports_external.number().int().min(1).optional(),
       lexicalDiversityCheck: exports_external.boolean().optional(),
-      randomClusterFraction: exports_external.number().min(0).max(1).optional()
+      randomClusterFraction: exports_external.number().min(0).max(1).optional(),
+      mergeInformationFloor: exports_external.boolean().optional(),
+      minSpecificityRetention: exports_external.number().min(0).max(1).optional()
     }).passthrough().optional(),
     cls: exports_external.object({
       enabled: exports_external.boolean().optional(),
@@ -16068,12 +16188,23 @@ var init_config_schema = __esm(() => {
     salienceThreshold: exports_external.number().min(0).max(1).optional(),
     replayBudget: exports_external.number().int().min(0).optional()
   }).passthrough();
+  ImproveCollapseDetectorSchema = exports_external.object({
+    enabled: exports_external.boolean().optional(),
+    canaryCount: exports_external.number().int().min(3).max(200).optional(),
+    k: exports_external.number().int().min(1).max(100).optional(),
+    windowCycles: exports_external.number().int().min(2).max(50).optional(),
+    recallDropThreshold: exports_external.number().min(0).max(1).optional(),
+    entropyDropThreshold: exports_external.number().min(0).max(1).optional(),
+    churnMinAcceptedActions: exports_external.number().int().min(1).optional(),
+    retentionDays: exports_external.number().int().min(1).optional()
+  }).passthrough();
   ImproveConfigSchema = exports_external.object({
     utilityDecay: ImproveUtilityDecaySchema.optional(),
     eventRetentionDays: nonNegativeNumber.optional(),
     calibration: ImproveCalibrationSchema.optional(),
     exploration: ImproveExplorationSchema.optional(),
-    salience: ImproveSalienceSchema.optional()
+    salience: ImproveSalienceSchema.optional(),
+    collapseDetector: ImproveCollapseDetectorSchema.optional()
   }).passthrough();
   GRAPH_EXTRACTION_INCLUDE_TYPES_ALLOWED = [
     "memory",
@@ -16332,6 +16463,7 @@ __export(exports_config, {
   getIndexPassConfig: () => getIndexPassConfig,
   getEffectiveRegistries: () => getEffectiveRegistries,
   getDefaultLlmConfig: () => getDefaultLlmConfig,
+  _setSaveConfigForTests: () => _setSaveConfigForTests,
   VALID_HARNESS_IDS: () => VALID_HARNESS_IDS,
   FEEDBACK_FAILURE_MODES: () => FEEDBACK_FAILURE_MODES,
   DEFAULT_GRAPH_EXTRACTION_BATCH_SIZE: () => DEFAULT_GRAPH_EXTRACTION_BATCH_SIZE,
@@ -16474,7 +16606,17 @@ function loadConfig() {
   warnIfProjectConfigPresent(process.cwd());
   return loadUserConfig();
 }
+function _setSaveConfigForTests(fake) {
+  saveConfigOverride = fake;
+}
 function saveConfig(config) {
+  if (saveConfigOverride) {
+    saveConfigOverride(config);
+    return;
+  }
+  saveConfigReal(config);
+}
+function saveConfigReal(config) {
   cachedConfig = undefined;
   const configPath = getConfigPath();
   const dir = path11.dirname(configPath);
@@ -16643,7 +16785,7 @@ function isFile(filePath) {
     return false;
   }
 }
-var FEEDBACK_FAILURE_MODES, DEFAULT_GRAPH_EXTRACTION_BATCH_SIZE = 4, GRAPH_EXTRACTION_CHARS_PER_BODY = 1500, DEFAULT_CONFIG, PROJECT_CONFIG_RELATIVE_PATH, cachedConfig, INDEX_RESERVED_KEYS, PROJECT_CONFIG_DEPRECATION_WARNED;
+var FEEDBACK_FAILURE_MODES, DEFAULT_GRAPH_EXTRACTION_BATCH_SIZE = 4, GRAPH_EXTRACTION_CHARS_PER_BODY = 1500, DEFAULT_CONFIG, PROJECT_CONFIG_RELATIVE_PATH, cachedConfig, saveConfigOverride, INDEX_RESERVED_KEYS, PROJECT_CONFIG_DEPRECATION_WARNED;
 var init_config2 = __esm(() => {
   init_errors();
   init_config_io();
@@ -17801,6 +17943,7 @@ function getRetrievalCounts(db, refs) {
          FROM usage_events
          WHERE event_type IN ('search','show','curate')
            AND entry_ref IS NOT NULL
+           AND (source IS NULL OR source NOT IN ('improve','task'))
            AND CASE
                  WHEN instr(entry_ref, '//') > 0
                    THEN substr(entry_ref, instr(entry_ref, '//') + 2)

package/dist/scripts/migrations/import-fs-improve-runs-to-db.js

@@ -53,12 +53,16 @@ function appendToLogFile(level, args) {
   }
 }
 function warn(...args) {
+  if (sinkOverride) {
+    sinkOverride("warn", args);
+    return;
+  }
   appendToLogFile("WARN", args);
   if (!quiet) {
     console.warn(...args);
   }
 }
-var quiet = false, logFilePath;
+var quiet = false, logFilePath, sinkOverride;
 var init_warn = () => {};
 
 // node_modules/dotenv/lib/main.js
@@ -455,21 +459,11 @@ function scanKeys(text) {
   }
   return keys;
 }
-function scanComments(text) {
-  const comments = [];
-  for (const line of text.split(/\r?\n/)) {
-    const trimmed = line.trimStart();
-    if (trimmed.startsWith("#")) {
-      comments.push(trimmed.slice(1).trimStart());
-    }
-  }
-  return comments;
-}
 function listKeys(envPath) {
   if (!fs3.existsSync(envPath))
-    return { keys: [], comments: [] };
+    return { keys: [] };
   const text = fs3.readFileSync(envPath, "utf8");
-  return { keys: scanKeys(text), comments: scanComments(text) };
+  return { keys: scanKeys(text) };
 }
 var import_dotenv, ASSIGN_RE;
 var init_env = __esm(() => {
@@ -8211,12 +8205,7 @@ function applyScriptMetadata(entry, ctx) {
   }
 }
 function applyEnvMetadata(entry, ctx) {
-  const { keys, comments } = listKeys(ctx.absPath);
-  if (comments.length > 0 && !entry.description) {
-    entry.description = comments.join(" ").slice(0, 500);
-    entry.source = "comments";
-    entry.confidence = 0.7;
-  }
+  const { keys } = listKeys(ctx.absPath);
   if (keys.length > 0) {
     entry.searchHints = keys;
   }
@@ -9142,6 +9131,44 @@ var MIGRATIONS = [
     up: `
       ALTER TABLE asset_salience ADD COLUMN encoding_source TEXT DEFAULT NULL;
     `
+  },
+  {
+    id: "016-collapse-churn-detector",
+    up: `
+      CREATE TABLE IF NOT EXISTS canary_queries (
+        id            INTEGER PRIMARY KEY AUTOINCREMENT,
+        canary_set_id TEXT    NOT NULL,
+        anchor_ref    TEXT    NOT NULL,
+        query         TEXT    NOT NULL,
+        source        TEXT    NOT NULL DEFAULT 'auto',
+        active        INTEGER NOT NULL DEFAULT 1,
+        created_at    TEXT    NOT NULL
+      );
+      CREATE INDEX IF NOT EXISTS idx_canary_queries_active
+        ON canary_queries(active, canary_set_id);
+
+      CREATE TABLE IF NOT EXISTS improve_cycle_metrics (
+        id                      INTEGER PRIMARY KEY AUTOINCREMENT,
+        run_id                  TEXT    NOT NULL,
+        ts                      TEXT    NOT NULL,
+        pass                    TEXT    NOT NULL,
+        canary_set_id           TEXT    NOT NULL,
+        mean_recall             REAL    NOT NULL,
+        mean_ndcg               REAL    NOT NULL,
+        mean_mrr                REAL    NOT NULL,
+        canary_ranks_json       TEXT    NOT NULL,
+        store_total             INTEGER NOT NULL,
+        store_by_type_json      TEXT    NOT NULL,
+        distinct_content_ratio  REAL    NOT NULL,
+        mean_bigram_diversity   REAL    NOT NULL,
+        over_generation_count   INTEGER NOT NULL,
+        accepted_actions        INTEGER NOT NULL,
+        merge_floor_violations  INTEGER NOT NULL DEFAULT 0,
+        alerts_json             TEXT    NOT NULL DEFAULT '[]'
+      );
+      CREATE INDEX IF NOT EXISTS idx_improve_cycle_metrics_ts
+        ON improve_cycle_metrics(ts);
+    `
   }
 ];
 function runMigrations2(db) {

package/dist/setup/detect.js

@@ -14,6 +14,11 @@ import { defaultWhich } from "../integrations/agent/detect.js";
 import { SESSION_LOG_HARNESSES } from "../integrations/harnesses/index.js";
 import { spawn } from "../runtime.js";
 import { detectHarnessConfigs } from "./harness-config-import.js";
+let detectOverrides;
+/** TEST-ONLY. Swap the network/host probes; pass undefined to restore. */
+export function _setDetectForTests(fakes) {
+    detectOverrides = fakes;
+}
 // ── Ollama Detection ────────────────────────────────────────────────────────
 const OLLAMA_BASE = "http://localhost:11434";
 /**
@@ -23,6 +28,8 @@ const OLLAMA_BASE = "http://localhost:11434";
  * via subprocess. Returns available models sorted alphabetically.
  */
 export async function detectOllama() {
+    if (detectOverrides?.detectOllama)
+        return detectOverrides.detectOllama();
     const result = { available: false, models: [], endpoint: OLLAMA_BASE };
     // Try HTTP API first
     try {
@@ -120,6 +127,8 @@ const AGENT_PLATFORMS = SESSION_LOG_HARNESSES.filter((h) => h.setupDetectionDir)
  * Supports both HOME (Unix) and USERPROFILE (Windows).
  */
 export function detectAgentPlatforms() {
+    if (detectOverrides?.detectAgentPlatforms)
+        return detectOverrides.detectAgentPlatforms();
     const home = process.env.HOME?.trim() || process.env.USERPROFILE?.trim();
     if (!home)
         return [];

package/dist/setup/registry-stash-loader.js

@@ -44,6 +44,13 @@ const FALLBACK_STASHES = [
         defaultSelected: false,
     },
 ];
+// ── Test seam ────────────────────────────────────────────────────────────────
+// Swap-and-restore override. Inert in production; only tests call the setter.
+let loadSetupStashesOverride;
+/** TEST-ONLY. Swap the implementation of `loadSetupStashes`; pass undefined to restore. */
+export function _setLoadSetupStashesForTests(fake) {
+    loadSetupStashesOverride = fake;
+}
 // ── Loader ──────────────────────────────────────────────────────────────────
 /**
  * Fetch available stashes from the registry and map to SetupStashEntry[].
@@ -55,6 +62,11 @@ const FALLBACK_STASHES = [
  * @param timeoutMs    Fetch timeout in ms (default: 4000).
  */
 export async function loadSetupStashes(registryUrl, timeoutMs = 4000) {
+    if (loadSetupStashesOverride)
+        return loadSetupStashesOverride(registryUrl, timeoutMs);
+    return loadSetupStashesReal(registryUrl, timeoutMs);
+}
+async function loadSetupStashesReal(registryUrl, timeoutMs = 4000) {
     try {
         const response = await fetch(registryUrl, {
             signal: AbortSignal.timeout(timeoutMs),

package/dist/setup/setup.js

@@ -12,7 +12,7 @@ import { promises as dnsPromises } from "node:dns";
 import fs from "node:fs";
 import os from "node:os";
 import path from "node:path";
-import * as p from "@clack/prompts";
+import * as p from "../cli/clack.js";
 import { akmInit } from "../commands/sources/init.js";
 import { detectServerDefault, isCiEnvironment, registerDefaultTasks } from "../commands/tasks/default-tasks.js";
 import { akmTasksAdd, akmTasksList, akmTasksSetEnabled, akmTasksSync } from "../commands/tasks/tasks.js";

package/dist/storage/repositories/index-db.js

@@ -3,6 +3,12 @@
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 import { closeDatabase, openExistingDatabase } from "../../indexer/db/db.js";
 import { resolveStorageLocations } from "../locations.js";
+/**
+ * Busy-timeout (ms) for read-path telemetry writers. Small on purpose: a
+ * usage-event insert contending with a background reindex should be dropped,
+ * not waited on for the default 30s.
+ */
+export const TELEMETRY_BUSY_TIMEOUT_MS = 250;
 /**
  * Scoped-resource (loan pattern) helper for the index database (`index.db`).
  *
@@ -32,9 +38,12 @@ import { resolveStorageLocations } from "../locations.js";
  * @param fn Receives the open index database; must finish all DB work before returning.
  * @returns Whatever `fn` returns.
  */
-export function withIndexDb(fn) {
+export function withIndexDb(fn, opts) {
     const db = openExistingDatabase(resolveStorageLocations().indexDb);
     try {
+        if (opts?.busyTimeoutMs !== undefined) {
+            db.exec(`PRAGMA busy_timeout = ${Math.max(0, Math.floor(opts.busyTimeoutMs))}`);
+        }
         return fn(db);
     }
     finally {

package/dist/tasks/backends/index.js

@@ -4,7 +4,14 @@
 import { CRON_BACKEND } from "./cron.js";
 import { LAUNCHD_BACKEND } from "./launchd.js";
 import { SCHTASKS_BACKEND } from "./schtasks.js";
+let backendsOverrides;
+/** TEST-ONLY. Swap backend selection; pass undefined to restore the real implementations. */
+export function _setBackendsForTests(fakes) {
+    backendsOverrides = fakes;
+}
 export function selectBackend(options = {}) {
+    if (backendsOverrides?.selectBackend)
+        return backendsOverrides.selectBackend(options);
     const platform = options.platform ?? process.platform;
     switch (platform) {
         case "win32":
@@ -16,6 +23,8 @@ export function selectBackend(options = {}) {
     }
 }
 export function backendNameForPlatform(platform = process.platform) {
+    if (backendsOverrides?.backendNameForPlatform)
+        return backendsOverrides.backendNameForPlatform(platform);
     if (platform === "win32")
         return "schtasks";
     if (platform === "darwin")

package/dist/tasks/runner.js

@@ -131,6 +131,11 @@ async function runCommandTask(input) {
             stdout: "pipe",
             stderr: "pipe",
             cwd: process.env.HOME ?? "/tmp",
+            // Stamp task-runner provenance so any akm invocation in the command tree
+            // records usage events as machine traffic, not user demand (DRIFT-6).
+            // A more specific stamp already in the environment (e.g. improve's
+            // AKM_EVENT_SOURCE=improve on its child spawns) still wins in children.
+            env: { ...process.env, AKM_EVENT_SOURCE: process.env.AKM_EVENT_SOURCE ?? "task" },
         });
         let timer;
         let timedOut = false;
@@ -351,6 +356,10 @@ async function runPromptTask(input) {
         timeoutMs: agentTimeoutMs,
         cwd: stashDir,
         ...agentOptions,
+        // Stamp task-runner provenance for any akm invocation the agent makes
+        // (DRIFT-6: agent-task traffic must not be recorded as user demand).
+        // Caller-supplied env still wins on conflicts.
+        env: { AKM_EVENT_SOURCE: "task", ...agentOptions?.env },
     });
     const finishedAt = now();
     const log = renderPromptLog({ task, profileName: profile.name, result });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "akm-cli",
-  "version": "0.9.0-beta.52",
+  "version": "0.9.0-beta.54",
   "type": "module",
   "description": "akm (Agent Knowledge Management) — A package manager for AI agent skills, commands, tools, and knowledge. Works with Claude Code, OpenCode, Cursor, and any AI coding assistant.",
   "keywords": [
@@ -58,9 +58,7 @@
     "sweep:tmp": "bun scripts/sweep-test-tmp.ts",
     "test": "bash scripts/test-unit.sh",
     "test:unit": "bash scripts/test-unit.sh",
-    "test:integration": "bun run sweep:tmp && bun test --isolate --timeout=30000 ./tests/integration ./tests/commands ./tests/workflows",
-    "test:unit:shard": "bun run sweep:tmp && bun test --isolate --timeout=30000 ./tests --path-ignore-patterns=tests/integration --shard=${SHARD:?set SHARD=k/N}",
-    "test:integration:shard": "bun run sweep:tmp && bun test --isolate --timeout=30000 ./tests/integration ./tests/commands ./tests/workflows --shard=${SHARD:?set SHARD=k/N}",
+    "test:integration": "bun run sweep:tmp && bun test --timeout=30000 ./tests/integration",
     "test:node-smoke": "bun scripts/node-smoke.ts",
     "test:node-compat": "AKM_NODE_COMPAT_TESTS=1 bun test --timeout=120000 tests/integration/node-compat.test.ts",
     "test:time": "bun scripts/test-timing-report.ts",