From db81f77b0916c8065f9bf679c015eee30f5f921b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=85=AC=E6=98=8E?=
 <83812544+Ed1s0nZ@users.noreply.github.com>
Date: Tue, 25 Feb 2025 14:20:40 +0800
Subject: [PATCH] Update scan.go

---
 scan.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/scan.go b/scan.go
index 1f11ab0..e2462a2 100644
--- a/scan.go
+++ b/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 {