fix(utils.py): add exception mapping for gemini

This commit is contained in:
Krrish Dholakia 2024-02-21 21:31:26 -08:00
parent fb2ae3a032
commit 6ba1a5f6b2

View file

@ -6813,7 +6813,7 @@ def exception_type(
llm_provider="vertex_ai",
request=original_exception.request,
)
elif custom_llm_provider == "palm":
elif custom_llm_provider == "palm" or custom_llm_provider == "gemini":
if "503 Getting metadata" in error_str:
# auth errors look like this
# 503 Getting metadata from plugin failed with error: Reauthentication is needed. Please run `gcloud auth application-default login` to reauthenticate.