'commit'

src/main/java/Tools/MaxKb/Model/MaxKbModel.java

@@ -188,4 +188,27 @@ public class MaxKbModel {
         sql = "delete from dataset where \"desc\"='【代码生成】'";
         Db.update(sql);
     }
+
+    /**
+     * 获取文档
+     *
+     * @param documentId
+     * @return
+     */
+    public Record getDocument(String documentId) {
+        String sql = "select * from document where id=?";
+        return Db.findFirst(sql, UUID.fromString(documentId));
+    }
+
+    /**
+     * 判断文档是否已经自动索引
+     *
+     * @param documentId 文档id
+     * @return
+     */
+    public boolean autoIndexed(String documentId) {
+        Record record = getDocument(documentId);
+        String status = record.getStr("status");
+        return status.equals("nn2");
+    }
 }

src/main/java/Tools/MaxKb/Service/MaxKbService.java

@@ -120,6 +120,7 @@ public class MaxKbService {
 
     /**
      * 生成问题
+     *
      * @param zskName    知识库名称
      * @param modelName  模型名称
      * @param documentId 文档Id
@@ -128,4 +129,14 @@ public class MaxKbService {
     public static JSONObject generateQuestion(String zskName, String modelName, String documentId) {
         return MaxKbImpl.generateQuestion(getZskId(zskName), getModelIdByModelName(modelName), documentId);
     }
+
+    /**
+     * 是否完成了自动索引
+     *
+     * @param documentId 文档id
+     * @return
+     */
+    public static boolean autoIndexed(String documentId) {
+        return mm.autoIndexed(documentId);
+    }
 }

src/main/java/Tools/MaxKb/UploadQA.java

@@ -1,8 +1,8 @@
 package Tools.MaxKb;
 
-import Tools.MaxKb.Service.Impl.MaxKbImpl;
 import Tools.MaxKb.Service.MaxKbService;
 import Tools.MaxKb.Util.MaxKbUtil;
+import com.dsideal.QingLong.Util.CommonUtil;
 import com.jfinal.kit.PropKit;
 
 public class UploadQA {
@@ -23,10 +23,16 @@ public class UploadQA {
         String model_id = MaxKbService.getModelIdByModelName(modelName);
 
         //3、等待模型生成问题完毕
-
+        while (true) {
-
+            boolean finish = MaxKbService.autoIndexed(documentId);
-        //4、手动生成问题
+            if (finish) {
+                //手动生成问题
                 MaxKbService.generateQuestion(zskName, model_id, documentId);
+                break;
+            }
+            Thread.sleep(1000);
+            System.out.println(CommonUtil.getCurrentTime() + "    正在等待模型生成问题完毕...");
+        }
         System.out.println("恭喜，所有操作成功完成！");
     }
 }

src/main/java/com/dsideal/QingLong/Util/CommonUtil.java

@@ -23,6 +23,8 @@ import java.nio.charset.StandardCharsets;
 import java.security.SecureRandom;
 import java.text.ParseException;
 import java.text.SimpleDateFormat;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
 import java.util.*;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
@@ -814,4 +816,12 @@ public class CommonUtil {
 
         return year + "-" + month + "-" + day;
     }
+
+    /**
+     * 获取当前时间
+     * @return
+     */
+    public static String getCurrentTime(){
+        return LocalDateTime.now().format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
+    }
 }