From d4c2eb1797e6b2b4274493c79acbd094f6210369 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 4 May 2024 22:04:53 -0700 Subject: [PATCH] fix post call rule test --- litellm/tests/test_rules.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_rules.py b/litellm/tests/test_rules.py index f359a6de353..7e2d7b81962 100644 --- a/litellm/tests/test_rules.py +++ b/litellm/tests/test_rules.py @@ -127,8 +127,8 @@ def test_post_call_rule_streaming(): print(type(e)) print(vars(e)) assert ( - e.message - == "OpenAIException - This violates LiteLLM Proxy Rules. Response too short" + "OpenAIException - This violates LiteLLM Proxy Rules. Response too short" + in e.message )