mirror of
https://github.com/Ed1s0nZ/PrivHunterAI.git
synced 2025-11-21 18:28:06 +00:00
Update scan.go
This commit is contained in:
parent
2388972a43
commit
db81f77b09
7
scan.go
7
scan.go
@ -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 {
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user