feat: publish sanitized custom.97 source

setting/operation_setting/quota_setting.go

@@ -1,14 +1,24 @@
 package operation_setting
 
-import "github.com/QuantumNous/new-api/setting/config"
+import (
+	"fmt"
+	"math"
+	"strconv"
+
+	"github.com/QuantumNous/new-api/setting/config"
+)
 
 type QuotaSetting struct {
-	EnableFreeModelPreConsume bool `json:"enable_free_model_pre_consume"` // 是否对免费模型启用预消耗
+	EnableFreeModelPreConsume bool    `json:"enable_free_model_pre_consume"` // 是否对免费模型启用预消耗
+	TrustQuotaUSD             float64 `json:"trust_quota_usd"`               // 钱包免预扣门槛，0 表示禁用
+	PreConsumeMultiplier      float64 `json:"pre_consume_multiplier"`        // 预计输入费用的预扣倍率，仅影响预留
 }
 
 // 默认配置
 var quotaSetting = QuotaSetting{
 	EnableFreeModelPreConsume: true,
+	TrustQuotaUSD:             10,
+	PreConsumeMultiplier:      1,
 }
 
 func init() {
@@ -19,3 +29,27 @@ func init() {
 func GetQuotaSetting() *QuotaSetting {
 	return &quotaSetting
 }
+
+// ValidateQuotaOption validates reservation settings before they are persisted.
+func ValidateQuotaOption(key, value string) error {
+	if key != "quota_setting.trust_quota_usd" && key != "quota_setting.pre_consume_multiplier" {
+		return nil
+	}
+	number, err := strconv.ParseFloat(value, 64)
+	if err != nil || math.IsNaN(number) || math.IsInf(number, 0) || number < 0 {
+		return fmt.Errorf("%s must be a finite non-negative number", key)
+	}
+	if key == "quota_setting.pre_consume_multiplier" && number == 0 {
+		return fmt.Errorf("%s must be greater than zero", key)
+	}
+	return nil
+}
+
+// InputPreConsumeMultiplier rejects invalid runtime settings as well as invalid saves.
+func InputPreConsumeMultiplier() (float64, error) {
+	multiplier := quotaSetting.PreConsumeMultiplier
+	if multiplier <= 0 || math.IsNaN(multiplier) || math.IsInf(multiplier, 0) {
+		return 0, fmt.Errorf("pre-consume multiplier must be a finite number greater than zero")
+	}
+	return multiplier, nil
+}