From 037378ece8fee6d1060631906c361489f2288d6c Mon Sep 17 00:00:00 2001 From: "devin-ai-integration[bot]" <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Wed, 7 Oct 2026 17:41:44 -0700 Subject: [PATCH] fix(ci): restore vertex model sets mutated by get_optional_params tests (#45209) * fix(ci): restore vertex model sets mutated by get_optional_params tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test: annotate monkeypatch params in vertex optional-params tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --------- Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- tests/unit/test_utils_get_optional_params.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_utils_get_optional_params.py b/tests/unit/test_utils_get_optional_params.py index 92ce66c6ebe..d9d049191a1 100644 --- a/tests/unit/test_utils_get_optional_params.py +++ b/tests/unit/test_utils_get_optional_params.py @@ -312,8 +312,8 @@ def test_azure_ai_mistral_optional_params(): assert "user" not in optional_params -def test_vertex_ai_llama_3_optional_params(): - litellm.vertex_llama3_models = ["meta/llama3-405b-instruct-maas"] +def test_vertex_ai_llama_3_optional_params(monkeypatch: pytest.MonkeyPatch) -> None: + monkeypatch.setattr(litellm, "vertex_llama3_models", {"meta/llama3-405b-instruct-maas"}) litellm.drop_params = True optional_params = get_optional_params( model="meta/llama3-405b-instruct-maas", @@ -325,8 +325,8 @@ def test_vertex_ai_llama_3_optional_params(): assert "user" not in optional_params -def test_vertex_ai_mistral_optional_params(): - litellm.vertex_mistral_models = ["mistral-large@2407"] +def test_vertex_ai_mistral_optional_params(monkeypatch: pytest.MonkeyPatch) -> None: + monkeypatch.setattr(litellm, "vertex_mistral_models", {"mistral-large@2407"}) litellm.drop_params = True optional_params = get_optional_params( model="mistral-large@2407",