phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(azure.py,-openai.py): raise the correct exceptions for image generation calls

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-09 11:55:38 +05:30 • committed by ishaan-jaff
parent 27e52794df
commit d03b886079
2 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/azure.py
View file

@ -702,7 +702,9 @@ class AzureChatCompletion(BaseLLM):
exception_mapping_worked = True
raise e
except Exception as e:
if exception_mapping_worked:
if hasattr(e, "status_code"):
raise e
elif exception_mapping_worked:
raise e
else:
import traceback

6
litellm/llms/openai.py
View file

@ -716,12 +716,14 @@ class OpenAIChatCompletion(BaseLLM):
exception_mapping_worked = True
raise e
except Exception as e:
if exception_mapping_worked:
if hasattr(e, "status_code"):
raise e
elif exception_mapping_worked:
raise e
else:
import traceback
raise OpenAIError(status_code=500, message=traceback.format_exc())
raise AzureOpenAIError(status_code=500, message=traceback.format_exc())
async def ahealth_check(
self,