UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-23 13:51:07 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: <think> tag

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-02-05 02:10:28 -08:00
parent 1ab66e83c6
commit c36c63f1e8
1 changed files with 26 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1233,6 +1233,16 @@ async def process_chat_response(
].replace(match.group(0), "")
if not content_blocks[-1]["content"]:
content_blocks.pop()
if not content_blocks:
# Append the new block
content_blocks.append(
{
"type": "text",
"content": "",
}
)
# Append the new block
content_blocks.append(
{
@ -1461,14 +1471,23 @@ async def process_chat_response(
log.debug("Error: ", e)
continue
# Clean up the last text block
if content_blocks[-1]["type"] == "text":
content_blocks[-1]["content"] = content_blocks[-1][
"content"
].strip()
if content_blocks:
# Clean up the last text block
if content_blocks[-1]["type"] == "text":
content_blocks[-1]["content"] = content_blocks[-1][
"content"
].strip()
if not content_blocks[-1]["content"]:
content_blocks.pop()
if not content_blocks[-1]["content"]:
content_blocks.pop()
if not content_blocks:
content_blocks.append(
{
"type": "text",
"content": "",
}
)
if response_tool_calls:
tool_calls.append(response_tool_calls)