Function init

plugin/chatgpt/main.go:42–281

()

)

func init() {
	engine.OnRegex(`^(?:chatgpt|//)([\s\S]*)$`, zero.OnlyToMe, getdb).SetBlock(false).
		Handle(func(ctx *zero.Ctx) {
			var messages []chatMessage
			args := ctx.State["regex_matched"].([]string)[1]
			key := sessionKey{
				group: ctx.Event.GroupID,
				user:  ctx.Event.UserID,
			}
			if args == "reset" || args == "重置记忆" {
				cache.Delete(key)
				ctx.SendChain(message.Text("已清除上下文！"))
				return
			}
			// 添加预设
			apiKey, err := getkey(ctx)
			if err != nil {
				ctx.SendChain(message.Text("ERROR：", err))
				return
			}
			gid := ctx.Event.GroupID
			if gid == 0 {
				gid = -ctx.Event.UserID
			}
			content, err := db.findgroupmode(gid)
			if err == nil {
				messages = append(messages, chatMessage{
					Role:    "system",
					Content: content,
				})
				if len(cache.Get(key)) > 1 {
					messages = append(messages, cache.Get(key)[1:]...)
				}
			} else {
				c, err := db.findgroupmode(-1)
				if err != nil {
					messages = append(messages, cache.Get(key)...)
				} else {
					messages = append(messages, chatMessage{
						Role:    "system",
						Content: c,
					})
					if len(cache.Get(key)) > 1 {
						messages = append(messages, cache.Get(key)[1:]...)
					}
				}
			}
			messages = append(messages, chatMessage{
				Role:    "user",
				Content: args,
			})
			resp, err := completions(messages, apiKey)
			if err != nil {
				ctx.SendChain(message.Text("请求ChatGPT失败: ", err))
				return
			}
			reply := resp.Choices[0].Message
			reply.Content = strings.TrimSpace(reply.Content)