allow setting litellm.api_key

This commit is contained in:
ishaan-jaff 2023-08-14 17:17:18 -07:00
parent 4f6958dc5b
commit 8b408bb83b
3 changed files with 4 additions and 3 deletions

View file

@ -5,6 +5,7 @@ set_verbose=False
telemetry=True
max_tokens = 256 # OpenAI Defaults
retry = True
api_key = None
openai_key = None
azure_key = None
anthropic_key = None

View file

@ -717,8 +717,8 @@ def get_secret(secret_name):
# if secret manager fails default to using .env variables
os.environ[secret_name] = secret # set to env to be safe
return secret
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)

View file

@ -1,6 +1,6 @@
[tool.poetry]
name = "litellm"
version = "0.1.389"
version = "0.1.390"
description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"]
license = "MIT License"