♻️ refactor: change test logic

middleware/auth.go

@@ -1,12 +1,13 @@
 package middleware
 
 import (
-	"github.com/gin-contrib/sessions"
-	"github.com/gin-gonic/gin"
 	"net/http"
 	"one-api/common"
 	"one-api/model"
 	"strings"
+
+	"github.com/gin-contrib/sessions"
+	"github.com/gin-gonic/gin"
 )
 
 func authHelper(c *gin.Context, minRole int) {
@@ -114,7 +115,7 @@ func TokenAuth() func(c *gin.Context) {
 		c.Set("token_id", token.Id)
 		requestURL := c.Request.URL.String()
 		consumeQuota := true
-		if strings.HasPrefix(requestURL, "/v1/models") {
+		if strings.HasPrefix(requestURL, "/v1/models") || token.Id == 0 {
 			consumeQuota = false
 		}
 		c.Set("consume_quota", consumeQuota)

middleware/distributor.go

@@ -2,11 +2,12 @@ package middleware
 
 import (
 	"fmt"
-	"github.com/gin-gonic/gin"
 	"net/http"
 	"one-api/common"
 	"one-api/model"
 	"strconv"
+
+	"github.com/gin-gonic/gin"
 )
 
 func Distribute() func(c *gin.Context) {
@@ -36,7 +37,8 @@ func Distribute() func(c *gin.Context) {
 				c.Abort()
 				return
 			}
-			if channel.Status != common.ChannelStatusEnabled {
+			tokenId := c.GetInt("token_id") // If use ServerToken, don't check disabled
+			if channel.Status != common.ChannelStatusEnabled && tokenId != 0 {
 				c.JSON(200, gin.H{
 					"error": gin.H{
 						"message": "该渠道已被禁用",