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()
# update model list
model_list = data["model_list"]
# check if all model providers are in environment
model_providers = data["model_providers"]
missing_llm_provider = None
for item in model_providers:
if f"{item.upper()}_API_KEY" not in os.environ:
missing_llm_provider = item
break
# update environment - if required
threading.Thread(target=get_all_keys, args=(missing_llm_provider)).start()
# # check if all model providers are in environment
# model_providers = data["model_providers"]
# missing_llm_provider = None
# for item in model_providers:
# if f"{item.upper()}_API_KEY" not in os.environ:
# missing_llm_provider = item
# break
# # update environment - if required
# threading.Thread(target=get_all_keys, args=(missing_llm_provider)).start()
return model_list
return [] # return empty list by default
except:
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 ################

2
pyproject.toml
View file

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