feat 尝试将回复意愿加入思维流

This commit is contained in:
SengokuCola
2025-03-24 23:22:03 +08:00
parent 6c9b04c1be
commit 5dea0c3a16
5 changed files with 44 additions and 11 deletions

View File

@@ -145,7 +145,10 @@ class ChatBot:
interested_rate=interested_rate,
sender_id=str(message.message_info.user_info.user_id),
)
current_willing = willing_manager.get_willing(chat_stream=chat)
current_willing_old = willing_manager.get_willing(chat_stream=chat)
current_willing_new = (subheartflow_manager.get_subheartflow(chat.stream_id).current_state.willing-5)/4
print(f"旧回复意愿:{current_willing_old},新回复意愿:{current_willing_new}")
current_willing = (current_willing_old + current_willing_new) / 2
logger.info(
f"[{current_time}][{chat.group_info.group_name if chat.group_info else '私聊'}]"

View File

@@ -59,6 +59,7 @@ class BotConfig:
llm_topic_judge: Dict[str, str] = field(default_factory=lambda: {})
llm_summary_by_topic: Dict[str, str] = field(default_factory=lambda: {})
llm_emotion_judge: Dict[str, str] = field(default_factory=lambda: {})
llm_outer_world: Dict[str, str] = field(default_factory=lambda: {})
embedding: Dict[str, str] = field(default_factory=lambda: {})
vlm: Dict[str, str] = field(default_factory=lambda: {})
moderation: Dict[str, str] = field(default_factory=lambda: {})
@@ -237,6 +238,7 @@ class BotConfig:
"llm_topic_judge",
"llm_summary_by_topic",
"llm_emotion_judge",
"llm_outer_world",
"vlm",
"embedding",
"moderation",