From 20be76123b4cabcad732bd8ed4f777e3aae99a01 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 24 Aug 2024 23:15:39 -0700 Subject: [PATCH] fix(openai.py): fix exception raised --- litellm/llms/OpenAI/openai.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/litellm/llms/OpenAI/openai.py b/litellm/llms/OpenAI/openai.py index 54bac565772..1b90d5825dc 100644 --- a/litellm/llms/OpenAI/openai.py +++ b/litellm/llms/OpenAI/openai.py @@ -1160,11 +1160,18 @@ class OpenAIChatCompletion(BaseLLM): "complete_input_dict": data, }, ) - headers, response = self.make_sync_openai_chat_completion_request( - openai_client=openai_client, - data=data, - timeout=timeout, - ) + 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 + ) logging_obj.model_call_details["response_headers"] = headers streamwrapper = CustomStreamWrapper( @@ -1263,6 +1270,7 @@ class OpenAIChatCompletion(BaseLLM): error_headers = getattr(e, "headers", None) if response is not None and hasattr(response, "text"): + error_headers = getattr(e, "headers", None) raise OpenAIError( status_code=500, message=f"{str(e)}\n\nOriginal Response: {response.text}",