From 2a3bc8c19029d7f7584a25631f0858a509c9949a Mon Sep 17 00:00:00 2001
From: Ishaan Jaff <ishaanjaffer0324@gmail.com>
Date: Thu, 22 Aug 2024 10:59:18 -0700
Subject: [PATCH] add azure_ad_token_provider as all litellm params

---
 litellm/types/utils.py | 1 +
 litellm/utils.py       | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/litellm/types/utils.py b/litellm/types/utils.py
index 8efbe5a11..62a3a5bcd 100644
--- a/litellm/types/utils.py
+++ b/litellm/types/utils.py
@@ -1116,6 +1116,7 @@ all_litellm_params = [
     "cooldown_time",
     "cache_key",
     "max_retries",
+    "azure_ad_token_provider",
 ]
 
 
diff --git a/litellm/utils.py b/litellm/utils.py
index 0e9e531e9..ea6b99524 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -2323,6 +2323,7 @@ def get_litellm_params(
     output_cost_per_second=None,
     cooldown_time=None,
     text_completion=None,
+    azure_ad_token_provider=None,
 ):
     litellm_params = {
         "acompletion": acompletion,
@@ -2347,6 +2348,7 @@ def get_litellm_params(
         "output_cost_per_second": output_cost_per_second,
         "cooldown_time": cooldown_time,
         "text_completion": text_completion,
+        "azure_ad_token_provider": azure_ad_token_provider,
     }
 
     return litellm_params