scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-08 22:21:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test: fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-21 22:30:41 -07:00
parent c5e030481a
commit 25820680eb
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_amazing_vertex_completion.py
View file

@ -957,6 +957,8 @@ async def test_partner_models_httpx(model, sync_mode):
assert isinstance(response._hidden_params["response_cost"], float)
except litellm.RateLimitError as e:
pass
except litellm.InternalServerError as e:
pass
except Exception as e:
if "429 Quota exceeded" in str(e):
pass
@ -1006,7 +1008,9 @@ async def test_partner_models_httpx_streaming(model, sync_mode):
idx += 1
print(f"response: {response}")
except litellm.RateLimitError:
except litellm.RateLimitError as e:
pass
except litellm.InternalServerError as e:
pass
except Exception as e:
if "429 Quota exceeded" in str(e):