scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 154

fix(vertex_llm_base.py): handle default vertex location

Browse source
This commit is contained in:
Krrish Dholakia 2025-05-28 11:33:39 -07:00
parent 18f7316b08
commit cc780ecf5a
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/vertex_ai/vertex_llm_base.py
View file

@ -135,15 +135,20 @@ class VertexBase:
return google_auth.default(scopes=scopes)
def get_default_vertex_location(self) -> str:
return "us-central1"
def get_api_base(
self, api_base: Optional[str], vertex_location: Optional[str]
) -> str:
if api_base:
return api_base
elif vertex_location and vertex_location == "global":
elif vertex_location == "global":
return "https://aiplatform.googleapis.com"
else:
elif vertex_location:
return f"https://{vertex_location}-aiplatform.googleapis.com"
else:
return f"https://{self.get_default_vertex_location()}-aiplatform.googleapis.com"
def refresh_auth(self, credentials: Any) -> None:
from google.auth.transport.requests import (