This commit is contained in:
Krrish Dholakia 2023-08-21 06:05:45 -07:00
parent a1ad773667
commit ecdc7abfd8
3 changed files with 3 additions and 3 deletions

View file

@ -186,7 +186,7 @@ def completion(
# set API KEY # set API KEY
if not api_key and litellm.openai_key: if not api_key and litellm.openai_key:
api_key = litellm.openai_key api_key = litellm.openai_key
elif not api_key and get_secret("AZURE_API_KEY"): elif not api_key and get_secret("OPENAI_API_KEY"):
api_key = get_secret("OPENAI_API_KEY") api_key = get_secret("OPENAI_API_KEY")
openai.api_key = api_key openai.api_key = api_key
@ -218,7 +218,7 @@ def completion(
# set API KEY # set API KEY
if not api_key and litellm.openai_key: if not api_key and litellm.openai_key:
api_key = litellm.openai_key api_key = litellm.openai_key
elif not api_key and get_secret("AZURE_API_KEY"): elif not api_key and get_secret("OPENAI_API_KEY"):
api_key = get_secret("OPENAI_API_KEY") api_key = get_secret("OPENAI_API_KEY")
openai.api_key = api_key openai.api_key = api_key

View file

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