main
黄海 7 months ago
parent a1c2130873
commit 77459148e2

@ -17,24 +17,24 @@ public class TestMaxKB {
//初始化数据库 //初始化数据库
MaxKbUtil.Init(); MaxKbUtil.Init();
//获取模型的id // //获取模型的id
String model_name = "DeepSeek"; // String model_name = "DeepSeek";
Record record = MaxKbUtil.getModel(model_name); // Record record = MaxKbUtil.getModel(model_name);
String model_id = record.getStr("id"); // String model_id = record.getStr("id");
System.out.println(model_id); // System.out.println(model_id);
//拷贝生成应用 //1、拷贝生成应用
String sourceApplicationName = "应用【模板】"; String sourceApplicationName = "应用【模板】";
String targetApplicationName = "XXX的应用"; String targetApplicationName = "XXX的应用";
String application_id = MaxKbUtil.addApplication(sourceApplicationName, targetApplicationName); String application_id = MaxKbUtil.addApplication(sourceApplicationName, targetApplicationName);
//拷贝生成知识库 //2、拷贝生成知识库
String sourceDataSetName = "知识库【模板】"; String sourceDataSetName = "知识库【模板】";
String targetDataSetName = "XXX的知识库"; String targetDataSetName = "XXX的知识库";
String dataset_id = MaxKbUtil.addDataSet(sourceDataSetName, targetDataSetName); String dataset_id = MaxKbUtil.addDataSet(sourceDataSetName, targetDataSetName);
//应用与知识库关联 //3、应用与知识库关联
record = new Record(); Record record = new Record();
record.set("create_time", LocalDateTime.now()); record.set("create_time", LocalDateTime.now());
record.set("update_time", LocalDateTime.now()); record.set("update_time", LocalDateTime.now());
record.set("id", UUID.randomUUID()); record.set("id", UUID.randomUUID());
@ -42,7 +42,7 @@ public class TestMaxKB {
record.set("dataset_id", UUID.fromString(dataset_id)); record.set("dataset_id", UUID.fromString(dataset_id));
Db.save("application_dataset_mapping", "id", record); Db.save("application_dataset_mapping", "id", record);
//生成应用的访问Token //4、生成应用的访问Token
MaxKbUtil.addAccessToken(application_id); MaxKbUtil.addAccessToken(application_id);
System.out.println("增加成功!"); System.out.println("增加成功!");

@ -68,6 +68,8 @@ public class MaxKbUtil {
record.set("id", UUID.randomUUID()); record.set("id", UUID.randomUUID());
//名称 //名称
record.set("name", targetApplicationName); record.set("name", targetApplicationName);
//描述
record.set("desc","【代码生成】");
//增加 //增加
Db.save("application", "id", record); Db.save("application", "id", record);
@ -98,12 +100,13 @@ public class MaxKbUtil {
System.exit(0); System.exit(0);
} }
//克隆出对象 //克隆出对象
record = new com.jfinal.plugin.activerecord.Record().setColumns(record.getColumns()); record = new Record().setColumns(record.getColumns());
//生成一个uuid //生成一个uuid
record.set("id", UUID.randomUUID()); record.set("id", UUID.randomUUID());
//名称 //名称
record.set("name", targetDatasetName); record.set("name", targetDatasetName);
record.set("desc", targetDatasetName); //描述
record.set("desc","【代码生成】");
//增加 //增加
Db.save("dataset", "id", record); Db.save("dataset", "id", record);
return record.getStr("id"); return record.getStr("id");

Loading…
Cancel
Save