mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-17 16:06:38 +08:00
42 lines
1000 B
Go
42 lines
1000 B
Go
package constant
|
||
|
||
import "strings"
|
||
|
||
var CheckSensitiveEnabled = true
|
||
var CheckSensitiveOnPromptEnabled = true
|
||
var CheckSensitiveOnCompletionEnabled = true
|
||
|
||
// StopOnSensitiveEnabled 如果检测到敏感词，是否立刻停止生成，否则替换敏感词
|
||
var StopOnSensitiveEnabled = true
|
||
|
||
// StreamCacheQueueLength 流模式缓存队列长度，0表示无缓存
|
||
var StreamCacheQueueLength = 0
|
||
|
||
// SensitiveWords 敏感词
|
||
// var SensitiveWords []string
|
||
var SensitiveWords = []string{
|
||
"test",
|
||
}
|
||
|
||
func SensitiveWordsToString() string {
|
||
return strings.Join(SensitiveWords, "\n")
|
||
}
|
||
|
||
func SensitiveWordsFromString(s string) {
|
||
sw := strings.Split(s, "\n")
|
||
for _, w := range sw {
|
||
w = strings.TrimSpace(w)
|
||
if w != "" {
|
||
SensitiveWords = append(SensitiveWords, w)
|
||
}
|
||
}
|
||
}
|
||
|
||
func ShouldCheckPromptSensitive() bool {
|
||
return CheckSensitiveEnabled && CheckSensitiveOnPromptEnabled
|
||
}
|
||
|
||
func ShouldCheckCompletionSensitive() bool {
|
||
return CheckSensitiveEnabled && CheckSensitiveOnCompletionEnabled
|
||
}
|