From fcf65d5215970a579dd8175eacab8b32c0adfceb Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 1 Jul 2024 21:05:37 -0700
Subject: [PATCH] fix exception provider not known

---
 litellm/utils.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/litellm/utils.py b/litellm/utils.py
index 103f854b6..f8e8566f8 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -5810,6 +5810,18 @@ def exception_type(
                 _model_group = _metadata.get("model_group")
                 _deployment = _metadata.get("deployment")
                 extra_information = f"\nModel: {model}"
+
+                exception_provider = "Unknown"
+                if (
+                    isinstance(custom_llm_provider, str)
+                    and len(custom_llm_provider) > 0
+                ):
+                    exception_provider = (
+                        custom_llm_provider[0].upper()
+                        + custom_llm_provider[1:]
+                        + "Exception"
+                    )
+
                 if _api_base:
                     extra_information += f"\nAPI Base: `{_api_base}`"
                 if (