fix: 完善令牌预扣费逻辑

This commit is contained in:
CaIon
2024-01-11 14:12:38 +08:00
parent 701a28d0da
commit 64e9e9cc20
2 changed files with 4 additions and 4 deletions

View File

@@ -115,10 +115,9 @@ func TokenAuth() func(c *gin.Context) {
c.Set("id", token.UserId)
c.Set("token_id", token.Id)
c.Set("token_name", token.Name)
c.Set("token_unlimited_quota", token.UnlimitedQuota)
if !token.UnlimitedQuota {
c.Set("token_quota", token.RemainQuota)
} else {
c.Set("token_quota", -1)
}
if token.ModelLimitsEnabled {
c.Set("token_model_limit_enabled", true)