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-01-10 16:27:37 -08:00
parent dd08b457d8
commit 21abe43f3b
1 changed files with 20 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -770,14 +770,17 @@ async def process_chat_response(
)
if res and isinstance(res, dict):
title = (
res.get("choices", [])[0]
.get("message", {})
.get(
"content",
message.get("content", "New Chat"),
)
).strip()
if len(res.get("choices", [])) == 1:
title = (
res.get("choices", [])[0]
.get("message", {})
.get(
"content",
message.get("content", "New Chat"),
)
).strip()
else:
title = None
if not title:
title = messages[0].get("content", "New Chat")
@ -814,11 +817,14 @@ async def process_chat_response(
)
if res and isinstance(res, dict):
tags_string = (
res.get("choices", [])[0]
.get("message", {})
.get("content", "")
)
if len(res.get("choices", [])) == 1:
tags_string = (
res.get("choices", [])[0]
.get("message", {})
.get("content", "")
)
else:
tags_string = ""
tags_string = tags_string[
tags_string.find("{") : tags_string.rfind("}") + 1
@ -837,7 +843,7 @@ async def process_chat_response(
}
)
except Exception as e:
print(f"Error: {e}")
pass
event_emitter = None
if (