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) {
System.out.println("发送的body="+body);
String url = baseUrl + "/api/dataset/" + zskId + "/document/_bach";
HttpResponse response = HttpRequest.put(url)
.header("Accept", "application/json, text/plain, */*")

@ -20,7 +20,7 @@ public class MaxKbService {
* @param 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) {
System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行!");
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);
JSONArray ja = jo.getJSONArray("data");

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