Kain344/History_Api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Azen Kain
2026-05-05 10:28:44 +07:00
parent 44ebdeedc3
commit 8b440ad5c8
1 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
pkg/ai/rag.go
View File

@@ -99,15 +99,14 @@ func (u *RagUtils) GenerateResponse(ctx context.Context, prompt string) (string,
func stripThinking(raw string) string { func stripThinking(raw string) string {
startTag := "<answer>" startTag := "<answer>"
endTag := "</answer>" endTag := "</answer>"
startIdx := strings.Index(raw, startTag)
endIdx := strings.LastIndex(raw, endTag)
if startIdx != -1 && endIdx != -1 && endIdx > startIdx { lastStart := strings.LastIndex(raw, startTag)
return strings.TrimSpace(raw[startIdx+len(startTag) : endIdx]) if lastStart != -1 {
} content := raw[lastStart+len(startTag):]
if endIdx := strings.Index(content, endTag); endIdx != -1 {
if startIdx != -1 { return strings.TrimSpace(content[:endIdx])
return strings.TrimSpace(raw[startIdx+len(startTag):]) }
return strings.TrimSpace(content)
} }
if !strings.Contains(raw, "* ") { if !strings.Contains(raw, "* ") {