From 0a15d3b3c3c8f4f4fc31a0f3f01b1dc38bde93ae Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Mon, 26 Aug 2024 15:42:55 -0700 Subject: [PATCH] fix(utils.py): fix message replace --- litellm/utils.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/litellm/utils.py b/litellm/utils.py index 68eac7c3a..81ba95087 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -6649,9 +6649,14 @@ def exception_type( else: message = str(original_exception) - if message is not None and isinstance(message, str): + if message is not None and isinstance( + message, str + ): # done to prevent user-confusion. Relevant issue - https://github.com/BerriAI/litellm/issues/1414 message = message.replace("OPENAI", custom_llm_provider.upper()) - message = message.replace("OpenAI", custom_llm_provider) + message = message.replace( + "openai.OpenAIError", + "{}.{}Error".format(custom_llm_provider, custom_llm_provider), + ) if custom_llm_provider == "openai": exception_provider = "OpenAI" + "Exception" else: