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: title generation

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-01-29 14:40:36 -08:00
parent 4abede9a2b
commit 5420c165c6
3 changed files with 35 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -892,16 +892,24 @@ async def process_chat_response(
if res and isinstance(res, dict):
if len(res.get("choices", [])) == 1:
title = (
title_string = (
res.get("choices", [])[0]
.get("message", {})
.get(
"content",
message.get("content", "New Chat"),
)
).strip()
.get("content", message.get("content", "New Chat"))
)
else:
title = None
title_string = ""
title_string = title_string[
title_string.find("{") : title_string.rfind("}") + 1
]
try:
title = json.loads(title_string).get(
"title", "New Chat"
)
except Exception as e:
pass
if not title:
title = messages[0].get("content", "New Chat")