From b91e5d3887f3241615b2caa04ac1e56b157b9b26 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 27 Aug 2024 14:25:33 -0700 Subject: [PATCH] fix(openai.py): fix post call error logging for aembedding calls --- litellm/llms/openai.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py index d62e89f8afd..941b498437b 100644 --- a/litellm/llms/openai.py +++ b/litellm/llms/openai.py @@ -1376,8 +1376,22 @@ class OpenAIChatCompletion(BaseLLM): _response_headers=headers, ) # type: ignore except OpenAIError as e: + ## LOGGING + logging_obj.post_call( + input=input, + 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=input, + api_key=api_key, + additional_args={"complete_input_dict": data}, + original_response=str(e), + ) status_code = getattr(e, "status_code", 500) error_headers = getattr(e, "headers", None) raise OpenAIError(