From 53ed8f3b7e0bdc61814f21ddceb04de3cb518378 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 14 Nov 2023 12:38:01 -0800 Subject: [PATCH] (feat) pass azure_ad_token to embedding --- litellm/main.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/main.py b/litellm/main.py index 47e09447c14..cb78d56d11f 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -1728,6 +1728,11 @@ def embedding( get_secret("AZURE_API_VERSION") ) + azure_ad_token = ( + kwargs.pop("azure_ad_token", None) or + get_secret("AZURE_AD_TOKEN") + ) + api_key = ( api_key or litellm.api_key or