From ab5a8092f7513c353a882980af979d0a618818c3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= Date: Fri, 19 Apr 2024 09:54:50 +0800 Subject: [PATCH] =?UTF-8?q?:art:=20=E4=BC=98=E5=8C=96AI=E5=AF=B9=E8=AF=9D?= =?UTF-8?q?=E8=A7=92=E8=89=B2=E6=8F=90=E7=A4=BA=E8=AF=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tasks/summary/summary.go | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/tasks/summary/summary.go b/tasks/summary/summary.go index 432a0f92..0ec5052b 100644 --- a/tasks/summary/summary.go +++ b/tasks/summary/summary.go @@ -35,13 +35,18 @@ func AiSummary() { // 组装对话记录为字符串 var content []string for _, record := range records { - content = append(content, fmt.Sprintf("%s: %s\n-----end-----", record.Nickname, record.Message)) + content = append(content, fmt.Sprintf(`{"%s": "%s"}--end--`, record.Nickname, strings.ReplaceAll(record.Message, "\n", "。。"))) } - msg := fmt.Sprintf("请帮我总结一下一下的群聊内容的梗概,生成的梗概尽可能详细,并且最好带上名字。\n"+ - "注意,他们可能是多个话题,请仔细甄别。\n"+ - "每一行代表一个人的发言,每一行的的格式为: \n{nickname}: {content}\n-----end-----"+ - "\n\n聊天记录如下: \n%s", strings.Join(content, "\n")) + msgTmp := `请帮我总结一下一下的群聊内容的梗概,生成的梗概需要尽可能详细,需要带上一些聊天关键信息,并且带上群友名字。 +注意,他们可能是多个话题,请仔细甄别。 +每一行代表一个人的发言,每一行的的格式为: {"{nickname}": "{content}"}--end-- + +聊天记录如下: +%s +` + + msg := fmt.Sprintf(msgTmp, strings.Join(content, "\n")) // AI总结 messages := []openai.ChatCompletionMessage{