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

fixed get_secret

Browse source
This commit is contained in:
ishaan-jaff 2023-08-14 17:34:28 -07:00
parent f5b6c04de4
commit 952e994ad1
2 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/utils.py
View file

@ -713,14 +713,14 @@ 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:
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
elif litellm.api_key != None: # if users use litellm default key
return litellm.api_key
else:
return os.environ.get(secret_name)
elif litellm.api_key != None: # if users use litellm default key
return litellm.api_key
else:
return os.environ.get(secret_name)
######## Streaming Class ############################
# wraps the completion stream to return the correct format for the model