From 8a3dbc6a3b81cb6424ce19e25b3f009490b2fa46 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 17:15:53 +0800
Subject: [PATCH] 'commit'

---
 src/main/java/Tools/MaxKb/TestMaxKB.java      | 11 ++---------
 src/main/java/Tools/MaxKb/Util/MaxKbUtil.java |  3 ++-
 2 files changed, 4 insertions(+), 10 deletions(-)

diff --git a/src/main/java/Tools/MaxKb/TestMaxKB.java b/src/main/java/Tools/MaxKb/TestMaxKB.java
index 9089df80..4fff2f53 100644
--- a/src/main/java/Tools/MaxKb/TestMaxKB.java
+++ b/src/main/java/Tools/MaxKb/TestMaxKB.java
@@ -18,12 +18,6 @@ public class TestMaxKB {
         //初始化数据库
         MaxKbUtil.Init();
 
-//        //获取模型的id
-//        String model_name = "DeepSeek";
-//        Record record = MaxKbUtil.getModel(model_name);
-//        String model_id = record.getStr("id");
-//        System.out.println(model_id);
-
         //1、删除代码生成的应用程序与知识库
         MaxKbUtil.delCodeGenerateApplicationDataSet();
 
@@ -47,8 +41,7 @@ public class TestMaxKB {
         Db.save("application_dataset_mapping", "id", record);
 
         //5、生成应用的访问Token
-        MaxKbUtil.addAccessToken(application_id);
-
-        System.out.println("增加成功！");
+        String access_token = MaxKbUtil.addAccessToken(application_id);
+        System.out.println("增加成功！access_token=" + access_token);
     }
 }
\ No newline at end of file
diff --git a/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java b/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java
index e3862918..cf787e95 100644
--- a/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java
+++ b/src/main/java/Tools/MaxKb/Util/MaxKbUtil.java
@@ -118,7 +118,7 @@ public class MaxKbUtil {
      *
      * @param application_id 应用的id
      */
-    public static void addAccessToken(String application_id) {
+    public static String addAccessToken(String application_id) {
         com.jfinal.plugin.activerecord.Record record = new Record();
         record.set("create_time", LocalDateTime.now());
         record.set("update_time", LocalDateTime.now());
@@ -131,6 +131,7 @@ public class MaxKbUtil {
         record.set("white_list", new String[]{""});
         record.set("show_source", false);
         Db.save("application_access_token", "application_id", record);
+        return record.getStr("access_token");
     }
 
     /**