main
黄海 7 months ago
parent 03e638919d
commit c7ca96823c

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

@ -2,6 +2,7 @@ package Tools.MaxKb;
import Tools.MaxKb.Service.MaxKbService; import Tools.MaxKb.Service.MaxKbService;
import Tools.MaxKb.Util.MaxKbUtil; import Tools.MaxKb.Util.MaxKbUtil;
import com.alibaba.fastjson.JSONObject;
import com.dsideal.QingLong.Util.CommonUtil; import com.dsideal.QingLong.Util.CommonUtil;
import com.jfinal.kit.PropKit; import com.jfinal.kit.PropKit;
@ -20,14 +21,14 @@ public class UploadQA1 {
String documentId = MaxKbService.uploadQA(zskName, uploadFile); String documentId = MaxKbService.uploadQA(zskName, uploadFile);
//2、使用哪个模型 //2、使用哪个模型
String modelName = PropKit.get("modelName"); String modelName = PropKit.get("modelName");
String model_id = MaxKbService.getModelIdByModelName(modelName); Thread.sleep(30000);
//3、等待模型生成问题完毕 //3、等待模型生成问题完毕
while (true) { while (true) {
boolean finish = MaxKbService.autoIndexed(documentId); boolean finish = MaxKbService.autoIndexed(documentId);
if (finish) { if (finish) {
//手动生成问题 //手动生成问题
MaxKbService.generateQuestion(zskName, model_id, documentId); JSONObject res = MaxKbService.generateQuestion(zskName, modelName, documentId);
System.out.println(res);
break; break;
} }
Thread.sleep(10000); Thread.sleep(10000);

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