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

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 00454df83f
commit 1b723d4694
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")