QingLong/src/main/java/Tools/MaxKb/Service/MaxKbService.java

package Tools.MaxKb.Service;

import Tools.MaxKb.Service.Impl.MaxKbImpl;
import Tools.MaxKb.Model.MaxKbModel;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.Record;

import java.io.File;
import java.util.List;

public class MaxKbService {
    public static MaxKbModel mm = new MaxKbModel();

    /**
     * 上传文件到知识库
     * @param dataSetName 数据集名称
     * @param uploadFile 文件路径
     */
    public static void uploadQA(String dataSetName,String uploadFile) {
        List<com.jfinal.plugin.activerecord.Record> list = mm.getDataSetList(dataSetName);
        if (list.size() > 1) {
            System.out.println("数据集:【" + dataSetName + "】名称重复,请检查后重新运行！");
            return;
        }
        if (list.isEmpty()) {
            System.out.println("数据集:【" + dataSetName + "】不存在,请检查后重新运行！");
            return;
        }
        //数据集ID
        String zskId = list.getFirst().getStr("id");

        //在上传文档前，需要删除掉旧的同名文档
        File fi = new File(uploadFile);
        String fileName = fi.getName();
        //获取身份识别
        String authCode = MaxKbImpl.getAuthorization();
        //上传文档
        List<com.jfinal.plugin.activerecord.Record> listDocuments = mm.getDocumentListByDataSetId(zskId);
        for (Record record : listDocuments) {
            String documentName = record.getStr("name");
            if (documentName.equals(fileName)) {
                String documentId = record.getStr("id");
                //1、需要找出根据此文档有哪些已经生成的问题，并且需要删除掉所有已经生成的问题
                mm.delDocumentProblem(documentId);
                //准备删除掉旧的文档
                MaxKbImpl.delDocument(authCode, zskId, documentId);
                System.out.println("删除旧的文档成功：" + documentName);
            }
        }
        //上传新的文档
        String documentId = MaxKbImpl.uploadQA(authCode, zskId, uploadFile);
        //2、需要重新生成问题
        String modelName = PropKit.get("modelName");
        String model_id = getModelIdByModelName(modelName);
        //生成问题
        MaxKbImpl.generateQuestion(authCode, zskId, model_id, documentId);
    }


    /**
     * 根据模型名称获取模型ID
     *
     * @param modelName 模型名称
     * @return
     */
    public static String getModelIdByModelName(String modelName) {
        List<Record> list = mm.getModel(modelName);
        if (list.isEmpty()) return null;
        return list.getFirst().getStr("id");
    }
}