Update format.go (#312)

pkg/chatgpt/format.go

@@ -1,12 +1,21 @@
 package chatgpt
 
+import (
+	"regexp"
+	"strings"
+)
+
+// 适配 deepseek r1
 func formatAnswer(answer string) string {
-	for len(answer) > 0 {
+	answer = strings.TrimSpace(answer)
-		if answer[:1] == "\n" || answer[0] == ' ' {
+
-			answer = answer[1:]
+	re := regexp.MustCompile(`(?s)<think>.*?</think>`)
-		} else {
+	answer = re.ReplaceAllString(answer, "")
-			break
+
-		}
+	answer = strings.ReplaceAll(answer, "<think>", "")
-	}
+	answer = strings.ReplaceAll(answer, "</think>", "")
+
+	answer = strings.TrimSpace(answer)
+
 	return answer
 }