fix(openai.py): fix error handling

This commit is contained in:
Krrish Dholakia 2024-08-24 23:37:41 -07:00
parent 20be76123b
commit 4bf893afe1

View file

@ -1160,18 +1160,11 @@ class OpenAIChatCompletion(BaseLLM):
"complete_input_dict": data,
},
)
try:
headers, response = self.make_sync_openai_chat_completion_request(
openai_client=openai_client,
data=data,
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
)
headers, response = self.make_sync_openai_chat_completion_request(
openai_client=openai_client,
data=data,
timeout=timeout,
)
logging_obj.model_call_details["response_headers"] = headers
streamwrapper = CustomStreamWrapper(