From e8b844abaebe4baf74c7cfbe0477c583b2596009 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 23 Nov 2023 15:36:51 -0800
Subject: [PATCH] (fix) azure: better debugging

---
 litellm/llms/azure.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/litellm/llms/azure.py b/litellm/llms/azure.py
index 93b46bb129..660617f0cc 100644
--- a/litellm/llms/azure.py
+++ b/litellm/llms/azure.py
@@ -122,7 +122,10 @@ class AzureChatCompletion(BaseLLM):
                 input=messages,
                 api_key=api_key,
                 additional_args={
-                    "headers": headers,
+                    "headers": {
+                        "api_key": api_key, 
+                        "azure_ad_token": azure_ad_token
+                    },
                     "api_version": api_version,
                     "api_base": api_base,
                     "complete_input_dict": data,
@@ -302,7 +305,13 @@ class AzureChatCompletion(BaseLLM):
             logging_obj.pre_call(
                     input=input,
                     api_key=api_key,
-                    additional_args={"complete_input_dict": data},
+                    additional_args={
+                        "complete_input_dict": data, 
+                        "headers": {
+                            "api_key": api_key, 
+                            "azure_ad_token": azure_ad_token
+                        }
+                    },
                 )
             ## COMPLETION CALL            
             response = azure_client.embeddings.create(**data) # type: ignore