From 32f22adf8b21e9fa84e6eaa37e937150aae1b427 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 17 Nov 2023 14:43:17 -0800 Subject: [PATCH] (feat) openai improve logging post_call --- litellm/llms/openai.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/litellm/llms/openai.py b/litellm/llms/openai.py index 726c2b3b975..3ecc9061a4c 100644 --- a/litellm/llms/openai.py +++ b/litellm/llms/openai.py @@ -209,6 +209,12 @@ class OpenAIChatCompletion(BaseLLM): else: openai_client = OpenAI(api_key=api_key, base_url=api_base, http_client=litellm.client_session, timeout=timeout) response = openai_client.chat.completions.create(**data) # type: ignore + logging_obj.post_call( + input=None, + api_key=api_key, + original_response=response, + additional_args={"complete_input_dict": data}, + ) return convert_to_model_response_object(response_object=json.loads(response.model_dump_json()), model_response_object=model_response) except Exception as e: if "Conversation roles must alternate user/assistant" in str(e) or "user and assistant roles should be alternating" in str(e):