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

Merge pull request #2281 from mkhludnev/fix#2260-2261

Browse source 
fix #2260 #2261
This commit is contained in:
Krish Dholakia 2024-03-01 13:23:01 -08:00 • committed by GitHub
commit f9ef3ce32d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/utils.py
View file

@ -5512,7 +5512,7 @@ def validate_environment(model: Optional[str] = None) -> dict:
} }
## EXTRACT LLM PROVIDER - if model name provided ## EXTRACT LLM PROVIDER - if model name provided
try: try:
custom_llm_provider = get_llm_provider(model=model) _, custom_llm_provider, _, _ = get_llm_provider(model=model)
except: except:
custom_llm_provider = None custom_llm_provider = None
# # check if llm provider part of model name # # check if llm provider part of model name
@ -5606,7 +5606,7 @@ def validate_environment(model: Optional[str] = None) -> dict:
## openai - chatcompletion + text completion ## openai - chatcompletion + text completion
if ( if (
model in litellm.open_ai_chat_completion_models model in litellm.open_ai_chat_completion_models
or litellm.open_ai_text_completion_models or model in litellm.open_ai_text_completion_models
): ):
if "OPENAI_API_KEY" in os.environ: if "OPENAI_API_KEY" in os.environ:
keys_in_environment = True keys_in_environment = True