diff --git a/litellm/main.py b/litellm/main.py index cbf2bde4a..a8f7fbd5e 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -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 diff --git a/pyproject.toml b/pyproject.toml index dab74454d..aebd15b8b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -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"