fix(openai.py): fix post call error logging for aembedding calls

2024-08-27 14:25:33 -07:00 · 2024-08-27 14:25:33 -07:00 · b91e5d3887
commit b91e5d3887
parent 3dee98dc47
1 changed files with 14 additions and 0 deletions
--- 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(