OpenWebUI/open-webui-custom
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
2025-12-29 01:31:27 +04:00
parent 5d1459df16
commit 2453b75ff0
5 changed files with 28 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -360,7 +360,7 @@ def check_model_access(user, model):
raise Exception("Model not found")
def get_filtered_models(models, user):
def get_filtered_models(models, user, db=None):
# Filter out models that the user does not have access to
if (
user.role == "user"
@@ -373,7 +373,7 @@ def get_filtered_models(models, user):
}
filtered_models = []
user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
for model in models:
if model.get("arena"):
if has_access(