41 lines
994 B
Go
41 lines
994 B
Go
package setting
|
|
|
|
import "strings"
|
|
|
|
var CheckSensitiveEnabled = false
|
|
var CheckSensitiveOnPromptEnabled = false
|
|
|
|
//var CheckSensitiveOnCompletionEnabled = true
|
|
|
|
// StopOnSensitiveEnabled 如果检测到敏感词，是否立刻停止生成，否则替换敏感词
|
|
var StopOnSensitiveEnabled = true
|
|
|
|
// StreamCacheQueueLength 流模式缓存队列长度，0表示无缓存
|
|
var StreamCacheQueueLength = 0
|
|
|
|
// SensitiveWords 敏感词
|
|
var SensitiveWords []string
|
|
|
|
func SensitiveWordsToString() string {
|
|
return strings.Join(SensitiveWords, "\n")
|
|
}
|
|
|
|
func SensitiveWordsFromString(s string) {
|
|
SensitiveWords = []string{}
|
|
sw := strings.SplitSeq(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
|
|
//}
|