fix:优化no——reply计算

This commit is contained in:
SengokuCola
2025-07-11 11:44:13 +08:00
parent 3768fb22ad
commit 375f28242a
5 changed files with 21 additions and 54 deletions

View File

@@ -54,7 +54,7 @@ async def _calculate_interest(message: MessageRecv) -> Tuple[float, bool]:
with Timer("记忆激活"):
interested_rate = await hippocampus_manager.get_activate_from_text(
message.processed_plain_text,
fast_retrieval=True,
fast_retrieval=False,
)
logger.debug(f"记忆激活率: {interested_rate:.2f}")
@@ -106,20 +106,19 @@ class HeartFCMessageReceiver:
group_info=groupinfo,
)
interested_rate, is_mentioned = await _calculate_interest(message)
message.interest_value = interested_rate
await self.storage.store_message(message, chat)
subheartflow = await heartflow.get_or_create_subheartflow(chat.stream_id)
message.update_chat_stream(chat)
# 6. 兴趣度计算与更新
interested_rate, is_mentioned = await _calculate_interest(message)
subheartflow.add_message_to_normal_chat_cache(message, interested_rate, is_mentioned)
chat_mood = mood_manager.get_mood_by_chat_id(subheartflow.chat_id)
asyncio.create_task(chat_mood.update_mood_by_message(message, interested_rate))
with open("interested_rates.txt", "a", encoding="utf-8") as f:
f.write(f"{interested_rate}\n")
# 7. 日志记录
mes_name = chat.group_info.group_name if chat.group_info else "私聊"

View File

@@ -113,6 +113,7 @@ class MessageRecv(Message):
self.is_mentioned = None
self.priority_mode = "interest"
self.priority_info = None
self.interest_value = None
def update_chat_stream(self, chat_stream: "ChatStream"):
self.chat_stream = chat_stream
@@ -337,6 +338,8 @@ class MessageSending(MessageProcessBase):
# 用于显示发送内容与显示不一致的情况
self.display_message = display_message
self.interest_value = 0.0
def build_reply(self):
"""设置回复消息"""
if self.reply:

View File

@@ -1,4 +1,5 @@
import re
import traceback
from typing import Union
# from ...common.database.database import db # db is now Peewee's SqliteDatabase instance
@@ -36,11 +37,11 @@ class MessageStorage:
filtered_display_message = re.sub(pattern, "", display_message, flags=re.DOTALL)
else:
filtered_display_message = ""
interest_value = 0
reply_to = message.reply_to
else:
filtered_display_message = ""
interest_value = message.interest_value
reply_to = ""
chat_info_dict = chat_stream.to_dict()
@@ -80,9 +81,11 @@ class MessageStorage:
processed_plain_text=filtered_processed_plain_text,
display_message=filtered_display_message,
memorized_times=message.memorized_times,
interest_value=interest_value,
)
except Exception:
logger.exception("存储消息失败")
traceback.print_exc()
@staticmethod
async def store_recalled_message(message_id: str, time: str, chat_stream: ChatStream) -> None: