From 28a0c61c513d85610ccac94210baff772a1f1ac4 Mon Sep 17 00:00:00 2001 From: Julio Quinteros Pro Date: Sun, 15 Feb 2026 18:26:51 -0300 Subject: [PATCH] fix(test): add environment cleanup for Vertex AI rerank tests Add setup_method and teardown_method to clean up Google/Vertex AI environment variables that may be left by previous tests. Previous tests may set GOOGLE_APPLICATION_CREDENTIALS or other Vertex environment variables and not clean them up, causing this test to attempt real Google authentication instead of using mocks. This fix: - Saves and clears Google/Vertex env vars in setup_method - Restores them in teardown_method - Prevents "DefaultCredentialsError" in CI when run with other tests Test passes in isolation but fails in CI due to test ordering. This is another test isolation issue, NOT related to PR #21217. Co-Authored-By: Claude Sonnet 4.5 --- .../test_vertex_ai_rerank_transformation.py | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/test_litellm/llms/vertex_ai/rerank/test_vertex_ai_rerank_transformation.py b/tests/test_litellm/llms/vertex_ai/rerank/test_vertex_ai_rerank_transformation.py index c1de7933f9..fbf5239797 100644 --- a/tests/test_litellm/llms/vertex_ai/rerank/test_vertex_ai_rerank_transformation.py +++ b/tests/test_litellm/llms/vertex_ai/rerank/test_vertex_ai_rerank_transformation.py @@ -15,9 +15,30 @@ from litellm.types.rerank import RerankResponse class TestVertexAIRerankTransform: def setup_method(self): + # Save and clear Google/Vertex AI environment variables to prevent + # test isolation issues where previous tests leave credentials set + self._saved_env = {} + env_vars_to_clear = [ + "GOOGLE_APPLICATION_CREDENTIALS", + "GOOGLE_CLOUD_PROJECT", + "VERTEXAI_PROJECT", + "VERTEX_PROJECT", + "VERTEX_LOCATION", + "VERTEX_AI_PROJECT", + ] + for var in env_vars_to_clear: + if var in os.environ: + self._saved_env[var] = os.environ[var] + del os.environ[var] + self.config = VertexAIRerankConfig() self.model = "semantic-ranker-default@latest" + def teardown_method(self): + # Restore saved environment variables + for var, value in self._saved_env.items(): + os.environ[var] = value + @patch('litellm.llms.vertex_ai.rerank.transformation.VertexAIRerankConfig._ensure_access_token') def test_get_complete_url(self, mock_ensure_access_token): """Test URL generation for Vertex AI Discovery Engine rerank API."""