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

fix(proxy_server.py): fix user model returned in /models

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-27 08:04:29 -08:00
parent e9947be7b2
commit aafba24e84
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/proxy_server.py
View file

@ -632,7 +632,8 @@ def model_list():
print(f"llm model list: {llm_model_list}") print(f"llm model list: {llm_model_list}")
all_models += [m["model_name"] for m in llm_model_list] all_models += [m["model_name"] for m in llm_model_list]
if user_model is not None: if user_model is not None:
all_models += user_model all_models += [user_model]
print(f"all_models: {all_models}")
### CHECK OLLAMA MODELS ### ### CHECK OLLAMA MODELS ###
try: try:
response = requests.get("http://0.0.0.0:11434/api/tags") response = requests.get("http://0.0.0.0:11434/api/tags")