UPDATE: Fix chatbot
All checks were successful
Build and Release / release (push) Successful in 1m24s

This commit is contained in:
2026-05-05 10:21:47 +07:00
parent de2e1cddf3
commit 44ebdeedc3
2 changed files with 18 additions and 5 deletions

View File

@@ -97,12 +97,24 @@ func (u *RagUtils) GenerateResponse(ctx context.Context, prompt string) (string,
}
func stripThinking(raw string) string {
startTag := "<answer>"
endTag := "</answer>"
startIdx := strings.Index(raw, startTag)
endIdx := strings.LastIndex(raw, endTag)
if startIdx != -1 && endIdx != -1 && endIdx > startIdx {
return strings.TrimSpace(raw[startIdx+len(startTag) : endIdx])
}
if startIdx != -1 {
return strings.TrimSpace(raw[startIdx+len(startTag):])
}
if !strings.Contains(raw, "* ") {
return strings.TrimSpace(raw)
}
lines := strings.Split(raw, "\n")
answerStart := len(lines)
for i := len(lines) - 1; i >= 0; i-- {
trimmed := strings.TrimSpace(lines[i])