Update scan.go

This commit is contained in:
公明 2025-02-25 14:20:40 +08:00 committed by GitHub
parent 2388972a43
commit db81f77b09
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -9,6 +9,7 @@ import (
"net/url"
"strings"
"time"
aiapis "yuequanScan/AIAPIS"
)
type Result struct {
@ -120,11 +121,11 @@ func detectPrivilegeEscalation(AI string, resp1, resp2 string) (string, error) {
switch AI {
case "kimi":
result, err = kimi(resp1, resp2) // 调用 kimi 检测是否越权
result, err = aiapis.Kimi(resp1, resp2) // 调用 kimi 检测是否越权
case "deepseek":
result, err = deepSeek(resp1, resp2) // 调用 deepSeek 检测是否越权
result, err = aiapis.DeepSeek(resp1, resp2) // 调用 deepSeek 检测是否越权
default:
result, err = kimi(resp1, resp2) // 默认调用 kimi 检测是否越权
result, err = aiapis.Kimi(resp1, resp2) // 默认调用 kimi 检测是否越权
}
if err != nil {