From 8142a12acb3004ae82d8542e3adc903fbc49a90c Mon Sep 17 00:00:00 2001 From: Windpicker-owo <3431391539@qq.com> Date: Tue, 7 Oct 2025 17:50:18 +0800 Subject: [PATCH] =?UTF-8?q?feat(chat):=20=E6=B7=BB=E5=8A=A0=E6=9C=BA?= =?UTF-8?q?=E5=99=A8=E4=BA=BA=E8=BA=AB=E4=BB=BD=E4=BF=A1=E6=81=AF=E5=88=B0?= =?UTF-8?q?=E6=8F=90=E7=A4=BA=E8=AF=8D=E6=A8=A1=E6=9D=BF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 在多个提示词模板中添加机器人名称和昵称信息,让AI能够正确识别自己的身份。同时更新PromptParameters类以支持这些新参数。 --- src/chat/replyer/default_generator.py | 12 ++++++++++++ src/chat/utils/prompt.py | 4 ++++ 2 files changed, 16 insertions(+) diff --git a/src/chat/replyer/default_generator.py b/src/chat/replyer/default_generator.py index f4a436797..bc6ffa54b 100644 --- a/src/chat/replyer/default_generator.py +++ b/src/chat/replyer/default_generator.py @@ -67,6 +67,9 @@ def init_prompt(): {moderation_prompt} 不要复读你前面发过的内容,意思相近也不行。 不要浮夸,不要夸张修辞,平淡且不要输出多余内容(包括前后缀,冒号和引号,括号,表情包,at或 @等 ),只输出一条回复就好。 + +*你叫{bot_name},也有人叫你{bot_nickname}* + 现在,你说: """, "default_expressor_prompt", @@ -147,6 +150,8 @@ def init_prompt(): 请注意不要输出多余内容(包括前后缀,冒号和引号,at或 @等 )。只输出回复内容。 {moderation_prompt} +*你叫{bot_name},也有人叫你{bot_nickname}* + 现在,你说: """, "s4u_style_prompt", @@ -214,6 +219,9 @@ If you need to use the search tool, please directly call the function "lpmm_sear 2. 可以分享你的看法、提出相关问题,或者开个合适的玩笑。 3. 目的是让对话更有趣、更深入。 最终请输出一条简短、完整且口语化的回复。 + +*你叫{bot_name},也有人叫你{bot_nickname}* + 现在,你说: """, "normal_style_prompt", @@ -1399,6 +1407,8 @@ class DefaultReplyer: reply_target_block=reply_target_block, mood_prompt=mood_prompt, action_descriptions=action_descriptions, + bot_name=global_config.bot.nickname, + bot_nickname=",".join(global_config.bot.alias_names) if global_config.bot.alias_names else "", ) # 使用新的统一Prompt系统 - 使用正确的模板名称 @@ -1554,6 +1564,8 @@ class DefaultReplyer: # 添加已构建的表达习惯和关系信息 expression_habits_block=expression_habits_block, relation_info_block=relation_info, + bot_name=global_config.bot.nickname, + bot_nickname=",".join(global_config.bot.alias_names) if global_config.bot.alias_names else "", ) # 使用新的统一Prompt系统 - Expressor模式 diff --git a/src/chat/utils/prompt.py b/src/chat/utils/prompt.py index 9f24d6e29..f39f69b81 100644 --- a/src/chat/utils/prompt.py +++ b/src/chat/utils/prompt.py @@ -35,6 +35,8 @@ class PromptParameters: reply_to: str = "" extra_info: str = "" prompt_mode: Literal["s4u", "normal", "minimal"] = "s4u" + bot_name: str = "" + bot_nickname: str = "" # 功能开关 enable_tool: bool = True @@ -450,6 +452,8 @@ class Prompt: "reply_target_block": self.parameters.reply_target_block, "mood_state": self.parameters.mood_prompt, "action_descriptions": self.parameters.action_descriptions, + "bot_name": self.parameters.bot_name, + "bot_nickname": self.parameters.bot_nickname, } )