fix: raise on unparseable skill-comply classifier output

_parse_classification returned {} for garbage or non-dict output, so a failed classification looked like zero matched steps and graded as non-compliant. Raise ClassifierParseError (a RuntimeError, like the non-zero-rc path) instead.
This commit is contained in:
drakeo338 committed 2026-10-11 19:21:15 +00:00
1 parent d4a44ded8c
commit 7b34638500
2 files changed
+42 -4

No files matched your search

+10 -4
View File
@@ -14,6 +14,10 @@ from scripts.parser import ComplianceSpec, ObservationEvent
PROMPTS_DIR = Path(__file__).parent.parent / "prompts"
class ClassifierParseError(RuntimeError):
"""Raised when the classifier output cannot be parsed."""
def classify_events(
spec: ComplianceSpec,
trace: list[ObservationEvent],
@@ -73,13 +77,15 @@ def _parse_classification(text: str) -> dict[str, list[int]]:
try:
parsed = json.loads(cleaned)
if not isinstance(parsed, dict):
logger.warning("Classifier returned non-dict JSON: %s", type(parsed).__name__)
return {}
raise ClassifierParseError(
f"classifier returned non-dict JSON: {type(parsed).__name__}"
)
return {
k: [int(i) for i in v]
for k, v in parsed.items()
if isinstance(v, list)
}
except (json.JSONDecodeError, ValueError, TypeError) as e:
logger.warning("Failed to parse classification output: %s", e)
return {}
raise ClassifierParseError(
f"failed to parse classification output: {e}"
) from e
@@ -0,0 +1,32 @@
"""Tests for scripts.classifier._parse_classification."""
import pytest
from scripts import classifier
from scripts.classifier import _parse_classification
class TestParseClassification:
def test_plain_json(self):
assert _parse_classification('{"a": [0, 2]}') == {"a": [0, 2]}
def test_fenced_json(self):
assert _parse_classification('```json\n{"a": [1]}\n```') == {"a": [1]}
def test_dict_without_list_values_is_empty(self):
assert _parse_classification('{"a": "x"}') == {}
def test_garbage_raises(self):
with pytest.raises(RuntimeError):
_parse_classification("sorry, I cannot classify these")
def test_non_dict_json_raises(self):
with pytest.raises(RuntimeError):
_parse_classification("[1, 2, 3]")
def test_non_integer_indices_raise(self):
with pytest.raises(RuntimeError):
_parse_classification('{"a": ["x"]}')
def test_is_runtime_error(self):
assert issubclass(classifier.ClassifierParseError, RuntimeError)