From bbd94f504cd087ed7c657790117daf268efdb7a7 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 29 Mar 2024 13:12:16 -0700 Subject: [PATCH] test(test_rules.py): fix assert --- litellm/tests/test_rules.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_rules.py b/litellm/tests/test_rules.py index 0997143bfa..f359a6de35 100644 --- a/litellm/tests/test_rules.py +++ b/litellm/tests/test_rules.py @@ -126,7 +126,10 @@ def test_post_call_rule_streaming(): print("Got exception", e) print(type(e)) print(vars(e)) - assert e.message == "This violates LiteLLM Proxy Rules. Response too short" + assert ( + e.message + == "OpenAIException - This violates LiteLLM Proxy Rules. Response too short" + ) def test_post_call_processing_error_async_response():