github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-05-01 17:59:28 -05:00
Code Issues 915 Packages Projects Releases 126 Wiki Activity

enh: models endpoint optimization

Browse Source 
Co-Authored-By: Classic298 <27028174+Classic298@users.noreply.github.com>

#20010
This commit is contained in:
Timothy Jaeryang Baek
2025-12-21 15:43:02 +04:00
parent 53c1ca64b7
commit 0dd2cfe1f2
2 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/open_webui/utils/models.py
View File

@@ -366,6 +366,12 @@ def get_filtered_models(models, user):
user.role == "user"
or (user.role == "admin" and not BYPASS_ADMIN_ACCESS_CONTROL)
) and not BYPASS_MODEL_ACCESS_CONTROL:
model_ids = [model["id"] for model in models if not model.get("arena")]
model_infos = {
model_info.id: model_info
for model_info in Models.get_models_by_ids(model_ids)
}
filtered_models = []
user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
for model in models:
@@ -381,7 +387,7 @@ def get_filtered_models(models, user):
filtered_models.append(model)
continue
model_info = Models.get_model_by_id(model["id"])
model_info = model_infos.get(model["id"], None)
if model_info:
if (
(user.role == "admin" and BYPASS_ADMIN_ACCESS_CONTROL)