xmengnet/go-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

feat(ai): 添加stripMarkdown函数以移除消息中的Markdown格式

Browse source
This commit is contained in:
liyp 2024-09-07 16:18:20 +08:00
parent 554949feb7
commit cbdaba63a9
1 changed files with 18 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
workers/ai.go
View file

@ -79,7 +79,7 @@ func (a *AI) GetMsg() string {
return "请求失败" return "请求失败"
} }
// println(resp.Choices[0].Message.Content) // println(resp.Choices[0].Message.Content)
return msg + resp.Choices[0].Message.Content return msg + stripMarkdown(resp.Choices[0].Message.Content)
} else { } else {
// 匹配回复消息 // 匹配回复消息
pattern := `^\[CQ:reply,id=(-?\d+)\]` pattern := `^\[CQ:reply,id=(-?\d+)\]`
@ -159,7 +159,7 @@ func (a *AI) GetMsg() string {
msg += resp.Choices[0].Message.Content msg += resp.Choices[0].Message.Content
} }
return msg return stripMarkdown(msg)
} }
models, err := client.ListModels(context.Background()) models, err := client.ListModels(context.Background())
if err != nil { if err != nil {
@ -178,6 +178,22 @@ func (a *AI) GetMsg() string {
// return handleModelRequest(OPENAI_API_KEY, OPENAI_BaseURL) // return handleModelRequest(OPENAI_API_KEY, OPENAI_BaseURL)
} }
func stripMarkdown(text string) string {
// 移除代码块，但保留代码内容
re := regexp.MustCompile("```(.*?)```")
text = re.ReplaceAllString(text, "$1")
// 移除粗体
re = regexp.MustCompile(`\*\*(.*?)\*\*`)
text = re.ReplaceAllString(text, "$1")
// 移除下划线
re = regexp.MustCompile("__(.*?)__")
text = re.ReplaceAllString(text, "$1")
return text
}
func getConfig() (string, string, string) { func getConfig() (string, string, string) {
var OPENAI_API_KEY, OPENAI_BaseURL, MODEL string var OPENAI_API_KEY, OPENAI_BaseURL, MODEL string