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 }