phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

add azure_ad_token_provider as all litellm params

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

1
litellm/types/utils.py
View file

@ -1116,6 +1116,7 @@ all_litellm_params = [
"cooldown_time", "cooldown_time",
"cache_key", "cache_key",
"max_retries", "max_retries",
"azure_ad_token_provider",
] ]

2
litellm/utils.py
View file

@ -2323,6 +2323,7 @@ def get_litellm_params(
output_cost_per_second=None, output_cost_per_second=None,
cooldown_time=None, cooldown_time=None,
text_completion=None, text_completion=None,
azure_ad_token_provider=None,
): ):
litellm_params = { litellm_params = {
"acompletion": acompletion, "acompletion": acompletion,
@ -2347,6 +2348,7 @@ def get_litellm_params(
"output_cost_per_second": output_cost_per_second, "output_cost_per_second": output_cost_per_second,
"cooldown_time": cooldown_time, "cooldown_time": cooldown_time,
"text_completion": text_completion, "text_completion": text_completion,
"azure_ad_token_provider": azure_ad_token_provider,
} }
return litellm_params return litellm_params