fixed bug for interface convert to int

This commit is contained in:
RockYang 2023-04-04 11:06:04 +08:00
parent a78fb83f30
commit d81d9af986
2 changed files with 5 additions and 5 deletions

View File

@ -32,7 +32,7 @@ func (s *Server) ChatHandle(c *gin.Context) {
c.Abort() c.Abort()
return return
} }
logger.Infof("New websocket connected, IP: %s, Username: %s", c.Request.RemoteAddr, session.Username) logger.Infof("New websocket connected, IP: %s, Username: %s", c.Request.RemoteAddr, session.Username)
client := NewWsClient(ws) client := NewWsClient(ws)
var roles = GetChatRoles() var roles = GetChatRoles()

View File

@ -50,14 +50,14 @@ func (s *Server) ConfigSetHandle(c *gin.Context) {
} }
// max_users // max_users
if maxTokens, ok := data["max_tokens"]; ok { if maxTokens, ok := data["max_tokens"]; ok {
s.Config.Chat.MaxTokens = maxTokens.(int) s.Config.Chat.MaxTokens = int(maxTokens.(float64))
} }
// enable Context // enable Context
if enableContext, ok := data["enable_context"]; ok { if enableContext, ok := data["enable_context"]; ok {
s.Config.Chat.EnableContext = enableContext.(bool) s.Config.Chat.EnableContext = enableContext.(bool)
} }
if expireTime, ok := data["chat_context_expire_time"]; ok { if expireTime, ok := data["chat_context_expire_time"]; ok {
s.Config.Chat.ChatContextExpireTime = expireTime.(int) s.Config.Chat.ChatContextExpireTime = int(expireTime.(float64))
} }
// enable auth // enable auth
if enableAuth, ok := data["enable_auth"]; ok { if enableAuth, ok := data["enable_auth"]; ok {
@ -168,7 +168,7 @@ func (s *Server) SetUserHandle(c *gin.Context) {
} }
var maxCalls int var maxCalls int
if v, ok := data["max_calls"]; ok { if v, ok := data["max_calls"]; ok {
maxCalls = v.(int) maxCalls = int(v.(float64))
} }
if maxCalls < 0 { if maxCalls < 0 {
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"}) c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"})
@ -187,7 +187,7 @@ func (s *Server) SetUserHandle(c *gin.Context) {
user.EnableHistory = v.(bool) user.EnableHistory = v.(bool)
} }
if v, ok := data["remaining_calls"]; ok { if v, ok := data["remaining_calls"]; ok {
user.RemainingCalls = v.(int) user.RemainingCalls = int(v.(float64))
} }
if v, ok := data["api_key"]; ok { if v, ok := data["api_key"]; ok {
user.ApiKey = v.(string) user.ApiKey = v.(string)