Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-02-24 12:11:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-01-05 03:46:46 +04:00
parent 2bb13d5dbc
commit b55a46ae99
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -337,7 +337,7 @@ async def get_all_models(request, refresh: bool = False, user: UserModel = None)
return models
def check_model_access(user, model):
def check_model_access(user, model, db=None):
if model.get("arena"):
if not has_access(
user.id,
@@ -345,16 +345,17 @@ def check_model_access(user, model):
access_control=model.get("info", {})
.get("meta", {})
.get("access_control", {}),
db=db,
):
raise Exception("Model not found")
else:
model_info = Models.get_model_by_id(model.get("id"))
model_info = Models.get_model_by_id(model.get("id"), db=db)
if not model_info:
raise Exception("Model not found")
elif not (
user.id == model_info.user_id
or has_access(
user.id, type="read", access_control=model_info.access_control
user.id, type="read", access_control=model_info.access_control, db=db
)
):
raise Exception("Model not found")
@@ -373,7 +374,9 @@ def get_filtered_models(models, user, db=None):
}
filtered_models = []
user_group_ids = {group.id for group in Groups.get_groups_by_member_id(user.id, db=db)}
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(