Merge pull request #91 from songquanpeng/main

Fork Sync: Update from parent repository
This commit is contained in:
quzard 2023-06-29 11:28:30 +08:00 committed by GitHub
commit 252c8e366e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 2 deletions

View File

@ -457,5 +457,7 @@
"提示": "Prompt", "提示": "Prompt",
"补全": "Completion", "补全": "Completion",
"消耗额度": "Used Quota", "消耗额度": "Used Quota",
"可选值": "Optional Values" "可选值": "Optional Values",
"渠道不存在:%d": "Channel does not exist: %d",
"数据库一致性已被破坏,请联系管理员": "Database consistency has been broken, please contact the administrator"
} }

View File

@ -78,9 +78,14 @@ func Distribute() func(c *gin.Context) {
} }
channel, err = model.CacheGetRandomSatisfiedChannel(userGroup, modelRequest.Model) channel, err = model.CacheGetRandomSatisfiedChannel(userGroup, modelRequest.Model)
if err != nil { if err != nil {
message := "无可用渠道"
if channel != nil {
common.SysError(fmt.Sprintf("渠道不存在:%d", channel.Id))
message = "数据库一致性已被破坏,请联系管理员"
}
c.JSON(http.StatusServiceUnavailable, gin.H{ c.JSON(http.StatusServiceUnavailable, gin.H{
"error": gin.H{ "error": gin.H{
"message": "无可用渠道", "message": message,
"type": "one_api_error", "type": "one_api_error",
}, },
}) })

View File

@ -24,6 +24,7 @@ func GetRandomSatisfiedChannel(group string, model string) (*Channel, error) {
return nil, err return nil, err
} }
channel := Channel{} channel := Channel{}
channel.Id = ability.ChannelId
err = DB.First(&channel, "id = ?", ability.ChannelId).Error err = DB.First(&channel, "id = ?", ability.ChannelId).Error
return &channel, err return &channel, err
} }