Merge branch 'PFC-test' of https://github.com/smartmita/MaiBot into PFC-test

src/plugins/PFC/action_planner.py

@@ -81,6 +81,24 @@ block_and_ignore: 更加极端的结束对话方式，直接结束对话并在
 
 注意：请严格按照JSON格式输出，不要包含任何其他内容。"""
 
+# 新增：Prompt(3): 决定是否在结束对话前发送告别语
+PROMPT_END_DECISION = """{persona_text}。刚刚你决定结束一场 QQ 私聊。
+
+【你们之前的聊天记录】
+{chat_history_text}
+
+你觉得你们的对话已经完整结束了吗？有时候，在对话自然结束后再说点什么可能会有点奇怪，但有时也可能需要一条简短的消息来圆满结束。
+如果觉得确实有必要再发一条简短、自然、符合你人设的告别消息（比如 "好，下次再聊~" 或 "嗯，先这样吧"），就输出 "yes"。
+如果觉得当前状态下直接结束对话更好，没有必要再发消息，就输出 "no"。
+
+请以 JSON 格式输出你的选择：
+{{
+    "say_bye": "yes/no",
+    "reason": "选择 yes 或 no 的原因和内心想法 (简要说明)"
+}}
+
+注意：请严格按照 JSON 格式输出，不要包含任何其他内容。"""
+
 
 # ActionPlanner 类定义，顶格
 class ActionPlanner:
@@ -336,9 +354,10 @@ class ActionPlanner:
         logger.debug(f"[私聊][{self.private_name}]发送到LLM的最终提示词:\n------\n{prompt}\n------")
         try:
             content, _ = await self.llm.generate_response_async(prompt)
-            logger.debug(f"[私聊][{self.private_name}]LLM原始返回内容: {content}")
+            logger.debug(f"[私聊][{self.private_name}]LLM (行动规划) 原始返回内容: {content}")
 
-            success, result = get_items_from_json(
+            # --- 初始行动规划解析 ---
+            success, initial_result = get_items_from_json(
                 content,
                 self.private_name,
                 "action",
@@ -346,20 +365,73 @@ class ActionPlanner:
                 default_values={"action": "wait", "reason": "LLM返回格式错误或未提供原因，默认等待"},
             )
 
-            action = result.get("action", "wait")
+            initial_action = initial_result.get("action", "wait")
-            reason = result.get("reason", "LLM未提供原因，默认等待")
+            initial_reason = initial_result.get("reason", "LLM未提供原因，默认等待")
 
-            # 验证action类型
+            # 检查是否需要进行结束对话决策 ---
-            # 更新 valid_actions 列表以包含 send_new_message
+            if initial_action == "end_conversation":
+                logger.info(f"[私聊][{self.private_name}]初步规划结束对话，进入告别决策...")
+
+                # 使用新的 PROMPT_END_DECISION
+                end_decision_prompt = PROMPT_END_DECISION.format(
+                    persona_text=persona_text, # 复用之前的 persona_text
+                    chat_history_text=chat_history_text # 复用之前的 chat_history_text
+                )
+
+                logger.debug(f"[私聊][{self.private_name}]发送到LLM的结束决策提示词:\n------\n{end_decision_prompt}\n------")
+                try:
+                    end_content, _ = await self.llm.generate_response_async(end_decision_prompt) # 再次调用LLM
+                    logger.debug(f"[私聊][{self.private_name}]LLM (结束决策) 原始返回内容: {end_content}")
+
+                    # 解析结束决策的JSON
+                    end_success, end_result = get_items_from_json(
+                        end_content,
+                        self.private_name,
+                        "say_bye",
+                        "reason",
+                        default_values={"say_bye": "no", "reason": "结束决策LLM返回格式错误，默认不告别"},
+                        required_types={"say_bye": str, "reason": str} # 明确类型
+                    )
+
+                    say_bye_decision = end_result.get("say_bye", "no").lower() # 转小写方便比较
+                    end_decision_reason = end_result.get("reason", "未提供原因")
+
+                    if end_success and say_bye_decision == "yes":
+                        # 决定要告别，返回新的 'say_goodbye' 动作
+                        logger.info(f"[私聊][{self.private_name}]结束决策: yes, 准备生成告别语. 原因: {end_decision_reason}")
+                        # 注意：这里的 reason 可以考虑拼接初始原因和结束决策原因，或者只用结束决策原因
+                        final_action = "say_goodbye"
+                        final_reason = f"决定发送告别语。决策原因: {end_decision_reason} (原结束理由: {initial_reason})"
+                        return final_action, final_reason
+                    else:
+                        # 决定不告别 (包括解析失败或明确说no)
+                        logger.info(f"[私聊][{self.private_name}]结束决策: no, 直接结束对话. 原因: {end_decision_reason}")
+                        # 返回原始的 'end_conversation' 动作
+                        final_action = "end_conversation"
+                        final_reason = initial_reason # 保持原始的结束理由
+                        return final_action, final_reason
+
+                except Exception as end_e:
+                    logger.error(f"[私聊][{self.private_name}]调用结束决策LLM或处理结果时出错: {str(end_e)}")
+                    # 出错时，默认执行原始的结束对话
+                    logger.warning(f"[私聊][{self.private_name}]结束决策出错，将按原计划执行 end_conversation")
+                    return "end_conversation", initial_reason # 返回原始动作和原因
+
+            else:
+                action = initial_action
+                reason = initial_reason
+
+                # 验证action类型 (保持不变)
                 valid_actions = [
                     "direct_reply",
-                "send_new_message",  # 添加新动作
+                    "send_new_message",
                     "fetch_knowledge",
                     "wait",
                     "listening",
                     "rethink_goal",
-                "end_conversation",
+                    "end_conversation", # 仍然需要验证，因为可能从上面决策后返回
                     "block_and_ignore",
+                    "say_goodbye" # 也要验证这个新动作
                 ]
                 if action not in valid_actions:
                     logger.warning(f"[私聊][{self.private_name}]LLM返回了未知的行动类型: '{action}'，强制改为 wait")
@@ -371,5 +443,6 @@ class ActionPlanner:
                 return action, reason
 
         except Exception as e:
+            # 外层异常处理保持不变
             logger.error(f"[私聊][{self.private_name}]规划行动时调用 LLM 或处理结果出错: {str(e)}")
             return "wait", f"行动规划处理中发生错误，暂时等待: {str(e)}"

src/plugins/PFC/conversation.py

@@ -564,10 +564,48 @@ class Conversation:
                 )
                 self.conversation_info.last_successful_reply_action = None  # 重置状态
 
-        elif action == "end_conversation":
+        elif action == "say_goodbye":
+            self.state = ConversationState.GENERATING # 也可以定义一个新的状态，如 ENDING
+            logger.info(f"[私聊][{self.private_name}]执行行动: 生成并发送告别语...")
+            try:
+                # 1. 生成告别语 (使用 'say_goodbye' action_type)
+                self.generated_reply = await self.reply_generator.generate(
+                    observation_info, conversation_info, action_type="say_goodbye"
+                )
+                logger.info(f"[私聊][{self.private_name}]生成的告别语: {self.generated_reply}")
+
+                # 2. 直接发送告别语 (不经过检查)
+                if self.generated_reply: # 确保生成了内容
+                    await self._send_reply() # 调用发送方法
+                    # 发送成功后，标记动作成功
+                    action_successful = True
+                    logger.info(f"[私聊][{self.private_name}]告别语已发送。")
+                else:
+                    logger.warning(f"[私聊][{self.private_name}]未能生成告别语内容，无法发送。")
+                    action_successful = False # 标记动作失败
+                    conversation_info.done_action[action_index].update(
+                        {"status": "recall", "final_reason": "未能生成告别语内容"}
+                    )
+
+                # 3. 无论是否发送成功，都准备结束对话
                 self.should_continue = False
-            logger.info(f"[私聊][{self.private_name}]决定结束对话...")
+                logger.info(f"[私聊][{self.private_name}]发送告别语流程结束，即将停止对话实例。")
-            action_successful = True  # 标记动作成功
+
+            except Exception as goodbye_err:
+                logger.error(f"[私聊][{self.private_name}]生成或发送告别语时出错: {goodbye_err}")
+                logger.error(f"[私聊][{self.private_name}]{traceback.format_exc()}")
+                # 即使出错，也结束对话
+                self.should_continue = False
+                action_successful = False # 标记动作失败
+                conversation_info.done_action[action_index].update(
+                    {"status": "recall", "final_reason": f"生成或发送告别语时出错: {goodbye_err}"}
+                )
+
+        elif action == "end_conversation":
+            # 这个分支现在只会在 action_planner 最终决定不告别时被调用
+            self.should_continue = False
+            logger.info(f"[私聊][{self.private_name}]收到最终结束指令，停止对话...")
+            action_successful = True  # 标记这个指令本身是成功的
 
         elif action == "block_and_ignore":
             logger.info(f"[私聊][{self.private_name}]不想再理你了...")

src/plugins/PFC/reply_generator.py

@@ -57,6 +57,23 @@ PROMPT_SEND_NEW_MESSAGE = """{persona_text}。现在你在参与一场QQ私聊
 
 请直接输出回复内容，不需要任何额外格式。"""
 
+# Prompt for say_goodbye (告别语生成)
+PROMPT_FAREWELL = """{persona_text}。你在参与一场 QQ 私聊，现在对话似乎已经结束，你决定再发一条最后的消息来圆满结束。
+
+最近的聊天记录：
+{chat_history_text}
+
+请根据上述信息，结合聊天记录，构思一条**简短、自然、符合你人设**的最后的消息。
+这条消息应该：
+1. 从你自己的角度发言。
+2. 符合你的性格特征和身份细节。
+3. 通俗易懂，自然流畅，通常很简短。
+4. 自然地为这场对话画上句号，避免开启新话题或显得冗长、刻意。
+
+请像真人一样随意自然，**简洁是关键**。
+不要输出多余内容（包括前后缀、冒号、引号、括号、表情包、at或@等）。
+
+请直接输出最终的告别消息内容，不需要任何额外格式。"""
 
 class ReplyGenerator:
     """回复生成器"""
@@ -135,10 +152,14 @@ class ReplyGenerator:
         if action_type == "send_new_message":
             prompt_template = PROMPT_SEND_NEW_MESSAGE
             logger.info(f"[私聊][{self.private_name}]使用 PROMPT_SEND_NEW_MESSAGE (追问生成)")
-        else:  # 默认使用 direct_reply 的 prompt
+        elif action_type == "say_goodbye": # 处理告别动作
+            prompt_template = PROMPT_FAREWELL
+            logger.info(f"[私聊][{self.private_name}]使用 PROMPT_FAREWELL (告别语生成)")
+        else:  # 默认使用 direct_reply 的 prompt (包括 'direct_reply' 或其他未明确处理的类型)
             prompt_template = PROMPT_DIRECT_REPLY
             logger.info(f"[私聊][{self.private_name}]使用 PROMPT_DIRECT_REPLY (首次/非连续回复生成)")
 
+
         # --- 格式化最终的 Prompt ---
         prompt = prompt_template.format(
             persona_text=persona_text, goals_str=goals_str, chat_history_text=chat_history_text