test(test_rules.py): fix assert

2024-03-29 13:12:16 -07:00 · 2024-03-29 13:12:16 -07:00 · bbd94f504c
commit bbd94f504c
parent b6cf3321f7
1 changed files with 4 additions and 1 deletions
--- 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():