fix: fix quota not consumed

songquanpeng · May 14, 2023 · 4a0e81f · 4a0e81f
1 parent 976c29e
commit 4a0e81f
Showing 1 changed file with 3 additions and 8 deletions.
diff --git 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 {