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

vertex_ai/codestral@2405 is very unstable - handle their instability in our tests

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-17 18:17:14 -08:00
parent 585b54e70c
commit e1ca95672a
2 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/local_testing/test_amazing_vertex_completion.py
View file

@ -995,6 +995,8 @@ async def test_partner_models_httpx(model, sync_mode):
pass pass
except litellm.APIConnectionError as e: except litellm.APIConnectionError as e:
pass pass
except litellm.ServiceUnavailableError as e:
pass
except Exception as e: except Exception as e:
if "429 Quota exceeded" in str(e): if "429 Quota exceeded" in str(e):
pass pass

3
tests/local_testing/test_text_completion.py
View file

@ -4193,6 +4193,9 @@ async def test_completion_codestral_fim_api_stream(model):
except litellm.APIConnectionError as e: except litellm.APIConnectionError as e:
print(e) print(e)
pass pass
except litellm.ServiceUnavailableError as e:
print(e)
pass
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")