增加提问上下问，接近官网效果。

README.md

@@ -2,9 +2,10 @@
 最近chatGPT异常火爆，想到将其接入到个人微信是件比较有趣的事，所以有了这个项目。项目基于[openwechat](https://github.com/eatmoreapple/openwechat)
 开发
 ### 目前实现了以下功能
- + 机器人群聊@回复
- + 机器人私聊回复
- + 好友添加自动通过
+ * 提问增加上下文，更接近官网效果 
+ * 机器人群聊@回复
+ * 机器人私聊回复
+ * 好友添加自动通过
 
 # 注册openai
 chatGPT注册可以参考[这里](https://juejin.cn/post/7173447848292253704)

go.mod

@@ -2,4 +2,7 @@ module github.com/869413421/wechatbot
 
 go 1.16
 
-require github.com/eatmoreapple/openwechat v1.2.1
+require (
+	github.com/eatmoreapple/openwechat v1.2.1
+	github.com/patrickmn/go-cache v2.1.0+incompatible // indirect
+)

go.sum

@@ -1,2 +1,4 @@
 github.com/eatmoreapple/openwechat v1.2.1 h1:ez4oqF/Y2NSEX/DbPV8lvj7JlfkYqvieeo4awx5lzfU=
 github.com/eatmoreapple/openwechat v1.2.1/go.mod h1:61HOzTyvLobGdgWhL68jfGNwTJEv0mhQ1miCXQrvWU8=
+github.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc=
+github.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=

gtp/gtp.go

@@ -50,7 +50,7 @@ func Completions(msg string) (string, error) {
 	requestBody := ChatGPTRequestBody{
 		Model:            "text-davinci-003",
 		Prompt:           msg,
-		MaxTokens:        2048,
+		MaxTokens:        1024,
 		Temperature:      0.7,
 		TopP:             1,
 		FrequencyPenalty: 0,

handlers/group_msg_handler.go

@@ -38,12 +38,13 @@ func (g *GroupMessageHandler) ReplyText(msg *openwechat.Message) error {
 		return nil
 	}
 
-	// 替换掉@文本，然后向GPT发起请求
+	// 替换掉@文本，设置会话上下文，然后向GPT发起请求。
 	replaceText := "@" + sender.Self.NickName
 	requestText := strings.TrimSpace(strings.ReplaceAll(msg.Content, replaceText, ""))
 	if requestText == "" {
 		return nil
 	}
+	requestText = UserService.GetUserSessionContext(sender.ID()) + requestText
 	reply, err := gtp.Completions(requestText)
 	if err != nil {
 		log.Printf("gtp request error: %v \n", err)
@@ -67,6 +68,8 @@ func (g *GroupMessageHandler) ReplyText(msg *openwechat.Message) error {
 	// 回复@我的用户
 	reply = strings.TrimSpace(reply)
 	reply = strings.Trim(reply, "\n")
+	// 设置上下文
+	UserService.SetUserSessionContext(sender.ID(), requestText, reply)
 	atText := "@" + groupSender.NickName
 	replyText := atText + reply
 	_, err = msg.ReplyText(replyText)

handlers/handler.go

@@ -2,6 +2,7 @@ package handlers
 
 import (
 	"github.com/869413421/wechatbot/config"
+	"github.com/869413421/wechatbot/service"
 	"github.com/eatmoreapple/openwechat"
 	"log"
 )
@@ -21,11 +22,14 @@ const (
 
 // handlers 所有消息类型类型的处理器
 var handlers map[HandlerType]MessageHandlerInterface
+var UserService service.UserServiceInterface
 
 func init() {
 	handlers = make(map[HandlerType]MessageHandlerInterface)
 	handlers[GroupHandler] = NewGroupMessageHandler()
 	handlers[UserHandler] = NewUserMessageHandler()
+
+	UserService = service.NewUserService()
 }
 
 // Handler 全局处理入口

handlers/user_msg_handler.go

@@ -32,9 +32,11 @@ func (g *UserMessageHandler) ReplyText(msg *openwechat.Message) error {
 	sender, err := msg.Sender()
 	log.Printf("Received User %v Text Msg : %v", sender.NickName, msg.Content)
 
-	// 向GPT发起请求
+	// 获取上下文，向GPT发起请求
 	requestText := strings.TrimSpace(msg.Content)
 	requestText = strings.Trim(msg.Content, "\n")
+
+	requestText = UserService.GetUserSessionContext(sender.ID()) + requestText
 	reply, err := gtp.Completions(requestText)
 	if err != nil {
 		log.Printf("gtp request error: %v \n", err)
@@ -45,9 +47,10 @@ func (g *UserMessageHandler) ReplyText(msg *openwechat.Message) error {
 		return nil
 	}
 
-	// 回复用户
+	// 设置上下文，回复用户
 	reply = strings.TrimSpace(reply)
 	reply = strings.Trim(reply, "\n")
+	UserService.SetUserSessionContext(sender.ID(), requestText, reply)
 	reply = "本消息由 chatGPT Bot回复：\n" + reply
 	_, err = msg.ReplyText(reply)
 	if err != nil {

service/user.go

@@ -0,0 +1,40 @@
+package service
+
+import (
+	"github.com/patrickmn/go-cache"
+	"time"
+)
+
+// UserServiceInterface 用户业务接口
+type UserServiceInterface interface {
+	GetUserSessionContext(userId string) string
+	SetUserSessionContext(userId string, question, reply string)
+}
+
+var _ UserServiceInterface = (*UserService)(nil)
+
+// UserService 用戶业务
+type UserService struct {
+	// 缓存
+	cache *cache.Cache
+}
+
+// NewUserService 创建新的业务层
+func NewUserService() UserServiceInterface {
+	return &UserService{cache: cache.New(time.Minute*5, time.Minute*10)}
+}
+
+// GetUserSessionContext 获取用户会话上下文文本
+func (s *UserService) GetUserSessionContext(userId string) string {
+	sessionContext, ok := s.cache.Get(userId)
+	if !ok {
+		return ""
+	}
+	return sessionContext.(string)
+}
+
+// SetUserSessionContext 设置用户会话上下文文本，question用户提问内容，GTP回复内容
+func (s *UserService) SetUserSessionContext(userId string, question, reply string) {
+	value := question + "\n" + reply
+	s.cache.Set(userId, value, time.Minute*5)
+}