mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-05 08:07:05 +00:00
test(test_rules.py): fix assert
This commit is contained in:
parent
b6cf3321f7
commit
bbd94f504c
1 changed files with 4 additions and 1 deletions
|
|
@ -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():
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue