test: update test

This commit is contained in:
Krrish Dholakia 2025-04-10 21:45:23 -07:00
parent 72ad9f92b8
commit e7d49eb387

View file

@ -1410,7 +1410,7 @@ def test_litellm_proxy_thinking_param():
custom_llm_provider="litellm_proxy",
thinking={"type": "enabled", "budget_tokens": 1024},
)
assert optional_params["thinking"] == {"type": "enabled", "budget_tokens": 1024}
assert optional_params["extra_body"]["thinking"] == {"type": "enabled", "budget_tokens": 1024}
def test_gemini_modalities_param():
optional_params = get_optional_params(