From 4d1806bc9514ebde61eb1e7f4305254c20e452b1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 4 May 2024 21:32:10 -0700 Subject: [PATCH] fix - vertex ai exceptions --- litellm/tests/test_exceptions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_exceptions.py b/litellm/tests/test_exceptions.py index 0db8c2cf410..777fb3f9b36 100644 --- a/litellm/tests/test_exceptions.py +++ b/litellm/tests/test_exceptions.py @@ -585,7 +585,7 @@ def test_router_completion_vertex_exception(): pytest.fail("Request should have failed - bad api key") except Exception as e: print("exception: ", e) - assert "model: vertex_ai/gemini-pro" in str(e) + assert "Model: gemini-pro" in str(e) assert "model_group: vertex-gemini-pro" in str(e) assert "deployment: vertex_ai/gemini-pro" in str(e) @@ -604,7 +604,7 @@ def test_litellm_completion_vertex_exception(): pytest.fail("Request should have failed - bad api key") except Exception as e: print("exception: ", e) - assert "model: vertex_ai/gemini-pro" in str(e) + assert "Model: gemini-pro" in str(e) assert "vertex_project: bad-project" in str(e)