nano-brain 2026.7.1 → 2026.7.3

package/bin/cli.js

@@ -4,26 +4,45 @@ import { execFileSync } from 'node:child_process';
 import { fileURLToPath } from 'node:url';
 import { dirname, join } from 'node:path';
 import { createRequire } from 'node:module';
+import { existsSync } from 'node:fs';
 
 const __filename = fileURLToPath(import.meta.url);
 const __dirname = dirname(__filename);
 
-const entry = join(__dirname, '..', 'src', 'index.ts');
 const args = process.argv.slice(2);
 
-let tsxBin;
-try {
-  const require = createRequire(join(__dirname, '..', 'package.json'));
-  tsxBin = join(dirname(require.resolve('tsx/package.json')), 'dist', 'cli.mjs');
-} catch {
-  tsxBin = join(__dirname, '..', 'node_modules', '.bin', 'tsx');
-}
+// Prefer the pre-compiled dist/index.js — works on any platform without esbuild.
+// Fall back to tsx (TypeScript source) only when dist is absent (e.g. dev checkout
+// without a build step).
+const distEntry = join(__dirname, '..', 'dist', 'index.js');
+
+if (existsSync(distEntry)) {
+  // Run compiled JS directly — no tsx / esbuild dependency at all.
+  try {
+    execFileSync('node', [distEntry, ...args], { stdio: 'inherit' });
+  } catch (err) {
+    if (err.message && !err.status) {
+      console.error('nano-brain: failed to start —', err.message);
+    }
+    process.exit(err.status || 1);
+  }
+} else {
+  // Dev fallback: run TypeScript source via tsx.
+  const tsEntry = join(__dirname, '..', 'src', 'index.ts');
+  let tsxBin;
+  try {
+    const require = createRequire(join(__dirname, '..', 'package.json'));
+    tsxBin = join(dirname(require.resolve('tsx/package.json')), 'dist', 'cli.mjs');
+  } catch {
+    tsxBin = join(__dirname, '..', 'node_modules', '.bin', 'tsx');
+  }
 
-try {
-  execFileSync('node', [tsxBin, entry, ...args], { stdio: 'inherit' });
-} catch (err) {
-  if (err.message && !err.status) {
-    console.error('nano-brain: failed to start —', err.message);
+  try {
+    execFileSync('node', [tsxBin, tsEntry, ...args], { stdio: 'inherit' });
+  } catch (err) {
+    if (err.message && !err.status) {
+      console.error('nano-brain: failed to start —', err.message);
+    }
+    process.exit(err.status || 1);
   }
-  process.exit(err.status || 1);
 }

package/dist/bandits.d.ts

@@ -0,0 +1,36 @@
+export interface BanditVariant {
+    value: number;
+    successes: number;
+    failures: number;
+}
+export interface BanditConfig {
+    parameterName: string;
+    variants: BanditVariant[];
+    minObservations: number;
+    dampeningFactor: number;
+    bounds: {
+        min: number;
+        max: number;
+    };
+}
+export interface BanditState {
+    configs: BanditConfig[];
+    version: number;
+    updatedAt: string;
+}
+export declare class ThompsonSampler {
+    private configs;
+    private seed;
+    private rng;
+    constructor(configs: BanditConfig[], seed?: number);
+    selectVariant(parameterName: string): number;
+    recordReward(parameterName: string, variantValue: number, success: boolean): void;
+    getState(): BanditConfig[];
+    selectSearchConfig(): Record<string, number>;
+    private betaSample;
+    private gammaSample;
+    private normalSample;
+    private seededRandom;
+}
+export declare const DEFAULT_BANDIT_CONFIGS: BanditConfig[];
+//# sourceMappingURL=bandits.d.ts.map

package/dist/bandits.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bandits.d.ts","sourceRoot":"","sources":["../src/bandits.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,YAAY;IAC3B,aAAa,EAAE,MAAM,CAAC;IACtB,QAAQ,EAAE,aAAa,EAAE,CAAC;IAC1B,eAAe,EAAE,MAAM,CAAC;IACxB,eAAe,EAAE,MAAM,CAAC;IACxB,MAAM,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,GAAG,EAAE,MAAM,CAAA;KAAE,CAAC;CACtC;AAED,MAAM,WAAW,WAAW;IAC1B,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,qBAAa,eAAe;IAC1B,OAAO,CAAC,OAAO,CAAiB;IAChC,OAAO,CAAC,IAAI,CAAgB;IAC5B,OAAO,CAAC,GAAG,CAAe;gBAEd,OAAO,EAAE,YAAY,EAAE,EAAE,IAAI,CAAC,EAAE,MAAM;IAMlD,aAAa,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM;IAuB5C,YAAY,CAAC,aAAa,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IASjF,QAAQ,IAAI,YAAY,EAAE;IAI1B,kBAAkB,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAU5C,OAAO,CAAC,UAAU;IAMlB,OAAO,CAAC,WAAW;IAmBnB,OAAO,CAAC,YAAY;IAMpB,OAAO,CAAC,YAAY;CAOrB;AAED,eAAO,MAAM,sBAAsB,EAAE,YAAY,EAwBhD,CAAC"}

package/dist/bandits.js

@@ -0,0 +1,117 @@
+export class ThompsonSampler {
+    configs;
+    seed;
+    rng;
+    constructor(configs, seed) {
+        this.configs = configs;
+        this.seed = seed ?? null;
+        this.rng = seed !== undefined ? this.seededRandom(seed) : Math.random.bind(Math);
+    }
+    selectVariant(parameterName) {
+        const config = this.configs.find(c => c.parameterName === parameterName);
+        if (!config)
+            throw new Error(`Unknown parameter: ${parameterName}`);
+        const totalObs = config.variants.reduce((sum, v) => sum + v.successes + v.failures, 0);
+        if (totalObs < config.minObservations * config.variants.length) {
+            const idx = Math.floor(this.rng() * config.variants.length);
+            return config.variants[idx].value;
+        }
+        let bestValue = -1;
+        let bestVariant = config.variants[0].value;
+        for (const variant of config.variants) {
+            const sample = this.betaSample(variant.successes, variant.failures);
+            if (sample > bestValue) {
+                bestValue = sample;
+                bestVariant = variant.value;
+            }
+        }
+        return bestVariant;
+    }
+    recordReward(parameterName, variantValue, success) {
+        const config = this.configs.find(c => c.parameterName === parameterName);
+        if (!config)
+            return;
+        const variant = config.variants.find(v => v.value === variantValue);
+        if (!variant)
+            return;
+        if (success)
+            variant.successes++;
+        else
+            variant.failures++;
+    }
+    getState() {
+        return this.configs;
+    }
+    selectSearchConfig() {
+        const result = {};
+        for (const config of this.configs) {
+            let value = this.selectVariant(config.parameterName);
+            value = Math.max(config.bounds.min, Math.min(config.bounds.max, value));
+            result[config.parameterName] = value;
+        }
+        return result;
+    }
+    betaSample(alpha, beta) {
+        const x = this.gammaSample(alpha);
+        const y = this.gammaSample(beta);
+        return x / (x + y);
+    }
+    gammaSample(shape) {
+        if (shape < 1) {
+            return this.gammaSample(shape + 1) * Math.pow(this.rng(), 1 / shape);
+        }
+        const d = shape - 1 / 3;
+        const c = 1 / Math.sqrt(9 * d);
+        while (true) {
+            let x, v;
+            do {
+                x = this.normalSample();
+                v = 1 + c * x;
+            } while (v <= 0);
+            v = v * v * v;
+            const u = this.rng();
+            if (u < 1 - 0.0331 * (x * x) * (x * x))
+                return d * v;
+            if (Math.log(u) < 0.5 * x * x + d * (1 - v + Math.log(v)))
+                return d * v;
+        }
+    }
+    normalSample() {
+        const u1 = this.rng();
+        const u2 = this.rng();
+        return Math.sqrt(-2 * Math.log(u1)) * Math.cos(2 * Math.PI * u2);
+    }
+    seededRandom(seed) {
+        let s = seed;
+        return () => {
+            s = (s * 1664525 + 1013904223) & 0xFFFFFFFF;
+            return (s >>> 0) / 0xFFFFFFFF;
+        };
+    }
+}
+export const DEFAULT_BANDIT_CONFIGS = [
+    {
+        parameterName: 'rrf_k',
+        variants: [
+            { value: 30, successes: 1, failures: 1 },
+            { value: 60, successes: 1, failures: 1 },
+            { value: 90, successes: 1, failures: 1 },
+        ],
+        minObservations: 100,
+        dampeningFactor: 0.1,
+        bounds: { min: 10, max: 120 },
+    },
+    {
+        parameterName: 'centrality_weight',
+        variants: [
+            { value: 0.0, successes: 1, failures: 1 },
+            { value: 0.05, successes: 1, failures: 1 },
+            { value: 0.1, successes: 1, failures: 1 },
+            { value: 0.2, successes: 1, failures: 1 },
+        ],
+        minObservations: 100,
+        dampeningFactor: 0.1,
+        bounds: { min: 0.0, max: 0.5 },
+    },
+];
+//# sourceMappingURL=bandits.js.map

package/dist/bandits.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bandits.js","sourceRoot":"","sources":["../src/bandits.ts"],"names":[],"mappings":"AAsBA,MAAM,OAAO,eAAe;IAClB,OAAO,CAAiB;IACxB,IAAI,CAAgB;IACpB,GAAG,CAAe;IAE1B,YAAY,OAAuB,EAAE,IAAa;QAChD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,IAAI,CAAC;QACzB,IAAI,CAAC,GAAG,GAAG,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACnF,CAAC;IAED,aAAa,CAAC,aAAqB;QACjC,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,KAAK,aAAa,CAAC,CAAC;QACzE,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,aAAa,EAAE,CAAC,CAAC;QAEpE,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QAEvF,IAAI,QAAQ,GAAG,MAAM,CAAC,eAAe,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YAC5D,OAAO,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC;QACpC,CAAC;QAED,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;QACnB,IAAI,WAAW,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;QAC3C,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;YACtC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YACpE,IAAI,MAAM,GAAG,SAAS,EAAE,CAAC;gBACvB,SAAS,GAAG,MAAM,CAAC;gBACnB,WAAW,GAAG,OAAO,CAAC,KAAK,CAAC;YAC9B,CAAC;QACH,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,YAAY,CAAC,aAAqB,EAAE,YAAoB,EAAE,OAAgB;QACxE,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,aAAa,KAAK,aAAa,CAAC,CAAC;QACzE,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,YAAY,CAAC,CAAC;QACpE,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,OAAO;YAAE,OAAO,CAAC,SAAS,EAAE,CAAC;;YAC5B,OAAO,CAAC,QAAQ,EAAE,CAAC;IAC1B,CAAC;IAED,QAAQ;QACN,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED,kBAAkB;QAChB,MAAM,MAAM,GAA2B,EAAE,CAAC;QAC1C,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAClC,IAAI,KAAK,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;YACrD,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAAC;YACxE,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,GAAG,KAAK,CAAC;QACvC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,UAAU,CAAC,KAAa,EAAE,IAAY;QAC5C,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAClC,MAAM,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACjC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;IACrB,CAAC;IAEO,WAAW,CAAC,KAAa;QAC/B,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;YACd,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC;QACvE,CAAC;QACD,MAAM,CAAC,GAAG,KAAK,GAAG,CAAC,GAAC,CAAC,CAAC;QACtB,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/B,OAAO,IAAI,EAAE,CAAC;YACZ,IAAI,CAAS,EAAE,CAAS,CAAC;YACzB,GAAG,CAAC;gBACF,CAAC,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;gBACxB,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAChB,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;YACjB,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACd,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACrB,IAAI,CAAC,GAAG,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC;gBAAE,OAAO,CAAC,GAAG,CAAC,CAAC;YACrD,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAAE,OAAO,CAAC,GAAG,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAEO,YAAY;QAClB,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;IACnE,CAAC;IAEO,YAAY,CAAC,IAAY;QAC/B,IAAI,CAAC,GAAG,IAAI,CAAC;QACb,OAAO,GAAG,EAAE;YACV,CAAC,GAAG,CAAC,CAAC,GAAG,OAAO,GAAG,UAAU,CAAC,GAAG,UAAU,CAAC;YAC5C,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,UAAU,CAAC;QAChC,CAAC,CAAC;IACJ,CAAC;CACF;AAED,MAAM,CAAC,MAAM,sBAAsB,GAAmB;IACpD;QACE,aAAa,EAAE,OAAO;QACtB,QAAQ,EAAE;YACR,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;YACxC,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;YACxC,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;SACzC;QACD,eAAe,EAAE,GAAG;QACpB,eAAe,EAAE,GAAG;QACpB,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE;KAC9B;IACD;QACE,aAAa,EAAE,mBAAmB;QAClC,QAAQ,EAAE;YACR,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;YACzC,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;YAC1C,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;YACzC,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE;SAC1C;QACD,eAAe,EAAE,GAAG;QACpB,eAAe,EAAE,GAAG;QACpB,MAAM,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE;KAC/B;CACF,CAAC"}

package/dist/bench.d.ts

@@ -0,0 +1,3 @@
+import type { GlobalOptions } from './index.js';
+export declare function handleBench(globalOpts: GlobalOptions, commandArgs: string[]): Promise<void>;
+//# sourceMappingURL=bench.d.ts.map

package/dist/bench.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bench.d.ts","sourceRoot":"","sources":["../src/bench.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AA8/BhD,wBAAsB,WAAW,CAAC,UAAU,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAwGjG"}