mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 10:14:26 +00:00
test: update test
This commit is contained in:
parent
72ad9f92b8
commit
e7d49eb387
1 changed files with 1 additions and 1 deletions
|
@ -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(
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue