scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-11 03:38:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

fix(test): use valid AnthropicThinkingParam type in thinking param test

Browse source 
'disabled' is not a valid Literal value for AnthropicThinkingParam.type.
Use 'adaptive' to correctly reflect the type contract.
This commit is contained in:
Christian Flasshoff 2026-04-17 11:44:41 -07:00
parent b4a0e0fd7d
commit 3d327d6883
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py
View file

@ -938,12 +938,12 @@ def test_vertex_ai_map_thinking_param_with_budget_tokens_0():
def test_vertex_ai_map_thinking_param_disabled():
"""
type="disabled" must produce includeThoughts: False with no thinkingBudget field.
type="adaptive" (non-enabled) must produce includeThoughts: False with no thinkingBudget field.
"""
from litellm.types.llms.anthropic import AnthropicThinkingParam
v = VertexGeminiConfig()
thinking_param: AnthropicThinkingParam = {"type": "disabled"}
thinking_param: AnthropicThinkingParam = {"type": "adaptive"}
result = v._map_thinking_param(thinking_param=thinking_param)
assert result == {"includeThoughts": False}
assert "thinkingBudget" not in result