phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use azure_ad_token_provider to init clients

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-22 11:03:49 -07:00
parent 26354fbb9d
commit 525d152d85
3 changed files with 35 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -943,6 +943,7 @@ def completion(
output_cost_per_token=output_cost_per_token,
cooldown_time=cooldown_time,
text_completion=kwargs.get("text_completion"),
azure_ad_token_provider=kwargs.get("azure_ad_token_provider"),
)
logging.update_environment_variables(
model=model,
@ -3229,6 +3230,7 @@ def embedding(
"model_config",
"cooldown_time",
"tags",
"azure_ad_token_provider",
]
default_params = openai_params + litellm_params
non_default_params = {