From 6ba1a5f6b2dfcefa0884abfd072652ad44877073 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 21 Feb 2024 21:31:26 -0800 Subject: [PATCH] fix(utils.py): add exception mapping for gemini --- litellm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index 0bb7bd2b30e..4260ee6e16c 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -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.