main
黄海 7 months ago
parent b523d79c41
commit a92bdeadf6

@ -179,6 +179,7 @@ public class MaxKbImpl {
} }
public static String _bach(String authCode, String zskId, String body) { public static String _bach(String authCode, String zskId, String body) {
System.out.println("发送的body="+body);
String url = baseUrl + "/api/dataset/" + zskId + "/document/_bach"; String url = baseUrl + "/api/dataset/" + zskId + "/document/_bach";
HttpResponse response = HttpRequest.put(url) HttpResponse response = HttpRequest.put(url)
.header("Accept", "application/json, text/plain, */*") .header("Accept", "application/json, text/plain, */*")

@ -20,7 +20,7 @@ public class MaxKbService {
* @param uploadFile * @param uploadFile
*/ */
public static void uploadQA(String dataSetName, String uploadFile) { public static void uploadQA(String dataSetName, String uploadFile) {
List<com.jfinal.plugin.activerecord.Record> list = mm.getDataSetList(dataSetName); List<Record> list = mm.getDataSetList(dataSetName);
if (list.size() > 1) { if (list.size() > 1) {
System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行!"); System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行!");
return; return;
@ -79,7 +79,19 @@ public class MaxKbService {
/** /**
* *
*/ */
public static String _bach(String zskId, String uploadFile) { public static String _bach(String dataSetName, String uploadFile) {
List<Record> list = mm.getDataSetList(dataSetName);
if (list.size() > 1) {
System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行!");
return null;
}
if (list.isEmpty()) {
System.out.println("数据集:【" + dataSetName + "】不存在,请检查后重新运行!");
return null;
}
//数据集ID
String zskId = list.getFirst().getStr("id");
//上传文件 //上传文件
cn.hutool.json.JSONObject jo = uploadDocumentSplit(uploadFile); cn.hutool.json.JSONObject jo = uploadDocumentSplit(uploadFile);
JSONArray ja = jo.getJSONArray("data"); JSONArray ja = jo.getJSONArray("data");

@ -1,5 +1,6 @@
package Tools.MaxKb; package Tools.MaxKb;
import Tools.MaxKb.Service.MaxKbService;
import Tools.MaxKb.Util.MaxKbUtil; import Tools.MaxKb.Util.MaxKbUtil;
import com.jfinal.kit.PropKit; import com.jfinal.kit.PropKit;
@ -12,5 +13,10 @@ public class UploadDocument {
PropKit.use("MaxKb.properties"); PropKit.use("MaxKb.properties");
//初始化数据库 //初始化数据库
MaxKbUtil.Init(); MaxKbUtil.Init();
//测试上传
String dataSetName = "XXX的知识库";
String res = MaxKbService._bach(dataSetName, uploadFile);
System.out.println(res);
} }
} }
Loading…
Cancel
Save