grail/gf-lt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: collapsing thinking while thinking

Browse Source
This commit is contained in:
Grail Finder
2026-02-21 20:28:12 +03:00
parent deece322ef
commit 66ccb7a732
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bot.go
View File

@@ -1197,6 +1197,16 @@ func chatToText(messages []models.RoleMsg, showSys bool) string {
} }
return "[yellow::i][thinking... (press Alt+T to expand)][-:-:-]" return "[yellow::i][thinking... (press Alt+T to expand)][-:-:-]"
}) })
// Handle incomplete thinking blocks (during streaming when </think> hasn't arrived yet)
if strings.Contains(text, "<think>") && !strings.Contains(text, "</think>") {
// Find the incomplete thinking block and replace it
startIdx := strings.Index(text, "<think>")
if startIdx != -1 {
content := text[startIdx+len("<think>"):]
placeholder := fmt.Sprintf("[yellow::i][thinking... (%d chars) (press Alt+T to expand)][-:-:-]", len(content))
text = text[:startIdx] + placeholder
}
}
} }
return text return text