forked from phoenix/litellm-mirror
test: handle service unavailable error
This commit is contained in:
parent
51ec270501
commit
8bf0f57b59
1 changed files with 1 additions and 1 deletions
|
@ -3333,7 +3333,7 @@ async def test_acompletion_function_call_with_streaming(model):
|
||||||
validate_final_streaming_function_calling_chunk(chunk=chunk)
|
validate_final_streaming_function_calling_chunk(chunk=chunk)
|
||||||
idx += 1
|
idx += 1
|
||||||
# raise Exception("it worked! ")
|
# raise Exception("it worked! ")
|
||||||
except litellm.InternalServerError:
|
except litellm.InternalServerError or litellm.ServiceUnavailableError:
|
||||||
pass
|
pass
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
pytest.fail(f"Error occurred: {e}")
|
pytest.fail(f"Error occurred: {e}")
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue