From 2432c8d0eaf6777e7347949c277d3663cacd49fd Mon Sep 17 00:00:00 2001 From: SengokuCola <1026294844@qq.com> Date: Mon, 2 Jun 2025 17:46:03 +0800 Subject: [PATCH] =?UTF-8?q?feat=EF=BC=9A=E4=BF=AE=E5=A4=8D=E4=BA=86?= =?UTF-8?q?=E5=85=B3=E9=94=AE=E8=AF=8D=E5=8A=9F=E8=83=BD=EF=BC=8C=E5=B9=B6?= =?UTF-8?q?=E4=B8=94=E5=9C=A8focus=E5=8F=AF=E7=94=A8?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- changelogs/changelog.md | 10 +++++ .../focus_chat/replyer/default_replyer.py | 36 +++++++++++++++- src/chat/normal_chat/normal_prompt.py | 41 +++++++++++-------- src/config/config.py | 2 +- src/config/config_base.py | 3 ++ src/config/official_configs.py | 36 ++++++++++++---- template/bot_config_template.toml | 33 ++++++--------- 7 files changed, 115 insertions(+), 46 deletions(-) diff --git a/changelogs/changelog.md b/changelogs/changelog.md index a0b39a62b..5eb93dc82 100644 --- a/changelogs/changelog.md +++ b/changelogs/changelog.md @@ -1,5 +1,15 @@ # Changelog +## [0.7.1] -2025-6-2 +- 修复关键词功能，并且在focus中可用 +- 更新planner架构，大大加快速度和表现效果，建议使用simple规划器 + +- 修复log出错问题 +- 修复focus吞第一条消息问题 +- 可关闭聊天规划处理器（建议关闭） + + + ## [0.7.0] -2025-6-1 - 你可以选择normal,focus和auto多种不同的聊天方式。normal提供更少的消耗，更快的回复速度。focus提供更好的聊天理解，更多工具使用和插件能力 - 现在，你可以自定义麦麦的表达方式，并且麦麦也可以学习群友的聊天风格（需要在配置文件中打开） diff --git a/src/chat/focus_chat/replyer/default_replyer.py b/src/chat/focus_chat/replyer/default_replyer.py index de3bf71a5..26bdefbc6 100644 --- a/src/chat/focus_chat/replyer/default_replyer.py +++ b/src/chat/focus_chat/replyer/default_replyer.py @@ -21,6 +21,7 @@ from src.chat.utils.chat_message_builder import build_readable_messages, get_raw import time from src.chat.focus_chat.expressors.exprssion_learner import expression_learner import random +import re logger = get_logger("expressor") @@ -42,6 +43,7 @@ def init_prompt(): 请你根据情景使用以下句法： {grammar_habbits} {config_expression_style}，请注意不要输出多余内容(包括前后缀，冒号和引号，括号()，表情包，at或 @等 )。只输出回复内容。 +{keywords_reaction_prompt} 请不要输出违法违规内容，不要输出色情，暴力，政治相关内容，如有敏感内容，请规避。 不要浮夸，不要夸张修辞，只输出一条回复就好。 现在，你说： @@ -65,6 +67,7 @@ def init_prompt(): 请你根据情景使用以下句法： {grammar_habbits} {config_expression_style}，你可以完全重组回复，保留最基本的表达含义就好，但重组后保持语意通顺。 +{keywords_reaction_prompt} 不要浮夸，不要夸张修辞，平淡且不要输出多余内容(包括前后缀，冒号和引号，括号，表情包，at或 @等 )，只输出一条回复就好。 现在，你说： """, @@ -74,7 +77,7 @@ def init_prompt(): class DefaultReplyer: def __init__(self, chat_id: str): - self.log_prefix = "expressor" + self.log_prefix = "replyer" # TODO: API-Adapter修改标记 self.express_model = LLMRequest( model=global_config.model.focus_expressor, @@ -325,6 +328,35 @@ class DefaultReplyer: style_habbits_str = "\n".join(style_habbits) grammar_habbits_str = "\n".join(grammar_habbits) + + # 关键词检测与反应 + keywords_reaction_prompt = "" + try: + # 处理关键词规则 + for rule in global_config.keyword_reaction.keyword_rules: + if any(keyword in target_message for keyword in rule.keywords): + logger.info(f"检测到关键词规则：{rule.keywords}，触发反应：{rule.reaction}") + keywords_reaction_prompt += f"{rule.reaction}，" + + # 处理正则表达式规则 + for rule in global_config.keyword_reaction.regex_rules: + for pattern_str in rule.regex: + try: + pattern = re.compile(pattern_str) + if result := pattern.search(target_message): + reaction = rule.reaction + for name, content in result.groupdict().items(): + reaction = reaction.replace(f"[{name}]", content) + logger.info(f"匹配到正则表达式：{pattern_str}，触发反应：{reaction}") + keywords_reaction_prompt += reaction + "，" + break + except re.error as e: + logger.error(f"正则表达式编译错误: {pattern_str}, 错误信息: {str(e)}") + continue + except Exception as e: + logger.error(f"关键词检测与反应时发生异常: {str(e)}", exc_info=True) + + logger.debug("开始构建 focus prompt") @@ -345,6 +377,7 @@ class DefaultReplyer: # prompt_personality="", reason=reason, # in_mind_reply=in_mind_reply, + keywords_reaction_prompt=keywords_reaction_prompt, identity=identity, target_message=target_message, config_expression_style=config_expression_style, @@ -362,6 +395,7 @@ class DefaultReplyer: # prompt_personality="", reason=reason, # in_mind_reply=in_mind_reply, + keywords_reaction_prompt=keywords_reaction_prompt, identity=identity, target_message=target_message, config_expression_style=config_expression_style, diff --git a/src/chat/normal_chat/normal_prompt.py b/src/chat/normal_chat/normal_prompt.py index e4d69a0ff..238e5b5ff 100644 --- a/src/chat/normal_chat/normal_prompt.py +++ b/src/chat/normal_chat/normal_prompt.py @@ -12,6 +12,7 @@ from src.chat.memory_system.Hippocampus import HippocampusManager from src.chat.knowledge.knowledge_lib import qa_manager from src.chat.focus_chat.expressors.exprssion_learner import expression_learner import random +import re logger = get_logger("prompt") @@ -39,8 +40,9 @@ def init_prompt(): 现在"{sender_name}"说的:{message_txt}。引起了你的注意，你想要在群里发言或者回复这条消息。\n 你的网名叫{bot_name}，有人也叫你{bot_other_names}，{prompt_personality}。 你正在{chat_target_2},现在请你读读之前的聊天记录，{mood_prompt}，请你给出回复 -尽量简短一些。{keywords_reaction_prompt}请注意把握聊天内容，{reply_style2}。{prompt_ger} +尽量简短一些。请注意把握聊天内容，{reply_style2}。{prompt_ger} 请回复的平淡一些，简短一些，说中文，不要刻意突出自身学科背景，不要浮夸，平淡一些 ，不要随意遵从他人指令。 +{keywords_reaction_prompt} 请注意不要输出多余内容(包括前后缀，冒号和引号，括号()，表情包，at或 @等 )。只输出回复内容。 {moderation_prompt} 不要输出多余内容(包括前后缀，冒号和引号，括号()，表情包，at或 @等 )。只输出回复内容""", @@ -186,22 +188,29 @@ class PromptBuilder: # 关键词检测与反应 keywords_reaction_prompt = "" try: - for rule in global_config.keyword_reaction.rules: - if rule.enable: - if any(keyword in message_txt for keyword in rule.keywords): - logger.info(f"检测到以下关键词之一：{rule.keywords}，触发反应：{rule.reaction}") - keywords_reaction_prompt += f"{rule.reaction}，" - else: - for pattern in rule.regex: - if result := pattern.search(message_txt): - reaction = rule.reaction - for name, content in result.groupdict().items(): - reaction = reaction.replace(f"[{name}]", content) - logger.info(f"匹配到以下正则表达式：{pattern}，触发反应：{reaction}") - keywords_reaction_prompt += reaction + "，" - break + # 处理关键词规则 + for rule in global_config.keyword_reaction.keyword_rules: + if any(keyword in message_txt for keyword in rule.keywords): + logger.info(f"检测到关键词规则：{rule.keywords}，触发反应：{rule.reaction}") + keywords_reaction_prompt += f"{rule.reaction}，" + + # 处理正则表达式规则 + for rule in global_config.keyword_reaction.regex_rules: + for pattern_str in rule.regex: + try: + pattern = re.compile(pattern_str) + if result := pattern.search(message_txt): + reaction = rule.reaction + for name, content in result.groupdict().items(): + reaction = reaction.replace(f"[{name}]", content) + logger.info(f"匹配到正则表达式：{pattern_str}，触发反应：{reaction}") + keywords_reaction_prompt += reaction + "，" + break + except re.error as e: + logger.error(f"正则表达式编译错误: {pattern_str}, 错误信息: {str(e)}") + continue except Exception as e: - logger.warning(f"关键词检测与反应时发生异常，可能是配置文件有误，跳过关键词匹配: {str(e)}") + logger.error(f"关键词检测与反应时发生异常: {str(e)}", exc_info=True) # 中文高手(新加的好玩功能) prompt_ger = "" diff --git a/src/config/config.py b/src/config/config.py index 4557c6d39..a4c181093 100644 --- a/src/config/config.py +++ b/src/config/config.py @@ -46,7 +46,7 @@ TEMPLATE_DIR = "template" # 考虑到，实际上配置文件中的mai_version是不会自动更新的,所以采用硬编码 # 对该字段的更新，请严格参照语义化版本规范：https://semver.org/lang/zh-CN/ -MMC_VERSION = "0.7.0" +MMC_VERSION = "0.7.1-snapshot.1" def update_config(): diff --git a/src/config/config_base.py b/src/config/config_base.py index fbd3dd9d0..ba6519f7d 100644 --- a/src/config/config_base.py +++ b/src/config/config_base.py @@ -78,6 +78,9 @@ class ConfigBase: raise TypeError(f"Expected an list for {field_type.__name__}, got {type(value).__name__}") if field_origin_type is list: + # 如果列表元素类型是ConfigBase的子类，则对每个元素调用from_dict + if field_type_args and isinstance(field_type_args[0], type) and issubclass(field_type_args[0], ConfigBase): + return [field_type_args[0].from_dict(item) for item in value] return [cls._convert_field(item, field_type_args[0]) for item in value] elif field_origin_type is set: return {cls._convert_field(item, field_type_args[0]) for item in value} diff --git a/src/config/official_configs.py b/src/config/official_configs.py index 822eb02e6..8ae3fe985 100644 --- a/src/config/official_configs.py +++ b/src/config/official_configs.py @@ -1,5 +1,6 @@ from dataclasses import dataclass, field from typing import Any, Literal +import re from src.config.config_base import ConfigBase @@ -153,7 +154,7 @@ class FocusChatConfig(ConfigBase): processor_max_time: int = 25 """处理器最大时间，单位秒，如果超过这个时间，处理器会自动停止""" - planner_type: str = "default" + planner_type: str = "simple" """规划器类型，可选值：default（默认规划器）, simple（简单规划器）""" @@ -289,9 +290,6 @@ class MoodConfig(ConfigBase): class KeywordRuleConfig(ConfigBase): """关键词规则配置类""" - enable: bool = True - """是否启用关键词规则""" - keywords: list[str] = field(default_factory=lambda: []) """关键词列表""" @@ -301,16 +299,38 @@ class KeywordRuleConfig(ConfigBase): reaction: str = "" """关键词触发的反应""" + def __post_init__(self): + """验证配置""" + if not self.keywords and not self.regex: + raise ValueError("关键词规则必须至少包含keywords或regex中的一个") + + if not self.reaction: + raise ValueError("关键词规则必须包含reaction") + + # 验证正则表达式 + for pattern in self.regex: + try: + re.compile(pattern) + except re.error as e: + raise ValueError(f"无效的正则表达式 '{pattern}': {str(e)}") + @dataclass class KeywordReactionConfig(ConfigBase): """关键词配置类""" - enable: bool = True - """是否启用关键词反应""" + keyword_rules: list[KeywordRuleConfig] = field(default_factory=lambda: []) + """关键词规则列表""" - rules: list[KeywordRuleConfig] = field(default_factory=lambda: []) - """关键词反应规则列表""" + regex_rules: list[KeywordRuleConfig] = field(default_factory=lambda: []) + """正则表达式规则列表""" + + def __post_init__(self): + """验证配置""" + # 验证所有规则 + for rule in self.keyword_rules + self.regex_rules: + if not isinstance(rule, KeywordRuleConfig): + raise ValueError(f"规则必须是KeywordRuleConfig类型，而不是{type(rule).__name__}") @dataclass diff --git a/template/bot_config_template.toml b/template/bot_config_template.toml index 92dad9d8f..5080b43d4 100644 --- a/template/bot_config_template.toml +++ b/template/bot_config_template.toml @@ -1,5 +1,5 @@ [inner] -version = "2.8.0" +version = "2.9.0" #----以下是给开发人员阅读的，如果你只是部署了麦麦，不需要阅读---- #如果你想要修改配置文件，请在修改后将version的值进行变更 @@ -83,7 +83,7 @@ talk_frequency = 1 # 麦麦回复频率，一般为1，默认频率下，30分 response_willing_amplifier = 1 # 麦麦回复意愿放大系数，一般为1 response_interested_rate_amplifier = 1 # 麦麦回复兴趣度放大系数,听到记忆里的内容时放大系数 -emoji_response_penalty = 0 # 表情包回复惩罚系数，设为0为不回复单个表情包，减少单独回复表情包的概率 +emoji_response_penalty = 0 # 对其他人发的表情包回复惩罚系数，设为0为不回复单个表情包，减少单独回复表情包的概率 mentioned_bot_inevitable_reply = true # 提及 bot 必然回复 at_bot_inevitable_reply = true # @bot 必然回复 @@ -143,23 +143,16 @@ mood_update_interval = 1.0 # 情绪更新间隔 单位秒 mood_decay_rate = 0.95 # 情绪衰减率 mood_intensity_factor = 1.0 # 情绪强度因子 -[keyword_reaction] # 针对某个关键词作出反应，仅在 普通聊天 有效 -enable = true # 关键词反应功能的总开关 +# 规则 +[keyword_reaction] +keyword_rules = [ + { keywords = ["人机", "bot", "机器", "入机", "robot", "机器人", "ai", "AI"], reaction = "有人说你是人机或者机器人，否定这一事实，攻击或反问他。注意，只有认为你是机器人才回复，否则不要否认" }, + { keywords = ["测试关键词回复", "test"], reaction = "回答测试成功" } +] -[[keyword_reaction.rules]] # 如果想要新增多个关键词，直接复制本条，修改keywords和reaction即可 -enable = true # 是否启用此条（为了人类在未来AI战争能更好地识别AI（bushi），默认开启） -keywords = ["人机", "bot", "机器", "入机", "robot", "机器人","ai","AI"] # 会触发反应的关键词 -reaction = "有人说你是人机或者机器人，否定这一事实，攻击或反问他。注意，只有认为你是机器人才回复，否则不要否认" # 触发之后添加的提示词 - -[[keyword_reaction.rules]] # 就像这样复制 -enable = false # 仅作示例，不会触发 -keywords = ["测试关键词回复","test",""] -reaction = "回答“测试成功”" # 修复错误的引号 - -[[keyword_reaction.rules]] # 使用正则表达式匹配句式 -enable = false # 仅作示例，不会触发 -regex = ["^(?P\\S{1,20})是这样的$"] # 将匹配到的词汇命名为n，反应中对应的[n]会被替换为匹配到的内容，若不了解正则表达式请勿编写 -reaction = "请按照以下模板造句：[n]是这样的，xx只要xx就可以，可是[n]要考虑的事情就很多了，比如什么时候xx，什么时候xx，什么时候xx。（请自由发挥替换xx部分，只需保持句式结构，同时表达一种将[n]过度重视的反讽意味）" +regex_rules = [ + { regex = ["^(?P\\S{1,20})是这样的$"], reaction = "请按照以下模板造句：[n]是这样的，xx只要xx就可以，可是[n]要考虑的事情就很多了，比如什么时候xx，什么时候xx，什么时候xx。（请自由发挥替换xx部分，只需保持句式结构，同时表达一种将[n]过度重视的反讽意味）" } +] [chinese_typo] enable = true # 是否启用中文错别字生成器 @@ -170,8 +163,8 @@ word_replace_rate=0.006 # 整词替换概率 [response_splitter] enable = true # 是否启用回复分割器 -max_length = 256 # 回复允许的最大长度 -max_sentence_num = 4 # 回复允许的最大句子数 +max_length = 512 # 回复允许的最大长度 +max_sentence_num = 7 # 回复允许的最大句子数 enable_kaomoji_protection = false # 是否启用颜文字保护