scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-07 02:59:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

catch gemini 429 errors

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-05 14:02:01 -07:00
parent 34a9b8648d
commit 3dbb547af0
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_amazing_vertex_completion.py
View file

@ -521,6 +521,8 @@ def test_gemini_pro_function_calling():
print(f"completion: {completion}")
assert completion.choices[0].message.content is None
assert len(completion.choices[0].message.tool_calls) == 1
except litellm.RateLimitError as e:
pass
except Exception as e:
if "429 Quota exceeded" in str(e):
pass