main
黄海 7 months ago
parent 03e638919d
commit c7ca96823c

@ -159,7 +159,7 @@ public class MaxKbImpl {
* @return
*/
public static JSONObject generateQuestion(String zskId, String modelId, String documentId) {
String authCode=MaxKbImpl.getAuthorization();
String authCode = MaxKbImpl.getAuthorization();
String url = baseUrl + "/api/dataset/" + zskId + "/document/batch_generate_related";
JSONObject jo = new JSONObject();
jo.put("model_id", modelId);
@ -167,6 +167,10 @@ public class MaxKbImpl {
List<String> list = new ArrayList<>();
list.add(documentId);
jo.put("document_id_list", list);
System.out.println("modelId=" + modelId);
System.out.println(jo.toString());
HttpResponse response = HttpRequest.put(url)
.header("Accept", "application/json, text/plain, */*")
.header("Accept-Encoding", "gzip,deflate")

@ -2,6 +2,7 @@ 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;
@ -20,14 +21,14 @@ public class UploadQA1 {
String documentId = MaxKbService.uploadQA(zskName, uploadFile);
//2、使用哪个模型
String modelName = PropKit.get("modelName");
String model_id = MaxKbService.getModelIdByModelName(modelName);
Thread.sleep(30000);
//3、等待模型生成问题完毕
while (true) {
boolean finish = MaxKbService.autoIndexed(documentId);
if (finish) {
//手动生成问题
MaxKbService.generateQuestion(zskName, model_id, documentId);
JSONObject res = MaxKbService.generateQuestion(zskName, modelName, documentId);
System.out.println(res);
break;
}
Thread.sleep(10000);

@ -9,4 +9,4 @@ baseUrl=http://10.10.14.206:8080
MaxKBUsername=admin
MaxKBPassword=Dsideal4r5t6y7u!@#
# 大语言模型名称【通义千问qwen-max】
modelName=我的模型
modelName=DeepSeek
Loading…
Cancel
Save