(feat) add ollama_chat exception mapping

2023-12-25 23:43:14 +05:30 · 2023-12-25 23:43:14 +05:30 · 67ed1ee10e
commit 67ed1ee10e
parent f27a93a4dc
1 changed files with 3 additions and 1 deletions
--- 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: