diff --git a/litellm/llms/azure.py b/litellm/llms/azure.py
index 0d15a09825c..b2490f8f950 100644
--- a/litellm/llms/azure.py
+++ b/litellm/llms/azure.py
@@ -214,6 +214,7 @@ class AzureChatCompletion(BaseLLM):
                 input: list,
                 api_key: str,
                 api_base: str,
+                azure_ad_token: str,
                 api_version: str,
                 logging_obj=None,
                 model_response=None,
@@ -221,7 +222,7 @@ class AzureChatCompletion(BaseLLM):
         super().embedding()
         exception_mapping_worked = False
         try: 
-            headers = self.validate_environment(api_key)
+            headers = self.validate_environment(api_key, azure_ad_token=azure_ad_token)
             # Ensure api_base ends with a trailing slash
             if not api_base.endswith('/'):
                 api_base += '/'