mirror/nanobot
1
0
Fork 0
mirror of https://github.com/HKUDS/nanobot.git synced 2026-04-10 21:23:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: strip <thought> blocks from Gemma 4 and similar models

Browse Source
This commit is contained in:
flobo3 2026-04-09 17:51:29 +03:00 committed by Xubin Ren
parent 69d748bf8f
commit 6b7e78a8e0
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nanobot/utils/helpers.py
View File

@ -15,9 +15,12 @@ from loguru import logger
def strip_think(text: str) -> str:
"""Remove <think>…</think> blocks and any unclosed trailing <think> tag."""
"""Remove thinking blocks and any unclosed trailing tag."""
text = re.sub(r"<think>[\s\S]*?</think>", "", text)
text = re.sub(r"<think>[\s\S]*$", "", text)
# Gemma 4 and similar models use <thought>...</thought> blocks
text = re.sub(r"<thought>[\s\S]*?</thought>", "", text)
text = re.sub(r"<thought>[\s\S]*$", "", text)
return text.strip()