phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #3861 from Manouchehri/aks-oidc-1852

Browse source 
feat(util.py/azure.py): Add OIDC support when running LiteLLM on Azure + Azure Upstream caching
This commit is contained in:
Krish Dholakia 2024-06-12 12:47:08 -07:00 • committed by GitHub
commit 821d32fe17
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 41 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

8
litellm/utils.py
View file

@ -10066,6 +10066,14 @@ def get_secret(
return oidc_token
else:
raise ValueError("Github OIDC provider failed")
elif oidc_provider == "azure":
# https://azure.github.io/azure-workload-identity/docs/quick-start.html
azure_federated_token_file = os.getenv("AZURE_FEDERATED_TOKEN_FILE")
if azure_federated_token_file is None:
raise ValueError("AZURE_FEDERATED_TOKEN_FILE not found in environment")
with open(azure_federated_token_file, "r") as f:
oidc_token = f.read()
return oidc_token
else:
raise ValueError("Unsupported OIDC provider")