fix exception provider not known

2025-04-25 10:44:24 +00:00 · 2024-07-01 21:05:37 -07:00 · 2024-07-01 21:05:37 -07:00 · fcf65d5215
commit fcf65d5215
parent 4c95782f74
1 changed files with 12 additions and 0 deletions
--- 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 (