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

fix(azure.py): fix raw response dump

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-27 13:44:38 -07:00
parent 18731cf42b
commit 63adb3f940
2 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/openai.py
View file

@ -2126,7 +2126,8 @@ class OpenAITextCompletion(BaseLLM):
openai_client = client
try:
response = openai_client.completions.with_raw_response.create(**data)
raw_response = openai_client.completions.with_raw_response.create(**data)
response = raw_response.parse()
except Exception as e:
status_code = getattr(e, "status_code", 500)
error_headers = getattr(e, "headers", None)
@ -2170,8 +2171,8 @@ class OpenAITextCompletion(BaseLLM):
else:
openai_client = client
response = await openai_client.completions.with_raw_response.create(**data)
raw_response = await openai_client.completions.with_raw_response.create(**data)
response = raw_response.parse()
streamwrapper = CustomStreamWrapper(
completion_stream=response,
model=model,