Skip to content

Commit 4405dfa

Browse files
author
Dylan Huang
committed
more robust klavis test
1 parent c6317f7 commit 4405dfa

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

tests/pytest/test_pytest_klavis_mcp.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,12 @@ async def test_pytest_klavis_mcp(row: EvaluationRow) -> EvaluationRow:
5252
)
5353
response_text = response.choices[0].message.content
5454
logger.info("response_text: %s", response_text)
55-
score = json.loads(response_text or "{}")["score"]
55+
try:
56+
parsed = json.loads(response_text or "{}")
57+
score = parsed.get("score", 0.0)
58+
except (json.JSONDecodeError, TypeError):
59+
logger.warning("Failed to parse response as JSON: %s", response_text)
60+
score = 0.0
5661

5762
row.evaluation_result = EvaluateResult(
5863
score=score,

0 commit comments

Comments
 (0)