Merge branch 'dev' of https://github.com/MaiM-with-u/MaiBot into dev

src/heart_flow/mai_state_manager.py

@@ -13,8 +13,8 @@ mai_state_config = LogConfig(
 logger = get_module_logger("mai_state_manager", config=mai_state_config)
 
 
-enable_unlimited_hfc_chat = True
+# enable_unlimited_hfc_chat = True
-# enable_unlimited_hfc_chat = False
+enable_unlimited_hfc_chat = False
 
 
 class MaiState(enum.Enum):

src/plugins/PFC/conversation.py

@@ -1,7 +1,9 @@
 import time
 import asyncio
 import datetime
-from .message_storage import MongoDBMessageStorage
+
+# from .message_storage import MongoDBMessageStorage
+from src.plugins.utils.chat_message_builder import get_raw_msg_before_timestamp_with_chat
 from ...config.config import global_config
 from typing import Dict, Any
 from ..chat.message import Message
@@ -75,13 +77,10 @@ class Conversation:
             raise
         try:
             logger.info(f"为 {self.stream_id} 加载初始聊天记录...")
-            storage = MongoDBMessageStorage()  # 创建存储实例
+            initial_messages = await get_raw_msg_before_timestamp_with_chat(  #
-            # 获取当前时间点之前最多 N 条消息 (比如 30 条)
-            # get_messages_before 返回的是按时间正序排列的列表
-            initial_messages = await storage.get_messages_before(
                 chat_id=self.stream_id,
-                time_point=time.time(),
+                timestamp=time.time(),
-                limit=30,  # 加载最近20条作为初始上下文，可以调整
+                limit=30,  # 加载最近30条作为初始上下文，可以调整
             )
             if initial_messages:
                 # 将加载的消息填充到 ObservationInfo 的 chat_history