diff --git a/pkg/chatgpt/format.go b/pkg/chatgpt/format.go
index 492e5e2..9dbc294 100644
--- a/pkg/chatgpt/format.go
+++ b/pkg/chatgpt/format.go
@@ -1,12 +1,21 @@
package chatgpt
+import (
+ "regexp"
+ "strings"
+)
+
+// 适配 deepseek r1
func formatAnswer(answer string) string {
- for len(answer) > 0 {
- if answer[:1] == "\n" || answer[0] == ' ' {
- answer = answer[1:]
- } else {
- break
- }
- }
+ answer = strings.TrimSpace(answer)
+
+ re := regexp.MustCompile(`(?s).*?`)
+ answer = re.ReplaceAllString(answer, "")
+
+ answer = strings.ReplaceAll(answer, "", "")
+ answer = strings.ReplaceAll(answer, "", "")
+
+ answer = strings.TrimSpace(answer)
+
return answer
}