Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: fix quota not consumed

Browse Source
This commit is contained in:
JustSong 2023-05-14 20:36:28 +08:00
parent 976c29ea9f
commit 4a0e81fe83
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
middleware/auth.go
View File

@ -111,14 +111,9 @@ func TokenAuth() func(c *gin.Context) {
c.Set("id", token.UserId) c.Set("id", token.UserId)
c.Set("token_id", token.Id) c.Set("token_id", token.Id)
requestURL := c.Request.URL.String() requestURL := c.Request.URL.String()
consumeQuota := false consumeQuota := !token.UnlimitedQuota
switch requestURL { if strings.HasPrefix(requestURL, "/models") {
case "/v1/chat/completions": consumeQuota = false
consumeQuota = !token.UnlimitedQuota
case "/v1/completions":
consumeQuota = !token.UnlimitedQuota
case "/v1/edits":
consumeQuota = !token.UnlimitedQuota
} }
c.Set("consume_quota", consumeQuota) c.Set("consume_quota", consumeQuota)
if len(parts) > 1 { if len(parts) > 1 {