bhetherman/open-webui
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
2026-08-24 18:35:04 -04:00
parent b96d2b12da
commit 536b9edec0
2 changed files with 36 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/socket/main.py
+11 -23
View File
@@ -1074,15 +1074,12 @@ async def get_event_emitter(request_info, update_db=True):
embeds = event_payload.get('embeds', [])
if not event_payload.get('replace', False):
message = await Chats.get_message_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
)
embeds.extend(message.get('embeds', []))
existing_embeds = await Chats.get_message_metadata_list(chat_id, message_id, 'embeds')
embeds.extend(existing_embeds)
await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
chat_id,
message_id,
{
'embeds': embeds,
},
@@ -1090,17 +1087,13 @@ async def get_event_emitter(request_info, update_db=True):
)
elif event_type == 'files':
message = await Chats.get_message_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
)
files = event_data.get('data', {}).get('files', [])
files.extend(message.get('files', []))
existing_files = await Chats.get_message_metadata_list(chat_id, message_id, 'files')
files.extend(existing_files)
await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
chat_id,
message_id,
{
'files': files,
},
@@ -1110,17 +1103,12 @@ async def get_event_emitter(request_info, update_db=True):
elif event_type in ('source', 'citation'):
data = event_data.get('data', {})
if data.get('type') is None:
message = await Chats.get_message_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
)
sources = message.get('sources', [])
sources = await Chats.get_message_metadata_list(chat_id, message_id, 'sources')
sources.append(data)
await Chats.upsert_message_to_chat_by_id_and_message_id(
request_info['chat_id'],
request_info['message_id'],
chat_id,
message_id,
{
'sources': sources,
},