From ceaf944cd4557ab7f6f611f0b2cc2a43f12ad9da Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 14 Nov 2023 12:09:28 -0800 Subject: [PATCH] (feat) completion:azure - add azure_ad_token --- litellm/main.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/main.py b/litellm/main.py index 49416fdec..1cc5352d4 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -456,6 +456,10 @@ def completion( get_secret("AZURE_API_KEY") ) + azure_ad_token = ( + get_secret("AZURE_AD_TOKEN") + ) + headers = ( headers or litellm.headers @@ -476,6 +480,7 @@ def completion( api_base=api_base, api_version=api_version, api_type=api_type, + azure_ad_token=azure_ad_token, model_response=model_response, print_verbose=print_verbose, optional_params=optional_params,