diff --git a/lib/crewai/tests/test_task_guardrails.py b/lib/crewai/tests/test_task_guardrails.py index 7ceaf847b..b06cd2dcb 100644 --- a/lib/crewai/tests/test_task_guardrails.py +++ b/lib/crewai/tests/test_task_guardrails.py @@ -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")