azure - log post api call

2024-06-27 17:38:03 -07:00 · 2024-06-27 17:38:03 -07:00 · 57ba0a46b7
commit 57ba0a46b7
parent a012f231b6
1 changed files with 9 additions and 1 deletions
--- 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: