update exception mapping

This commit is contained in:
Krrish Dholakia 2023-08-18 14:35:59 -07:00
parent 83c6d22848
commit cb963dec68
2 changed files with 2 additions and 2 deletions

View file

@ -844,7 +844,7 @@ def embedding(model, input=[], azure=False, force_timeout=60, logger_fn=None):
# log the original exception
logging(model=model, input=input, azure=azure, logger_fn=logger_fn, exception=e)
## Map to OpenAI Exception
raise exception_type(model=model, original_exception=e)
raise exception_type(model=model, original_exception=e, custom_llm_provider="azure" if azure==True else None)
raise e