phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix(openai.py): fix error handling

Browse source
This commit is contained in:
Krrish Dholakia 2024-08-24 23:37:41 -07:00
parent 20be76123b
commit 4bf893afe1
1 changed files with 5 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/OpenAI/openai.py
View file

@ -1160,18 +1160,11 @@ class OpenAIChatCompletion(BaseLLM):
"complete_input_dict": data, "complete_input_dict": data,
}, },
) )
try:
headers, response = self.make_sync_openai_chat_completion_request( headers, response = self.make_sync_openai_chat_completion_request(
openai_client=openai_client, openai_client=openai_client,
data=data, data=data,
timeout=timeout, timeout=timeout,
) )
except Exception as e:
status_code = getattr(e, "status_code", 500)
error_headers = getattr(e, "headers", None)
raise OpenAIError(
status_code=status_code, message=str(e), headers=error_headers
)
logging_obj.model_call_details["response_headers"] = headers logging_obj.model_call_details["response_headers"] = headers
streamwrapper = CustomStreamWrapper( streamwrapper = CustomStreamWrapper(