From 25820680ebd9457cbcfc12761023b447158031ef Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 21 Aug 2024 22:30:41 -0700 Subject: [PATCH] test: fix test --- litellm/tests/test_amazing_vertex_completion.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index e30a9ed362b..73824758fa8 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -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):