mirror of
https://github.com/norandom/OpenUltraSAST.git
synced 2026-10-16 19:59:26 +00:00
feat(learning-harness): failure facts and the two proposers
Task 3.5. A proposer is handed facts, never the scores it is about to be judged on: the misses and leaks of one family on the train split, what each run actually did, and the hypotheses this family already tried and lost. A holdout pair never appears in those facts. Self-diagnosis from free narrative confabulates; structured facts are what make a proposal answerable. A proposal is one hypothesis, one lever and one file inside one family's directory, and the narrowness is the safety property: a round can only touch what it targets, so every other family is frozen by construction rather than by promise. A proposal that names two files, an unknown file, a path leaving the directory, or a prompt over the cap is refused by name, and the write roots exclude the verifier code and the corpus. The scripted proposer runs out rather than repeating itself. The HarnessX proposer reports its write roots and degrades with a recorded reason when the extra is absent; its evolve call lands with the live rounds. Co-Authored-By: Claude Opus 5 (1M context) <[email protected]> Claude-Session: https://claude.ai/code/session_01T7XD7x5KonMdmGRkp63EQG
This commit is contained in:
co-authored by
Claude Opus 5
parent
127dbef8fc
commit
ad7c98eb50
@@ -0,0 +1,213 @@
|
||||
"""What a round proposes, and who may propose it (learning-harness, Req 9.1, 9.2, 5.4, 7.6).
|
||||
|
||||
A proposer is handed facts, never scores it is about to be judged on: the misses and leaks of one family
|
||||
on the train split, what each run actually did, and the hypotheses this family already tried and lost.
|
||||
Self-diagnosis from free narrative confabulates; structured facts are what make a proposal answerable.
|
||||
|
||||
A proposal is one hypothesis, one lever, and one file inside one family's directory. The narrowness is
|
||||
the safety property: a round can only touch what it targets, so every other family is frozen by
|
||||
construction rather than by promise, and the verifier code and the corpus stay outside every write root.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from collections.abc import Mapping, Sequence
|
||||
from dataclasses import dataclass, field
|
||||
from pathlib import Path
|
||||
from typing import Literal, Protocol
|
||||
|
||||
from ..harness_ext import has_harnessx
|
||||
from .detectors import MAX_PROMPT_CHARS, FamilyConfig
|
||||
from .journal import LearningJournal
|
||||
from .scoring import PairFamilyScore
|
||||
|
||||
Lever = Literal["tool", "checklist", "counterexample", "memory", "prompt"]
|
||||
LEVERS: tuple[Lever, ...] = ("tool", "checklist", "counterexample", "memory", "prompt")
|
||||
# The only files a round may write, one per lever. Anything else is a proposal reaching beyond its family.
|
||||
EDITABLE_FILES = {
|
||||
"tool": "family.toml",
|
||||
"checklist": "checklist.md",
|
||||
"counterexample": "counterexamples.jsonl",
|
||||
"memory": "hard_negatives.jsonl",
|
||||
"prompt": "prompt.md",
|
||||
}
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class PairFact:
|
||||
"""What one pair did, in terms a proposer can act on."""
|
||||
|
||||
pair: str
|
||||
outcome: str
|
||||
runs: tuple[str, ...] = ()
|
||||
other_findings: int = 0
|
||||
rung: str = "suspicion"
|
||||
|
||||
def to_dict(self) -> dict[str, object]:
|
||||
return {
|
||||
"pair": self.pair,
|
||||
"outcome": self.outcome,
|
||||
"runs": list(self.runs),
|
||||
"other_findings": self.other_findings,
|
||||
"rung": self.rung,
|
||||
}
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class FailureFacts:
|
||||
family: str
|
||||
config: FamilyConfig
|
||||
misses: tuple[PairFact, ...] = ()
|
||||
leaks: tuple[PairFact, ...] = ()
|
||||
rejected: tuple[str, ...] = ()
|
||||
|
||||
def to_dict(self) -> dict[str, object]:
|
||||
return {
|
||||
"family": self.family,
|
||||
"config_version": self.config.version,
|
||||
"tools": list(self.config.tools),
|
||||
"misses": [item.to_dict() for item in self.misses],
|
||||
"leaks": [item.to_dict() for item in self.leaks],
|
||||
"rejected": list(self.rejected),
|
||||
}
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class Proposal:
|
||||
hypothesis: str
|
||||
lever: Lever
|
||||
change: Mapping[str, str] # file name inside the family directory -> its new contents
|
||||
predicted_affected: tuple[str, ...] = ()
|
||||
predicted_at_risk: tuple[str, ...] = ()
|
||||
|
||||
def to_dict(self) -> dict[str, object]:
|
||||
return {
|
||||
"hypothesis": self.hypothesis,
|
||||
"lever": self.lever,
|
||||
"change": dict(self.change),
|
||||
"predicted_affected": list(self.predicted_affected),
|
||||
"predicted_at_risk": list(self.predicted_at_risk),
|
||||
}
|
||||
|
||||
|
||||
class Proposer(Protocol):
|
||||
def propose(self, facts: FailureFacts) -> Proposal | None: ...
|
||||
|
||||
|
||||
def build_failure_facts(
|
||||
scores: Sequence[PairFamilyScore],
|
||||
*,
|
||||
family: str,
|
||||
config: FamilyConfig,
|
||||
journal: LearningJournal,
|
||||
train_pairs: set[str],
|
||||
) -> FailureFacts:
|
||||
"""The facts of one family's failures on the train split. A holdout pair never appears here (Req 5.4)."""
|
||||
mine = [item for item in scores if item.family == family and item.pair in train_pairs and item.outcome != "unscorable"]
|
||||
return FailureFacts(
|
||||
family=family,
|
||||
config=config,
|
||||
misses=tuple(_fact(item, item.other_findings_vuln) for item in mine if item.outcome in {"both_silent", "reversed"}),
|
||||
leaks=tuple(_fact(item, item.other_findings_fixed) for item in mine if item.outcome in {"both_flagged", "reversed"}),
|
||||
rejected=journal.rejected_buffer(family),
|
||||
)
|
||||
|
||||
|
||||
def refuse_proposal(proposal: Proposal, config: FamilyConfig) -> str | None:
|
||||
"""None when the proposal is one legal change to its own family; otherwise why it is refused."""
|
||||
if proposal.lever not in LEVERS:
|
||||
return f"unknown lever {proposal.lever!r}; expected one of {', '.join(LEVERS)}"
|
||||
if len(proposal.change) != 1:
|
||||
return f"a round changes one file, not {len(proposal.change)}"
|
||||
name, content = next(iter(proposal.change.items()))
|
||||
if "/" in name or "\\" in name or name.startswith("."):
|
||||
return f"{name!r} reaches outside the family directory"
|
||||
if name != EDITABLE_FILES[proposal.lever]:
|
||||
return f"{name!r} is not the file the {proposal.lever!r} lever writes ({EDITABLE_FILES[proposal.lever]})"
|
||||
if name == "prompt.md" and len(content) + len(config.checklist) > MAX_PROMPT_CHARS:
|
||||
return f"the proposed prompt exceeds the cap of {MAX_PROMPT_CHARS} characters"
|
||||
if name == "checklist.md" and len(content) + len(config.prompt) > MAX_PROMPT_CHARS:
|
||||
return f"the proposed checklist exceeds the cap of {MAX_PROMPT_CHARS} characters"
|
||||
return None
|
||||
|
||||
|
||||
def apply_proposal(proposal: Proposal, directory: Path) -> None:
|
||||
"""Write the one file, inside the one directory. Callers snapshot first so a rejection can revert."""
|
||||
directory.mkdir(parents=True, exist_ok=True)
|
||||
for name, content in proposal.change.items():
|
||||
(directory / name).write_text(content, encoding="utf-8")
|
||||
|
||||
|
||||
def write_roots(configs_dir: Path, family: str) -> tuple[Path, ...]:
|
||||
"""Everything a round may write: exactly one family's directory, and never the verifiers or the corpus."""
|
||||
return (configs_dir / family,)
|
||||
|
||||
|
||||
@dataclass
|
||||
class ScriptedProposer:
|
||||
"""A fixed sequence of proposals, for tests and for the first offline rounds."""
|
||||
|
||||
proposals: list[Proposal] = field(default_factory=list)
|
||||
used: int = 0
|
||||
|
||||
def propose(self, facts: FailureFacts) -> Proposal | None:
|
||||
del facts
|
||||
if self.used >= len(self.proposals):
|
||||
return None
|
||||
self.used += 1
|
||||
return self.proposals[self.used - 1]
|
||||
|
||||
|
||||
@dataclass
|
||||
class HarnessXProposer:
|
||||
"""The meta-agent, with write access limited to the family directory it is targeting."""
|
||||
|
||||
configs_dir: Path
|
||||
model: str
|
||||
reason: str = ""
|
||||
|
||||
def write_roots(self, family: str) -> tuple[Path, ...]:
|
||||
return write_roots(self.configs_dir, family)
|
||||
|
||||
def propose(self, facts: FailureFacts) -> Proposal | None:
|
||||
if not has_harnessx():
|
||||
self.reason = "harnessx_unavailable"
|
||||
return None
|
||||
try:
|
||||
from ..harness_ext import require_harnessx
|
||||
|
||||
require_harnessx()
|
||||
except Exception: # noqa: BLE001 — a proposer that cannot start proposes nothing and says so
|
||||
self.reason = "harnessx_unavailable"
|
||||
return None
|
||||
self.reason = "harnessx_proposer_not_wired" # the evolve call lands with the live rounds (task 5.3)
|
||||
del facts
|
||||
return None
|
||||
|
||||
|
||||
def _fact(score: PairFamilyScore, other: int) -> PairFact:
|
||||
return PairFact(pair=score.pair, outcome=score.outcome, runs=score.runs, other_findings=other, rung=score.rung)
|
||||
|
||||
|
||||
def facts_prompt(facts: FailureFacts) -> str:
|
||||
"""The facts as the proposer sees them: structured, and free of anything it will be judged on."""
|
||||
return json.dumps(facts.to_dict(), indent=2, sort_keys=True)
|
||||
|
||||
|
||||
__all__ = [
|
||||
"EDITABLE_FILES",
|
||||
"LEVERS",
|
||||
"FailureFacts",
|
||||
"HarnessXProposer",
|
||||
"Lever",
|
||||
"PairFact",
|
||||
"Proposal",
|
||||
"Proposer",
|
||||
"ScriptedProposer",
|
||||
"apply_proposal",
|
||||
"build_failure_facts",
|
||||
"facts_prompt",
|
||||
"refuse_proposal",
|
||||
"write_roots",
|
||||
]
|
||||
Reference in New Issue
Block a user