Merge pull request #21272 from BerriAI/fix/vertex-ai-gpt-oss-test-isolation

fix(test): add environment cleanup for Vertex AI GPT-OSS tests
This commit is contained in:
jquinter
2026-02-15 19:25:49 -03:00
committed by GitHub
@@ -16,6 +16,32 @@ from litellm.llms.vertex_ai.vertex_ai_partner_models.gpt_oss.transformation impo
)
@pytest.fixture(autouse=True)
def clean_vertex_env():
"""Clear Google/Vertex AI environment variables before each test to prevent test isolation issues."""
saved_env = {}
env_vars_to_clear = [
"GOOGLE_APPLICATION_CREDENTIALS",
"GOOGLE_CLOUD_PROJECT",
"VERTEXAI_PROJECT",
"VERTEXAI_LOCATION",
"VERTEXAI_CREDENTIALS",
"VERTEX_PROJECT",
"VERTEX_LOCATION",
"VERTEX_AI_PROJECT",
]
for var in env_vars_to_clear:
if var in os.environ:
saved_env[var] = os.environ[var]
del os.environ[var]
yield
# Restore saved environment variables
for var, value in saved_env.items():
os.environ[var] = value
class TestVertexAIGPTOSSTransformation:
"""Test class for VertexAI GPT-OSS transformation functionality."""