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

style(vertex): apply ruff format to the 3.7-flash thinking fix

Browse source
This commit is contained in:
gegemeimingzi 2026-08-19 22:50:39 +08:00
parent d246efcf41
commit 398ed330f1
1 changed files with 2 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
View file

@ -281,11 +281,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
HIGH for it, while earlier 3.x-flash models (3.0, 3.1, 3.5, 3.6) still accept it.
"""
lowered: Final = model.lower()
return (
"flash" in lowered
and "gemini-3" in lowered
and "gemini-3.7" not in lowered
)
return "flash" in lowered and "gemini-3" in lowered and "gemini-3.7" not in lowered
@staticmethod
def _forward_gemini_function_call_id(model: str) -> bool:
@ -967,9 +963,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
# Follow provider defaults unless explicitly opted into legacy behavior.
if litellm.enable_gemini_default_thinking_level_low is True:
params["thinkingLevel"] = (
"minimal"
if VertexGeminiConfig._is_gemini3_flash_supports_minimal(model)
else "low"
"minimal" if VertexGeminiConfig._is_gemini3_flash_supports_minimal(model) else "low"
)
else:
# Thinking disabled