(feat) add ollama_chat exception mapping

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