From 85b2137f9c41ffc2b505e92e7f1dcbaf8cd942da Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 4 May 2024 16:09:20 -0700 Subject: [PATCH] fix - test exceptions vertex ai --- litellm/tests/test_exceptions.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/litellm/tests/test_exceptions.py b/litellm/tests/test_exceptions.py index 9fcac2b48e..0db8c2cf41 100644 --- a/litellm/tests/test_exceptions.py +++ b/litellm/tests/test_exceptions.py @@ -605,8 +605,7 @@ def test_litellm_completion_vertex_exception(): except Exception as e: print("exception: ", e) assert "model: vertex_ai/gemini-pro" in str(e) - assert "model_group" not in str(e) - assert "deployment" not in str(e) + assert "vertex_project: bad-project" in str(e) # # test_invalid_request_error(model="command-nightly")