From 81137e0533b5b7023a21edf07103ed3998bd509d Mon Sep 17 00:00:00 2001 From: "1808837298@qq.com" <1808837298@qq.com> Date: Fri, 7 Mar 2025 19:59:00 +0800 Subject: [PATCH] refactor: Remove redundant user quota retrieval in audio relay --- relay/relay-audio.go | 5 ----- relay/relay-text.go | 1 + 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/relay/relay-audio.go b/relay/relay-audio.go index 6263dcb9..b77ee80e 100644 --- a/relay/relay-audio.go +++ b/relay/relay-audio.go @@ -7,7 +7,6 @@ import ( "net/http" "one-api/common" "one-api/dto" - "one-api/model" relaycommon "one-api/relay/common" relayconstant "one-api/relay/constant" "one-api/relay/helper" @@ -80,10 +79,6 @@ func AudioHelper(c *gin.Context) (openaiErr *dto.OpenAIErrorWithStatusCode) { return service.OpenAIErrorWrapperLocal(err, "model_price_error", http.StatusInternalServerError) } - userQuota, err := model.GetUserQuota(relayInfo.UserId, false) - if err != nil { - return service.OpenAIErrorWrapperLocal(err, "get_user_quota_failed", http.StatusInternalServerError) - } preConsumedQuota, userQuota, openaiErr := preConsumeQuota(c, priceData.ShouldPreConsumedQuota, relayInfo) if openaiErr != nil { return openaiErr diff --git a/relay/relay-text.go b/relay/relay-text.go index bf6c5fd3..57b13ca7 100644 --- a/relay/relay-text.go +++ b/relay/relay-text.go @@ -110,6 +110,7 @@ func TextHelper(c *gin.Context) (openaiErr *dto.OpenAIErrorWithStatusCode) { if err != nil { return service.OpenAIErrorWrapperLocal(err, "model_price_error", http.StatusInternalServerError) } + // pre-consume quota 预消耗配额 preConsumedQuota, userQuota, openaiErr := preConsumeQuota(c, priceData.ShouldPreConsumedQuota, relayInfo) if openaiErr != nil {