89 lines
3.8 KiB
JavaScript
89 lines
3.8 KiB
JavaScript
#!/usr/bin/env node
|
|
// build-judge-payloads.mjs — deterministic fan-out prep for the per-claim
|
|
// groundedness judge bake-off. Turns the blind claim manifest + a judge prompt
|
|
// template into 45 ready-to-dispatch subagent payloads (one per file), so the
|
|
// fan-out is reproducible instead of hand-assembled in main context.
|
|
//
|
|
// The v3 fan-out grouped claims by file MANUALLY at dispatch time; that made the
|
|
// exact payloads non-reproducible. This script fixes the construction: same prompt,
|
|
// same per-file claim grouping, every run — so v3.1 (and any future arm) is
|
|
// apples-to-apples and a fresh session can resume with one command, no improvising.
|
|
//
|
|
// Pure string assembly — no LLM, no network, no math. Reads:
|
|
// data/judge-bakeoff-claims.json (blind manifest from extract-judge-claims.mjs)
|
|
// <--prompt> (judge-claim-prompt-vN.md, with <FILE>/<CLAIMS>)
|
|
// Writes (with --write):
|
|
// data/<--out> (array of {file, claim_count, prompt})
|
|
//
|
|
// Usage:
|
|
// node scripts/kb-eval/build-judge-payloads.mjs --prompt judge-claim-prompt-v3.1.md \
|
|
// --out judge-bakeoff-payloads-v3.1.json [--write]
|
|
// (default: print per-file claim counts + a sanity sample; --write persists)
|
|
|
|
import fs from 'node:fs';
|
|
import path from 'node:path';
|
|
import { fileURLToPath } from 'node:url';
|
|
|
|
const __dirname = path.dirname(fileURLToPath(import.meta.url));
|
|
const DATA = path.join(__dirname, 'data');
|
|
|
|
function flag(name, def) {
|
|
const i = process.argv.indexOf(name);
|
|
return i >= 0 ? process.argv[i + 1] : def;
|
|
}
|
|
|
|
const promptName = flag('--prompt');
|
|
if (!promptName) {
|
|
console.error('error: --prompt <judge-claim-prompt-vN.md> is required');
|
|
process.exit(2);
|
|
}
|
|
const promptPath = path.isAbsolute(promptName) ? promptName : path.join(__dirname, promptName);
|
|
if (!fs.existsSync(promptPath)) {
|
|
console.error(`error: prompt template not found: ${promptPath}`);
|
|
process.exit(2);
|
|
}
|
|
const template = fs.readFileSync(promptPath, 'utf8');
|
|
if (!template.includes('<FILE>') || !template.includes('<CLAIMS>')) {
|
|
console.error('error: prompt template must contain both <FILE> and <CLAIMS> placeholders');
|
|
process.exit(2);
|
|
}
|
|
|
|
const manifest = JSON.parse(fs.readFileSync(path.join(DATA, 'judge-bakeoff-claims.json'), 'utf8'));
|
|
const claims = manifest.claims || [];
|
|
|
|
// Group by file, preserving manifest order (deterministic).
|
|
const byFile = new Map();
|
|
for (const c of claims) {
|
|
if (!byFile.has(c.file)) byFile.set(c.file, []);
|
|
byFile.get(c.file).push({
|
|
id: c.id,
|
|
claim: c.claim,
|
|
claim_type: c.claim_type,
|
|
evidence_url: c.evidence_url || null,
|
|
});
|
|
}
|
|
|
|
const payloads = [];
|
|
for (const [file, fileClaims] of byFile) {
|
|
const claimsBlock = JSON.stringify(fileClaims, null, 2);
|
|
const prompt = template.split('<FILE>').join(file).split('<CLAIMS>').join(claimsBlock);
|
|
payloads.push({ file, claim_count: fileClaims.length, prompt });
|
|
}
|
|
|
|
const totalClaims = payloads.reduce((s, p) => s + p.claim_count, 0);
|
|
|
|
if (process.argv.includes('--write')) {
|
|
const outName = flag('--out', `judge-bakeoff-payloads-${path.basename(promptName).replace(/^judge-claim-prompt-/, '').replace(/\.md$/, '')}.json`);
|
|
const outPath = path.join(DATA, outName);
|
|
fs.writeFileSync(outPath, JSON.stringify(payloads, null, 2) + '\n');
|
|
console.log(`wrote ${outPath}`);
|
|
console.log(`${payloads.length} per-file payloads, ${totalClaims} claims total, prompt=${promptName}`);
|
|
} else {
|
|
console.log(`prompt template: ${promptName}`);
|
|
console.log(`${payloads.length} files, ${totalClaims} claims total`);
|
|
console.log('per-file claim counts:');
|
|
for (const p of payloads) console.log(` ${p.claim_count.toString().padStart(2)} ${p.file}`);
|
|
console.log(`\nsample payload[0] head (file=${payloads[0].file}):`);
|
|
console.log(payloads[0].prompt.slice(0, 200) + ' …');
|
|
console.log('\n(dry run — pass --write to persist)');
|
|
}
|