李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
81 lines
2.1 KiB
Go
81 lines
2.1 KiB
Go
package service
|
||
|
||
import (
|
||
"fmt"
|
||
"math/rand/v2"
|
||
"regexp"
|
||
"strconv"
|
||
)
|
||
|
||
// monitorChallengePromptTemplate 1:1 复刻 BingZi-233/check-cx 的 few-shot 模板。
|
||
const monitorChallengePromptTemplate = `Calculate and respond with ONLY the number, nothing else.
|
||
|
||
Q: 3 + 5 = ?
|
||
A: 8
|
||
|
||
Q: 12 - 7 = ?
|
||
A: 5
|
||
|
||
Q: %d %s %d = ?
|
||
A:`
|
||
|
||
// monitorChallengeNumberRegex 提取响应中的所有整数（含负号）。
|
||
var monitorChallengeNumberRegex = regexp.MustCompile(`-?\d+`)
|
||
|
||
// monitorChallenge 一次 challenge 的 prompt + 期望答案。
|
||
type monitorChallenge struct {
|
||
Prompt string
|
||
Expected string
|
||
}
|
||
|
||
// generateChallenge 生成一次随机算术 challenge：
|
||
// - 随机两个 [monitorChallengeMin, monitorChallengeMax] 整数
|
||
// - 50% 加 / 50% 减；减法用 max - min 保证非负
|
||
// - 渲染 few-shot 模板
|
||
//
|
||
// 不强求加密随机：math/rand/v2 足够分散，避免 crypto/rand 的开销。
|
||
func generateChallenge() monitorChallenge {
|
||
a := randIntInRange(monitorChallengeMin, monitorChallengeMax)
|
||
b := randIntInRange(monitorChallengeMin, monitorChallengeMax)
|
||
|
||
if rand.IntN(2) == 0 { //nolint:gosec // 仅用于生成测试问题，无安全影响
|
||
// 加法
|
||
return monitorChallenge{
|
||
Prompt: fmt.Sprintf(monitorChallengePromptTemplate, a, "+", b),
|
||
Expected: strconv.Itoa(a + b),
|
||
}
|
||
}
|
||
|
||
// 减法，保证非负
|
||
hi, lo := a, b
|
||
if lo > hi {
|
||
hi, lo = lo, hi
|
||
}
|
||
return monitorChallenge{
|
||
Prompt: fmt.Sprintf(monitorChallengePromptTemplate, hi, "-", lo),
|
||
Expected: strconv.Itoa(hi - lo),
|
||
}
|
||
}
|
||
|
||
// randIntInRange 返回 [min, max] 闭区间的随机整数。
|
||
func randIntInRange(minVal, maxVal int) int {
|
||
if maxVal <= minVal {
|
||
return minVal
|
||
}
|
||
return minVal + rand.IntN(maxVal-minVal+1) //nolint:gosec
|
||
}
|
||
|
||
// validateChallenge 在响应文本中查找 expected 整数答案，返回是否通过校验。
|
||
func validateChallenge(responseText, expected string) bool {
|
||
if responseText == "" || expected == "" {
|
||
return false
|
||
}
|
||
matches := monitorChallengeNumberRegex.FindAllString(responseText, -1)
|
||
for _, m := range matches {
|
||
if m == expected {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|