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

updates to get model list

Browse source
This commit is contained in:
Krrish Dholakia 2023-08-24 19:06:44 -07:00
parent 7d16df98ac
commit acccf17e44
3 changed files with 9 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

BIN
litellm/__pycache__/utils.cpython-311.pyc
View file

Binary file not shown.

9
litellm/utils.py
View file

@ -1197,8 +1197,15 @@ 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
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 # update environment - if required
threading.Thread(target=get_all_keys, args=()).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:

2
pyproject.toml
View file

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