phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test(test_exceptions.py): fix test to handle service unavailable error

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-15 18:19:52 -08:00
parent aa78f55dff
commit 4c4ced18b2
1 changed files with 8 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/tests/test_exceptions.py
View file

@ -76,11 +76,14 @@ def test_context_window_with_fallbacks(model):
sample_text = "how does a court case get to the Supreme Court?" * 1000 sample_text = "how does a court case get to the Supreme Court?" * 1000
messages = [{"content": sample_text, "role": "user"}] messages = [{"content": sample_text, "role": "user"}]
completion( try:
model=model, completion(
messages=messages, model=model,
context_window_fallback_dict=ctx_window_fallback_dict, messages=messages,
) context_window_fallback_dict=ctx_window_fallback_dict,
)
except litellm.ServiceUnavailableError as e:
pass
# for model in litellm.models_by_provider["bedrock"]: # for model in litellm.models_by_provider["bedrock"]: