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

krrishdholakia · krrishdholakia · commit 647e867cf8ae · 2024-02-26T22:07:33.000-08:00
diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py
@@ -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:

Original file line number	Diff line number	Diff line change
`@@ -222,6 +222,8 @@ async def test_async_vertexai_response():`
`222`	`222`	`model=model, messages=messages, temperature=0.7, timeout=5`
`223`	`223`	`)`
`224`	`224`	`print(f"response: {response}")`
	`225`	`+ except litellm.RateLimitError as e:`
	`226`	`+ pass`
`225`	`227`	`except litellm.Timeout as e:`
`226`	`228`	`pass`
`227`	`229`	`except Exception as e:`