ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 12:25:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Merge pull request #19097 from adam-skalicky/api_models_perf_optimization

Browse source 
perf: Fetched user_group_ids prior to looping through models with has_access to reduce DB hits for group membership
This commit is contained in:
Tim Baek 2025-11-11 00:16:14 -05:00 committed by GitHub
commit 8bff76f745
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -12,6 +12,7 @@ from open_webui.functions import get_function_models
from open_webui.models.functions import Functions from open_webui.models.functions import Functions
from open_webui.models.models import Models from open_webui.models.models import Models
from open_webui.models.groups import Groups
from open_webui.utils.plugin import ( from open_webui.utils.plugin import (
@ -356,6 +357,7 @@ def get_filtered_models(models, 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:
filtered_models = [] filtered_models = []
user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id)}
for model in models: for model in models:
if model.get("arena"): if model.get("arena"):
if has_access( if has_access(
@ -364,6 +366,7 @@ def get_filtered_models(models, user):
access_control=model.get("info", {}) access_control=model.get("info", {})
.get("meta", {}) .get("meta", {})
.get("access_control", {}), .get("access_control", {}),
user_group_ids=user_group_ids,
): ):
filtered_models.append(model) filtered_models.append(model)
continue continue
@ -377,6 +380,7 @@ def get_filtered_models(models, user):
user.id, user.id,
type="read", type="read",
access_control=model_info.access_control, access_control=model_info.access_control,
user_group_ids=user_group_ids,
) )
): ):
filtered_models.append(model) filtered_models.append(model)