diff --git a/scan.go b/scan.go index 676d5cc..1be881f 100644 --- a/scan.go +++ b/scan.go @@ -68,11 +68,29 @@ func sendHTTPAndKimi(r *RequestResponseLog) { // 输出响应体字符串 fmt.Println("Response1 Body:", resp1) fmt.Println("Response2 Body:", resp2) - result, err := kimi(resp1, resp2) //调用kimi检测是否越权 - if err != nil { - fmt.Println(err) + switch AI { + case "kimi": + result, err := kimi(resp1, resp2) //调用kimi检测是否越权 + if err != nil { + fmt.Println(err) + } + log.Println("kimi:") + log.Println(result) + case "deepseek": + result, err := deepSeek(resp1, resp2) //调用kimi检测是否越权 + if err != nil { + fmt.Println(err) + } + log.Println("deepseek:") + log.Println(result) + default: + result, err := kimi(resp1, resp2) //调用kimi检测是否越权 + if err != nil { + fmt.Println(err) + } + log.Println("kimi:") + log.Println(result) } - log.Println(result) } }