phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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 ################