tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-18 04:28:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(main.py): read azure ad token from optional params extra body

Browse Source
This commit is contained in:
Krrish Dholakia
2024-01-18 17:14:03 -08:00
parent a8ba5df90e
commit e0aaa94f28
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/main.py
+3 -3
View File
@@ -692,9 +692,9 @@ def completion(
or get_secret("AZURE_API_KEY")
)
azure_ad_token = optional_params.pop("azure_ad_token", None) or get_secret(
"AZURE_AD_TOKEN"
)
azure_ad_token = optional_params.get("extra_body", {}).pop(
"azure_ad_token", None
) or get_secret("AZURE_AD_TOKEN")
headers = headers or litellm.headers
litellm/tests/test_completion.py
+3 -3
View File
@@ -268,7 +268,7 @@ def test_completion_azure_gpt4_vision():
pytest.fail(f"Error occurred: {e}")
test_completion_azure_gpt4_vision()
# test_completion_azure_gpt4_vision()
@pytest.mark.skip(reason="this test is flaky")
@@ -990,9 +990,9 @@ def test_azure_openai_ad_token():
print("azure ad token respoonse\n")
print(response)
litellm.input_callback = []
except:
except Exception as e:
litellm.input_callback = []
pass
pytest.fail(f"An exception occurs - {str(e)}")
# test_azure_openai_ad_token()