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

fix(azure.py): improve error handling for azure image gen responses

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-08 16:39:05 -07:00
parent 3045a2d9b3
commit 381347082d
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/azure.py
View file

@ -1108,7 +1108,10 @@ class AzureChatCompletion(BaseLLM):
"api-key": api_key, "api-key": api_key,
}, },
) )
operation_location_url = response.headers["operation-location"] if "operation-location" in response.headers:
operation_location_url = response.headers["operation-location"]
else:
raise AzureOpenAIError(status_code=500, message=response.text)
response = await async_handler.get( response = await async_handler.get(
url=operation_location_url, url=operation_location_url,
headers={ headers={
@ -1220,7 +1223,10 @@ class AzureChatCompletion(BaseLLM):
"api-key": api_key, "api-key": api_key,
}, },
) )
operation_location_url = response.headers["operation-location"] if "operation-location" in response.headers:
operation_location_url = response.headers["operation-location"]
else:
raise AzureOpenAIError(status_code=500, message=response.text)
response = sync_handler.get( response = sync_handler.get(
url=operation_location_url, url=operation_location_url,
headers={ headers={