braintrust-lite 0.1.7 → 0.1.9

package/bin/consult

@@ -1,79 +0,0 @@
-#!/usr/bin/env node
-'use strict';
-
-// CLI entry point for braintrust-lite.
-// Usage: consult [options] "prompt"
-//        cat file | consult "review this"
-
-import { readFileSync } from 'fs';
-import { resolve } from 'path';
-import { consult } from '../src/consult.js';
-import { formatAsMarkdown, formatAsJson } from '../src/format.js';
-
-// ─── Arg parsing ──────────────────────────────────────────────────────────────
-
-const flags = { skip: [] };
-const positional = [];
-const argv = process.argv.slice(2);
-
-for (let i = 0; i < argv.length; i++) {
-  const a = argv[i];
-  if (a === '--skip') { flags.skip.push(argv[++i]); continue; }
-  if (a === '--only') { flags.only = argv[++i]; continue; }
-  if (a === '--timeout') { flags.timeout = Number(argv[++i]); continue; }
-  if (a === '--dir') { flags.dir = argv[++i]; continue; }
-  if (a === '--json') { flags.json = true; continue; }
-  if (a === '--help' || a === '-h') { printHelp(); process.exit(0); }
-  positional.push(a);
-}
-
-function printHelp() {
-  console.error(`Usage: consult [options] "your question"
-       cat file | consult "explain this"
-
-Options:
-  --only <model>     Only run one model: codex | gemini | claude
-  --skip <model>     Skip a model (repeatable)
-  --timeout <sec>    Per-model timeout in seconds (default: 90)
-  --dir <path>       Working directory for CLI subprocesses
-  --json             Output full JSON instead of markdown
-  --help             Show this help`);
-}
-
-// ─── Prompt ───────────────────────────────────────────────────────────────────
-
-let prompt = positional.join(' ');
-
-if (!process.stdin.isTTY) {
-  const stdin = readFileSync(0, 'utf8').trim();
-  if (stdin) prompt = prompt ? `${prompt}\n\n<context>\n${stdin}\n</context>` : stdin;
-}
-
-if (!prompt) {
-  printHelp();
-  process.exit(1);
-}
-
-// ─── Run ─────────────────────────────────────────────────────────────────────
-
-const { results, mapping } = await consult({
-  prompt,
-  only: flags.only,
-  skip: flags.skip,
-  timeoutMs: flags.timeout != null ? flags.timeout * 1000 : 90_000,
-  cwd: flags.dir ? resolve(flags.dir) : undefined,
-  blind: !flags.only, // blind mode only useful when multiple providers run
-});
-
-// Progress summary to stderr
-for (const r of results) {
-  const status = r.error ? `⚠ ${r.error}` : `✓ ${(r.duration_ms / 1000).toFixed(1)}s`;
-  process.stderr.write(`[${r.provider}: ${status}]\n`);
-}
-
-// Output to stdout
-if (flags.json) {
-  console.log(formatAsJson(prompt, results, mapping));
-} else {
-  console.log('\n' + formatAsMarkdown(results, mapping));
-}

package/scripts/setup.js

@@ -1,66 +0,0 @@
-#!/usr/bin/env node
-import { mkdirSync, writeFileSync, existsSync, readFileSync, copyFileSync } from 'fs';
-import { join, dirname } from 'path';
-import { fileURLToPath } from 'url';
-import { spawnSync } from 'child_process';
-import https from 'https';
-
-const __dirname = dirname(fileURLToPath(import.meta.url));
-const SKILL_DIR  = join(process.env.HOME || '/tmp', '.claude', 'skills', 'consult');
-const SKILL_PATH = join(SKILL_DIR, 'SKILL.md');
-const SKILL_URL  = 'https://raw.githubusercontent.com/HongjieRen/braintrust-lite/main/skills/consult/SKILL.md';
-const G = '\x1b[32m✓\x1b[0m', R = '\x1b[31m✗\x1b[0m';
-
-function fetch(url) {
-  return new Promise((resolve, reject) => {
-    const req = https.get(url, { timeout: 10000 }, res => {
-      if (res.statusCode === 301 || res.statusCode === 302) return fetch(res.headers.location).then(resolve).catch(reject);
-      if (res.statusCode !== 200) return reject(new Error(`HTTP ${res.statusCode}`));
-      let body = ''; res.on('data', d => { body += d; }); res.on('end', () => resolve(body));
-    });
-    req.on('error', reject); req.on('timeout', () => { req.destroy(); reject(new Error('timeout')); });
-  });
-}
-
-function getVersion(p) { try { return (readFileSync(p,'utf8').match(/^version:\s*(.+)$/m)||[])[1]?.trim(); } catch { return null; } }
-
-console.log('\nbraintrust-lite setup\n');
-console.log('Installing consult skill:');
-mkdirSync(SKILL_DIR, { recursive: true });
-if (existsSync(SKILL_PATH)) copyFileSync(SKILL_PATH, SKILL_PATH + '.bak');
-
-const bundled = join(__dirname, '..', 'skills', 'consult', 'SKILL.md');
-if (existsSync(bundled)) {
-  copyFileSync(bundled, SKILL_PATH);
-  console.log(`  ${G}  SKILL.md installed  (v${getVersion(SKILL_PATH)})`);
-} else {
-  try {
-    const content = await fetch(SKILL_URL);
-    writeFileSync(SKILL_PATH, content, 'utf8');
-    console.log(`  ${G}  SKILL.md downloaded from GitHub  (v${getVersion(SKILL_PATH)})`);
-  } catch(e) {
-    console.log(`  ${R}  SKILL.md download failed: ${e.message}`);
-    process.exitCode = 1;
-  }
-}
-
-console.log('\nRegistering MCP server:');
-const claudeOk = spawnSync('claude', ['--version'], { timeout: 5000 }).status === 0;
-if (!claudeOk) {
-  console.log(`  ${R}  claude CLI not found — install Claude Code first`);
-  console.log(`       Then: claude mcp add braintrust-lite -- npx -y braintrust-lite@~0.1`);
-  process.exitCode = 1;
-} else {
-  const list = spawnSync('claude', ['mcp', 'list'], { encoding: 'utf8', timeout: 5000 });
-  if ((list.stdout || '').includes('braintrust-lite')) {
-    console.log(`  ${G}  MCP server already registered`);
-  } else {
-    const r = spawnSync('claude', ['mcp', 'add', 'braintrust-lite', '--', 'npx', '-y', 'braintrust-lite@~0.1'], { encoding: 'utf8', timeout: 10000 });
-    if (r.status === 0) console.log(`  ${G}  MCP server registered  (npx braintrust-lite@~0.1)`);
-    else { console.log(`  ${R}  MCP registration failed`); console.log(`       Manual: claude mcp add braintrust-lite -- npx -y braintrust-lite@~0.1`); process.exitCode = 1; }
-  }
-}
-
-console.log();
-console.log(process.exitCode ? '  \x1b[33mSetup done with warnings — fix issues above.\x1b[0m' : '  \x1b[32mDone! Restart Claude Code, then use /consult.\x1b[0m');
-console.log();

package/src/consult.js

@@ -1,81 +0,0 @@
-import {
-  runProcess,
-  adaptCodex, adaptGemini, adaptClaude,
-  CODEX_ARGS_PREFIX, GEMINI_ARGS_PREFIX, CLAUDE_ARGS_PREFIX,
-} from './providers.js';
-
-const SYSTEM_PROMPT = `你是一个独立思考的高级专家。请基于自己的判断给出高质量、可执行的回答。
-要求：独立思考，不假设其他专家会补充；区分结论、依据、假设、风险；简洁但完整。`;
-
-const PROVIDERS = {
-  codex:  { cmd: 'codex',  buildArgs: p => [...CODEX_ARGS_PREFIX,  `${SYSTEM_PROMPT}\n\n${p}`], adapt: adaptCodex  },
-  gemini: { cmd: 'gemini', buildArgs: p => ['-p', `${SYSTEM_PROMPT}\n\n${p}`, ...GEMINI_ARGS_PREFIX], adapt: adaptGemini },
-  claude: { cmd: 'claude', buildArgs: p => [...CLAUDE_ARGS_PREFIX,  `${SYSTEM_PROMPT}\n\n${p}`], adapt: adaptClaude },
-};
-
-function shuffle(arr) {
-  for (let i = arr.length - 1; i > 0; i--) {
-    const j = Math.floor(Math.random() * (i + 1));
-    [arr[i], arr[j]] = [arr[j], arr[i]];
-  }
-  return arr;
-}
-
-function anonymize(results) {
-  const labels = ['Model A', 'Model B', 'Model C', 'Model D', 'Model E'];
-  const shuffled = shuffle([...results]);
-  const mapping = {};
-  const anonymized = shuffled.map((r, i) => {
-    mapping[labels[i]] = r.provider;
-    return { ...r, provider: labels[i] };
-  });
-  return { results: anonymized, mapping };
-}
-
-async function runOne(name, prompt, { cwd, timeoutMs }) {
-  const p = PROVIDERS[name];
-  const start = Date.now();
-  const raw = await runProcess(p.cmd, p.buildArgs(prompt), { cwd, timeoutMs });
-  const duration_ms = Date.now() - start;
-
-  const error_type = raw.error_type || null;
-  const error = error_type === 'enoent' ? 'not installed'
-    : error_type === 'timeout' ? 'timeout'
-    : error_type === 'nonzero' ? `exit ${raw.code}`
-    : error_type ? error_type
-    : null;
-
-  const { content } = error ? { content: '' } : p.adapt(raw);
-  return { provider: name, content, duration_ms, error, error_type };
-}
-
-/**
- * Consult Codex, Gemini, and/or Claude in parallel.
- * Returns { results, mapping, successCount, totalCount }
- */
-export async function consult({ prompt, only, skip = [], timeoutMs = 90_000, cwd, blind = true } = {}) {
-  const targets = Object.keys(PROVIDERS)
-    .filter(name => (only ? name === only : true))
-    .filter(name => !skip.includes(name));
-
-  if (targets.length === 0) throw new Error('No providers selected — check --only / --skip flags.');
-
-  const settled = await Promise.allSettled(
-    targets.map(name => runOne(name, prompt, { cwd, timeoutMs }))
-  );
-
-  const results = targets.map((name, i) =>
-    settled[i].status === 'fulfilled'
-      ? settled[i].value
-      : { provider: name, content: '', duration_ms: 0, error: 'rejected', error_type: 'rejected' }
-  );
-
-  const successCount = results.filter(r => !r.error).length;
-  const totalCount = results.length;
-
-  if (blind) {
-    const { results: anon, mapping } = anonymize(results);
-    return { results: anon, mapping, successCount, totalCount };
-  }
-  return { results, mapping: null, successCount, totalCount };
-}

package/src/providers.js

@@ -1,91 +0,0 @@
-import { spawn } from 'child_process';
-
-// ─── Provider argv constants ──────────────────────────────────────────────────
-
-export const CODEX_ARGS_PREFIX = ['exec', '--json', '--skip-git-repo-check', '--ephemeral'];
-export const GEMINI_ARGS_PREFIX = ['-o', 'json'];
-export const CLAUDE_ARGS_PREFIX = ['--output-format', 'json', '-p'];
-
-// ─── Process runner ───────────────────────────────────────────────────────────
-
-/**
- * Spawn a subprocess with an AbortController-based timeout.
- * Returns { stdout, stderr, code, error_type } — never rejects.
- * error_type: 'enoent' | 'timeout' | 'nonzero' | 'spawn_error' | null
- */
-export function runProcess(cmd, args, { cwd, timeoutMs } = {}) {
-  const ac = new AbortController();
-  const proc = spawn(cmd, args, {
-    signal: ac.signal,
-    stdio: ['ignore', 'pipe', 'pipe'],
-    ...(cwd ? { cwd } : {}),
-  });
-
-  let stdout = '';
-  let stderr = '';
-  proc.stdout.on('data', d => { stdout += d; });
-  proc.stderr.on('data', d => { stderr += d; });
-
-  const timer = timeoutMs ? setTimeout(() => ac.abort(), timeoutMs) : null;
-
-  return new Promise(resolve => {
-    let resolved = false;
-    const done = (code, error_type = null) => {
-      if (resolved) return;
-      resolved = true;
-      if (timer) clearTimeout(timer);
-      resolve({ stdout, stderr, code, error_type });
-    };
-    proc.on('close', code => done(code, code !== 0 ? 'nonzero' : null));
-    proc.on('error', err => {
-      if (err.name === 'AbortError') done('timeout', 'timeout');
-      else if (err.code === 'ENOENT') done(-1, 'enoent');
-      else done(-1, 'spawn_error');
-    });
-  });
-}
-
-// ─── Adapters ─────────────────────────────────────────────────────────────────
-
-export function fallback(rawStdout) {
-  return { content: rawStdout.slice(-2000).trim() || '[no output]', parse_mode: 'fallback' };
-}
-
-export function adaptCodex(raw) {
-  try {
-    const events = raw.stdout.trim().split('\n').flatMap(l => {
-      try { return [JSON.parse(l)]; } catch { return []; }
-    });
-    const msg = events.filter(e => e.type === 'item.completed' && e.item?.type === 'agent_message').pop()
-      ?? events.filter(e => e.type === 'item.completed').pop();
-    if (msg?.item?.text) return { content: msg.item.text, parse_mode: 'jsonl' };
-  } catch { /* fall through */ }
-  return fallback(raw.stdout);
-}
-
-export function parseGeminiResponse(stdout) {
-  const jsonStart = stdout.indexOf('{');
-  if (jsonStart === -1) return null;
-  const j = JSON.parse(stdout.slice(jsonStart));
-  if (typeof j.response === 'string') return j.response;
-  for (const v of Object.values(j)) {
-    if (v && typeof v === 'object' && typeof v.response === 'string') return v.response;
-  }
-  return null;
-}
-
-export function adaptGemini(raw) {
-  try {
-    const response = parseGeminiResponse(raw.stdout);
-    if (response) return { content: response, parse_mode: 'json' };
-  } catch { /* fall through */ }
-  return fallback(raw.stdout);
-}
-
-export function adaptClaude(raw) {
-  try {
-    const j = JSON.parse(raw.stdout);
-    if (typeof j.result === 'string') return { content: j.result, parse_mode: 'json' };
-  } catch { /* fall through */ }
-  return fallback(raw.stdout);
-}