You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.4 KiB
39 lines
1.4 KiB
package Tools.MaxKb;
|
|
|
|
import Tools.MaxKb.Service.MaxKbService;
|
|
import Tools.MaxKb.Util.MaxKbUtil;
|
|
import com.alibaba.fastjson.JSONObject;
|
|
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");
|
|
Thread.sleep(30000);
|
|
//3、等待模型生成问题完毕
|
|
while (true) {
|
|
boolean finish = MaxKbService.autoIndexed(documentId);
|
|
if (finish) {
|
|
//手动生成问题
|
|
JSONObject res = MaxKbService.generateQuestion(zskName, modelName, documentId);
|
|
System.out.println(res);
|
|
break;
|
|
}
|
|
Thread.sleep(10000);
|
|
System.out.println(CommonUtil.getCurrentTime() + " 正在等待模型生成问题完毕...");
|
|
}
|
|
System.out.println("恭喜，所有操作成功完成！");
|
|
}
|
|
} |