chatgpt-dingtalk/pkg/chatgpt/format.go

package chatgpt

import (
	"regexp"
	"strings"
)

// 适配 deepseek r1
func formatAnswer(answer string) string {
	answer = strings.TrimSpace(answer)

	re := regexp.MustCompile(`(?s)<think>.*?</think>`)
	answer = re.ReplaceAllString(answer, "")

	answer = strings.ReplaceAll(answer, "<think>", "")
	answer = strings.ReplaceAll(answer, "</think>", "")

	answer = strings.TrimSpace(answer)

	return answer
}