scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 136

(fix) undo change

Browse source
This commit is contained in:
ishaan-jaff 2024-02-02 12:38:29 -08:00
parent e37294c921
commit 556ca48448
1 changed files with 6 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
litellm/main.py
View file

@ -807,16 +807,12 @@ def completion(
# set API KEY
api_key = (
api_key or litellm.api_key
) # for deepinfra/perplexity/anyscale we check in get_llm_provider and pass in the api key from there
# only for OpenAI we check env OPENAI_API_KEY, and litellm.openai_key
if (
custom_llm_provider == "openai"
or custom_llm_provider == "custom_openai"
or model in litellm.open_ai_chat_completion_models
or "ft:gpt-3.5-turbo" in model
):
api_key = api_key or litellm.openai_key or get_secret("OPENAI_API_KEY")
api_key
or litellm.api_key # for deepinfra/perplexity/anyscale we check in get_llm_provider and pass in the api key from there
or litellm.openai_key
or get_secret("OPENAI_API_KEY")
)
headers = headers or litellm.headers
## LOAD CONFIG - if set