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
This commit is contained in:
Timothy Jaeryang Baek 2026-10-16 04:20:14 +04:00
parent 8c85f75d67
commit 788a61180b
2 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
backend/open_webui/models/chats.py
View file

@ -1270,10 +1270,8 @@ class ChatTable:
parent_id = None
if user_message:
existing = messages.get(user_message_id)
if existing and (
existing.get('role') != 'user' or (existing.get('user_id') or chat.user_id) != user.id
):
raise HTTPException(403, 'You can only regenerate your own messages.')
if existing and existing.get('role') != 'user':
raise HTTPException(403, 'You can only regenerate replies to user messages.')
parent_id = existing.get('parentId') if existing else user_message.get('parentId')
if parent_id is not None and parent_id not in messages:
raise HTTPException(409, 'Parent message no longer exists.')
@ -1292,7 +1290,7 @@ class ChatTable:
user_id=user.id,
user={'id': user.id, 'name': user.name},
)
elif 'models' in user_message:
elif 'models' in user_message and (existing.get('user_id') or chat.user_id) == user.id:
message = {**existing, 'models': user_message['models']}
turn[user_message_id] = self.upsert_message_to_history(
history, user_message_id, self._clean_null_bytes(message)
@ -1320,7 +1318,9 @@ class ChatTable:
chat.current_message_id = history['currentId']
chat.updated_at = int(time.time())
for mid, message in turn.items():
await ChatMessages.upsert_message(mid, chat_id, user.id, message, db=db)
await ChatMessages.upsert_message(
mid, chat_id, message.get('user_id') or chat.user_id, message, db=db
)
if parent_id:
turn[parent_id] = history['messages'][parent_id]
return {'messages': turn, 'currentId': history['currentId']}

4
src/lib/components/chat/Messages/ResponseMessage.svelte
View file

@ -1421,7 +1421,9 @@
</button>
</Tooltip>
{/if}
{/if}
{#if (!readOnly && (!message.user_id || message.user_id === $user?.id)) || shareMode === 'continue'}
{#if $user?.role === 'admin' || ($user?.permissions?.chat?.regenerate_response ?? true)}
{#if $settings?.regenerateMenu ?? true}
<button
@ -1569,7 +1571,9 @@
</Tooltip>
{/if}
{/if}
{/if}
{#if (!readOnly && (!message.user_id || message.user_id === $user?.id)) || (shareMode === 'continue' && message.user_id === $user?.id)}
{#each (!readOnly && model?.actions) || [] as action}
<Tooltip
content={resolveLocalizedFunction(