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

(feat) completion:azure - add azure_ad_token

Browse source
This commit is contained in:
ishaan-jaff 2023-11-14 12:09:28 -08:00
parent 7a6586fa75
commit ceaf944cd4
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
litellm/main.py
View file

@ -456,6 +456,10 @@ def completion(
get_secret("AZURE_API_KEY") get_secret("AZURE_API_KEY")
) )
azure_ad_token = (
get_secret("AZURE_AD_TOKEN")
)
headers = ( headers = (
headers or headers or
litellm.headers litellm.headers
@ -476,6 +480,7 @@ def completion(
api_base=api_base, api_base=api_base,
api_version=api_version, api_version=api_version,
api_type=api_type, api_type=api_type,
azure_ad_token=azure_ad_token,
model_response=model_response, model_response=model_response,
print_verbose=print_verbose, print_verbose=print_verbose,
optional_params=optional_params, optional_params=optional_params,