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