Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-02-16 23:52:32 -06:00
parent aede1b7a08
commit 34cd3d79e8
1 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/utils/models.py
+10 -8
View File
@@ -395,11 +395,13 @@ def get_filtered_models(models, user, db=None):
user.role == "user" user.role == "user"
or (user.role == "admin" and not BYPASS_ADMIN_ACCESS_CONTROL) or (user.role == "admin" and not BYPASS_ADMIN_ACCESS_CONTROL)
) and not BYPASS_MODEL_ACCESS_CONTROL: ) and not BYPASS_MODEL_ACCESS_CONTROL:
model_ids = [model["id"] for model in models if not model.get("arena")] model_infos = {}
model_infos = { for model in models:
model_info.id: model_info if model.get("arena"):
for model_info in Models.get_models_by_ids(model_ids) continue
} info = model.get("info")
if info:
model_infos[model["id"]] = info
user_group_ids = { user_group_ids = {
group.id for group in Groups.get_groups_by_member_id(user.id, db=db) group.id for group in Groups.get_groups_by_member_id(user.id, db=db)
@@ -429,12 +431,12 @@ def get_filtered_models(models, user, db=None):
filtered_models.append(model) filtered_models.append(model)
continue continue
model_info = model_infos.get(model["id"], None) model_info = model_infos.get(model["id"])
if model_info: if model_info:
if ( if (
(user.role == "admin" and BYPASS_ADMIN_ACCESS_CONTROL) (user.role == "admin" and BYPASS_ADMIN_ACCESS_CONTROL)
or user.id == model_info.user_id or user.id == model_info["user_id"]
or model_info.id in accessible_model_ids or model["id"] in accessible_model_ids
): ):
filtered_models.append(model) filtered_models.append(model)