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

fix(azure.py): fix httpx call for dall e 3

Browse source
This commit is contained in:
Krrish Dholakia 2024-07-02 18:28:45 -07:00
parent 540db53ccb
commit ac00639301
2 changed files with 19 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/llms/azure.py
View file

@ -1150,7 +1150,13 @@ class AzureChatCompletion(BaseLLM):
error_data = response.json()
raise AzureOpenAIError(status_code=400, message=json.dumps(error_data))
return response
result = response.json()["result"]
return httpx.Response(
status_code=200,
headers=response.headers,
content=json.dumps(result).encode("utf-8"),
request=httpx.Request(method="POST", url="https://api.openai.com/v1"),
)
return await async_handler.post(
url=api_base,
json=data,
@ -1249,7 +1255,13 @@ class AzureChatCompletion(BaseLLM):
error_data = response.json()
raise AzureOpenAIError(status_code=400, message=json.dumps(error_data))
return response
result = response.json()["result"]
return httpx.Response(
status_code=200,
headers=response.headers,
content=json.dumps(result).encode("utf-8"),
request=httpx.Request(method="POST", url="https://api.openai.com/v1"),
)
return sync_handler.post(
url=api_base,
json=data,
@ -1324,7 +1336,7 @@ class AzureChatCompletion(BaseLLM):
api_key=api_key,
data=data,
)
response = httpx_response.json()["result"]
response = httpx_response.json()
stringified_response = response
## LOGGING
@ -1431,7 +1443,7 @@ class AzureChatCompletion(BaseLLM):
api_key=api_key or "",
data=data,
)
response = httpx_response.json()["result"]
response = httpx_response.json()
## LOGGING
logging_obj.post_call(