add azure_ad_token_provider as all litellm params

This commit is contained in:
Ishaan Jaff 2024-08-22 10:59:18 -07:00
parent a120135dd1
commit 2a3bc8c190
2 changed files with 3 additions and 0 deletions

View file

@ -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