phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(utils.py): fix message replace

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-26 15:42:55 -07:00
parent 71bf5b31b2
commit 0a15d3b3c3
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
litellm/utils.py
View file

@ -6649,9 +6649,14 @@ def exception_type(
else: else:
message = str(original_exception) 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.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": if custom_llm_provider == "openai":
exception_provider = "OpenAI" + "Exception" exception_provider = "OpenAI" + "Exception"
else: else: