ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rename _is_model_gemini_spec_model

Browse Source
This commit is contained in:
Ishaan Jaff 2025-03-26 11:14:51 -07:00
parent 1822c45c0e
commit 793a920caa
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
litellm/llms/vertex_ai/common_utils.py
View File

@ -31,7 +31,7 @@ def get_supports_system_message(
)
# Vertex Models called in the `/gemini` request/response format also support system messages
if litellm.VertexGeminiConfig._is_model_gemini_gemini_spec_model(model):
if litellm.VertexGeminiConfig._is_model_gemini_spec_model(model):
supports_system_message = True
except Exception as e:
verbose_logger.warning(

2
litellm/llms/vertex_ai/gemini/vertex_and_google_ai_studio_gemini.py
View File

@ -420,7 +420,7 @@ class VertexGeminiConfig(VertexAIBaseConfig, BaseConfig):
]
@staticmethod
def _is_model_gemini_gemini_spec_model(model: Optional[str]) -> bool:
def _is_model_gemini_spec_model(model: Optional[str]) -> bool:
"""
Returns true if user is trying to call custom model in `/gemini` request/response format
"""