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

ci/cd fix predibase 500 errors

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-11 23:15:48 -07:00
parent a9e61ef96d
commit 5ebeb4d067
2 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/utils.py
View file

@ -8771,6 +8771,13 @@ def exception_type(
response=original_exception.response,
litellm_debug_info=extra_information,
)
if "Request failed during generation" in error_str:
# this is an internal server error from predibase
raise litellm.InternalServerError(
message=f"PredibaseException - {error_str}",
llm_provider="predibase",
model=model,
)
elif hasattr(original_exception, "status_code"):
if original_exception.status_code == 500:
exception_mapping_worked = True