Дмитрий
106 lines
3.7 KiB
JavaScript
106 lines
3.7 KiB
JavaScript
#!/usr/bin/env node
|
|
/**
|
|
* Rule #8 — Classifier-mismatch enforce.
|
|
*
|
|
* Stop hook. Reads classifier output from router-state. If classifier recommended
|
|
* a node with confidence >= threshold AND the turn DIDN'T invoke a matching
|
|
* skill/task — block.
|
|
*
|
|
* Override: "без скилов" / "direct ok" / explicit "override: <reason>" line in
|
|
* assistant text.
|
|
*
|
|
* Spec: docs/superpowers/specs/2026-05-25-enforce-hard-rules-design.md
|
|
*/
|
|
|
|
import {
|
|
readStdin,
|
|
parseEventJson,
|
|
readTranscript,
|
|
lastUserPromptText,
|
|
lastAssistantText,
|
|
turnToolUses,
|
|
findOverride,
|
|
logOverride,
|
|
exitDecision,
|
|
readRouterState,
|
|
} from './enforce-hook-helpers.mjs';
|
|
|
|
const RULE_KEY = 'classifier-mismatch';
|
|
const CONFIDENCE_THRESHOLD = 0.7;
|
|
|
|
const MUTATING_TOOLS = new Set(['Edit', 'Write', 'MultiEdit', 'NotebookEdit', 'Bash']);
|
|
|
|
/** Normalize a node id: strip "superpowers:" / "skill:" prefix; allow #ID. */
|
|
function normalizeNode(s) {
|
|
if (typeof s !== 'string') return '';
|
|
return s.toLowerCase().replace(/^skill:/, '').replace(/^superpowers:/, '');
|
|
}
|
|
|
|
function nodeMatches(recommendation, toolUse) {
|
|
if (!recommendation || !toolUse) return false;
|
|
const rec = normalizeNode(recommendation);
|
|
if (!rec) return false;
|
|
if (toolUse.name === 'Skill') {
|
|
const s = normalizeNode(String(toolUse.input && toolUse.input.skill || ''));
|
|
if (s && (s === rec || s.includes(rec) || rec.includes(s))) return true;
|
|
}
|
|
if (toolUse.name === 'Task') {
|
|
const sub = String(toolUse.input && toolUse.input.subagent_type || '').toLowerCase();
|
|
if (sub && rec.includes(sub)) return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
export function decide({ toolUses, recommendation, confidence, assistantText, override }) {
|
|
// Pure conversation: skip.
|
|
const hasMutating = toolUses.some((u) => MUTATING_TOOLS.has(u.name));
|
|
if (!hasMutating) return { block: false };
|
|
if (override) return { block: false };
|
|
|
|
if (!recommendation) return { block: false };
|
|
if (typeof confidence === 'number' && confidence < CONFIDENCE_THRESHOLD) return { block: false };
|
|
|
|
const matched = toolUses.some((u) => nodeMatches(recommendation, u));
|
|
if (matched) return { block: false };
|
|
|
|
// Allow explicit override: lines like "override: <reason>" in assistant text.
|
|
if (assistantText && /\boverride:\s+\S/i.test(assistantText)) return { block: false };
|
|
|
|
return {
|
|
block: true,
|
|
message: [
|
|
`[enforce-classifier-match] Classifier recommended "${recommendation}" (confidence=${confidence ?? 'n/a'}) but turn did not invoke that skill/node.`,
|
|
`Either:`,
|
|
` - Invoke ${recommendation} via Skill / Task tool, OR`,
|
|
` - Add an explicit "override: <reason>" line in your response, OR`,
|
|
` - Include "без скилов" / "direct ok" in the next user prompt.`,
|
|
].join('\n'),
|
|
};
|
|
}
|
|
|
|
async function main() {
|
|
try {
|
|
const raw = await readStdin();
|
|
const event = parseEventJson(raw);
|
|
const transcript = readTranscript(event.transcript_path);
|
|
const userPrompt = lastUserPromptText(transcript);
|
|
const override = findOverride(userPrompt, RULE_KEY);
|
|
if (override) logOverride(RULE_KEY, override, event.session_id);
|
|
|
|
const state = readRouterState(event.session_id);
|
|
const cls = state && state.classification;
|
|
const recommendation = cls && (cls.recommended_node || cls.recommendedNode);
|
|
const confidence = cls && typeof cls.confidence === 'number' ? cls.confidence : null;
|
|
const toolUses = turnToolUses(transcript);
|
|
const assistantText = lastAssistantText(transcript);
|
|
|
|
const result = decide({ toolUses, recommendation, confidence, assistantText, override });
|
|
exitDecision(result);
|
|
} catch {
|
|
exitDecision({ block: false });
|
|
}
|
|
}
|
|
|
|
const isCli = process.argv[1] && process.argv[1].replace(/\\/g, '/').endsWith('/enforce-classifier-match.mjs');
|
|
if (isCli) main();
|