From cb963dec682aabcf2cd46aa464a1557193a62608 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Fri, 18 Aug 2023 14:35:59 -0700
Subject: [PATCH] update exception mapping

---
 litellm/main.py | 2 +-
 pyproject.toml  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/litellm/main.py b/litellm/main.py
index cbf2bde4a2..a8f7fbd5e0 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 dab74454d7..aebd15b8b0 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"