xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-10 02:23:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: atomatically migrate old ratio to new

Browse Source
This commit is contained in:
WqyJh
2025-01-20 17:51:26 +08:00
parent 215e59b76d
commit 56df7c097a
2 changed files with 85 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
model/option.go
View File

@@ -83,15 +83,35 @@ func InitOptionMap() {
func loadOptionsFromDatabase() {
options, _ := AllOption()
var oldModelRatio string
var oldCompletionRatio string
for _, option := range options {
if option.Key == "ModelRatio" {
oldModelRatio = option.Value
option.Value = billingratio.AddNewMissingRatio(option.Value)
}
if option.Key == "CompletionRatio" {
oldCompletionRatio = option.Value
}
err := updateOptionMap(option.Key, option.Value)
if err != nil {
logger.SysError("failed to update option map: " + err.Error())
}
}
for _, option := range options {
if option.Key == "Ratio" {
option.Value = billingratio.AddOldRatio(oldModelRatio, oldCompletionRatio)
err := updateOptionMap(option.Key, option.Value)
if err != nil {
logger.SysError("failed to update option map: " + err.Error())
}
err = UpdateOption(option.Key, option.Value)
if err != nil {
logger.SysError("failed to update option map: " + err.Error())
}
logger.SysLog("ratio merged")
}
}
}
func SyncOptions(frequency int) {