gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

将错别字生成提取到配置，一句一个错别字太烦了！

Browse Source
This commit is contained in:
ChangingSelf
2025-03-09 11:07:20 +08:00
parent 909441ea65
commit 39bb99cef3
3 changed files with 32 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/plugins/chat/config.py
View File

@@ -66,6 +66,12 @@ class BotConfig:
keywords_reaction_rules = [] # 关键词回复规则
chinese_typo_enable=True # 是否启用中文错别字生成器
chinese_typo_error_rate=0.03 # 单字替换概率
chinese_typo_min_freq=7 # 最小字频阈值
chinese_typo_tone_error_rate=0.2 # 声调错误概率
chinese_typo_word_replace_rate=0.02 # 整词替换概率
# 默认人设
PROMPT_PERSONALITY=[
"曾经是一个学习地质的女大学生，现在学习心理学和脑科学，你会刷贴吧",
@@ -203,6 +209,15 @@ class BotConfig:
if keywords_reaction_config.get("enable", False):
config.keywords_reaction_rules = keywords_reaction_config.get("rules", config.keywords_reaction_rules)
if "chinese_typo_generator" in toml_dict:
# 读取中文错别字生成器配置
chinese_typo_generator_config = toml_dict["chinese_typo_generator"]
config.chinese_typo_enable = chinese_typo_generator_config.get("enable", config.chinese_typo_enable)
config.chinese_typo_error_rate = chinese_typo_generator_config.get("error_rate", config.chinese_typo_error_rate)
config.chinese_typo_min_freq = chinese_typo_generator_config.get("min_freq", config.chinese_typo_min_freq)
config.chinese_typo_tone_error_rate = chinese_typo_generator_config.get("tone_error_rate", config.chinese_typo_tone_error_rate)
config.chinese_typo_word_replace_rate = chinese_typo_generator_config.get("word_replace_rate", config.chinese_typo_word_replace_rate)
# 群组配置
if "groups" in toml_dict:
groups_config = toml_dict["groups"]

17
src/plugins/chat/utils.py
View File

@@ -330,13 +330,16 @@ def process_llm_response(text: str) -> List[str]:
print(f"回复过长 ({len(text)} 字符)，返回默认回复")
return ['懒得说']
# 处理长消息
typo_generator = ChineseTypoGenerator(
error_rate=0.03,
min_freq=7,
tone_error_rate=0.2,
word_replace_rate=0.02
)
typoed_text = typo_generator.create_typo_sentence(text)[0]
if global_config.chinese_typo_enable:
typo_generator = ChineseTypoGenerator(
error_rate=global_config.chinese_typo_error_rate,
min_freq=global_config.chinese_typo_min_freq,
tone_error_rate=global_config.chinese_typo_tone_error_rate,
word_replace_rate=global_config.chinese_typo_word_replace_rate
)
typoed_text = typo_generator.create_typo_sentence(text)[0]
else:
typoed_text = text
sentences = split_into_sentences_w_remove_punctuation(typoed_text)
# 检查分割后的消息数量是否过多（超过3条）
if len(sentences) > 4: