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

package/dist/commands/read/show.js

@@ -28,10 +28,10 @@ import { NotFoundError, rethrowIfTestIsolationError, UsageError } from "../../co
 import { appendEvent, readEvents } from "../../core/events.js";
 import { closeDatabase, computeBodyHash, findEntryIdByRef, openExistingDatabase } from "../../indexer/db/db.js";
 import { hasGraphData } from "../../indexer/db/graph-db.js";
-import { ensureIndex } from "../../indexer/ensure-index.js";
 import { listRelatedPathsForFile } from "../../indexer/graph/graph-boost.js";
 import { extractGraphForSingleFile } from "../../indexer/graph/graph-extraction.js";
 import { lookup } from "../../indexer/indexer.js";
+import { ensurePrimaryIndexForRead, resolveReadSources } from "../../indexer/read-preflight.js";
 import { buildEditHint, findSourceForPath, isEditable, resolveSourceEntries } from "../../indexer/search/search-source.js";
 import { insertUsageEvent } from "../../indexer/usage/usage-events.js";
 import { buildFileContext, buildRenderContext, getRenderer, runMatchers } from "../../indexer/walk/file-context.js";
@@ -147,10 +147,8 @@ export async function akmShowUnified(input) {
         }
     }
     // Auto-index when stale so the index is current before lookup.
-    const allSources = resolveSourceEntries();
-    if (allSources.length > 0) {
-        await ensureIndex(allSources[0].path);
-    }
+    const { primarySource } = resolveReadSources();
+    await ensurePrimaryIndexForRead(primarySource);
     // Try local filesystem (FTS5 index lookup)
     const result = await showLocal(input);
     // Scope filter narrows resolution: if --scope was supplied, the asset's

package/dist/commands/sources/add-cli.js

@@ -3,8 +3,8 @@
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 import fs from "node:fs";
 import path from "node:path";
-import * as p from "@clack/prompts";
 import { defineCommand } from "citty";
+import * as p from "../../cli/clack.js";
 import { output, runWithJsonErrors } from "../../cli/shared.js";
 import { UsageError } from "../../core/errors.js";
 import { appendEvent } from "../../core/events.js";

package/dist/commands/sources/init.js

@@ -55,7 +55,19 @@ function assertInitSandbox(stashDir, dirExplicitlyProvided) {
 function isUnderTestRunner() {
     return process.env.BUN_TEST === "1" || process.env.NODE_ENV === "test";
 }
+// ── Test seam ────────────────────────────────────────────────────────────────
+// Swap-and-restore override. Inert in production; only tests call the setter.
+let akmInitOverride;
+/** TEST-ONLY. Swap the implementation of `akmInit`; pass undefined to restore. */
+export function _setAkmInitForTests(fake) {
+    akmInitOverride = fake;
+}
 export async function akmInit(options) {
+    if (akmInitOverride)
+        return akmInitOverride(options);
+    return akmInitReal(options);
+}
+async function akmInitReal(options) {
     const dirExplicitlyProvided = options?.dir != null;
     const setDefault = options?.setDefault === true;
     const stashDir = options?.dir ? path.resolve(options.dir) : getDefaultStashDir();

package/dist/commands/sources/stash-cli.js

@@ -25,8 +25,8 @@
  * SIGINT/SIGTERM handlers in a try/finally — left byte-for-byte untouched.
  */
 import path from "node:path";
-import * as p from "@clack/prompts";
 import { defineCommand } from "citty";
+import * as p from "../../cli/clack.js";
 import { defineJsonCommand, output, runWithJsonErrors } from "../../cli/shared.js";
 import { assertFlatAssetName } from "../../core/asset/asset-create.js";
 import { isHttpUrl } from "../../core/common.js";

package/dist/commands/tasks/default-tasks.js

@@ -76,12 +76,19 @@ const DEFAULT_DEPS = {
     list: akmTasksList,
     add: akmTasksAdd,
 };
+let defaultTasksOverrides;
+/** TEST-ONLY. Swap the CI/server/register functions; pass undefined to restore. */
+export function _setDefaultTasksForTests(fakes) {
+    defaultTasksOverrides = fakes;
+}
 /**
  * Decide whether `akm setup` is running in a CI environment, where it must
  * register NO scheduled tasks. Mirrors the common `CI=true` convention used by
  * GitHub Actions, GitLab CI, CircleCI, etc.
  */
 export function isCiEnvironment(env = process.env) {
+    if (defaultTasksOverrides?.isCiEnvironment)
+        return defaultTasksOverrides.isCiEnvironment(env);
     const ci = env.CI;
     if (ci === undefined || ci === null)
         return false;
@@ -95,6 +102,8 @@ export function isCiEnvironment(env = process.env) {
  * Used as the default when setup is non-interactive (no TTY / --yes / CI).
  */
 export function detectServerDefault() {
+    if (defaultTasksOverrides?.detectServerDefault)
+        return defaultTasksOverrides.detectServerDefault();
     if (os.platform() !== "linux")
         return false;
     // A laptop exposes a battery under /sys/class/power_supply/BAT*. Absence of
@@ -121,6 +130,9 @@ export function detectServerDefault() {
  *   never re-disable a user-enabled task).
  */
 export async function registerDefaultTasks(options = {}) {
+    if (defaultTasksOverrides?.registerDefaultTasks) {
+        return defaultTasksOverrides.registerDefaultTasks(options);
+    }
     if (isCiEnvironment()) {
         return { skipped: true, reason: "ci", created: [], existing: [], toggled: [] };
     }

package/dist/core/config/config.js

@@ -278,7 +278,19 @@ export function loadConfig() {
     warnIfProjectConfigPresent(process.cwd());
     return loadUserConfig();
 }
+let saveConfigOverride;
+/** TEST-ONLY. Swap the implementation of `saveConfig`; pass undefined to restore. */
+export function _setSaveConfigForTests(fake) {
+    saveConfigOverride = fake;
+}
 export function saveConfig(config) {
+    if (saveConfigOverride) {
+        saveConfigOverride(config);
+        return;
+    }
+    saveConfigReal(config);
+}
+function saveConfigReal(config) {
     cachedConfig = undefined;
     const configPath = getConfigPath();
     const dir = path.dirname(configPath);

package/dist/core/warn.js

@@ -17,6 +17,11 @@ import path from "node:path";
 let quiet = false;
 let verbose = false;
 let logFilePath;
+let sinkOverride;
+/** TEST-ONLY. Swap the output sink; pass undefined to restore real output. */
+export function _setWarnSinkForTests(fake) {
+    sinkOverride = fake;
+}
 export function setQuiet(value) {
     quiet = value;
 }
@@ -96,6 +101,10 @@ function appendToLogFile(level, args) {
  * Use for progress counters and status lines (replaces console.error used for progress).
  */
 export function info(...args) {
+    if (sinkOverride) {
+        sinkOverride("info", args);
+        return;
+    }
     appendToLogFile("INFO", args);
     if (!quiet) {
         console.warn(...args);
@@ -107,6 +116,10 @@ export function info(...args) {
  * Drop-in replacement for console.warn() across the codebase.
  */
 export function warn(...args) {
+    if (sinkOverride) {
+        sinkOverride("warn", args);
+        return;
+    }
     appendToLogFile("WARN", args);
     if (!quiet) {
         console.warn(...args);
@@ -118,6 +131,10 @@ export function warn(...args) {
  * Drop-in replacement for console.error() used for diagnostic failures.
  */
 export function error(...args) {
+    if (sinkOverride) {
+        sinkOverride("error", args);
+        return;
+    }
     appendToLogFile("ERROR", args);
     if (!quiet) {
         console.error(...args);
@@ -129,6 +146,10 @@ export function error(...args) {
  * default verbosity (e.g. registry-content workflow validation errors).
  */
 export function warnVerbose(...args) {
+    if (sinkOverride) {
+        sinkOverride("warnVerbose", args);
+        return;
+    }
     if (isVerbose()) {
         warn(...args);
     }

package/dist/indexer/db/db.js

@@ -1633,6 +1633,11 @@ function bareRef(ref) {
  * entry_ref populated (see logCurateEvent), so curation is a real retrieval
  * signal here. Legacy summary-only curate rows with a NULL entry_ref simply
  * contribute nothing.
+ *
+ * Machine-sourced events (`source` = 'improve' or 'task') are EXCLUDED: this
+ * count feeds salience/ranking, and pipeline probe traffic counting as demand
+ * creates a self-reinforcing loop (meta-review 05 DRIFT-6). NULL sources
+ * (pre-column rows) count as user demand.
  */
 export function getRetrievalCounts(db, refs) {
     if (refs.length === 0)
@@ -1671,6 +1676,7 @@ export 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/indexer/ensure-index.js

@@ -185,7 +185,7 @@ async function runInlineReindex(stashDir) {
     }
     catch (error) {
         warn("Auto-index failed, proceeding with existing index:", error instanceof Error ? error.message : String(error));
-        return true;
+        return false;
     }
 }
 /**
@@ -200,7 +200,8 @@ async function runInlineReindex(stashDir) {
  * trigger and waits for it. Use this for callers like `improve` whose
  * planning logic depends on a current `entries` table in the same process.
  *
- * Returns `true` if an index run was attempted.
+ * Returns `true` only when an inline index run succeeds.
+ * A rebuild attempt that fails (throws) resolves to `false`.
  */
 export async function ensureIndex(stashDir, options = {}) {
     if (options.mode === "blocking") {

package/dist/indexer/index-writer-lock.js

@@ -7,6 +7,7 @@ import { probeLock, releaseLock, releaseLockIfOwned, tryAcquireLockSync } from "
 import { getDbPath, getIndexWriterLockPath } from "../core/paths.js";
 const INDEX_WRITER_LOCK_STALE_AFTER_MS = 12 * 60 * 60 * 1000;
 const INDEX_WRITER_WAIT_MS = 100;
+const DEFAULT_INDEX_WRITER_MAX_WAIT_MS = 10 * 60 * 1000;
 const heldLocks = new Map();
 function buildPayload(purpose, pid = process.pid) {
     return JSON.stringify({
@@ -49,6 +50,8 @@ function retainHeldLock(lockPath) {
 export async function acquireIndexWriterLease(options) {
     const mode = options.mode ?? "wait";
     const lockPath = getIndexWriterLockPath();
+    const startedAt = Date.now();
+    const maxWaitMs = options.maxWaitMs ?? DEFAULT_INDEX_WRITER_MAX_WAIT_MS;
     fs.mkdirSync(path.dirname(lockPath), { recursive: true });
     if (heldLocks.has(lockPath)) {
         return retainHeldLock(lockPath);
@@ -68,6 +71,12 @@ export async function acquireIndexWriterLease(options) {
         }
         if (mode === "try")
             return undefined;
+        // Held by another live process. Time out only *after* a real acquisition
+        // attempt, so a caller with maxWaitMs:0 still gets one chance at a free lock
+        // instead of throwing before it ever tries.
+        if (maxWaitMs >= 0 && Date.now() - startedAt >= maxWaitMs) {
+            throw new Error(`timed out waiting for index writer lease for ${options.purpose}`);
+        }
         await delay(INDEX_WRITER_WAIT_MS);
     }
 }

package/dist/indexer/indexer.js

@@ -118,7 +118,7 @@ async function runWalkPhase(ctx) {
     ctx.timing.tWalkEnd = Date.now();
     throwIfAborted(signal);
     // LLM enrichment for directories that need it
-    await enhanceDirsWithLlm(db, config, dirsNeedingLlm, onProgress, signal, true, reEnrich);
+    await enhanceDirsWithLlm(db, config, dirsNeedingLlm, onProgress, signal, reEnrich);
     onProgress({
         phase: "llm",
         message: resolveIndexPassLLM("enrichment", config)
@@ -226,7 +226,19 @@ function runCleanPass(db, dryRun) {
     };
 }
 // ── Indexer ──────────────────────────────────────────────────────────────────
+// ── Test seam ────────────────────────────────────────────────────────────────
+// Swap-and-restore override. Inert in production; only tests call the setter.
+let akmIndexOverride;
+/** TEST-ONLY. Swap the implementation of `akmIndex`; pass undefined to restore. */
+export function _setAkmIndexForTests(fake) {
+    akmIndexOverride = fake;
+}
 export async function akmIndex(options) {
+    if (akmIndexOverride)
+        return akmIndexOverride(options);
+    return akmIndexReal(options);
+}
+async function akmIndexReal(options) {
     return withIndexWriterLease({ purpose: "akm-index", signal: options?.signal }, async () => {
         const stashDir = options?.stashDir || resolveStashDir();
         const onProgress = options?.onProgress ?? (() => { });
@@ -640,7 +652,7 @@ async function indexEntries(db, allSourceEntries, isIncremental, builtAtMs, hadR
     insertTransaction();
     return { scannedDirs, skippedDirs, generatedCount, warnings, dirsNeedingLlm };
 }
-async function enhanceDirsWithLlm(db, config, dirsNeedingLlm, onProgress, signal, _enrich = false, reEnrich = false) {
+async function enhanceDirsWithLlm(db, config, dirsNeedingLlm, onProgress, signal, reEnrich = false) {
     // Resolve per-pass LLM config via the unified shim. Returns undefined when
     // either no `akm.llm` is configured or the user opted this pass out via
     // `index.enrichment.llm = false`. (#208)
@@ -977,7 +989,7 @@ function resolveIndexedFiles(dirPath, files, stash) {
     for (const entry of stash.entries) {
         const entryPath = entry.filename
             ? path.join(dirPath, entry.filename)
-            : matchEntryToFile(entry.name, fileBasenameMap, files);
+            : matchEntryToFile(entry.name, fileBasenameMap);
         if (entryPath)
             resolved.add(entryPath);
     }
@@ -1096,7 +1108,7 @@ export function buildFileBasenameMap(files) {
  *      try matching the last segment
  *   3. No implicit file fallback: ambiguous legacy entries are skipped
  */
-export function matchEntryToFile(entryName, fileMap, _files) {
+export function matchEntryToFile(entryName, fileMap) {
     // Exact match on entry name
     const exact = fileMap.get(entryName);
     if (exact)

package/dist/indexer/read-preflight.js

@@ -0,0 +1,23 @@
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at https://mozilla.org/MPL/2.0/.
+import { ensureIndex } from "./ensure-index.js";
+import { resolveSourceEntries } from "./search/search-source.js";
+/** Resolve the active read sources using the same resolution rules as search/show. */
+export function resolveReadSources(overrideStashDir, existingConfig) {
+    const sources = resolveSourceEntries(overrideStashDir, existingConfig);
+    return { sources, primarySource: sources[0] };
+}
+/** Ensure the primary source index is readable for reads, when a primary exists. */
+export async function ensurePrimaryIndexForRead(primarySource) {
+    if (!primarySource?.path)
+        return false;
+    return ensureIndex(primarySource.path);
+}
+/**
+ * Convenience helper for callers that only need to ensure a read index from a
+ * configured stash path and default config.
+ */
+export async function ensurePrimaryIndexFromConfig(overrideStashDir, existingConfig) {
+    return ensurePrimaryIndexForRead(resolveReadSources(overrideStashDir, existingConfig).primarySource);
+}

package/dist/indexer/walk/walker.js

@@ -148,20 +148,28 @@ function isInsideGitRepo(dir) {
  * read (e.g. permission errors).
  */
 export function* walkMarkdownFiles(root) {
-    let entries;
-    try {
-        entries = fs.readdirSync(root, { withFileTypes: true });
-    }
-    catch {
-        return;
-    }
-    for (const entry of entries) {
-        const full = path.join(root, entry.name);
-        if (entry.isDirectory()) {
-            yield* walkMarkdownFiles(full);
+    const stack = [root];
+    while (stack.length > 0) {
+        const current = stack.pop();
+        if (!current)
+            continue;
+        let entries;
+        try {
+            entries = fs.readdirSync(current, { withFileTypes: true });
         }
-        else if (entry.isFile() && entry.name.toLowerCase().endsWith(".md")) {
-            yield full;
+        catch {
+            continue;
+        }
+        for (const entry of entries) {
+            const full = path.join(current, entry.name);
+            if (entry.isSymbolicLink())
+                continue;
+            if (entry.isDirectory()) {
+                stack.push(full);
+            }
+            else if (entry.isFile() && entry.name.toLowerCase().endsWith(".md")) {
+                yield full;
+            }
         }
     }
 }

package/dist/integrations/agent/detect.js

@@ -55,6 +55,11 @@ export function defaultWhich(bin, envSource = process.env) {
     }
     return undefined;
 }
+let detectOverrides;
+/** TEST-ONLY. Swap the detection implementations; pass undefined to restore. */
+export function _setAgentDetectForTests(fakes) {
+    detectOverrides = fakes;
+}
 /**
  * Probe every resolvable agent profile (built-ins plus user overrides)
  * for an installed CLI.
@@ -64,6 +69,8 @@ export function defaultWhich(bin, envSource = process.env) {
  * @param whichFn  Binary lookup. Tests should inject a stub.
  */
 export function detectAgentCliProfiles(agent, whichFn = defaultWhich) {
+    if (detectOverrides?.detectAgentCliProfiles)
+        return detectOverrides.detectAgentCliProfiles(agent, whichFn);
     const profiles = listResolvedAgentProfiles(agent);
     return profiles.map((profile) => probeProfile(profile, whichFn));
 }
@@ -87,6 +94,8 @@ function probeProfile(profile, whichFn) {
  *      writing `agent.default`.
  */
 export function pickDefaultAgentProfile(results, existingDefault) {
+    if (detectOverrides?.pickDefaultAgentProfile)
+        return detectOverrides.pickDefaultAgentProfile(results, existingDefault);
     if (existingDefault) {
         const match = results.find((r) => r.name === existingDefault && r.available);
         if (match)

package/dist/integrations/agent/index.js

@@ -3,7 +3,7 @@
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 export { getCommandBuilder } from "./builders.js";
 export { DEFAULT_AGENT_TIMEOUT_MS, listAgentProfileNames, listResolvedAgentProfiles, requireAgentProfile, resolveAgentProfile, resolveDefaultProfileName, resolveProfileFromConfig, } from "./config.js";
-export { defaultWhich, detectAgentCliProfiles, pickDefaultAgentProfile } from "./detect.js";
+export { _setAgentDetectForTests, defaultWhich, detectAgentCliProfiles, pickDefaultAgentProfile } from "./detect.js";
 export { listBuiltinModelAliases, resolveModel } from "./model-aliases.js";
 export { BUILTIN_AGENT_PROFILE_NAMES, getBuiltinAgentProfile, listBuiltinAgentProfiles, } from "./profiles.js";
 export { buildProposePrompt, buildReflectPrompt, buildSchemaRepairPrompt, extractDraftConfidence, parseAgentProposalPayload, } from "./prompts.js";

package/dist/llm/client.js

@@ -165,7 +165,19 @@ function isRetryable(err) {
     }
     return false;
 }
+// ── Test seam ────────────────────────────────────────────────────────────────
+// Swap-and-restore override. Inert in production; only tests call the setter.
+let chatCompletionOverride;
+/** TEST-ONLY. Swap the implementation of `chatCompletion`; pass undefined to restore. */
+export function _setChatCompletionForTests(fake) {
+    chatCompletionOverride = fake;
+}
 export async function chatCompletion(config, messages, options) {
+    if (chatCompletionOverride)
+        return chatCompletionOverride(config, messages, options);
+    return chatCompletionReal(config, messages, options);
+}
+async function chatCompletionReal(config, messages, options) {
     const effectiveTimeoutMs = options?.timeoutMs ?? config.timeoutMs ?? 120_000;
     const started = Date.now();
     try {

package/dist/llm/embedder.js

@@ -3,11 +3,26 @@
 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
 import { embedCacheKey, getCachedEmbedding, setCachedEmbedding } from "./embedders/cache.js";
 import { DETERMINISTIC_EMBED_MODEL_ID, deterministicEmbed, isDeterministicEmbedEnabled, } from "./embedders/deterministic.js";
-import { DEFAULT_LOCAL_MODEL, isTransformersAvailable, LocalEmbedder } from "./embedders/local.js";
+import { DEFAULT_LOCAL_MODEL, isTransformersAvailable as isTransformersAvailableReal, LocalEmbedder, } from "./embedders/local.js";
 import { hasRemoteEndpoint, RemoteEmbedder } from "./embedders/remote.js";
 // ── Re-exports (public API) ─────────────────────────────────────────────────
 export { clearEmbeddingCache } from "./embedders/cache.js";
-export { DEFAULT_LOCAL_MODEL, isTransformersAvailable } from "./embedders/local.js";
+export { _setTransformersLoaderForTests, DEFAULT_LOCAL_MODEL } from "./embedders/local.js";
+let embedderOverrides;
+/** TEST-ONLY. Swap embedder implementations; pass undefined to restore. */
+export function _setEmbedderForTests(fakes) {
+    embedderOverrides = fakes;
+}
+/**
+ * Check whether the @huggingface/transformers package is importable.
+ * Delegating wrapper around `./embedders/local`'s probe so tests can swap it
+ * via {@link _setEmbedderForTests}.
+ */
+export function isTransformersAvailable() {
+    if (embedderOverrides?.isTransformersAvailable)
+        return embedderOverrides.isTransformersAvailable();
+    return isTransformersAvailableReal();
+}
 // ── Singleton local embedder ────────────────────────────────────────────────
 // `_localEmbedder` is an intentional module-level singleton but constructed
 // lazily on first use. The underlying @huggingface/transformers pipeline is
@@ -40,6 +55,8 @@ export function resetLocalEmbedder() {
  * and embedding config. Repeated identical queries return the cached vector.
  */
 export async function embed(text, embeddingConfig, signal) {
+    if (embedderOverrides?.embed)
+        return embedderOverrides.embed(text, embeddingConfig, signal);
     // Deterministic mode (env-gated, test/bench only): model-free, stable.
     if (isDeterministicEmbedEnabled()) {
         return deterministicEmbed(text);
@@ -61,6 +78,8 @@ export async function embed(text, embeddingConfig, signal) {
  * which processes texts in chunks of 32 for genuine batched inference.
  */
 export async function embedBatch(texts, embeddingConfig, signal) {
+    if (embedderOverrides?.embedBatch)
+        return embedderOverrides.embedBatch(texts, embeddingConfig, signal);
     if (texts.length === 0)
         return [];
     // Deterministic mode (env-gated, test/bench only): model-free, stable.
@@ -104,6 +123,8 @@ export { cosineSimilarity } from "./embedders/types.js";
  *   - No config: use `DEFAULT_LOCAL_MODEL` (the shared singleton model).
  */
 export function resolveEmbeddingModelId(embeddingConfig) {
+    if (embedderOverrides?.resolveEmbeddingModelId)
+        return embedderOverrides.resolveEmbeddingModelId(embeddingConfig);
     if (isDeterministicEmbedEnabled())
         return DETERMINISTIC_EMBED_MODEL_ID;
     if (!embeddingConfig)
@@ -117,6 +138,9 @@ export function resolveEmbeddingModelId(embeddingConfig) {
  * Check whether embedding is available with a detailed reason on failure.
  */
 export async function checkEmbeddingAvailability(embeddingConfig) {
+    if (embedderOverrides?.checkEmbeddingAvailability) {
+        return embedderOverrides.checkEmbeddingAvailability(embeddingConfig);
+    }
     // Deterministic mode (env-gated): always available — no model, no network.
     if (isDeterministicEmbedEnabled()) {
         return { available: true };

package/dist/llm/embedders/local.js

@@ -28,6 +28,12 @@ function isBatchTensor(v) {
         Array.isArray(v.dims) &&
         v.dims.length >= 2);
 }
+const realTransformersLoader = () => import("@huggingface/transformers");
+let transformersLoader = realTransformersLoader;
+/** TEST-ONLY. Swap the transformers module loader; pass undefined to restore. */
+export function _setTransformersLoaderForTests(fake) {
+    transformersLoader = fake ?? realTransformersLoader;
+}
 const LOCAL_EMBEDDER_DTYPE = "fp32";
 const LOCAL_EMBEDDER_FALLBACK_DTYPE = "auto";
 /**
@@ -180,7 +186,7 @@ export class LocalEmbedder {
                 }
                 let pipeline;
                 try {
-                    const mod = await import("@huggingface/transformers");
+                    const mod = await transformersLoader();
                     pipeline = mod.pipeline;
                 }
                 catch (importError) {

package/dist/scripts/migrate-storage.js

@@ -7595,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
@@ -16451,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,
@@ -16593,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);
@@ -16762,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();
@@ -17920,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

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 [];