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

test: update test to handle overloaded error

Browse source
This commit is contained in:
Krrish Dholakia 2024-11-14 19:02:57 +05:30
parent fef31a1c01
commit b4f7771ed1
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/local_testing/test_streaming.py
View file

@ -3333,7 +3333,9 @@ 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 or litellm.ServiceUnavailableError:
except litellm.InternalServerError:
pass
except litellm.ServiceUnavailableError:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")