From 67ed1ee10eb49d68f3765c04512ee3a2bef404b7 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 25 Dec 2023 23:43:14 +0530 Subject: [PATCH] (feat) add ollama_chat exception mapping --- litellm/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index ee0de81c44..63e4a7fe29 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -6050,7 +6050,9 @@ def exception_type( ) raise original_exception raise original_exception - elif custom_llm_provider == "ollama": + elif ( + custom_llm_provider == "ollama" or custom_llm_provider == "ollama_chat" + ): if isinstance(original_exception, dict): error_str = original_exception.get("error", "") else: