From 1c263d057d0c5dd5f4815bb62bc7f6213ab03659 Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 27 Jun 2024 17:38:03 -0700
Subject: [PATCH] azure - log post api call

---
 litellm/llms/azure.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/litellm/llms/azure.py b/litellm/llms/azure.py
index 5d73b94350..fe10cc017c 100644
--- a/litellm/llms/azure.py
+++ b/litellm/llms/azure.py
@@ -660,8 +660,16 @@ class AzureChatCompletion(BaseLLM):
             response = await azure_client.chat.completions.create(
                 **data, timeout=timeout
             )
+
+            stringified_response = response.model_dump()
+            logging_obj.post_call(
+                input=data["messages"],
+                api_key=api_key,
+                original_response=stringified_response,
+                additional_args={"complete_input_dict": data},
+            )
             return convert_to_model_response_object(
-                response_object=response.model_dump(),
+                response_object=stringified_response,
                 model_response_object=model_response,
             )
         except AzureOpenAIError as e: