OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: model filter frontend integration

Browse Source
This commit is contained in:
Timothy J. Baek
2024-03-09 21:29:04 -08:00
parent b550e23bf6
commit 81daf4ceb9
4 changed files with 83 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/apps/ollama/main.py
View File

@@ -135,11 +135,14 @@ async def get_ollama_tags(
):
if url_idx == None:
models = await get_all_models()
if app.state.MODEL_FILTER_ENABLED:
if user.role == "user":
models["models"] = filter(
lambda model: model["name"] in app.state.MODEL_LIST,
models["models"],
models["models"] = list(
filter(
lambda model: model["name"] in app.state.MODEL_LIST,
models["models"],
)
)
return models
return models

8
backend/apps/openai/main.py
View File

@@ -196,9 +196,11 @@ async def get_models(url_idx: Optional[int] = None, user=Depends(get_current_use
models = await get_all_models()
if app.state.MODEL_FILTER_ENABLED:
if user.role == "user":
models["data"] = filter(
lambda model: model["id"] in app.state.MODEL_LIST,
models["data"],
models["data"] = list(
filter(
lambda model: model["id"] in app.state.MODEL_LIST,
models["data"],
)
)
return models
return models