fix(azure.py): correctly raise async exceptions

This commit is contained in:
Krrish Dholakia 2023-12-21 12:23:07 +05:30
parent 18a981b624
commit ee2a467708
2 changed files with 5 additions and 1 deletions

View file

@ -262,7 +262,10 @@ class AzureChatCompletion(BaseLLM):
exception_mapping_worked = True
raise e
except Exception as e:
raise AzureOpenAIError(status_code=500, message=str(e))
if hasattr(e, "status_code"):
raise e
else:
raise AzureOpenAIError(status_code=500, message=str(e))
def streaming(self,
logging_obj,