mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
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>
This commit is contained in:
parent
5531169710
commit
037378ece8
1 changed files with 4 additions and 4 deletions
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue