xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-09-21 18:56:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

auto remove API KEYS that execeed quota

Browse Source
This commit is contained in:
RockYang 2023-04-03 14:19:42 +08:00
parent 01d8a5b77d
commit 361c157e06
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server/chat_handler.go
View File

@ -172,7 +172,9 @@ func (s *Server) sendMessage(session types.ChatSession, role types.ChatRole, pro
} }
if len(line) < 20 { if len(line) < 20 {
continue continue
} else if strings.Contains(line, "This key is associated with a deactivated account") { } else if strings.Contains(line, "This key is associated with a deactivated account") || // 账号被禁用
strings.Contains(line, "You exceeded your current quota") { // 当前 KEY 余额被用尽
logger.Infof("API Key %s is deactivated", apiKey) logger.Infof("API Key %s is deactivated", apiKey)
// 移除当前 API key // 移除当前 API key
for i, v := range s.Config.Chat.ApiKeys { for i, v := range s.Config.Chat.ApiKeys {