From d755d509013d2e1eef434f2a0365fdbe8e0c1dc8 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Fri, 26 Jan 2024 21:05:49 -0800 Subject: [PATCH] fix(openai.py): fix openai image gen logging --- litellm/llms/openai.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py index 01887616ca..da89b7796e 100644 --- a/litellm/llms/openai.py +++ b/litellm/llms/openai.py @@ -718,8 +718,22 @@ class OpenAIChatCompletion(BaseLLM): return convert_to_model_response_object(response_object=response, model_response_object=model_response, response_type="image_generation") # type: ignore except OpenAIError as e: exception_mapping_worked = True + ## LOGGING + logging_obj.post_call( + input=prompt, + api_key=api_key, + additional_args={"complete_input_dict": data}, + original_response=str(e), + ) raise e except Exception as e: + ## LOGGING + logging_obj.post_call( + input=prompt, + api_key=api_key, + additional_args={"complete_input_dict": data}, + original_response=str(e), + ) if hasattr(e, "status_code"): raise OpenAIError(status_code=e.status_code, message=str(e)) else: