billing

type BillingBudgetChecker struct {
	// contains filtered or unexported fields
}

type BillingCheckRequest struct {
	VirtualKey string                `json:"virtual_key"`
	Provider   schemas.ModelProvider `json:"provider"`
	Model      string                `json:"model"`
	UserID     string                `json:"user_id,omitempty"`
}

type BillingCheckResult struct {
	Decision     Decision            `json:"decision"`
	Reason       string              `json:"reason,omitempty"`
	BudgetInfos  []*store.BudgetData `json:"budget_infos,omitempty"`
	TotalBalance float64             `json:"total_balance,omitempty"`
}

type BillingPlugin struct {
	// contains filtered or unexported fields
}

type BudgetChecker interface {
	// Check decides whether the request is allowed
	Check(ctx context.Context, entityWiseBudgets store.EntityWiseBudgets, baselines map[string]float64) (Decision, error)

	// Deduct updates usage after a successful request
	// Returns which budget IDs were actually charged and by how much
	Deduct(ctx context.Context, entityWiseBudgets store.EntityWiseBudgets, cost float64) (map[string]float64, error)
}

type BudgetWithInfo struct {
	Budget       *store.BudgetData
	IsPackage    bool    // 是否为套餐类型（有 ExpiresAt）
	DurationSecs float64 // 套餐时长（秒），用于排序
	Remaining    float64 // 剩余额度
}

type Decision string

type OffPeakDiscount struct {
	Percentage float64 `json:"percentage"`
	StartHour  int     `json:"start_hour,omitempty"`
	EndHour    int     `json:"end_hour,omitempty"`
	Days       []int   `json:"days,omitempty"`
	Timezone   string  `json:"timezone,omitempty"`
}

type OffPeakDiscountRule struct {
	StartHour int     `json:"start_hour"` // 0-23
	EndHour   int     `json:"end_hour"`   // 0-23
	Discount  float64 `json:"discount"`   // 0.0-1.0 (e.g., 0.5 means 50% off)
}

type UsageTracker struct {
	// contains filtered or unexported fields
}