修复私聊PFC

src/plugins/PFC/waiter.py

@@ -1,85 +1,74 @@
 from src.common.logger import get_module_logger
 from .chat_observer import ChatObserver
 from .conversation_info import ConversationInfo
-from src.individuality.individuality import Individuality
+# from src.individuality.individuality import Individuality # 不再需要
 from ...config.config import global_config
 import time
 import asyncio
 
 logger = get_module_logger("waiter")
 
+# --- 在这里设定你想要的超时时间（秒） ---
+# 例如： 120 秒 = 2 分钟
+DESIRED_TIMEOUT_SECONDS = 300 
 
 class Waiter:
-    """快 速 等 待"""
+    """等待处理类"""
 
     def __init__(self, stream_id: str):
         self.chat_observer = ChatObserver.get_instance(stream_id)
-        self.personality_info = Individuality.get_instance().get_prompt(type="personality", x_person=2, level=2)
         self.name = global_config.BOT_NICKNAME
-
-        self.wait_accumulated_time = 0
+        # self.wait_accumulated_time = 0 # 不再需要累加计时
 
     async def wait(self, conversation_info: ConversationInfo) -> bool:
-        """等待
-
-        Returns:
-            bool: 是否超时（True表示超时）
-        """
-        # 使用当前时间作为等待开始时间
+        """等待用户新消息或超时"""
         wait_start_time = time.time()
-        self.chat_observer.waiting_start_time = wait_start_time  # 设置等待开始时间
+        logger.info(f"进入常规等待状态 (超时: {DESIRED_TIMEOUT_SECONDS} 秒)...")
 
         while True:
             # 检查是否有新消息
             if self.chat_observer.new_message_after(wait_start_time):
                 logger.info("等待结束，收到新消息")
-                return False
+                return False # 返回 False 表示不是超时
 
             # 检查是否超时
-            if time.time() - wait_start_time > 300:
-                self.wait_accumulated_time += 300
-
-                logger.info("等待超过300秒，结束对话")
+            elapsed_time = time.time() - wait_start_time
+            if elapsed_time > DESIRED_TIMEOUT_SECONDS:
+                logger.info(f"等待超过 {DESIRED_TIMEOUT_SECONDS} 秒...添加思考目标。")
                 wait_goal = {
-                    "goal": f"你等待了{self.wait_accumulated_time / 60}分钟，思考接下来要做什么",
+                    "goal": f"你等待了{elapsed_time / 60:.1f}分钟，注意可能在对方看来聊天已经结束，思考接下来要做什么",
                     "reason": "对方很久没有回复你的消息了",
                 }
                 conversation_info.goal_list.append(wait_goal)
-                print(f"添加目标: {wait_goal}")
+                logger.info(f"添加目标: {wait_goal}")
+                return True # 返回 True 表示超时
 
-                return True
-
-            await asyncio.sleep(1)
-            logger.info("等待中...")
+            await asyncio.sleep(5) # 每 5 秒检查一次
+            logger.info("等待中...") # 可以考虑把这个频繁日志注释掉，只在超时或收到消息时输出
 
     async def wait_listening(self, conversation_info: ConversationInfo) -> bool:
-        """等待倾听
-
-        Returns:
-            bool: 是否超时（True表示超时）
-        """
-        # 使用当前时间作为等待开始时间
+        """倾听用户发言或超时"""
         wait_start_time = time.time()
-        self.chat_observer.waiting_start_time = wait_start_time  # 设置等待开始时间
+        logger.info(f"进入倾听等待状态 (超时: {DESIRED_TIMEOUT_SECONDS} 秒)...")
 
         while True:
             # 检查是否有新消息
             if self.chat_observer.new_message_after(wait_start_time):
-                logger.info("等待结束，收到新消息")
-                return False
+                logger.info("倾听等待结束，收到新消息")
+                return False # 返回 False 表示不是超时
 
             # 检查是否超时
-            if time.time() - wait_start_time > 300:
-                self.wait_accumulated_time += 300
-                logger.info("等待超过300秒，结束对话")
+            elapsed_time = time.time() - wait_start_time
+            if elapsed_time > DESIRED_TIMEOUT_SECONDS:
+                logger.info(f"倾听等待超过 {DESIRED_TIMEOUT_SECONDS} 秒...添加思考目标。")
                 wait_goal = {
-                    "goal": f"你等待了{self.wait_accumulated_time / 60}分钟，思考接下来要做什么",
+                    # 保持 goal 文本一致
+                    "goal": f"你等待了{elapsed_time / 60:.1f}分钟，注意可能在对方看来聊天已经结束，思考接下来要做什么", 
                     "reason": "对方话说一半消失了，很久没有回复",
                 }
                 conversation_info.goal_list.append(wait_goal)
-                print(f"添加目标: {wait_goal}")
+                logger.info(f"添加目标: {wait_goal}")
+                return True # 返回 True 表示超时
 
-                return True
-
-            await asyncio.sleep(1)
-            logger.info("等待中...")
+            await asyncio.sleep(5) # 每 5 秒检查一次
+            logger.info("倾听等待中...") # 同上，可以考虑注释掉