QingLong/src/main/java/Tools/MaxKb/UploadQA1.java

package Tools.MaxKb;

import Tools.MaxKb.Service.MaxKbService;
import Tools.MaxKb.Util.MaxKbUtil;
import com.dsideal.QingLong.Util.CommonUtil;
import com.jfinal.kit.PropKit;

public class UploadQA1 {
    //要上传的QA知识内容
    static String uploadFile = "D:\\dsWork\\QingLong\\Doc\\MaxKB\\黄海的个人简历【QA版本】.xlsx";

    public static void main(String[] args) throws Exception {
        //加载配置文件
        PropKit.use("MaxKb.properties");
        //初始化数据库
        MaxKbUtil.Init();

        //1、上传文件
        String zskName = "知识库【模板】";
        String documentId = MaxKbService.uploadQA(zskName, uploadFile);
        //2、使用哪个模型
        String modelName = PropKit.get("modelName");
        String model_id = MaxKbService.getModelIdByModelName(modelName);

        //3、等待模型生成问题完毕
        while (true) {
            boolean finish = MaxKbService.autoIndexed(documentId);
            if (finish) {
                //手动生成问题
                MaxKbService.generateQuestion(zskName, model_id, documentId);
                break;
            }
            Thread.sleep(10000);
            System.out.println(CommonUtil.getCurrentTime() + "    正在等待模型生成问题完毕...");
        }
        System.out.println("恭喜，所有操作成功完成！");
    }
}