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

update get model list

Browse source
This commit is contained in:
Krrish Dholakia 2023-09-07 13:15:50 -07:00
parent 072d93696c
commit c7e34bce5f
2 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/utils.py
View file

@ -1512,20 +1512,20 @@ def get_model_list():
data = response.json() data = response.json()
# update model list # update model list
model_list = data["model_list"] model_list = data["model_list"]
# check if all model providers are in environment # # check if all model providers are in environment
model_providers = data["model_providers"] # model_providers = data["model_providers"]
missing_llm_provider = None # missing_llm_provider = None
for item in model_providers: # for item in model_providers:
if f"{item.upper()}_API_KEY" not in os.environ: # if f"{item.upper()}_API_KEY" not in os.environ:
missing_llm_provider = item # missing_llm_provider = item
break # break
# update environment - if required # # update environment - if required
threading.Thread(target=get_all_keys, args=(missing_llm_provider)).start() # threading.Thread(target=get_all_keys, args=(missing_llm_provider)).start()
return model_list return model_list
return [] # return empty list by default return [] # return empty list by default
except: except:
print_verbose( print_verbose(
f"[Non-Blocking Error] get_all_keys error - {traceback.format_exc()}" f"[Non-Blocking Error] get_model_list error - {traceback.format_exc()}"
) )
####### EXCEPTION MAPPING ################ ####### EXCEPTION MAPPING ################

2
pyproject.toml
View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "litellm" name = "litellm"
version = "0.1.556" version = "0.1.557"
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"