mirror of https://github.com/Mai-with-u/MaiBot.git
适配推理模式
parent
280e488f71
commit
80759f3fa1
|
|
@ -17,6 +17,7 @@ from ...message import UserInfo, Seg
|
||||||
from src.common.logger import get_module_logger, CHAT_STYLE_CONFIG, LogConfig
|
from src.common.logger import get_module_logger, CHAT_STYLE_CONFIG, LogConfig
|
||||||
from ...chat.chat_stream import chat_manager
|
from ...chat.chat_stream import chat_manager
|
||||||
from ...person_info.relationship_manager import relationship_manager
|
from ...person_info.relationship_manager import relationship_manager
|
||||||
|
from ...chat.message_buffer import message_buffer
|
||||||
|
|
||||||
# 定义日志配置
|
# 定义日志配置
|
||||||
chat_config = LogConfig(
|
chat_config = LogConfig(
|
||||||
|
|
@ -143,6 +144,8 @@ class ReasoningChat:
|
||||||
userinfo = message.message_info.user_info
|
userinfo = message.message_info.user_info
|
||||||
messageinfo = message.message_info
|
messageinfo = message.message_info
|
||||||
|
|
||||||
|
# 消息加入缓冲池
|
||||||
|
await message_buffer.start_caching_messages(message)
|
||||||
|
|
||||||
# logger.info("使用推理聊天模式")
|
# logger.info("使用推理聊天模式")
|
||||||
|
|
||||||
|
|
@ -172,6 +175,17 @@ class ReasoningChat:
|
||||||
timer2 = time.time()
|
timer2 = time.time()
|
||||||
timing_results["记忆激活"] = timer2 - timer1
|
timing_results["记忆激活"] = timer2 - timer1
|
||||||
|
|
||||||
|
# 查询缓冲器结果,会整合前面跳过的消息,改变processed_plain_text
|
||||||
|
buffer_result = await message_buffer.query_buffer_result(message)
|
||||||
|
if not buffer_result:
|
||||||
|
if message.message_segment.type == "text":
|
||||||
|
logger.info(f"触发缓冲,已炸飞消息:{message.processed_plain_text}")
|
||||||
|
elif message.message_segment.type == "image":
|
||||||
|
logger.info(f"触发缓冲,已炸飞表情包/图片")
|
||||||
|
elif message.message_segment.type == "seglist":
|
||||||
|
logger.info(f"触发缓冲,已炸飞消息列")
|
||||||
|
return
|
||||||
|
|
||||||
is_mentioned = is_mentioned_bot_in_message(message)
|
is_mentioned = is_mentioned_bot_in_message(message)
|
||||||
|
|
||||||
# 计算回复意愿
|
# 计算回复意愿
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue