From c4964a9e21241ffbeca2ce153ae32509005f59f9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=9D=8E=E5=AF=BB=E6=AC=A2?= <lxh@cxh.cn>
Date: Tue, 18 Jun 2024 17:30:05 +0800
Subject: [PATCH] :bug: Fix a bug.

---
 plugin/plugins/ai.go | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/plugin/plugins/ai.go b/plugin/plugins/ai.go
index e632980..2619c11 100644
--- a/plugin/plugins/ai.go
+++ b/plugin/plugins/ai.go
@@ -48,8 +48,23 @@ func AI(m *plugin.MessageContext) {
 
 	// 处理预设角色，默认是配置文件里的，如果数据库配置不为空，则使用数据库配置
 	prompt := config.Conf.Ai.Personality
+	var dbPrompt entity.AiAssistant
 	if friendInfo.Prompt != "" {
-		prompt = friendInfo.Prompt
+		// 取出配置的角色
+		client.MySQL.First(&dbPrompt, "id = ?", friendInfo.Prompt)
+		if dbPrompt.Id != "" {
+			prompt = dbPrompt.Personality
+		}
+	}
+
+	// 配置模型
+	chatModel := openai.GPT3Dot5Turbo0613
+	if friendInfo.AiModel != "" {
+		chatModel = friendInfo.AiModel
+	} else if dbPrompt.Model != "" {
+		chatModel = dbPrompt.Model
+	} else if config.Conf.Ai.Model != "" {
+		chatModel = config.Conf.Ai.Model
 	}
 
 	// 组装消息体
@@ -101,14 +116,6 @@ func AI(m *plugin.MessageContext) {
 		Content: m.Content,
 	})
 
-	// 配置模型
-	chatModel := openai.GPT3Dot5Turbo0613
-	if friendInfo.AiModel != "" {
-		chatModel = friendInfo.AiModel
-	} else if config.Conf.Ai.Model != "" {
-		chatModel = config.Conf.Ai.Model
-	}
-
 	// 默认使用AI回复
 	conf := openai.DefaultConfig(config.Conf.Ai.ApiKey)
 	if config.Conf.Ai.BaseUrl != "" {