add memory time

Author: pwh-pwh

conf/.env.sample

@@ -10,6 +10,7 @@ WX_HELP_REPLY=输入以下命令进行对话\n/help：查看帮助\n/gpt：与GP
 
 # redis config
 KV_URL=redis://localhost:6479/0
+MSG_TIME=30  消息对话列表记忆时间(单位分钟)默认30分钟
 
 # maxOutput config
 # 最大输出tokens, 可选项

db/db.go

@@ -6,6 +6,7 @@ import (
 	"errors"
 	"fmt"
 	"os"
+	"strconv"
 	"sync"
 	"time"
 
@@ -81,7 +82,13 @@ func (r *RedisChatDb) SetMsgList(botType string, userId string, msgList []Msg) {
 		fmt.Println(err)
 		return
 	}
-	r.client.Set(context.Background(), fmt.Sprintf("%v:%v:%v", MSG_KEY, botType, userId), res, time.Minute*30)
+	msgTime := os.Getenv("MSG_TIME")
+	//转换为数字
+	msgT, err := strconv.Atoi(msgTime)
+	if err != nil || msgT <= 0 {
+		msgT = 30
+	}
+	r.client.Set(context.Background(), fmt.Sprintf("%v:%v:%v", MSG_KEY, botType, userId), res, time.Minute*time.Duration(msgT))
 }
 
 func GetChatDb() (ChatDb, error) {