perf: only return quota when it's not zero

fix: fix quota not return when error occurred (close #518 )
2025-11-06 16:53:42 +08:00 · 2023-09-13 22:05:10 +08:00 · 2023-09-13 21:50:45 +08:00
1 changed files with 9 additions and 0 deletions
--- a/controller/relay-text.go
+++ b/controller/relay-text.go
@@ -347,6 +347,15 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
 		isStream = isStream || strings.HasPrefix(resp.Header.Get("Content-Type"), "text/event-stream")

 		if resp.StatusCode != http.StatusOK {
+			if preConsumedQuota != 0 {
+				go func() {
+					// return pre-consumed quota
+					err := model.PostConsumeTokenQuota(tokenId, -preConsumedQuota)
+					if err != nil {
+						common.SysError("error return pre-consumed quota: " + err.Error())
+					}
+				}()
+			}
 			return relayErrorHandler(resp)
 		}
 	}
Author	SHA1	Message	Date
JustSong	420c375140	perf: only return quota when it's not zero	2023-09-13 22:05:10 +08:00
JustSong	01863d3e44	fix: fix quota not return when error occurred (close #518 )	2023-09-13 21:50:45 +08:00