xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-09-17 16:06:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: 保留功能

Browse Source
This commit is contained in:
CaIon 2024-03-20 20:15:32 +08:00
parent 64b9d3b58c
commit 2db4282666
3 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
constant/sensitive.go
View File

@ -9,6 +9,9 @@ var CheckSensitiveOnCompletionEnabled = true
// StopOnSensitiveEnabled 如果检测到敏感词，是否立刻停止生成，否则替换敏感词 // StopOnSensitiveEnabled 如果检测到敏感词，是否立刻停止生成，否则替换敏感词
var StopOnSensitiveEnabled = true var StopOnSensitiveEnabled = true
// StreamCacheQueueLength 流模式缓存队列长度，0表示无缓存
var StreamCacheQueueLength = 0
// SensitiveWords 敏感词 // SensitiveWords 敏感词
// var SensitiveWords []string // var SensitiveWords []string
var SensitiveWords = []string{ var SensitiveWords = []string{

3
model/option.go
View File

@ -95,6 +95,7 @@ func InitOptionMap() {
common.OptionMap["CheckSensitiveOnCompletionEnabled"] = strconv.FormatBool(constant.CheckSensitiveOnCompletionEnabled) common.OptionMap["CheckSensitiveOnCompletionEnabled"] = strconv.FormatBool(constant.CheckSensitiveOnCompletionEnabled)
common.OptionMap["StopOnSensitiveEnabled"] = strconv.FormatBool(constant.StopOnSensitiveEnabled) common.OptionMap["StopOnSensitiveEnabled"] = strconv.FormatBool(constant.StopOnSensitiveEnabled)
common.OptionMap["SensitiveWords"] = constant.SensitiveWordsToString() common.OptionMap["SensitiveWords"] = constant.SensitiveWordsToString()
common.OptionMap["StreamCacheQueueLength"] = strconv.Itoa(constant.StreamCacheQueueLength)
common.OptionMapRWMutex.Unlock() common.OptionMapRWMutex.Unlock()
loadOptionsFromDatabase() loadOptionsFromDatabase()
@ -288,6 +289,8 @@ func updateOptionMap(key string, value string) (err error) {
common.QuotaPerUnit, _ = strconv.ParseFloat(value, 64) common.QuotaPerUnit, _ = strconv.ParseFloat(value, 64)
case "SensitiveWords": case "SensitiveWords":
constant.SensitiveWordsFromString(value) constant.SensitiveWordsFromString(value)
case "StreamCacheQueueLength":
constant.StreamCacheQueueLength, _ = strconv.Atoi(value)
} }
return err return err
} }

14
web/src/components/OperationSetting.js
View File

@ -10,6 +10,7 @@ const OperationSetting = () => {
QuotaForInvitee: 0, QuotaForInvitee: 0,
QuotaRemindThreshold: 0, QuotaRemindThreshold: 0,
PreConsumedQuota: 0, PreConsumedQuota: 0,
StreamCacheQueueLength: 0,
ModelRatio: '', ModelRatio: '',
ModelPrice: '', ModelPrice: '',
GroupRatio: '', GroupRatio: '',
@ -307,6 +308,8 @@ const OperationSetting = () => {
name="CheckSensitiveOnCompletionEnabled" name="CheckSensitiveOnCompletionEnabled"
onChange={handleInputChange} onChange={handleInputChange}
/> />
</Form.Group>
<Form.Group inline>
<Form.Checkbox <Form.Checkbox
checked={inputs.StopOnSensitiveEnabled === 'true'} checked={inputs.StopOnSensitiveEnabled === 'true'}
label="在检测到屏蔽词时，立刻停止生成，否则替换屏蔽词" label="在检测到屏蔽词时，立刻停止生成，否则替换屏蔽词"
@ -314,6 +317,17 @@ const OperationSetting = () => {
onChange={handleInputChange} onChange={handleInputChange}
/> />
</Form.Group> </Form.Group>
{/*<Form.Group>*/}
{/* <Form.Input*/}
{/* label="流模式下缓存队列，默认不缓存，设置越大检测越准确，但是回复会有卡顿感"*/}
{/* name="StreamCacheTextLength"*/}
{/* onChange={handleInputChange}*/}
{/* value={inputs.StreamCacheQueueLength}*/}
{/* type="number"*/}
{/* min="0"*/}
{/* placeholder="例如：10"*/}
{/* />*/}
{/*</Form.Group>*/}
<Form.Group widths="equal"> <Form.Group widths="equal">
<Form.TextArea <Form.TextArea
label="屏蔽词列表，一行一个屏蔽词，不需要符号分割" label="屏蔽词列表，一行一个屏蔽词，不需要符号分割"