feat: 增加对消息自定义设置的支持,通过配置"maimcore_reply_probability_gain"字段可提高回复概率

This commit is contained in:
tcmofashi
2025-03-29 15:21:07 +08:00
parent 985b872828
commit 6dbee3dc56
2 changed files with 6 additions and 0 deletions

View File

@@ -147,6 +147,10 @@ class ChatBot:
) )
response = None response = None
if message.message_info.additional_config:
if "maimcore_reply_probability_gain" in message.message_info.additional_config.keys():
reply_probability += message.message_info.additional_config["maimcore_reply_probability_gain"]
# 开始组织语言 # 开始组织语言
if random() < reply_probability: if random() < reply_probability:
bot_user_info = UserInfo( bot_user_info = UserInfo(

View File

@@ -171,6 +171,7 @@ class BaseMessageInfo:
user_info: Optional[UserInfo] = None user_info: Optional[UserInfo] = None
format_info: Optional[FormatInfo] = None format_info: Optional[FormatInfo] = None
template_info: Optional[TemplateInfo] = None template_info: Optional[TemplateInfo] = None
additional_config: Optional[dict] = None
def to_dict(self) -> Dict: def to_dict(self) -> Dict:
"""转换为字典格式""" """转换为字典格式"""
@@ -201,6 +202,7 @@ class BaseMessageInfo:
platform=data.get("platform"), platform=data.get("platform"),
message_id=data.get("message_id"), message_id=data.get("message_id"),
time=data.get("time"), time=data.get("time"),
additional_config=data.get("additional_config", None),
group_info=group_info, group_info=group_info,
user_info=user_info, user_info=user_info,
format_info=format_info, format_info=format_info,