phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update get secret logic

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-28 07:21:02 -07:00
parent 841ac5ab20
commit 9497175a97
2 changed files with 8 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

13
litellm/utils.py
View file

@ -1444,13 +1444,12 @@ def get_secret(secret_name):
if litellm.secret_manager_client != None:
# TODO: check which secret manager is being used
# currently only supports Infisical
secret = litellm.secret_manager_client.get_secret(
secret_name).secret_value
if secret != None:
return secret # if secret found in secret manager return it
else:
raise ValueError(
f"Secret '{secret_name}' not found in secret manager")
try:
secret = litellm.secret_manager_client.get_secret(
secret_name).secret_value
except:
secret = None
return secret
elif litellm.api_key != None: # if users use litellm default key
return litellm.api_key
else: