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