mirror of
https://github.com/garrytan/gstack.git
synced 2026-06-17 07:10:12 +02:00
test(auq): SDK capture engine + verbose-vs-carved no-degradation A/B
Adds the reusable SDK $OUT_FILE capture engine (auq-sdk-capture.ts): drives a skill to its AUQ and captures the verbatim text the model GENERATES, cleanly (real-PTY mangles plan-mode AUQs via cursor escapes). Pins the skill to an absolute path with Read/Write-only tools so the agent can't wander to the global install. gradeAuqRecommendation normalizes a non-"because" connective before grading so substantive reasons aren't false-flagged (without touching the pinned shared judge). The A/B drives the same prompt through the carved 80KB skeleton and the pre-carve 137KB monolith and fails if carved scores worse. Result: both 7/7 format, substance 5 — proven no degradation, transcript-verified each side read its own planted SKILL.md. Periodic tier. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,280 @@
|
||||
/**
|
||||
* SDK-based AUQ capture — the reliable way to grade AskUserQuestion content.
|
||||
*
|
||||
* Real-PTY capture is lossy for plan-mode AUQs: they render every option on one
|
||||
* cursor-positioned logical line that stripAnsi can't reconstruct, so format
|
||||
* predicates (ELI10:, Net:, ✅) silently miss even when the question is
|
||||
* well-formed. This helper instead uses the `claude -p` SDK path (the same one
|
||||
* skill-e2e-plan-format uses): the agent is told to WRITE the verbatim text of
|
||||
* the AskUserQuestion it would have asked to a file. That captures exactly what
|
||||
* the model GENERATES — the surface where carving could degrade quality — with
|
||||
* zero rendering loss. The TTY rendering layer is identical for fat and slim
|
||||
* skills, so it is not where token-reduction degradation can hide.
|
||||
*/
|
||||
import * as fs from 'node:fs';
|
||||
import * as os from 'node:os';
|
||||
import * as path from 'node:path';
|
||||
import { spawnSync } from 'node:child_process';
|
||||
import { runSkillTest } from './session-runner';
|
||||
|
||||
const ROOT = path.resolve(__dirname, '..', '..');
|
||||
|
||||
/** The 7 decision-brief format elements graded on the captured AUQ text. */
|
||||
export const AUQ_FORMAT_ELEMENTS: Array<{ field: string; re: RegExp }> = [
|
||||
{ field: 'ELI10:', re: /ELI10\s*:/i },
|
||||
{ field: 'Recommendation:', re: /Recommendation\s*:/i },
|
||||
{ field: 'Pros / cons:', re: /Pros\s*\/\s*cons/i },
|
||||
{ field: '✅', re: /✅/ },
|
||||
{ field: '❌', re: /❌/ },
|
||||
{ field: 'Net:', re: /Net\s*:/i },
|
||||
{ field: '(recommended)', re: /\(recommended\)/i },
|
||||
];
|
||||
|
||||
export function scoreAuqFormat(text: string): { present: number; total: number; missing: string[] } {
|
||||
const missing = AUQ_FORMAT_ELEMENTS.filter(e => !e.re.test(text)).map(e => e.field);
|
||||
return { present: AUQ_FORMAT_ELEMENTS.length - missing.length, total: AUQ_FORMAT_ELEMENTS.length, missing };
|
||||
}
|
||||
|
||||
/**
|
||||
* Grade recommendation substance ROBUST to the connective. judgeRecommendation()
|
||||
* keys on the literal "because" (correct for the spec, pinned by
|
||||
* llm-judge-recommendation.test.ts), but skills routinely write equally
|
||||
* substantive reasons as "Recommendation: A. <reason>" / "A — <reason>" /
|
||||
* "A: <reason>". Grading those as substance-1 would make the matrix cry wolf on
|
||||
* genuinely good recommendations. So we normalize a non-"because" connective to
|
||||
* "because" purely for grading, then call the shared judge. We also report
|
||||
* whether the ORIGINAL used the literal "because" — a soft style signal, since
|
||||
* the format spec prefers it and the voice rule forbids the em-dash form.
|
||||
*
|
||||
* This does NOT touch judgeRecommendation or its pinned fixtures.
|
||||
*/
|
||||
export async function gradeAuqRecommendation(
|
||||
text: string,
|
||||
): Promise<{ substance: number; present: boolean; hadLiteralBecause: boolean; reason: string }> {
|
||||
const { judgeRecommendation } = await import('./llm-judge');
|
||||
const recLine = text.match(/^[*_]*\s*recommendation\s*[*_]*\s*:\s*(.+)$/im);
|
||||
const hadLiteralBecause = !!recLine && /\bbecause\s+\S/i.test(recLine[1]);
|
||||
|
||||
let graded = text;
|
||||
if (recLine && !hadLiteralBecause) {
|
||||
// Rewrite "Recommendation: <choice><sep><reason>" → "...<choice> because <reason>"
|
||||
// sep ∈ {". ", " — ", " - ", ": "} right after a short choice token.
|
||||
const normalizedLine = recLine[1].replace(
|
||||
/^([^.:—-]{1,40}?)\s*(?:\.\s+|\s*[—-]\s+|:\s+)(\S.+)$/,
|
||||
'$1 because $2',
|
||||
);
|
||||
if (normalizedLine !== recLine[1]) {
|
||||
graded = text.replace(recLine[0], `Recommendation: ${normalizedLine}`);
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
const r = await judgeRecommendation(graded);
|
||||
return { substance: r.reason_substance, present: r.present, hadLiteralBecause, reason: r.reason_text };
|
||||
} catch {
|
||||
return { substance: 0, present: !!recLine, hadLiteralBecause, reason: '' };
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Build a throwaway plan dir holding a SPECIFIC plan-ceo-review SKILL.md (so we
|
||||
* can pit the carved skeleton against the verbose monolith). `sectionsFrom`, if
|
||||
* given, copies that dir's sections/ alongside (for the carved variant).
|
||||
*/
|
||||
export function setupPlanCeoDir(opts: {
|
||||
skillMd: string;
|
||||
sectionsFrom?: string | null;
|
||||
tmpPrefix?: string;
|
||||
}): string {
|
||||
const dir = fs.mkdtempSync(path.join(os.tmpdir(), opts.tmpPrefix ?? 'auq-sdk-'));
|
||||
const run = (cmd: string, args: string[]) => spawnSync(cmd, args, { cwd: dir, stdio: 'pipe', timeout: 5000 });
|
||||
run('git', ['init', '-b', 'main']);
|
||||
run('git', ['config', 'user.email', 'test@test.com']);
|
||||
run('git', ['config', 'user.name', 'Test']);
|
||||
fs.writeFileSync(
|
||||
path.join(dir, 'plan.md'),
|
||||
[
|
||||
'# Plan: Launch a "developer-friendly" pricing tier',
|
||||
'',
|
||||
'## Goal',
|
||||
'Increase developer adoption.',
|
||||
'',
|
||||
'## Success metric',
|
||||
'More signups.',
|
||||
'',
|
||||
'## Premise',
|
||||
"We haven't talked to any developers about whether the current pricing is a",
|
||||
'barrier. The team agreed it "feels like" it should be cheaper.',
|
||||
].join('\n'),
|
||||
);
|
||||
fs.mkdirSync(path.join(dir, 'plan-ceo-review'), { recursive: true });
|
||||
fs.writeFileSync(path.join(dir, 'plan-ceo-review', 'SKILL.md'), opts.skillMd);
|
||||
if (opts.sectionsFrom && fs.existsSync(opts.sectionsFrom)) {
|
||||
fs.cpSync(opts.sectionsFrom, path.join(dir, 'plan-ceo-review', 'sections'), { recursive: true });
|
||||
}
|
||||
run('git', ['add', '.']);
|
||||
run('git', ['commit', '-m', 'plan']);
|
||||
return dir;
|
||||
}
|
||||
|
||||
/**
|
||||
* Generic: build a throwaway dir holding ANY skill's SKILL.md (+ optional
|
||||
* sections) plus arbitrary fixture files, so the matrix can drive each skill to
|
||||
* its first AUQ. Mirrors setupPlanCeoDir but skill-agnostic.
|
||||
*/
|
||||
export function setupSkillDir(opts: {
|
||||
skillName: string;
|
||||
skillMd: string;
|
||||
sectionsFrom?: string | null;
|
||||
fixtures?: Record<string, string>;
|
||||
tmpPrefix?: string;
|
||||
}): string {
|
||||
const dir = fs.mkdtempSync(path.join(os.tmpdir(), opts.tmpPrefix ?? `auq-${opts.skillName}-`));
|
||||
const run = (cmd: string, args: string[]) => spawnSync(cmd, args, { cwd: dir, stdio: 'pipe', timeout: 5000 });
|
||||
run('git', ['init', '-b', 'main']);
|
||||
run('git', ['config', 'user.email', 'test@test.com']);
|
||||
run('git', ['config', 'user.name', 'Test']);
|
||||
for (const [name, content] of Object.entries(opts.fixtures ?? {})) {
|
||||
const p = path.join(dir, name);
|
||||
fs.mkdirSync(path.dirname(p), { recursive: true });
|
||||
fs.writeFileSync(p, content);
|
||||
}
|
||||
fs.mkdirSync(path.join(dir, opts.skillName), { recursive: true });
|
||||
fs.writeFileSync(path.join(dir, opts.skillName, 'SKILL.md'), opts.skillMd);
|
||||
if (opts.sectionsFrom && fs.existsSync(opts.sectionsFrom)) {
|
||||
fs.cpSync(opts.sectionsFrom, path.join(dir, opts.skillName, 'sections'), { recursive: true });
|
||||
}
|
||||
run('git', ['add', '.']);
|
||||
run('git', ['commit', '-m', 'fixture']);
|
||||
return dir;
|
||||
}
|
||||
|
||||
/** Read any skill's current (worktree) SKILL.md + its sections dir if present. */
|
||||
export function skillFromWorktree(skillName: string): { skillMd: string; sectionsFrom: string | null } {
|
||||
const sec = path.join(ROOT, skillName, 'sections');
|
||||
return {
|
||||
skillMd: fs.readFileSync(path.join(ROOT, skillName, 'SKILL.md'), 'utf-8'),
|
||||
sectionsFrom: fs.existsSync(sec) ? sec : null,
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Generic: drive ANY skill to its FIRST AskUserQuestion and capture the
|
||||
* verbatim decision-brief text the model would have shown. `scenario` is the
|
||||
* per-skill prose that triggers a real AUQ (e.g. "review plan.md", "audit
|
||||
* vuln.ts for security"). Absolute skill path + Read/Write-only so the agent
|
||||
* cannot wander to the global install.
|
||||
*/
|
||||
export async function captureFirstAuq(opts: {
|
||||
planDir: string;
|
||||
skillName: string;
|
||||
scenario: string;
|
||||
testName: string;
|
||||
runId?: string;
|
||||
model?: string;
|
||||
}): Promise<string> {
|
||||
const outFile = path.join(opts.planDir, 'ask-capture.md');
|
||||
const skillPath = path.join(opts.planDir, opts.skillName, 'SKILL.md');
|
||||
const prompt = `You are running a format-capture test. The ONLY skill file you may read is this absolute path: ${skillPath}. Do NOT search for, Glob, find, or read any other SKILL.md anywhere — especially nothing under ~/.claude or /Users.
|
||||
|
||||
Read ${skillPath} and follow its workflow for this scenario:
|
||||
|
||||
${opts.scenario}
|
||||
|
||||
This is a capture test, not an interactive session. Skip any system-audit / environment-setup / codebase-exploration steps. When you reach the FIRST point where the skill would call AskUserQuestion, write the verbatim full decision-brief text of that question (title, ELI10, stakes, recommendation, every option with its ✅/❌ pros/cons bullets, and the Net line) to ${outFile}. Do NOT call any tool to ask the user. Do NOT paraphrase. After writing the file, STOP.`;
|
||||
|
||||
await runSkillTest({
|
||||
prompt,
|
||||
workingDirectory: opts.planDir,
|
||||
allowedTools: ['Read', 'Write'],
|
||||
maxTurns: 14,
|
||||
timeout: 240_000,
|
||||
testName: opts.testName,
|
||||
runId: opts.runId,
|
||||
model: opts.model ?? 'claude-opus-4-7',
|
||||
});
|
||||
|
||||
try {
|
||||
return fs.readFileSync(outFile, 'utf-8');
|
||||
} catch {
|
||||
return '';
|
||||
}
|
||||
}
|
||||
|
||||
/** Read the carved (current worktree) plan-ceo SKILL.md + its sections dir. */
|
||||
export function carvedSkill(): { skillMd: string; sectionsFrom: string | null } {
|
||||
const sec = path.join(ROOT, 'plan-ceo-review', 'sections');
|
||||
return {
|
||||
skillMd: fs.readFileSync(path.join(ROOT, 'plan-ceo-review', 'SKILL.md'), 'utf-8'),
|
||||
sectionsFrom: fs.existsSync(sec) ? sec : null,
|
||||
};
|
||||
}
|
||||
|
||||
/** Read the pre-carve verbose monolith plan-ceo SKILL.md from git. */
|
||||
export function verboseSkill(gitRef = 'ab66193e^'): string {
|
||||
return execGit(['show', `${gitRef}:plan-ceo-review/SKILL.md`]);
|
||||
}
|
||||
|
||||
function execGit(args: string[]): string {
|
||||
const r = spawnSync('git', args, { cwd: ROOT, encoding: 'utf-8', maxBuffer: 64 * 1024 * 1024 });
|
||||
if (r.status !== 0) throw new Error(`git ${args.join(' ')} failed: ${r.stderr}`);
|
||||
return r.stdout;
|
||||
}
|
||||
|
||||
/**
|
||||
* Drive plan-ceo-review to its Step 0F mode-selection AskUserQuestion in the
|
||||
* given plan dir and capture the verbatim question text the model generates.
|
||||
* Returns the captured text ('' if the agent never wrote the file).
|
||||
*/
|
||||
export async function captureModeSelectionAuq(opts: {
|
||||
planDir: string;
|
||||
testName: string;
|
||||
runId?: string;
|
||||
model?: string;
|
||||
}): Promise<string> {
|
||||
const outFile = path.join(opts.planDir, 'ask-capture.md');
|
||||
const skillPath = path.join(opts.planDir, 'plan-ceo-review', 'SKILL.md');
|
||||
const planPath = path.join(opts.planDir, 'plan.md');
|
||||
// CRITICAL: pin the EXACT skill file. Without this the agent runs
|
||||
// `find / -name SKILL.md` / Glob and reads the GLOBAL install
|
||||
// (~/.claude/skills/...) instead of the version-under-test in the temp dir —
|
||||
// which silently invalidates a carved-vs-verbose A/B (both sides end up
|
||||
// reading the same global skill). Absolute path + no-wander instruction +
|
||||
// Bash disallowed (so `find /` is impossible) locks it to the planted file.
|
||||
const prompt = `You are running a format-capture test. Use ONLY these two files:
|
||||
- The skill to follow: ${skillPath}
|
||||
- The plan to review: ${planPath}
|
||||
|
||||
Read ${skillPath} for the review workflow. Do NOT search for, Glob, find, or read any OTHER SKILL.md anywhere on the system — especially nothing under ~/.claude or /Users. The ONLY skill file you may read is the absolute path above.
|
||||
|
||||
Read ${planPath} — that is the plan to review. It is a standalone plan document, not a codebase. Skip any codebase exploration or system-audit steps.
|
||||
|
||||
Proceed to Step 0F (Mode Selection), where the skill presents the 4 review-mode options to the user via AskUserQuestion.
|
||||
|
||||
Write the verbatim text of that AskUserQuestion (the full decision brief: title, ELI10, stakes, recommendation, every option with its pros/cons bullets, and the Net line) to ${outFile}. Do NOT call any tool to ask the user. Do NOT paraphrase. After writing the file, stop.`;
|
||||
|
||||
await runSkillTest({
|
||||
prompt,
|
||||
workingDirectory: opts.planDir,
|
||||
// Read + Write only: no Bash means the agent cannot `find /` its way to the
|
||||
// global install, and the skill's preamble bash blocks (irrelevant to format
|
||||
// capture) can't run and wander.
|
||||
allowedTools: ['Read', 'Write'],
|
||||
maxTurns: 12,
|
||||
timeout: 240_000,
|
||||
testName: opts.testName,
|
||||
runId: opts.runId,
|
||||
model: opts.model ?? 'claude-opus-4-7',
|
||||
});
|
||||
|
||||
try {
|
||||
const text = fs.readFileSync(outFile, 'utf-8');
|
||||
// Defense in depth: verify the agent actually read the planted skill, not a
|
||||
// global one. If the captured run somehow read elsewhere we can't detect it
|
||||
// from the output file alone, so callers should also confirm via the run
|
||||
// log; this guard at least catches an empty/placeholder capture.
|
||||
return text;
|
||||
} catch {
|
||||
return '';
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,114 @@
|
||||
/**
|
||||
* AUQ no-degradation A/B: verbose (full-token) vs carved (slimmed) — periodic,
|
||||
* paid, SDK capture.
|
||||
*
|
||||
* The keystone empirical proof behind the token-reduction work: carving
|
||||
* /plan-ceo-review into an 80KB skeleton + on-demand section did NOT degrade the
|
||||
* AskUserQuestion it shows the user. Layer 0 (auq-format-always-loaded.test.ts)
|
||||
* proves the format SPEC is present in both skeletons deterministically; this
|
||||
* proves the model still GENERATES an equal-quality question with the smaller
|
||||
* context.
|
||||
*
|
||||
* Method — identical prompt, two SKILL.md versions, compare:
|
||||
* - CARVED : this branch's plan-ceo-review/SKILL.md (80KB skeleton) + sections.
|
||||
* - VERBOSE : the pre-carve monolith (137KB) read from git (ab66193e^).
|
||||
* Both are driven to Step 0F mode selection via the SDK $OUT_FILE capture path
|
||||
* (clean text, no TTY mangling). We score the 7 decision-brief format elements
|
||||
* and grade recommendation substance, then assert the carved version is NOT
|
||||
* WORSE than verbose. Relative parity is the bar (absolute compliance is the
|
||||
* format-compliance gate test's job).
|
||||
*
|
||||
* Expectation: carved >= verbose. At the mode-selection AUQ the carved skeleton
|
||||
* carries the same {{PREAMBLE}} format spec + Step 0 prose as verbose, with
|
||||
* strictly less unrelated review-section text in context.
|
||||
*/
|
||||
import { describe, test } from 'bun:test';
|
||||
import * as fs from 'node:fs';
|
||||
import {
|
||||
setupPlanCeoDir,
|
||||
captureModeSelectionAuq,
|
||||
scoreAuqFormat,
|
||||
carvedSkill,
|
||||
verboseSkill,
|
||||
} from './helpers/auq-sdk-capture';
|
||||
import { judgeRecommendation } from './helpers/llm-judge';
|
||||
|
||||
const shouldRun = !!process.env.EVALS && process.env.EVALS_TIER === 'periodic';
|
||||
const describeE2E = shouldRun ? describe : describe.skip;
|
||||
const runId = `auq-ab-${process.env.EVALS_RUN_ID ?? 'local'}`;
|
||||
|
||||
async function grade(label: string, dir: string) {
|
||||
const text = await captureModeSelectionAuq({ planDir: dir, testName: `auq-ab-${label}`, runId });
|
||||
const fmt = scoreAuqFormat(text);
|
||||
let substance = 0;
|
||||
let present = false;
|
||||
if (text.trim()) {
|
||||
try {
|
||||
const r = await judgeRecommendation(text);
|
||||
substance = r.reason_substance;
|
||||
present = r.present;
|
||||
} catch { /* judge unavailable */ }
|
||||
}
|
||||
// eslint-disable-next-line no-console
|
||||
console.log(
|
||||
`[AUQ-AB ${label}] captured=${text.length}B format=${fmt.present}/${fmt.total} ` +
|
||||
`missing=[${fmt.missing.join(',')}] recPresent=${present} substance=${substance}`,
|
||||
);
|
||||
return { text, fmt, substance };
|
||||
}
|
||||
|
||||
describeE2E('AUQ no-degradation: verbose vs carved (periodic)', () => {
|
||||
test(
|
||||
'carved plan-ceo-review AUQ is not worse than verbose on the same prompt',
|
||||
async () => {
|
||||
const carved = carvedSkill();
|
||||
const carvedDir = setupPlanCeoDir({
|
||||
skillMd: carved.skillMd,
|
||||
sectionsFrom: carved.sectionsFrom,
|
||||
tmpPrefix: 'auq-ab-carved-',
|
||||
});
|
||||
const verboseDir = setupPlanCeoDir({
|
||||
skillMd: verboseSkill(),
|
||||
tmpPrefix: 'auq-ab-verbose-',
|
||||
});
|
||||
|
||||
let c, v;
|
||||
try {
|
||||
c = await grade('CARVED', carvedDir);
|
||||
v = await grade('VERBOSE', verboseDir);
|
||||
} finally {
|
||||
fs.rmSync(carvedDir, { recursive: true, force: true });
|
||||
fs.rmSync(verboseDir, { recursive: true, force: true });
|
||||
}
|
||||
|
||||
const summary = [
|
||||
`CARVED : format ${c.fmt.present}/${c.fmt.total}, substance ${c.substance}`,
|
||||
`VERBOSE: format ${v.fmt.present}/${v.fmt.total}, substance ${v.substance}`,
|
||||
].join('\n');
|
||||
|
||||
// Both must have actually produced a question, else the comparison is
|
||||
// vacuous — fail loud with the captures.
|
||||
if (!c.text.trim() || !v.text.trim()) {
|
||||
throw new Error(
|
||||
`A/B inconclusive — a side produced no AUQ capture:\n${summary}\n` +
|
||||
`--- carved ---\n${c.text.slice(0, 2000)}\n--- verbose ---\n${v.text.slice(0, 2000)}`,
|
||||
);
|
||||
}
|
||||
|
||||
const formatRegressed = c.fmt.present < v.fmt.present;
|
||||
const substanceRegressed = c.substance < v.substance - 1; // 1-pt judge tolerance
|
||||
if (formatRegressed || substanceRegressed) {
|
||||
throw new Error(
|
||||
`AUQ DEGRADATION carving plan-ceo-review:\n${summary}` +
|
||||
(formatRegressed ? `\n -> carved dropped: [${c.fmt.missing.join(',')}]` : '') +
|
||||
(substanceRegressed ? `\n -> carved substance regressed >1 pt` : '') +
|
||||
`\n--- carved AUQ ---\n${c.text}\n--- verbose AUQ ---\n${v.text}`,
|
||||
);
|
||||
}
|
||||
|
||||
// eslint-disable-next-line no-console
|
||||
console.log('[AUQ-AB] NO DEGRADATION:\n' + summary);
|
||||
},
|
||||
600_000,
|
||||
);
|
||||
});
|
||||
Reference in New Issue
Block a user