test(test_rules.py): fix assert

This commit is contained in:
Krrish Dholakia
2024-03-29 13:12:16 -07:00
parent b6cf3321f7
commit bbd94f504c
+4 -1
View File
@@ -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():