fix: vertex llm base

This commit is contained in:
Ishaan Jaff 2025-05-20 10:50:18 -07:00
parent a676f69cab
commit f7efddf7c9
2 changed files with 3 additions and 3 deletions

View file

@ -312,7 +312,7 @@ class VertexBase:
)
except Exception as e:
verbose_logger.exception(
f"Failed to load vertex credentials. Check to see if credentials containing partial/invalid information. Error: {str(e)}"
f"Failed to load vertex credentials. Check to see if credentials containing partial/invalid information. Tried using project_id: {project_id}. Error: {str(e)}"
)
raise e

View file

@ -687,8 +687,8 @@ async def _base_vertex_proxy_route(
headers.pop("content-length", None)
headers.pop("host", None)
else:
vertex_project = vertex_credentials.vertex_project
vertex_location = vertex_credentials.vertex_location
vertex_project = vertex_credentials.vertex_project or vertex_project
vertex_location = vertex_credentials.vertex_location or vertex_location
vertex_credentials_str = vertex_credentials.vertex_credentials
_auth_header, vertex_project = await vertex_llm_base._ensure_access_token_async(