mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-03-30 04:03:18 +00:00
d7d3a2f763
* feat: channel affinity * feat: channel affinity -> model setting * fix: channel affinity * feat: channel affinity op * feat: channel_type setting * feat: clean * feat: cache supports both memory and Redis. * feat: Optimise ui/ux * feat: Optimise ui/ux * feat: Optimise codex usage ui/ux * feat: Optimise ui/ux * feat: Optimise ui/ux * feat: Optimise ui/ux * feat: If the affinitized channel fails and a retry succeeds on another channel, update the affinity to the successful channel
48 lines
1.5 KiB
Go
48 lines
1.5 KiB
Go
package operation_setting
|
|
|
|
import "github.com/QuantumNous/new-api/setting/config"
|
|
|
|
type ChannelAffinityKeySource struct {
|
|
Type string `json:"type"` // context_int, context_string, gjson
|
|
Key string `json:"key,omitempty"`
|
|
Path string `json:"path,omitempty"`
|
|
}
|
|
|
|
type ChannelAffinityRule struct {
|
|
Name string `json:"name"`
|
|
ModelRegex []string `json:"model_regex"`
|
|
PathRegex []string `json:"path_regex"`
|
|
UserAgentInclude []string `json:"user_agent_include,omitempty"`
|
|
KeySources []ChannelAffinityKeySource `json:"key_sources"`
|
|
|
|
ValueRegex string `json:"value_regex"`
|
|
TTLSeconds int `json:"ttl_seconds"`
|
|
|
|
IncludeUsingGroup bool `json:"include_using_group"`
|
|
IncludeRuleName bool `json:"include_rule_name"`
|
|
}
|
|
|
|
type ChannelAffinitySetting struct {
|
|
Enabled bool `json:"enabled"`
|
|
SwitchOnSuccess bool `json:"switch_on_success"`
|
|
MaxEntries int `json:"max_entries"`
|
|
DefaultTTLSeconds int `json:"default_ttl_seconds"`
|
|
Rules []ChannelAffinityRule `json:"rules"`
|
|
}
|
|
|
|
var channelAffinitySetting = ChannelAffinitySetting{
|
|
Enabled: false,
|
|
SwitchOnSuccess: true,
|
|
MaxEntries: 100_000,
|
|
DefaultTTLSeconds: 3600,
|
|
Rules: []ChannelAffinityRule{},
|
|
}
|
|
|
|
func init() {
|
|
config.GlobalConfig.Register("channel_affinity_setting", &channelAffinitySetting)
|
|
}
|
|
|
|
func GetChannelAffinitySetting() *ChannelAffinitySetting {
|
|
return &channelAffinitySetting
|
|
}
|