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

fix: Clear cached GCP creds on reauthentication error

Browse Source 
Co-authored-by: aider (vertex_ai/gemini-2.5-pro) <aider@aider.chat>
This commit is contained in:
Oz Ben-Ami 2025-07-29 11:14:17 -04:00
parent 33510120fd
commit 46eea13a9f
1 changed files with 22 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
litellm/llms/vertex_ai/vertex_llm_base.py
View File

@ -469,14 +469,28 @@ class VertexBase:
raise ValueError("Credentials are None after loading")
if _credentials.expired:
verbose_logger.debug(
f"Credentials expired, refreshing for project_id: {project_id}"
)
self.refresh_auth(_credentials)
self._credentials_project_mapping[credential_cache_key] = (
_credentials,
credential_project_id,
)
try:
verbose_logger.debug(
f"Credentials expired, refreshing for project_id: {project_id}"
)
self.refresh_auth(_credentials)
self._credentials_project_mapping[credential_cache_key] = (
_credentials,
credential_project_id,
)
except Exception as e:
# if refresh fails, it's possible the user has re-authenticated via `gcloud auth application-default login`
# in this case, we should try to reload the credentials by clearing the cache and retrying
if "Reauthentication is needed" in str(e):
verbose_logger.debug(
f"Credential refresh failed for project_id: {project_id}. Deleting from cache and retrying."
)
del self._credentials_project_mapping[credential_cache_key]
return self.get_access_token(
credentials=credentials,
project_id=project_id,
)
raise e
## VALIDATION STEP
if _credentials.token is None or not isinstance(_credentials.token, str):