xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-11 03:33:48 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

定期清理用户聊天会话上下文 ChatContext

Browse Source
This commit is contained in:
RockYang
2023-03-28 18:13:17 +08:00
parent 4e575d01db
commit cb2b01127b
6 changed files with 60 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
server/config_handler.go
View File

@@ -63,6 +63,18 @@ func (s *Server) ConfigSetHandle(c *gin.Context) {
s.Config.Chat.EnableContext = v
}
if expireTime, ok := data["chat_context_expire_time"]; ok {
v, err := strconv.Atoi(expireTime)
if err != nil {
c.JSON(http.StatusOK, types.BizVo{
Code: types.InvalidParams,
Message: "chat_context_expire_time must be a integer parameter",
})
return
}
s.Config.Chat.ChatContextExpireTime = v
}
// enable auth
if enableAuth, ok := data["enable_auth"]; ok {
v, err := strconv.ParseBool(enableAuth)
@@ -123,7 +135,7 @@ func (s *Server) AddUserHandle(c *gin.Context) {
return
}
user := types.User{Name: data.Name, MaxCalls: data.MaxCalls, RemainingCalls: data.MaxCalls}
user := types.User{Name: data.Name, MaxCalls: data.MaxCalls, RemainingCalls: data.MaxCalls, EnableHistory: data.EnableHistory}
err = PutUser(user)
if err != nil {
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"})