phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(feat) add ollama_chat exception mapping

Browse source
This commit is contained in:
ishaan-jaff 2023-12-25 23:43:14 +05:30
parent f27a93a4dc
commit 67ed1ee10e
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -6050,7 +6050,9 @@ def exception_type(
) )
raise original_exception raise original_exception
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): if isinstance(original_exception, dict):
error_str = original_exception.get("error", "") error_str = original_exception.get("error", "")
else: else: