fix: _extract_skill_json crashes on a truthy non-string teacher response (#1630)
This commit is contained in:
@@ -327,7 +327,7 @@ def _extract_skill_json(teacher_response: str) -> Optional[Dict[str, Any]]:
|
||||
treated as "teacher declined to write a skill", per the prompt
|
||||
contract.
|
||||
"""
|
||||
if not teacher_response:
|
||||
if not isinstance(teacher_response, str) or not teacher_response:
|
||||
return None
|
||||
import json
|
||||
m = re.search(r"```(?:json)?\s*\n(\{[\s\S]*?\})\s*\n```", teacher_response)
|
||||
|
||||
19
tests/test_extract_skill_json_nonstring.py
Normal file
19
tests/test_extract_skill_json_nonstring.py
Normal file
@@ -0,0 +1,19 @@
|
||||
"""Regression: _extract_skill_json must tolerate a non-string response.
|
||||
|
||||
The `if not teacher_response` guard only handled falsy values; a truthy
|
||||
non-string (e.g. a number or list from an unexpected LLM client) reached
|
||||
`re.search(..., teacher_response)` and raised TypeError. Non-strings now
|
||||
return None (treated as "no skill"), matching the documented contract.
|
||||
"""
|
||||
from src.teacher_escalation import _extract_skill_json
|
||||
|
||||
|
||||
def test_non_string_returns_none():
|
||||
assert _extract_skill_json(123) is None
|
||||
assert _extract_skill_json(["x"]) is None
|
||||
assert _extract_skill_json(None) is None
|
||||
|
||||
|
||||
def test_valid_json_block_parsed():
|
||||
resp = "sure:\n```json\n{\"name\": \"x\"}\n```\n"
|
||||
assert _extract_skill_json(resp) == {"name": "x"}
|
||||
Reference in New Issue
Block a user