diff --git a/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py b/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py index 818a82a8a18..3614db12aa0 100644 --- a/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py +++ b/tests/test_litellm/llms/vertex_ai/test_vertex_llm_base.py @@ -68,7 +68,7 @@ class TestVertexBase: project_id="different-project", custom_llm_provider="vertex_ai", ) - assert result == ("fake-token-1", "project-1") + assert result == ("fake-token-1", "different-project") @pytest.mark.parametrize("is_async", [True, False], ids=["async", "sync"]) @pytest.mark.asyncio