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

refactor setting keys

Browse source
This commit is contained in:
ishaan-jaff 2023-08-17 15:14:25 -07:00
parent 04c55b8f28
commit 3f5e47e3ce
2 changed files with 15 additions and 32 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/utils.py
View file

@ -833,10 +833,8 @@ def get_secret(secret_name):
# 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: # failsafe when secret manager fails
# if secret manager fails default to using .env variables
os.environ[secret_name] = secret # set to env to be safe
return secret
if secret != None:
return secret # if secret found in secret manager return it
elif litellm.api_key != None: # if users use litellm default key
return litellm.api_key
else: