From 0779dfbbd6f3c6ae72e2ce4f6f36ac39d269b800 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 14 Nov 2023 12:20:14 -0800
Subject: [PATCH] (fix) azure_ad_token

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

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 += '/'