|
|
@@ -4,6 +4,7 @@ import (
|
|
|
"one-api/common"
|
|
|
"one-api/setting"
|
|
|
"one-api/setting/config"
|
|
|
+ "one-api/setting/operation_setting"
|
|
|
"strconv"
|
|
|
"strings"
|
|
|
"time"
|
|
|
@@ -110,14 +111,14 @@ func InitOptionMap() {
|
|
|
common.OptionMap["MjForwardUrlEnabled"] = strconv.FormatBool(setting.MjForwardUrlEnabled)
|
|
|
common.OptionMap["MjActionCheckSuccessEnabled"] = strconv.FormatBool(setting.MjActionCheckSuccessEnabled)
|
|
|
common.OptionMap["CheckSensitiveEnabled"] = strconv.FormatBool(setting.CheckSensitiveEnabled)
|
|
|
- common.OptionMap["DemoSiteEnabled"] = strconv.FormatBool(setting.DemoSiteEnabled)
|
|
|
- common.OptionMap["SelfUseModeEnabled"] = strconv.FormatBool(setting.SelfUseModeEnabled)
|
|
|
+ common.OptionMap["DemoSiteEnabled"] = strconv.FormatBool(operation_setting.DemoSiteEnabled)
|
|
|
+ common.OptionMap["SelfUseModeEnabled"] = strconv.FormatBool(operation_setting.SelfUseModeEnabled)
|
|
|
common.OptionMap["ModelRequestRateLimitEnabled"] = strconv.FormatBool(setting.ModelRequestRateLimitEnabled)
|
|
|
common.OptionMap["CheckSensitiveOnPromptEnabled"] = strconv.FormatBool(setting.CheckSensitiveOnPromptEnabled)
|
|
|
common.OptionMap["StopOnSensitiveEnabled"] = strconv.FormatBool(setting.StopOnSensitiveEnabled)
|
|
|
common.OptionMap["SensitiveWords"] = setting.SensitiveWordsToString()
|
|
|
common.OptionMap["StreamCacheQueueLength"] = strconv.Itoa(setting.StreamCacheQueueLength)
|
|
|
- common.OptionMap["AutomaticDisableKeywords"] = setting.AutomaticDisableKeywordsToString()
|
|
|
+ common.OptionMap["AutomaticDisableKeywords"] = operation_setting.AutomaticDisableKeywordsToString()
|
|
|
|
|
|
// 自动添加所有注册的模型配置
|
|
|
modelConfigs := config.GlobalConfig.ExportAllConfigs()
|
|
|
@@ -243,9 +244,9 @@ func updateOptionMap(key string, value string) (err error) {
|
|
|
case "CheckSensitiveEnabled":
|
|
|
setting.CheckSensitiveEnabled = boolValue
|
|
|
case "DemoSiteEnabled":
|
|
|
- setting.DemoSiteEnabled = boolValue
|
|
|
+ operation_setting.DemoSiteEnabled = boolValue
|
|
|
case "SelfUseModeEnabled":
|
|
|
- setting.SelfUseModeEnabled = boolValue
|
|
|
+ operation_setting.SelfUseModeEnabled = boolValue
|
|
|
case "CheckSensitiveOnPromptEnabled":
|
|
|
setting.CheckSensitiveOnPromptEnabled = boolValue
|
|
|
case "ModelRequestRateLimitEnabled":
|
|
|
@@ -365,7 +366,7 @@ func updateOptionMap(key string, value string) (err error) {
|
|
|
case "SensitiveWords":
|
|
|
setting.SensitiveWordsFromString(value)
|
|
|
case "AutomaticDisableKeywords":
|
|
|
- setting.AutomaticDisableKeywordsFromString(value)
|
|
|
+ operation_setting.AutomaticDisableKeywordsFromString(value)
|
|
|
case "StreamCacheQueueLength":
|
|
|
setting.StreamCacheQueueLength, _ = strconv.Atoi(value)
|
|
|
}
|