phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor(bedrock.py): better exception mapping for bedrock + huggingface

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-04 16:12:12 -07:00
parent ab54262d37
commit 7c46e85ed6
2 changed files with 158 additions and 141 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/huggingface_restapi.py
View file

@ -141,6 +141,7 @@ def completion(
litellm_params=None,
logger_fn=None,
):
exception_mapping_worked = False
try:
headers = validate_environment(api_key, headers)
task = get_hf_task_for_model(model)
@ -365,10 +366,14 @@ def completion(
model_response._hidden_params["original_response"] = completion_response
return model_response
except HuggingfaceError as e:
exception_mapping_worked = True
raise e
except Exception as e:
import traceback
raise HuggingfaceError(status_code=500, message=traceback.format_exc())
if exception_mapping_worked:
raise e
else:
import traceback
raise HuggingfaceError(status_code=500, message=traceback.format_exc())
def embedding(