diff --git a/src/main/java/Tools/MaxKb/Model/MaxKbModel.java b/src/main/java/Tools/MaxKb/Model/MaxKbModel.java index 63050ffd..da7b074d 100644 --- a/src/main/java/Tools/MaxKb/Model/MaxKbModel.java +++ b/src/main/java/Tools/MaxKb/Model/MaxKbModel.java @@ -10,8 +10,18 @@ public class MaxKbModel { * @param dataSetName 数据集名称 * @return */ - public List getDataSetList(String dataSetName) { + public List getDataSetList(String dataSetName) { String sql = "select * from dataset where name=?"; return Db.find(sql, dataSetName); } + + /** + * 根据数据集id获取文档列表 + * @param dataset_id 数据集id + * @return + */ + public List getDocumentListByDataSetId(String dataset_id) { + String sql = "select * from document where dataset_id=? order by create_time desc"; + return Db.find(sql, dataset_id); + } } diff --git a/src/main/java/Tools/MaxKb/UploadMaxKBFile.java b/src/main/java/Tools/MaxKb/UploadMaxKBFile.java index 1800b4da..e22154c6 100644 --- a/src/main/java/Tools/MaxKb/UploadMaxKBFile.java +++ b/src/main/java/Tools/MaxKb/UploadMaxKBFile.java @@ -12,10 +12,8 @@ public class UploadMaxKBFile { public static MaxKbModel mm = new MaxKbModel(); - //知识库id,不知道id的可以从知识库的地址栏获取 - static String zskId = "0dbf473a-c352-11ef-a5a3-0242ac120003"; - //获取知识库上传文档接口 + //要上传的QA知识内容 static String uploadFile = "D:\\dsWork\\QingLong\\Doc\\MaxKB\\各学校人员和班级统计【结果】.xlsx"; public static void main(String[] args) throws Exception { diff --git a/src/main/resources/MaxKb.properties b/src/main/resources/MaxKb.properties index b28fde36..0c91815b 100644 --- a/src/main/resources/MaxKb.properties +++ b/src/main/resources/MaxKb.properties @@ -5,10 +5,8 @@ password=Password123@postgres jdbcUrl=jdbc:postgresql://10.10.14.206:5432/maxkb?reWriteBatchedInserts=true # MaxKB的配置信息 baseUrl=http://10.10.14.206:8080 - # MaxKB系统的管理员账号 MaxKBUsername=admin MaxKBPassword=Dsideal4r5t6y7u!@# - -# 知识库的名称 +# 知识库的名称【知识库的名称是固定的,变化的应该是知识库里面的文档】 dataSetName=测试用的知识库 \ No newline at end of file