scriptoriumadmin/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-10-18 04:43:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refac

Browse source 
Co-Authored-By: Classic298 <27028174+Classic298@users.noreply.github.com>
This commit is contained in:
Timothy Jaeryang Baek 2026-09-21 10:25:20 -04:00
parent c864e3ae7c
commit 754c4b5762
1 changed files with 24 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
backend/open_webui/routers/models.py
View file

@ -27,7 +27,7 @@ from open_webui.env import (
) )
from open_webui.events import EVENTS, publish_event from open_webui.events import EVENTS, publish_event
from open_webui.internal.db import get_async_session from open_webui.internal.db import get_async_session
from open_webui.models.access_grants import AccessGrants from open_webui.models.access_grants import AccessGrants, normalize_access_grants
from open_webui.models.config import Config from open_webui.models.config import Config
from open_webui.models.files import Files from open_webui.models.files import Files
from open_webui.models.groups import Groups from open_webui.models.groups import Groups
@ -985,13 +985,29 @@ async def update_model_by_id(
form_data.meta.background_image_url = model.meta.background_image_url form_data.meta.background_image_url = model.meta.background_image_url
await _verify_background_image(form_data.meta.background_image_url, user, db, model.meta.background_image_url) await _verify_background_image(form_data.meta.background_image_url, user, db, model.meta.background_image_url)
form_data.access_grants = await filter_allowed_access_grants( if form_data.access_grants is not None:
await Config.get('user.permissions'), # The editor resends every stored grant, so re-checking them would strip sharing this user cannot re-create.
user.id, existing_access_grants = {
user.role, (grant.principal_type, grant.principal_id, grant.permission) for grant in model.access_grants
form_data.access_grants, }
'sharing.public_models', submitted_access_grants_map = {
) (grant['principal_type'], grant['principal_id'], grant['permission']): grant
for grant in normalize_access_grants(form_data.access_grants)
}
preserved_access_grants = [
grant for key, grant in submitted_access_grants_map.items() if key in existing_access_grants
]
new_access_grants = [
grant for key, grant in submitted_access_grants_map.items() if key not in existing_access_grants
]
form_data.access_grants = preserved_access_grants + await filter_allowed_access_grants(
await Config.get('user.permissions'),
user.id,
user.role,
new_access_grants,
'sharing.public_models',
)
model = await Models.update_model_by_id(form_data.id, ModelForm(**form_data.model_dump()), db=db) model = await Models.update_model_by_id(form_data.id, ModelForm(**form_data.model_dump()), db=db)
if model: if model: