scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

test(test_completion.py): fix gemini rate limit error

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-21 07:17:27 -07:00
parent a6ee1e04fe
commit 8406016096
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_completion.py
View file

@ -707,6 +707,8 @@ def test_gemini_completion_call_error():
print(f"response: {response}")
for chunk in response:
print(chunk)
except litellm.RateLimitError:
pass
except litellm.InternalServerError:
pass
except Exception as e: