From 78f42921b9a8095a9bfc078cb626ca2e548f49aa Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 14 Nov 2023 12:36:32 -0800 Subject: [PATCH] (feat) pass azure_ad_token to completion --- litellm/main.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/main.py b/litellm/main.py index 1cc5352d4bd..47e09447c14 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -457,6 +457,7 @@ def completion( ) azure_ad_token = ( + optional_params.pop("azure_ad_token", None) or get_secret("AZURE_AD_TOKEN") ) @@ -1740,6 +1741,7 @@ def embedding( api_base=api_base, api_key=api_key, api_version=api_version, + azure_ad_token=None, logging_obj=logging, model_response=EmbeddingResponse(), optional_params=kwargs