From 3d327d6883e63c859cdad16ce3ed8524de52da0c Mon Sep 17 00:00:00 2001 From: Christian Flasshoff Date: Fri, 17 Apr 2026 11:44:41 -0700 Subject: [PATCH] fix(test): use valid AnthropicThinkingParam type in thinking param test 'disabled' is not a valid Literal value for AnthropicThinkingParam.type. Use 'adaptive' to correctly reflect the type contract. --- .../gemini/test_vertex_and_google_ai_studio_gemini.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py b/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py index 4c0115ef43e..86b3b4b3ec1 100644 --- a/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py +++ b/tests/test_litellm/llms/vertex_ai/gemini/test_vertex_and_google_ai_studio_gemini.py @@ -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