OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-04-14 01:08:16 -07:00
parent 403295600f
commit ec3b8fab5b
4 changed files with 69 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
backend/open_webui/socket/main.py
View File

@@ -339,16 +339,17 @@ def get_event_emitter(request_info, update_db=True):
request_info["message_id"],
)
content = message.get("content", "")
content += event_data.get("data", {}).get("content", "")
if message:
content = message.get("content", "")
content += event_data.get("data", {}).get("content", "")
Chats.upsert_message_to_chat_by_id_and_message_id(
request_info["chat_id"],
request_info["message_id"],
{
"content": content,
},
)
Chats.upsert_message_to_chat_by_id_and_message_id(
request_info["chat_id"],
request_info["message_id"],
{
"content": content,
},
)
if "type" in event_data and event_data["type"] == "replace":
content = event_data.get("data", {}).get("content", "")

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

@@ -537,8 +537,15 @@ async def chat_image_generation_handler(
for image in images:
await __event_emitter__(
{
"type": "message",
"data": {"content": f"![Generated Image]({image['url']})\n"},
"type": "files",
"data": {
"files": [
{
"type": "image",
"url": image["url"],
}
]
},
}
)