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

test: handle service unavailable error

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-14 18:13:56 +05:30
parent 51ec270501
commit 8bf0f57b59
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_streaming.py
View file

@ -3333,7 +3333,7 @@ async def test_acompletion_function_call_with_streaming(model):
validate_final_streaming_function_calling_chunk(chunk=chunk)
idx += 1
# raise Exception("it worked! ")
except litellm.InternalServerError:
except litellm.InternalServerError or litellm.ServiceUnavailableError:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")