feat:为s4u添加了优先队列和普通队列

This commit is contained in:
SengokuCola
2025-07-02 00:21:47 +08:00
parent 324b294b5f
commit bb2a95e388
6 changed files with 246 additions and 154 deletions

View File

@@ -1,7 +1,6 @@
from src.chat.message_receive.message import MessageRecv
from src.chat.message_receive.storage import MessageStorage
from src.chat.message_receive.chat_stream import get_chat_manager
from src.chat.utils.utils import is_mentioned_bot_in_message
from src.common.logger import get_logger
from .s4u_chat import get_s4u_chat_manager
@@ -47,13 +46,12 @@ class S4UMessageProcessor:
await self.storage.store_message(message, chat)
is_mentioned = is_mentioned_bot_in_message(message)
s4u_chat = get_s4u_chat_manager().get_or_create_chat(chat)
if userinfo.user_id in target_user_id_list:
await s4u_chat.response(message, is_mentioned=is_mentioned, interested_rate=1.0)
await s4u_chat.add_message(message)
else:
await s4u_chat.response(message, is_mentioned=is_mentioned, interested_rate=0.0)
await s4u_chat.add_message(message)
# 7. 日志记录
logger.info(f"[S4U]{userinfo.user_nickname}:{message.processed_plain_text}")