phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

_get_model_name_from_gemini_spec_model

Browse source
This commit is contained in:
Ishaan Jaff 2025-03-26 12:16:18 -07:00
parent 793a920caa
commit 93daf5cbac
2 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
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_spec_model(model):
if litellm.VertexGeminiConfig._is_model_gemini_gemini_spec_model(model):
supports_system_message = True
except Exception as e:
verbose_logger.warning(
@ -75,6 +75,8 @@ def _get_vertex_url(
) -> Tuple[str, str]:
url: Optional[str] = None
endpoint: Optional[str] = None
if litellm.VertexGeminiConfig._is_model_gemini_gemini_spec_model(model):
model = litellm.VertexGeminiConfig._get_model_name_from_gemini_spec_model(model)
if mode == "chat":
### SET RUNTIME ENDPOINT ###
endpoint = "generateContent"