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

test(test_amazing_vertex_completion.py): fix test to handle rate limit errors

Browse source
This commit is contained in:
Krrish Dholakia 2024-02-26 22:07:33 -08:00
parent 6b14c6287b
commit 647e867cf8
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

@ -222,6 +222,8 @@ async def test_async_vertexai_response():
model=model, messages=messages, temperature=0.7, timeout=5
)
print(f"response: {response}")
except litellm.RateLimitError as e:
pass
except litellm.Timeout as e:
pass
except Exception as e: