From db5d4b06c64f8746e696113aaf2608ee533d9d43 Mon Sep 17 00:00:00 2001
From: tzdwindows 7 <3076584115@qq.com>
Date: Sun, 9 Feb 2025 12:06:50 +0800
Subject: [PATCH] =?UTF-8?q?feat(box):=20=E5=AE=9E=E7=8E=B0=20AI=20?=
 =?UTF-8?q?=E5=AF=B9=E8=AF=9D=E7=AA=97=E5=8F=A3=E5=B9=B6=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E7=BD=91=E7=BB=9C=20API=20=E8=B0=83=E7=94=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- 新增 LMApi 类实现 API 调用获取 AI 回复
- 修改 LocalWindow 类，添加系统提示信息- 更新 Main 类，引入 Markdown处理库
- 在 build.gradle 中添加相关依赖
---
 build.gradle                                  |  26 ++---
 logo.ico                                      | Bin 4286 -> 0 bytes
 .../java/com/axis/innovators/box/Main.java    |   5 +-
 .../axis/innovators/box/gui/LocalWindow.java  |   1 +
 src/main/java/org/tzd/lm/LM.java              |  57 ++++++++---
 src/main/java/org/tzd/lm/LMApi.java           |  91 ++++++++++++++++++
 6 files changed, 150 insertions(+), 30 deletions(-)
 delete mode 100644 logo.ico
 create mode 100644 src/main/java/org/tzd/lm/LMApi.java

diff --git a/build.gradle b/build.gradle
index 3223498..3d15d65 100644
--- a/build.gradle
+++ b/build.gradle
@@ -24,28 +24,22 @@ repositories {
 dependencies {
     testImplementation platform('org.junit:junit-bom:5.10.0')
     testImplementation 'org.junit.jupiter:junit-jupiter'
+    // https://mvnrepository.com/artifact/org.commonmark/commonmark
+    implementation 'org.commonmark:commonmark:0.24.0'
+    // https://mvnrepository.com/artifact/org.commonjava.googlecode.markdown4j/markdown4j
+    implementation 'org.commonjava.googlecode.markdown4j:markdown4j:2.2-cj-1.1'
+    // https://mvnrepository.com/artifact/com.google.code.gson/gson
+    implementation 'com.google.code.gson:gson:2.8.9'
 }
 
 application {
     mainClass = 'com.axis.innovators.box.Main'
 }
 
-shadowJar {
-    archiveBaseName = 'app'
-    archiveClassifier = ''
-    archiveVersion = ''
-}
-
-launch4j {
-    mainClassName = 'com.axis.innovators.box.Main'
-    outfile = version + '.exe'
-    icon = "${projectDir}/logo.ico"
-    jar = shadowJar.archiveFile.get()
-    bundledJrePath = 'jre'
-}
-
-tasks.named('launch4j') {
-    dependsOn shadowJar
+jar {
+    manifest {
+        attributes 'Main-Class': 'com.axis.innovators.box.Main'
+    }
 }
 
 test {
diff --git a/logo.ico b/logo.ico
deleted file mode 100644
index 57b1602c89ab2216dedd78e4a8f16742b6b3bb13..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 4286
zcmds)KWh|G6vdyQBbmY^jaX#@rmzwOAp{F;L_dLzZ4d&nNR>)b_%UJ#NQw<uNyJXD
z3232UA@~JE!6HE=YNF%$?Yy%)S$Dvh-K>8$oV|D6yYHU+?!E8LCWK@9o1PAQht@<0
zr$Y$m1Pz5N0^{wE=FXlkPz?N|-$@!mKa65KUEVRpa5VzGi0P<fFi(;$h;uIjeboGG
z;>;t^|1W;;r&``C?SHki|0?^sebM!k$@jc)TUd<phUA9e{BOrO=52FJ`jpVDF2%Z&
zTE8in$MUtt9`+gcyd^Q75bEx}h&f3owf1Eno;Mo%9scLCJ0%Rl|042#liU*4g%zPC
zc%QrRnVxtg?8muhlEdQvC|`4ebF(D<hd>Npg*yUsmxNCO<HP2kl~41JOTQm^;QB1k
zr)B;@^FJYf&i{Am%YyllXN1Rs>+oLsUExC<dm@<%P3wPLYs=?+N6$Cg{<?9@=$Sk_
zhIi8G%?HADp@_F`1YVb<nS4AI?EkIwYaPFZ-9<ffCT#Oc`h`O_&H0HRU--0txZo_;
zf|!WSHpHBDY?||vLvl$@$*uj9b8=sff!a_b+fX}dSeM;C=^c6qfAkc+MUUA(y+;q)
z9!ZbVtJu)H^l(#t?heinzBtpIt$Bg7<`_7GoJGelE1kQ-{FCjux)Buq;CFAEA0F-+
z?i@I{gSd;#KPUYt`1j?PyOo-@gk8Zo;CV$@7q$fMYUarQp!;|J&DBrRo8|Lo8{S=m
z@bm865bUo?!p}RocYaf2yu<5O-?c<e3H+Au?~g_4U|TRBG4h+lHc0w36AE8-Bk;Ov
zxhmi0rzgrW|EkyJud?s*td;oDo9(lTr)u0hN&2Vx7qsWS2=r0!e@vXC|No1+c(1*(
ds%C%jc+TfbmghU;hj0+W<zb@Nf15MU^Pk5tJ}dwL

diff --git a/src/main/java/com/axis/innovators/box/Main.java b/src/main/java/com/axis/innovators/box/Main.java
index 855605a..261942c 100644
--- a/src/main/java/com/axis/innovators/box/Main.java
+++ b/src/main/java/com/axis/innovators/box/Main.java
@@ -6,12 +6,13 @@ import com.axis.innovators.box.events.SubscribeEvent;
 import com.axis.innovators.box.gui.FridaWindow;
 import com.axis.innovators.box.gui.LocalWindow;
 import com.axis.innovators.box.gui.MainWindow;
-import com.axis.innovators.box.tools.FolderCreator;
 import com.axis.innovators.box.tools.LibraryLoad;
+import org.markdown4j.Markdown4jProcessor;
 
 import javax.swing.*;
 import java.awt.*;
 import java.awt.event.ActionEvent;
+import java.io.IOException;
 
 /**
  * 主类
@@ -33,7 +34,7 @@ public class Main {
         // 我不想写这个了你们自己实现
     }
 
-    public static void main(String[] args) {
+    public static void main(String[] args) throws IOException {
         // 注册事件
         GlobalEventBus.EVENT_BUS.register(new Main());
         // 设置系统外观
diff --git a/src/main/java/com/axis/innovators/box/gui/LocalWindow.java b/src/main/java/com/axis/innovators/box/gui/LocalWindow.java
index 473ee6a..a86c077 100644
--- a/src/main/java/com/axis/innovators/box/gui/LocalWindow.java
+++ b/src/main/java/com/axis/innovators/box/gui/LocalWindow.java
@@ -210,6 +210,7 @@ public class LocalWindow extends JDialog {
                             contextHandles.getLast(),
                             temperature,
                             prompt,
+                            "你是一个乐于助人的AI助手，请用友好自然的语气回答用户问题。",
                             this::publish
                     );
                 }
diff --git a/src/main/java/org/tzd/lm/LM.java b/src/main/java/org/tzd/lm/LM.java
index 694fe09..6e1b4b0 100644
--- a/src/main/java/org/tzd/lm/LM.java
+++ b/src/main/java/org/tzd/lm/LM.java
@@ -9,7 +9,7 @@ import com.axis.innovators.box.tools.LibraryLoad;
  */
 public class LM {
     public static boolean CUDA = true;
-    public final static String DEEP_SEEK = "G:/deepseek/deepseek/DeepSeek-R1-Distill-Qwen-1.5B-Q8_0/DeepSeek-R1-Distill-Qwen-1.5B-Q8_0.gguf";
+    public final static String DEEP_SEEK = FolderCreator.getModelFolder() + "\\DeepSeek-R1-Distill-Qwen-1.5B-Q8_0.gguf";
 
     static {
         if (!CUDA) {
@@ -28,12 +28,32 @@ public class LM {
         }
         LibraryLoad.loadLibrary("LM");
     }
+    public static long llamaLoadModelFromFile(String pathModel){
+        return llamaLoadModelFromFile(pathModel,
+                false,
+                false,
+                false,
+                false,
+                null
+        );
+    }
+
     /**
      * 加载模型
      * @param pathModel 模型路径
+     * @param vocabOnly 只负重词汇，不负重
+     * @param useMmap 尽可能使用mmap
+     * @param useMlock 强制系统将模型保存在RAM中
+     * @param checkTensors 验证模型张量数据
+     * @param progressCallback 进度回调
      * @return 模型句柄
      */
-    public static native long llamaLoadModelFromFile(String pathModel);
+    public static native long llamaLoadModelFromFile(String pathModel,
+                                                     boolean vocabOnly,
+                                                     boolean useMmap,
+                                                     boolean useMlock,
+                                                     boolean checkTensors,
+                                                     ProgressCallback progressCallback);
 
     /**
      * 释放模型资源
@@ -97,19 +117,20 @@ public class LM {
                                    long ctxHandle,
                                    float temperature,
                                    String prompt,
+                                   String system,
                                    MessageCallback messageCallback){
         return inference(modelHandle,
                 ctxHandle,
                 temperature,
-                0.05f,
-                40,
-                0.95f,
+                0.1f,
+                100,
+                0.9f,
                 0,
-                5,
-                2.0f,
-                0,
-                0,
-                prompt,
+                64,
+                1.1f,
+                0.0f,
+                0.0f,
+                system + "\n用户：" + prompt + "\n助手：",
                 messageCallback
         );
     }
@@ -148,7 +169,7 @@ public class LM {
                                           MessageCallback messageCallback);
 
     /**
-     * 回调接口
+     * 消息回调接口
      */
     public interface MessageCallback {
         /**
@@ -158,12 +179,24 @@ public class LM {
         void onMessage(String message);
     }
 
+    /**
+     * 进度回调接口
+     */
+    public interface ProgressCallback {
+        /**
+         * 进度回调
+         * @param progress 进度（0.0到1.0之间）
+         * @return 如果提供的progress_callback返回true，模型将继续加载。如果返回false，则立即中止模型加载。
+         */
+        boolean onModelLoad(float progress);
+    }
+
     public static void main(String[] args) {
         // 加载模型
         long modelHandle = llamaLoadModelFromFile(DEEP_SEEK);
         // 创建新的上下文
         long ctxHandle = createContext(modelHandle);
-        inference(modelHandle, ctxHandle, 0.2f, "Tell me how gpt works in an easy way, in Chinese", new MessageCallback() {
+        inference(modelHandle, ctxHandle, 0.2f,  "Tell me who you are in English","你是一个乐于助人的AI助手", new MessageCallback() {
             @Override
             public void onMessage(String message) {
                 // 回调输出
diff --git a/src/main/java/org/tzd/lm/LMApi.java b/src/main/java/org/tzd/lm/LMApi.java
new file mode 100644
index 0000000..3f46beb
--- /dev/null
+++ b/src/main/java/org/tzd/lm/LMApi.java
@@ -0,0 +1,91 @@
+package org.tzd.lm;
+
+import java.io.*;
+import java.net.HttpURLConnection;
+import java.net.URI;
+import java.net.URL;
+import java.nio.charset.StandardCharsets;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+import com.google.gson.Gson;
+import com.google.gson.reflect.TypeToken;
+
+/**
+ * @author tzdwindows 7
+ */
+public class LMApi {
+    private static final String API_URL = "https://api.pearktrue.cn/api/deepseek/";
+    private static final Gson GSON = new Gson();
+
+    public static class Message {
+        private String role;
+        private String content;
+
+        public Message(String role, String content) {
+            this.role = role;
+            this.content = content;
+        }
+
+        public String getRole() { return role; }
+        public String getContent() { return content; }
+        public void setRole(String role) { this.role = role; }
+        public void setContent(String content) { this.content = content; }
+    }
+
+    /**
+     * 调用AI接口获取回复
+     * @param messages 消息列表
+     * @return AI回复内容
+     */
+    public String getAIResponse(List<Message> messages) throws IOException {
+        String jsonInput = GSON.toJson(Map.of("messages", messages));
+
+        URL url = URI.create(API_URL).toURL();
+        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
+        conn.setRequestMethod("POST");
+        conn.setRequestProperty("Content-Type", "application/json");
+        conn.setDoOutput(true);
+
+        try(OutputStream os = conn.getOutputStream()) {
+            byte[] input = jsonInput.getBytes(StandardCharsets.UTF_8);
+            os.write(input, 0, input.length);
+        }
+
+        int responseCode = conn.getResponseCode();
+        if (responseCode == HttpURLConnection.HTTP_OK) {
+            try (BufferedReader br = new BufferedReader(
+                    new InputStreamReader(conn.getInputStream(), "GBK"))) {
+                String response = readAllLines(br);
+                Map<String, String> result = GSON.fromJson(response,
+                        new TypeToken<Map<String, String>>(){}.getType());
+                return result.get("message");
+            }
+        } else {
+            throw new IOException("API请求失败，状态码：" + responseCode);
+        }
+    }
+
+    private String readAllLines(BufferedReader reader) throws IOException {
+        StringBuilder sb = new StringBuilder();
+        String line;
+        while ((line = reader.readLine()) != null) {
+            sb.append(line);
+        }
+        return sb.toString();
+    }
+
+    public static void main(String[] args) {
+        List<Message> messages = new ArrayList<>();
+        messages.add(new Message("system", "You are a helpful assistant."));
+        messages.add(new Message("user", "你好啊"));
+
+        try {
+            String reply = new LMApi().getAIResponse(messages);
+            System.out.println(reply);
+        } catch (IOException e) {
+            e.printStackTrace();
+        }
+    }
+}
\ No newline at end of file