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"
|
"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 {
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user