From 295b8e10689db2dae1747b33575244f5283433ed Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=E6=B5=B7?= <10402852@qq.com>
Date: Mon, 30 Dec 2024 16:21:33 +0800
Subject: [PATCH] 'commit'

---
 src/main/java/Tools/MaxKb/Test.java   | 61 +++++++++++++++++++++++++++
 src/main/java/UnitTest/TestMaxKB.java | 33 ++++++++-------
 2 files changed, 78 insertions(+), 16 deletions(-)
 create mode 100644 src/main/java/Tools/MaxKb/Test.java

diff --git a/src/main/java/Tools/MaxKb/Test.java b/src/main/java/Tools/MaxKb/Test.java
new file mode 100644
index 00000000..c5f19642
--- /dev/null
+++ b/src/main/java/Tools/MaxKb/Test.java
@@ -0,0 +1,61 @@
+package Tools.MaxKb;
+
+import Tools.MaxKb.Util.MaxKbDb;
+import com.dsideal.QingLong.Util.CommonUtil;
+import com.jfinal.kit.PropKit;
+import com.jfinal.plugin.activerecord.Db;
+import com.jfinal.plugin.activerecord.Record;
+import org.joda.time.DateTime;
+
+import java.util.UUID;
+
+public class Test {
+
+    /**
+     * 获取模型
+     *
+     * @param model_name 模型名称
+     * @return
+     */
+    public static Record getModel(String model_name) {
+        String sql = "select * from model where name=?";
+        return Db.findFirst(sql, model_name);
+    }
+
+    /**
+     * 获取应用
+     *
+     * @param applicationName 应用名称
+     * @return
+     */
+    public static Record getApplication(String applicationName) {
+        String sql = "select * from application where name=?";
+        return Db.findFirst(sql, applicationName);
+    }
+
+    public static void main(String[] args) throws Exception {
+        //加载配置文件
+        PropKit.use("MaxKb.properties");
+        //初始化数据库
+        MaxKbDb.Init();
+
+        //获取模型的id
+        String model_name = "DeepSeek";
+        Record record = getModel(model_name);
+        String model_id = record.getStr("id");
+        System.out.println(model_id);
+
+        //获取应用
+        String applicationName = "黄海的一个应用";
+        record = getApplication(applicationName);
+        //克隆出对象
+        record = new Record().setColumns(record.getColumns());
+        //生成一个uuid
+        record.set("id", UUID.randomUUID());
+        //名称
+        record.set("name", "XXX的拷贝应用");
+        //增加
+        Db.save("application", "id", record);
+        System.out.println("增加成功！");
+    }
+}
\ No newline at end of file
diff --git a/src/main/java/UnitTest/TestMaxKB.java b/src/main/java/UnitTest/TestMaxKB.java
index 8adbfdb9..dde4504a 100644
--- a/src/main/java/UnitTest/TestMaxKB.java
+++ b/src/main/java/UnitTest/TestMaxKB.java
@@ -1,8 +1,6 @@
 package UnitTest;
 
 import cn.hutool.http.HttpRequest;
-import com.alibaba.fastjson.JSON;
-import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 
 public class TestMaxKB {
@@ -23,21 +21,24 @@ public class TestMaxKB {
         System.out.println(chat_id);
 
         //对话
-//        url = "http://10.10.14.206:8080/api/application/chat_message/" + chat_id;
-//        JSONObject jo = new JSONObject();
-//        jo.put("message", "你好！");
-//        jo.put("stream", true);
-//        jo.put("re_chat", false);
-//        result = HttpRequest.post(url)
-//                .header("Authorization", apiKey)
-//                .header("Content-Type", "application/json")
-//                .body(jo.toString())
-//                .execute()
-//                .body();
-//
+        url = "http://10.10.14.206:8080/api/application/chat_message/" + chat_id;
+        JSONObject jo = new JSONObject();
+        jo.put("message", "你好！我的名字叫黄海，记住我!一会我会问你我叫什么名字！");
+        //jo.put("message", "你好！你知道我叫什么名字吗？");
+        jo.put("stream", true);
+        jo.put("re_chat", false);
+        result = HttpRequest.post(url)
+                .header("Authorization", apiKey)
+                .header("Content-Type", "application/json")
+                .body(jo.toString())
+                .execute()
+                .body();
+
+        System.out.println(result);
+
 //        for (String s : result.split("\n")) {
-//            JSONObject jsonObject= JSONObject.parseObject(s);
-//            System.out.println(jsonObject.getString("data"));
+//            //JSONObject jsonObject= JSONObject.parseObject(s);
+//            //System.out.println(jsonObject.getString("data"));
 //        }
 
     }