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 {