diff --git a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java
index bcc6dbd2..e97074c6 100644
--- a/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java
+++ b/mybatis-flex-processor/src/main/java/com/mybatisflex/processor/config/MybatisFlexConfig.java
@@ -58,10 +58,15 @@ public class MybatisFlexConfig {
 
             File moduleRoot = new File(aptConfigFileObject.toUri()).getParentFile().getParentFile().getParentFile();
 
-            // pom.xml      -> Maven 项目
-            // build.gradle -> Gradle 项目
-            if (new File(moduleRoot, "build.gradle").exists()) {
+            // pom.xml          -> Maven  项目
+            // build.gradle     -> Gradle 项目
+            // build.gradle.kts -> Gradle 项目
+            if (new File(moduleRoot, "pom.xml").exists()) {
+                FileUtil.setBuildFile("pom.xml");
+            } else if (new File(moduleRoot, "build.gradle").exists()) {
                 FileUtil.setBuildFile("build.gradle");
+            } else if (new File(moduleRoot, "build.gradle.kts").exists()) {
+                FileUtil.setBuildFile("build.gradle.kts");
             }
 
             while (FileUtil.existsBuildFile(moduleRoot)) {