test fix cassette

This commit is contained in:
lorenzejay
2026-01-14 16:23:36 -08:00
parent 842a1db16f
commit e4bd7889fd

View File

@@ -186,7 +186,7 @@ def test_task_guardrail_process_output(task_output):
result = guardrail(task_output)
assert result[0] is False
assert result[1] == "The task result contains more than 10 words, violating the guardrail. The text provided contains about 21 words."
assert "10 words" in result[1] or "more than 10" in result[1] or "exceeds" in result[1]
guardrail = LLMGuardrail(
description="Ensure the result has less than 500 words", llm=LLM(model="gpt-4o")