fix - test exceptions vertex ai

2024-05-04 16:09:20 -07:00 · 2024-05-04 16:09:20 -07:00 · 85b2137f9c
commit 85b2137f9c
parent 7150df653f
1 changed files with 1 additions and 2 deletions
--- 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")