From acdfcda8dbe44664c75fe4d471770fae70f3930a Mon Sep 17 00:00:00 2001
From: ddaodan <731882332@qq.com>
Date: Wed, 7 Aug 2024 17:55:29 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=20=E8=87=AA=E5=AE=9A?=
 =?UTF-8?q?=E4=B9=89prompt?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 changelog.md                                             | 2 ++
 .../java/com/ddaodan/MineChatGPT/CommandHandler.java     | 9 ++++++++-
 src/main/java/com/ddaodan/MineChatGPT/ConfigManager.java | 4 ++++
 src/main/resources/config.yml                            | 3 ++-
 4 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/changelog.md b/changelog.md
index ee13dda..f144e79 100644
--- a/changelog.md
+++ b/changelog.md
@@ -1,4 +1,6 @@
 # 更新日志
+## 2.4
+- 添加自定义prompt
 ## 2.3
 - 修复 #3：[BUG]提问乱码/答非所问
 ## 2.2
diff --git a/src/main/java/com/ddaodan/MineChatGPT/CommandHandler.java b/src/main/java/com/ddaodan/MineChatGPT/CommandHandler.java
index 1de7938..a5e0503 100644
--- a/src/main/java/com/ddaodan/MineChatGPT/CommandHandler.java
+++ b/src/main/java/com/ddaodan/MineChatGPT/CommandHandler.java
@@ -128,10 +128,17 @@ public class CommandHandler implements CommandExecutor {
         JSONObject json = new JSONObject();
         json.put("model", configManager.getDefaultModel());
         JSONArray messages = new JSONArray();
+        // 添加自定义 prompt
+        String customPrompt = configManager.getCustomPrompt();
+        if (!customPrompt.isEmpty()) {
+            JSONObject promptMessage = new JSONObject();
+            promptMessage.put("role", "system");
+            promptMessage.put("content", customPrompt);
+            messages.put(promptMessage);
+        }
         JSONObject message = new JSONObject();
         message.put("role", "user");
         message.put("content", utf8Question);
-        //message.put("content", question);
         messages.put(message);
         if (contextEnabled) {
             String history = conversationContext.getConversationHistory();
diff --git a/src/main/java/com/ddaodan/MineChatGPT/ConfigManager.java b/src/main/java/com/ddaodan/MineChatGPT/ConfigManager.java
index d9d0f9b..22e3085 100644
--- a/src/main/java/com/ddaodan/MineChatGPT/ConfigManager.java
+++ b/src/main/java/com/ddaodan/MineChatGPT/ConfigManager.java
@@ -56,6 +56,10 @@ public class ConfigManager {
         return config.getStringList("models");
     }
 
+    public String getCustomPrompt() {
+        return config.getString("prompt", "You are a helpful assistant.");
+    }
+
     public String getHelpMessage() {
         return translateColorCodes(config.getString("messages.help"));
     }
diff --git a/src/main/resources/config.yml b/src/main/resources/config.yml
index aa99f59..5605aaa 100644
--- a/src/main/resources/config.yml
+++ b/src/main/resources/config.yml
@@ -28,6 +28,7 @@ conversation:
   # Continuous conversation switch
   context_enabled: false
   max_history_size: 10
+prompt: "You are a helpful assistant."
 # Message settings
 messages:
   reload: "&aConfiguration reloaded successfully!"
@@ -53,4 +54,4 @@ messages:
 # If you don't know what this is, don't change it
 debug: false
 # DO NOT EDIT!!!!!
-version: 2.3
\ No newline at end of file
+version: 2.4
\ No newline at end of file