fix: rpm limit recore ignore empty channel id

labring · Jan 6, 2025 · 2baece4 · 2baece4
1 parent 9a7f5c1
commit 2baece4
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/service/aiproxy/model/utils.go b/service/aiproxy/model/utils.go
@@ -92,9 +92,11 @@ func BatchRecordConsume(
 	if err != nil {
 		errs = append(errs, fmt.Errorf("failed to update token used amount: %w", err))
 	}
-	err = UpdateChannelUsedAmount(channelID, amount, 1)
-	if err != nil {
-		errs = append(errs, fmt.Errorf("failed to update channel used amount: %w", err))
+	if channelID > 0 {
+		err = UpdateChannelUsedAmount(channelID, amount, 1)
+		if err != nil {
+			errs = append(errs, fmt.Errorf("failed to update channel used amount: %w", err))
+		}
 	}
 	if len(errs) == 0 {
 		return nil