fix: fix quota not consumed

2023-05-14 20:36:28 +08:00 · 2023-05-14 20:36:28 +08:00 · 4a0e81fe83
commit 4a0e81fe83
parent 976c29ea9f
1 changed files with 3 additions and 8 deletions
--- a/middleware/auth.go
+++ b/middleware/auth.go
@ -111,14 +111,9 @@ func TokenAuth() func(c *gin.Context) {
 		c.Set("id", token.UserId)
 		c.Set("token_id", token.Id)
 		requestURL := c.Request.URL.String()
-		consumeQuota := false
-		switch requestURL {
-		case "/v1/chat/completions":
-			consumeQuota = !token.UnlimitedQuota
-		case "/v1/completions":
-			consumeQuota = !token.UnlimitedQuota
-		case "/v1/edits":
-			consumeQuota = !token.UnlimitedQuota
+		consumeQuota := !token.UnlimitedQuota
+		if strings.HasPrefix(requestURL, "/models") {
+			consumeQuota = false
 		}
 		c.Set("consume_quota", consumeQuota)
 		if len(parts) > 1 {