tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-21 00:24:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

better exception raising for package not imported

Browse Source
This commit is contained in:
ishaan-jaff
2023-09-08 17:42:01 -07:00
parent a9cb373d04
commit c5ae0147d0
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/utils.py
+4 -3
View File
@@ -1578,7 +1578,7 @@ def exception_type(model, original_exception, custom_llm_provider):
model=model,
llm_provider=original_exception.llm_provider
)
raise original_exception
raise ValueError(original_exception)
elif model:
error_str = str(original_exception)
if isinstance(original_exception, BaseException):
@@ -1761,6 +1761,7 @@ def exception_type(model, original_exception, custom_llm_provider):
llm_provider="cohere",
model=model
)
raise ValueError(original_exception)
elif custom_llm_provider == "huggingface":
if "length limit exceeded" in error_str:
exception_mapping_worked = True
@@ -1912,7 +1913,7 @@ def exception_type(model, original_exception, custom_llm_provider):
model=model
)
else:
raise original_exception
raise ValueError(original_exception)
except Exception as e:
# LOGGING
exception_logging(
@@ -1929,7 +1930,7 @@ def exception_type(model, original_exception, custom_llm_provider):
):
threading.Thread(target=get_all_keys, args=(e.llm_provider,)).start()
# don't let an error with mapping interrupt the user from receiving an error from the llm api calls
raise original_exception
raise ValueError(original_exception)
####### CRASH REPORTING ################