Losita/mai-bot
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat：可以选择激活值计算模式，默认快速

Browse Source
This commit is contained in:
SengokuCola
2025-08-22 17:29:50 +08:00
parent e073ebcbbe
commit 1ddfa47e68
3 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/chat/heart_flow/heartflow_message_processor.py
View File

@@ -38,7 +38,7 @@ async def _calculate_interest(message: MessageRecv) -> Tuple[float, bool, list[s
interested_rate, keywords,keywords_lite = await hippocampus_manager.get_activate_from_text( interested_rate, keywords,keywords_lite = await hippocampus_manager.get_activate_from_text(
message.processed_plain_text, message.processed_plain_text,
max_depth= 4, max_depth= 4,
fast_retrieval=False, fast_retrieval=global_config.chat.interest_rate_mode == "fast",
) )
message.key_words = keywords message.key_words = keywords
message.key_words_lite = keywords_lite message.key_words_lite = keywords_lite
@@ -85,7 +85,7 @@ async def _calculate_interest(message: MessageRecv) -> Tuple[float, bool, list[s
message.interest_value = interested_rate message.interest_value = interested_rate
message.is_mentioned = is_mentioned message.is_mentioned = is_mentioned
return interested_rate, is_mentioned, keywords return interested_rate, keywords
class HeartFCMessageReceiver: class HeartFCMessageReceiver:

3
src/config/official_configs.py
View File

@@ -71,6 +71,9 @@ class ChatConfig(ConfigBase):
max_context_size: int = 18 max_context_size: int = 18
"""上下文长度""" """上下文长度"""
interest_rate_mode: Literal["fast", "accurate"] = "fast"
"""兴趣值计算模式，fast为快速计算，accurate为精确计算"""
mentioned_bot_inevitable_reply: bool = False mentioned_bot_inevitable_reply: bool = False
"""提及 bot 必然回复""" """提及 bot 必然回复"""

2
template/bot_config_template.toml
View File

@@ -65,6 +65,8 @@ focus_value = 0.5
max_context_size = 20 # 上下文长度 max_context_size = 20 # 上下文长度
interest_rate_mode = "fast" #激活值计算模式，可选fast或者accurate
mentioned_bot_inevitable_reply = true # 提及 bot 大概率回复 mentioned_bot_inevitable_reply = true # 提及 bot 大概率回复
at_bot_inevitable_reply = true # @bot 或 提及bot 大概率回复 at_bot_inevitable_reply = true # @bot 或 提及bot 大概率回复