From 8b440ad5c84a2e6bc46643e3f04374e5ba114b08 Mon Sep 17 00:00:00 2001
From: AzenKain <doduykhanh344@gmail.com>
Date: Tue, 5 May 2026 10:28:44 +0700
Subject: [PATCH] UPDATE: Fix chatbot

---
 pkg/ai/rag.go | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/pkg/ai/rag.go b/pkg/ai/rag.go
index 41aa07c..c2476dd 100644
--- a/pkg/ai/rag.go
+++ b/pkg/ai/rag.go
@@ -99,15 +99,14 @@ 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):])
+	lastStart := strings.LastIndex(raw, startTag)
+	if lastStart != -1 {
+		content := raw[lastStart+len(startTag):]
+		if endIdx := strings.Index(content, endTag); endIdx != -1 {
+			return strings.TrimSpace(content[:endIdx])
+		}
+		return strings.TrimSpace(content)
 	}
 
 	if !strings.Contains(raw, "*   ") {