From cb5043b87df5918fb80c4d9da7c9e2ed394869bc Mon Sep 17 00:00:00 2001
From: Hosigus <hosigus9733@gmail.com>
Date: Mon, 10 Mar 2025 14:33:30 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E5=88=AB=E5=90=8D=EF=BC=8C?=
 =?UTF-8?q?=E5=8F=AF=E4=BB=A5=E7=94=A8=E4=B8=8D=E5=90=8C=E5=90=8D=E7=A7=B0?=
 =?UTF-8?q?=E5=8F=AC=E5=94=A4=E6=9C=BA=E5=99=A8=E4=BA=BA?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docs/installation_standard.md      |  1 +
 src/plugins/chat/config.py         |  4 +++-
 src/plugins/chat/prompt_builder.py |  9 +++++----
 src/plugins/chat/utils.py          | 16 +++++-----------
 template/bot_config_template.toml  |  1 +
 5 files changed, 15 insertions(+), 16 deletions(-)

diff --git a/docs/installation_standard.md b/docs/installation_standard.md
index 5f52676d..d5866ce3 100644
--- a/docs/installation_standard.md
+++ b/docs/installation_standard.md
@@ -72,6 +72,7 @@ PLUGINS=["src2.plugins.chat"]
 [bot]
 qq = "机器人QQ号"  # 必填
 nickname = "麦麦"  # 机器人昵称
+alias_names = ["小麦", "阿麦"] # 机器人别名
 
 [personality]
 prompt_personality = [
diff --git a/src/plugins/chat/config.py b/src/plugins/chat/config.py
index c027753c..4802d043 100644
--- a/src/plugins/chat/config.py
+++ b/src/plugins/chat/config.py
@@ -1,6 +1,6 @@
 import os
 from dataclasses import dataclass, field
-from typing import Dict, Optional
+from typing import Dict, List, Optional
 
 import tomli
 from loguru import logger
@@ -15,6 +15,7 @@ class BotConfig:
 
     BOT_QQ: Optional[int] = 1
     BOT_NICKNAME: Optional[str] = None
+    BOT_ALIAS_NAMES: List[str] = field(default_factory=list) # 别名，可以通过这个叫它
     
     # 消息处理相关配置
     MIN_TEXT_LENGTH: int = 2  # 最小处理文本长度
@@ -186,6 +187,7 @@ class BotConfig:
             bot_qq = bot_config.get("qq")
             config.BOT_QQ = int(bot_qq)
             config.BOT_NICKNAME = bot_config.get("nickname", config.BOT_NICKNAME)
+            config.BOT_ALIAS_NAMES = bot_config.get("alias_names", config.BOT_ALIAS_NAMES)
 
         def response(parent: dict):
             response_config = parent["response"]
diff --git a/src/plugins/chat/prompt_builder.py b/src/plugins/chat/prompt_builder.py
index fdb887af..88576c7f 100644
--- a/src/plugins/chat/prompt_builder.py
+++ b/src/plugins/chat/prompt_builder.py
@@ -136,18 +136,19 @@ class PromptBuilder:
         probability_1 = global_config.PERSONALITY_1
         probability_2 = global_config.PERSONALITY_2
         probability_3 = global_config.PERSONALITY_3
-        prompt_personality = ''
+        
+        prompt_personality = f'{activate_prompt}你的网名叫{global_config.BOT_NICKNAME}，你还有很多别名:{"/".join(global_config.BOT_ALIAS_NAMES)}，'
         personality_choice = random.random()
         if personality_choice < probability_1:  # 第一种人格
-            prompt_personality = f'''{activate_prompt}你的网名叫{global_config.BOT_NICKNAME}，{personality[0]}, 你正在浏览qq群,{promt_info_prompt},
+            prompt_personality += f'''{personality[0]}, 你正在浏览qq群,{promt_info_prompt},
             现在请你给出日常且口语化的回复，平淡一些，尽量简短一些。{keywords_reaction_prompt}
             请注意把握群里的聊天内容，不要刻意突出自身学科背景，不要回复的太有条理，可以有个性。'''
         elif personality_choice < probability_1 + probability_2:  # 第二种人格
-            prompt_personality = f'''{activate_prompt}你的网名叫{global_config.BOT_NICKNAME}，{personality[1]}, 你正在浏览qq群，{promt_info_prompt},
+            prompt_personality += f'''{personality[1]}, 你正在浏览qq群，{promt_info_prompt},
             现在请你给出日常且口语化的回复，请表现你自己的见解，不要一昧迎合，尽量简短一些。{keywords_reaction_prompt}
             请你表达自己的见解和观点。可以有个性。'''
         else:  # 第三种人格
-            prompt_personality = f'''{activate_prompt}你的网名叫{global_config.BOT_NICKNAME}，{personality[2]}, 你正在浏览qq群，{promt_info_prompt},
+            prompt_personality += f'''{personality[2]}, 你正在浏览qq群，{promt_info_prompt},
             现在请你给出日常且口语化的回复，请表现你自己的见解，不要一昧迎合，尽量简短一些。{keywords_reaction_prompt}
             请你表达自己的见解和观点。可以有个性。'''
         
diff --git a/src/plugins/chat/utils.py b/src/plugins/chat/utils.py
index d166bcd2..f4f0ee53 100644
--- a/src/plugins/chat/utils.py
+++ b/src/plugins/chat/utils.py
@@ -52,19 +52,13 @@ def db_message_to_str(message_dict: Dict) -> str:
     return result
 
 
-def is_mentioned_bot_in_message(message: Message) -> bool:
-    """检查消息是否提到了机器人"""
-    keywords = [global_config.BOT_NICKNAME]
-    for keyword in keywords:
-        if keyword in message.processed_plain_text:
-            return True
-    return False
-
-
 def is_mentioned_bot_in_txt(message: str) -> bool:
     """检查消息是否提到了机器人"""
-    keywords = [global_config.BOT_NICKNAME]
-    for keyword in keywords:
+    if global_config.BOT_NICKNAME is None:
+        return True
+    if global_config.BOT_NICKNAME in message:
+        return True
+    for keyword in global_config.BOT_ALIAS_NAMES:
         if keyword in message:
             return True
     return False
diff --git a/template/bot_config_template.toml b/template/bot_config_template.toml
index d7c66d3f..feae628e 100644
--- a/template/bot_config_template.toml
+++ b/template/bot_config_template.toml
@@ -4,6 +4,7 @@ version = "0.0.3"
 [bot]
 qq = 123
 nickname = "麦麦"
+alias_names = ["小麦", "阿麦"]
 
 [personality]
 prompt_personality = [