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

View file

@ -1,6 +1,6 @@
[tool.poetry]
name = "litellm"
version = "0.1.425"
version = "0.1.426"
description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"]
license = "MIT License"